;;; yeetube.el --- YouTube/Invidious Front End -*- lexical-binding: t; -*- ;; Copyright (C) 2023 Thanos Apollo ;; Author: Thanos Apollo ;; Keywords: extensions youtube videos invidious ;; URL: https://git.sr.ht/~thanosapollo/yeetube.el ;; Version: 1.3.4 (defvar yeetube--version '1.3.4) ;; Package-Requires: ((emacs "27.2")) ;; This program is free software; you can redistribute it and/or modify ;; it under the terms of the GNU General Public License as published by ;; the Free Software Foundation, either version 3 of the License, or ;; (at your option) any later version. ;; This program is distributed in the hope that it will be useful, ;; but WITHOUT ANY WARRANTY; without even the implied warranty of ;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the ;; GNU General Public License for more details. ;; You should have received a copy of the GNU General Public License ;; along with this program. If not, see . ;;; Commentary: ;; Search, play and downlaod videos from YouTube or any Invidious instance, ;; including localhost. ;; This package does not use YouTube/Invidious API. ;;; Code: (require 'url) (require 'org-element) (require 'cl-lib) (defgroup yeetube nil "Youtube/Invidious Front-end." :group 'external :prefix "yeetube-") (defcustom yeetube-results-limit 15 "Define a limit for search results." :type 'number :safe #'numberp :group 'yeetube) (defcustom yeetube-results-prefix "+" "Define prefix to display results with." :type 'string :safe #'stringp :group 'yeetube) (defcustom yeetube-query-url "https://www.youtube.com" "Search URL." :type 'string :safe #'stringp :group 'yeetube) (defcustom yeetube-download-audio-format nil "Select download video as audio FORMAT. If nil download output will be the default format. Example Usage: (setq yeetube-download-audio-format \"m4a\")" :type 'string :safe #'stringp :group 'yeetube) (defcustom yeetube-display-info-keys t "Display default keybindings." :type 'boolean :safe #'booleanp :group 'yeetube) (defcustom yeetube-mpv-socket (concat temporary-file-directory "yeet-socket") "MPV Input Socket." :type 'string :safe #'stringp :group 'yeetube) (defcustom yeetube-player (concat (executable-find "mpv") " --input-ipc-server=" yeetube-mpv-socket) "Select default video player. It's recommended you keep it as the default value." :type 'string :safe #'stringp :group 'yeetube) (defcustom yeetube-download-directory "~/Downloads" "Default directory to downlaod videos." :type 'string :safe #'stringp :group 'yeetube) (define-minor-mode yeetube-mode "Yeetube mode." :init-value nil :lighter " yeetube-mode" :keymap (let ((yeetube-mode-map (make-sparse-keymap))) (define-key yeetube-mode-map (kbd "RET") 'yeetube-play) (define-key yeetube-mode-map (kbd "d") 'yeetube-download-video) (define-key yeetube-mode-map (kbd "u") 'yeetube-change-query-url) (define-key yeetube-mode-map (kbd "q") 'kill-current-buffer) (define-key yeetube-mode-map (kbd "D") 'yeetube-change-download-directory) (define-key yeetube-mode-map (kbd "a") 'yeetube-change-download-audio-format) (define-key yeetube-mode-map (kbd "p") 'yeetube-toggle-pause-mpv) (define-key yeetube-mode-map (kbd "v") 'yeetube-toggle-video-mpv) (define-key yeetube-mode-map (kbd "s") 'yeetube-save-video) (define-key yeetube-mode-map (kbd "P") 'yeetube-play-saved-video) yeetube-mode-map)) (defvar yeetube--yt-dlp (executable-find "yt-dlp")) (defvar yeetube--video-ids nil) (defvar yeetube--video-titles nil) (defvar yeetube-saved-videos nil) (defvar yeetube--last-played nil) (defun yeetube-youtube-p (url) "Check if it's a youtube URL." (if (string-match-p "youtube" url) t nil)) (defun yeetube-play-url (url) "Open URL using yeetube-player." (when (string-prefix-p "http" url) (if (string-match "mpv" yeetube-player) (shell-command (format "pkill -9 -f mpv")) (shell-command (format "pkill -9 -f %s" (shell-quote-argument yeetube-player)))) (call-process-shell-command (format "%s %s" yeetube-player url) nil 0) (message "Opening with %s" yeetube-player))) (defun yeetube-play () "Open the url at point in an `'org-mode buffer using ='yeetube-player'." (interactive) (let ((url (org-element-property :raw-link (org-element-context)))) (yeetube-play-url url))) (defun yeetube-load-saved-videos () "Load saved videos." (interactive) (let ((file-path (concat user-emacs-directory "yeetube"))) (if (file-exists-p file-path) (with-temp-buffer (insert-file-contents file-path) (goto-char (point-min)) (let ((contents (read (current-buffer)))) (setq yeetube-saved-videos contents))) (write-region "nil" nil file-path)))) (defun yeetube-save-video () "Save url at point." (interactive) (yeetube-load-saved-videos) (let ((name (read-string "Save as: ")) (url (org-element-property :raw-link (org-element-context)))) (push (cons name url) yeetube-saved-videos))) (defun yeetube-play-saved-video () "Select & Play a saved video." (interactive) (yeetube-load-saved-videos) (let ((video (completing-read "Select video: " yeetube-saved-videos nil t))) (yeetube-play-url (cdr (assoc video yeetube-saved-videos))))) (defun yeetube-remove-saved-video () "Select video to remove from saved videos." (interactive) (yeetube-load-saved-videos) (let ((video (completing-read "Select video: " yeetube-saved-videos nil t))) (setq yeetube-saved-videos (remove (assoc video yeetube-saved-videos) yeetube-saved-videos)))) (defun yeetube-remove-all-saved-videos ( ) "Clear yeetube saved." (interactive) (let ((clear-saved (y-or-n-p "Delete saved?"))) (when clear-saved (setq yeetube-saved-videos nil)))) (defun yeetube-toggle-video-mpv () "Toggle video on/off for mpv player." (interactive) (let ((socket (concat " --input-ipc-server=" yeetube-mpv-socket)) (no-video " --no-video") (mpv (executable-find "mpv"))) (setq yeetube-player (if (string-match no-video yeetube-player) (concat mpv socket) (concat mpv no-video socket))) (message (format "Yeetube Player: %s" yeetube-player)))) (defun yeetube-toggle-pause-mpv () "Play/Pause mpv." (interactive) (if (string-match "mpv" yeetube-player) (progn (shell-command (concat "echo '{ \"command\": [\"cycle\", \"pause\"] }' | socat - " yeetube-mpv-socket)) (message "mpv play/pause")) (error "To use this function you need to have mpv installed & set yeetube-player to the default value"))) ;; Usually titles from youtube get messed up, ;; This should be fix some of the common issues. (defun yeetube-fix-title (title) "Adjust TITLE." (let ((replacements '(("&" . "&") (""" . "\"") ("'" . "'") ("u0026" . "&")))) (mapc (lambda (replacement) (setq title (replace-regexp-in-string (car replacement) (cdr replacement) title))) replacements) title)) (defun yeetube-insert-content (prefix url video-titles video-ids) "Insert video links with titles into the buffer. Arguments: - PREFIX: The prefix string for each link. - URL: The base URL for the YouTube links. - VIDEO-TITLES: A list of video titles. - VIDEO-IDS: A list of video IDs. For each video ID and video title, inserts a link into the buffer in the format: PREFIX [[URL/watch?v=VIDEOID][VIDEOTITLE ]]" (cl-loop for (video-id . video-title) in (cl-mapcar #'cons (reverse video-ids) (reverse video-titles)) do (insert (format "%s [[%s/watch?v=%s][%s ]]\n" prefix url video-id (yeetube-fix-title video-title))))) (defun yeetube-create-buffer (query video-titles video-ids) "Create *Yeetube-Search* buffer for QUERY, using VIDEO-TITLES with VIDEO-IDS." (with-current-buffer (switch-to-buffer (get-buffer-create "*Yeetube Search*")) (setq buffer-read-only nil) (erase-buffer) (org-mode) (insert (format "searching: %s\nfor: %s \n* Search Results: \n \n" yeetube-query-url query)) (yeetube-insert-content yeetube-results-prefix yeetube-query-url video-titles video-ids) (yeetube-insert-info) (setq buffer-read-only t) (goto-char (point-min)) (search-forward yeetube-results-prefix) (yeetube-mode))) ;;;###autoload (defun yeetube-search (query) "Search for QUERY." (interactive "sYeetube Search: ") (let ((is-youtube? (yeetube-youtube-p yeetube-query-url))) (with-current-buffer (url-retrieve-synchronously (concat yeetube-query-url "/search?q=" (replace-regexp-in-string " " "+" query)) t t) (decode-coding-region (point-min) (point-max) 'utf-8) (goto-char (point-min)) (toggle-enable-multibyte-characters) (if is-youtube? (yeetube--get-content-youtube) (yeetube--get-content-invidious)) (yeetube-create-buffer query yeetube--video-titles yeetube--video-ids)))) (defun yeetube--get-content-youtube () "Get content from youtube." (setq yeetube--video-ids nil) (setq yeetube--video-titles nil) (while (and (< (length yeetube--video-ids) yeetube-results-limit) (search-forward "videoId" nil t)) (let* ((start (point)) (end (search-forward ",")) (videoid (buffer-substring (+ start 3) (- end 2)))) (unless (or (member videoid yeetube--video-ids) (not (and (>= (length videoid) 9) (<= (length videoid) 13) (string-match-p "^[a-zA-Z0-9_-]*$" videoid)))) (push videoid yeetube--video-ids) (search-forward "text") (let* ((start (point)) (end (search-forward ",\"")) (title (buffer-substring (+ start 3) (- end 5)))) (if (string-match-p "vssLoggingContext" title) (pop yeetube--video-ids) (push title yeetube--video-titles))))))) (defun yeetube--get-content-invidious () "Get content from an invidious instance." (setq yeetube--video-ids nil) (setq yeetube--video-titles nil) (while (and (< (length yeetube--video-ids) yeetube-results-limit) (search-forward "watch?v" nil t)) (let* ((start (point)) (end (search-forward ">")) (videoid (buffer-substring (+ start 1) (- end 2)))) (unless (or (member videoid yeetube--video-ids) (not (and (>= (length videoid) 9) (<= (length videoid) 13) (string-match-p "^[a-zA-Z0-9_-]*$" videoid)))) (push videoid yeetube--video-ids) (search-forward "\"auto\">") (let* ((start (point)) (end (search-forward ">")) (title (buffer-substring (+ start 0) (- end 4)))) (if (string-match-p "vssLoggingContext" title) (pop yeetube--video-ids) (push title yeetube--video-titles))))))) ;;;###autoload (defun yeetube-download-video () "Download using link at point in an `'org-mode buffer with yt-dlp." (interactive) (let ((url (org-element-property :raw-link (org-element-context)))) (when (string-prefix-p "http" url) (let ((default-directory yeetube-download-directory)) (async-shell-command (if yeetube-download-audio-format (format "%s %s --extract-audio --audio-format %s" (shell-quote-argument yeetube--yt-dlp) (shell-quote-argument url) (shell-quote-argument yeetube-download-audio-format)) (format "%s %s" (shell-quote-argument yeetube--yt-dlp) (shell-quote-argument url))) (message "Downloading %s " url)))))) ;;;###autoload (defun yeetube-download-videos () "Download one or multiple videos using yt-dlp. This command is not meant to be used in the *Yeetube Search* buffer. Usage Example: Open a Dired buffer and navigate where you want to download your videos, then run this command interactively." (interactive) (let ((links '()) (names '()) (url "") (name "") (buffer-counter 1) (name-counter 1)) ;; Read links and names until "q" is entered (while (not (string= url "q")) (setq url (read-string "Enter URL (q to quit): ")) (unless (string= url "q") (setq links (cons url links)) (setq name (read-string (format "Enter name (%d-NAME): " name-counter))) (while (get-buffer (format "download-video-%d" buffer-counter)) (setq buffer-counter (1+ buffer-counter))) (setq names (cons name names)) (setq buffer-counter (1+ buffer-counter)) (setq name-counter (1+ name-counter)))) ;; Process the collected links and names (setq links (reverse links)) (setq names (reverse names)) (dolist (pair (cl-mapcar 'cons links names)) (let ((url (car pair)) (name (cdr pair)) (buffer-name (format "download-video-%d" buffer-counter))) (async-shell-command (if yeetube-download-audio-format (format "%s %s --extract-audio --audio-format %s -o %s" (shell-quote-argument yeetube--yt-dlp) (shell-quote-argument url) (shell-quote-argument yeetube-download-audio-format) (shell-quote-argument name)) (format "%s %s -o %s" (shell-quote-argument yeetube--yt-dlp) (shell-quote-argument url) (shell-quote-argument name))) buffer-name) (setq buffer-counter (1+ buffer-counter)))))) (defun yeetube-insert-info () "Insert default keybindings at *Yeetube Search* buffer." (insert "\n\n** Info" (format "\nDownload Directory: %s" yeetube-download-directory) (format "\nDownload as audio format: %s" yeetube-download-audio-format)) (when yeetube-display-info-keys (insert "\n\n*** Keybindings" "\n" "\n~RET~ -> Play Video\n" "\n~v~ -> Toggle Video on/off\n" "\n~p~ -> Toggle Pause/Play\n" "\n~C-c C-o~ -> Open In Browser\n" "\n~d~ -> Download\n" "\n~D~ -> Change Download Directory\n" "\n~a~ -> Change Download (Audio) Format\n" "\n~u~ -> Change search URL (youtube.com, any invidious instance or localhost)\n" "\n~q~ -> Quit\n" "\n~s~ -> Save video\n" "\n~P~ -> Play Saved Video"))) (defun yeetube-change-download-directory () "Change download directory." (interactive) (setq yeetube-download-directory (read-directory-name "Select a directory: "))) (defun yeetube-change-download-audio-format (format) "Change download format to audio FORMAT." (interactive "sSpecify Audio Format(no for nil): ") (setq yeetube-download-audio-format format) (when (equal yeetube-download-audio-format "no") (setq yeetube-download-audio-format nil))) (defun yeetube-change-query-url () "Change `yeetube-query-url'." (interactive) (setq yeetube-query-url (read-string "URL: ")) (when (string-prefix-p "localhost" yeetube-query-url) (setq yeetube-query-url (concat "http://localhost:" (read-string "Port: ")))) (unless (or (string-prefix-p "http://" yeetube-query-url) (string-prefix-p "https://" yeetube-query-url)) (setq yeetube-query-url (concat "https://" yeetube-query-url))) (when (string-suffix-p "/" yeetube-query-url) (setq yeetube-query-url (substring yeetube-query-url 0 -1)))) (defun yeetube-update-info (symbol-name new-value _operation _where) "Update information for SYMBOL-NAME with NEW-VALUE. SYMBOL-NAME is the name of the symbol to update. NEW-VALUE is the new value for the symbol. OPERATION is the operation to perform (e.g., insert or replace). WHERE indicates where in the buffer the update should happen. OPERATION & WHERE are required to work with ='add-variable-watcher." (let ((to-change (pcase symbol-name ('yeetube-download-directory "Download Directory:") ('yeetube-download-audio-format "Download as audio format:") ('yeetube-query-url "searching:"))) (buffer-cur (buffer-name))) (when (get-buffer "*Yeetube Search*") (push-mark) (switch-to-buffer (get-buffer "*Yeetube Search*")) (setq buffer-read-only nil) (goto-char (point-min)) (search-forward to-change) (beginning-of-visual-line) (kill-region (point) (line-end-position)) (insert (format "%s %s" to-change new-value)) (setq-local buffer-read-only t) (switch-to-buffer buffer-cur)) (goto-char (mark)))) (defun yeetube-update-saved-videos-list (_symbol new-value _where _environment) "Updated saved videos. SYMBOL-NAME is the name of the symbol to update. NEW-VALUE is the new value for the symbol. OPERATION is the operation to perform (e.g., insert or replace). WHERE indicates where in the buffer the update should happen." (with-temp-buffer (find-file (concat user-emacs-directory "yeetube")) (erase-buffer) (setq yeetube-saved-videos new-value) (insert (pp-to-string yeetube-saved-videos)) (save-buffer) (kill-buffer))) (defun yeetube-version () "Show Yeetube Version." (interactive) (message "Yeetube Version: %s" yeetube--version)) ;; Variable to watch (add-variable-watcher 'yeetube-download-directory #'yeetube-update-info) (add-variable-watcher 'yeetube-download-audio-format #'yeetube-update-info) (add-variable-watcher 'yeetube-query-url #'yeetube-update-info) (add-variable-watcher 'yeetube-saved-videos #'yeetube-update-saved-videos-list) (provide 'yeetube) ;;; yeetube.el ends here