.local/bin/channelrefresh

@@ -1,61 +1,79 @@
 #!/bin/dash
-while ! ping -c 1 "9.9.9.9"; do sleep "0.5"; done
-dd="${XDG_CACHE_HOME}/ychannels"
-chl="${XDG_DATA_HOME}/channels"
-ctg="${XDG_DATA_HOME}/categories"
-n() { notify-send -i "${XDG_CONFIG_HOME}/dunst/yt.png" -u "critical" "${1}"; }
 
-eh() {
+while ! ping -c 1 9.9.9.9 > /dev/null 2>&1; do sleep 0.5; done
-	[ ! -s "${chl}" ] && [ ! -s "${ctg}" ] && {
+
-        	printf "%s\n" \
+# "~/.local/share/channels.txt" looks like below:
-                	"Luke Smith=https://www.youtube.com/@LukeSmithxyz/videos" \
+# Luke Smith=https://www.youtube.com/@LukeSmithxyz/videos
-                	"Mental Outlaw=https://www.youtube.com/@MentalOutlaw/videos" \
+# Mental Outlaw=https://www.youtube.com/@MentalOutlaw/videos
-                	> "${chl}"
+
-        	printf "%s\n" "Tech=Luke Smith|Mental Outlaw" > "${ctg}"
+# ~/.local/share/categories.txt looks like below:
-		sudo pacman -S --needed --noconfirm mpv yt-dlp jq || n "Failed to install mpv jq yt-dlp"
+# Tech=Luke Smith|Mental Outlaw
-	}
+
+INSTALLER="sudo pacman -S --noconfirm"
+DATA_DIR="$HOME/.cache/youtube_channels"
+CHANNEL_LIST="$HOME/.local/share/channels.txt"
+mkdir -p "$DATA_DIR" && touch "$CHANNEL_LIST"
+
+error_handling() {
+    [ -s "$CHANNEL_LIST" ] || {
+	    notify-send "You don't have any channels in 'channels.txt'."
+	    exit 1
+    }
+    grep -q "^.*=https://www.youtube.com/@[[:alnum:]]*/videos$" "$CHANNEL_LIST" || {
+	    notify-send "'channels.txt' formatting is wrong."
+	    exit 1
+    }
+    for pkg in yt-dlp mpv jq; do
+        command -v "$pkg" >/dev/null || {
+            notify-send "$pkg is not installed. Installing..."
+            $INSTALLER "$pkg" || {
+                notify-send "Failed to install $pkg."
+                exit 1
+            }
+    }
+    done
+}    
+
+compare_data() {
+  local channel_name="$1"
+  local data_file="${DATA_DIR}/${channel_name}.tsv"
+  local old_data_file="${DATA_DIR}/${channel_name}_old.tsv"
+
+    [ -e "$old_data_file" ] && {
+    old_urls=$(cut -f2 "$old_data_file")
+    new_urls=$(cut -f2 "$data_file")
+
+    echo "$old_urls" | sort > temp1
+    echo "$new_urls" | sort > temp2
+    new_videos=$(comm -13 temp1 temp2 | wc -l)
+    rm temp1 temp2
+
+    [ "$new_videos" -gt 0 ] && notify-send -u critical "$channel_name | $new_videos videos"
+  }
 }
 
-mkdir -p "${dd}" && touch "${chl}"
+update_data() {
-comd() {
+  local channel_name="$1"
-        chn="${1}"
+  local channel_url="$2"
-        df="${dd}/${chn}.tsv"
+  local data_file="${DATA_DIR}/${channel_name}.tsv"
-        odf="${dd}/${chn}_old.tsv"
+  local old_data_file="${DATA_DIR}/${channel_name}_old.tsv"
-        [ -f "${odf}" ] && {
-                ou="$(cut -f2 "${odf}")"
-                nu="$(cut -f2 "${df}")"
-                printf "%s\n" "${ou}" | sort > "t1"
-                printf "%s\n" "${nu}" | sort > "t2"
-                nv="$(comm -13 "t1" "t2" | wc -l)"
-                rm -f "t1" "t2"
-                [ "${nv}" -gt "0" ] && n "${chn} | ${nv} videos."
-        }
-}
-ud() {
-        chn="${1}"
-        cu="${2}"
-        df="${dd}/${chn}.tsv"
-        odf="${dd}/${chn}_old.tsv"
-        mv -f "${df}" "${odf}" 2> "/dev/null"
-        yt-dlp -j --flat-playlist --skip-download --extractor-args \
-                "youtubetab:approximate_date" "${cu}" |
-                jq -r '[.title, .url, .view_count, .duration, .upload_date] | @tsv' > "${df}"
-}
-uac() {
-        while IFS="=" read -r chn cu; do
-                ud "${chn}" "${cu}" &
-        done < "${chl}"
-        wait
-        while IFS="=" read -r chn cu; do
-                comd "${chn}"
-        done < "${chl}"
 
-        avf="${dd}/all_videos.tsv"
+  mv "$data_file" "$old_data_file" 2>/dev/null
-        rm -f "${avf}"
+
-        while IFS= read -r line; do
+  yt-dlp -j --flat-playlist --skip-download --extractor-args youtubetab:approximate_date "$channel_url" | jq -r '[.title, .url, .view_count, .duration, .upload_date] | @tsv' > "$data_file"
-                cn="${line%%=*}"
-                cat "${dd}/${cn}.tsv" >> "${avf}"
-        done < "${chl}"
 }
-eh
+
-uac
+update_all_channels() {
+  while IFS="=" read -r channel_name channel_url; do
+    update_data "$channel_name" "$channel_url" &
+  done < "$CHANNEL_LIST"
+
+  wait
+
+  while IFS="=" read -r channel_name channel_url; do
+    compare_data "$channel_name"
+  done < "$CHANNEL_LIST"
+}
+
+error_handling
+update_all_channels

.local/bin/ybrowser

@@ -1,186 +1,255 @@
-#!/bin/sh
+#!/bin/dash
-dd="${XDG_CACHE_HOME}/ychannels"
+
-yd="${HOME}/ytvideos"
+DATA_DIR="$HOME/.cache/youtube_channels"
-cl="${XDG_DATA_HOME}/categories"
+DOWNLOAD_DIR="$HOME/ytvideos"
-chl="${XDG_DATA_HOME}/channels"
+CATEGORY_LIST="$HOME/.local/share/categories.txt"
-cld="${XDG_DATA_HOME}/clists"
+CHANNEL_LIST="$HOME/.local/share/channels.txt"
-mkdir -p "${dd}" "${yd}" "${cld}"
+CUSTOM_LIST_DIR="$DATA_DIR/custom_lists"
-pr() { printf "%s\n" "${@}"; }
+
-d() {
+mkdir -p "$DATA_DIR" "$DOWNLOAD_DIR" "$CUSTOM_LIST_DIR"
-        i="$(cat)"
+
-        l="$(pr "${i}" | wc -l)"
+[ "$(find "$DATA_DIR" -maxdepth 1 -type f -name "*.tsv" 2>/dev/null)" ] || {
-        [ "${l}" -gt "21" ] && l="21"
+	notify-send "You need to run "channelrefresh" script first."
-        [ "${i}" ] || l="0"
+        exit 1
-        pr "${i}" | dmenu -i -l "${l}" -p "${1}"
 }
-n() { notify-send -i "${XDG_CONFIG_HOME}/dunst/yt.png" "${1}"; }
+
-sv() {
+DMENU() {
-        df="${1}"
+    dmenu -i -l "$1" -p "$2"
-        so="${2}"
-        case "${so}" in
-                "@@sv") sort -nr -t"	" -k3 "${df}" ;;
-                "@@sd") sort -nr -t"	" -k4 "${df}" ;;
-                *) sort -nr -t"	" -k5 "${df}" ;;
-        esac | cut -f1
 }
-gv() {
+
-        cn="${1}"
+sort_videos() {
-        so="${2}"
+  data_file="$1"
-        df="${dd}/${cn}.tsv"
+  sort_option="$2"
-        sv "${df}" "${so}"
+
+  case $sort_option in
+    "@@sv") sort -nr -t"	" -k3 "$data_file" ;;
+    "@@sd") sort -nr -t"	" -k4 "$data_file" ;;
+         *) sort -nr -t"	" -k5 "$data_file" ;;
+  esac | cut -f1
 }
-vu() {
+
-        cn="${1}"
+get_videos() {
-        vt="${2}"
+  channel_name="$1"
-        df="${dd}/${cn}.tsv"
+  sort_option="$2"
-        grep -F "${vt}" "${df}" | cut -f2
+  data_file="$DATA_DIR/$channel_name.tsv"
+  sort_videos "$data_file" "$sort_option"
 }
-ma() {
+
-        pr "WATCH" "DOWNLOAD" "SEND TO A LIST" | d "Actions"
+video_url() {
+  channel_name="$1"
+  video_title="$2"
+  data_file="$DATA_DIR/$channel_name.tsv"
+  grep -F "$video_title" "$data_file" | cut -f2
 }
-ca() {
+
-        lsts="$(pr "${cld}"/* | sed 's|.*/||')"
+rofi_action() {
-        [ "${lsts}" = "*" ] && lsts=""
+  echo "WATCH\nDOWNLOAD\nSEND TO A LIST" | DMENU 3 "Choose an action for the video"
-        pr "${lsts}" "## CREATE LIST ##" "## DELETE LIST ##" | d "Lists"
 }
-lva() {
+
-        pr "WATCH" "DOWNLOAD" "DELETE" | d "Actions"
+rofi_custom_list_action() {
+  echo "$(find "$CUSTOM_LIST_DIR" -maxdepth 1 -type f -exec basename {} \;)
+#### CREATE A LIST ####
+#### DELETE A LIST ####" | DMENU 10 "Choose an action or list"
 }
-atl() {
+
-        vt="${1}"
+list_video_action() {
-        cn="${2}"
+  echo "WATCH\nDOWNLOAD\nDELETE" | DMENU 3 "Choose an action for the video"
-        ln="${3}"
-        pr "${cn}: ${vt}" >> "${cld}/${ln}"
 }
-clm() {
+
-        while true; do
+add_to_list() {
-                lst="$(ca)"
+  video_title="$1"
-                case "${lst}" in
+  channel_name="$2"
-                        "## CREATE LIST ##")
+  list_name="$3"
-                                nlst="$(pr | d "Name")"
+  echo "$channel_name: $video_title" >> "$CUSTOM_LIST_DIR/$list_name"
-                                [ "${nlst}" ] && touch "${cld}/${nlst}"
-                                ;;
-                        "## DELETE LIST ##")
-                                dlst="$(find "${cld}" -mindepth "1" -exec basename {} \; | d "Delete List")"
-                                [ "${dlst}" ] && rm -f "${cld}/${dlst}"
-                                ;;
-                        "") return ;;
-                        *) cvm "${lst}" ;;
-                esac
-        done
 }
-cvm() {
+
-        ln="${1}"
+custom_list_menu() {
-        while true; do
+  while true; do
-                vi=$(d "Videos" < "${cld}/${ln}")
+    list="$(rofi_custom_list_action)"
-                [ "${vi}" ] || return
+    [ -z "$list" ] && return
-                cn="${vi%%: *}"
+    case "$list" in
-                vt="${vi##*: }"
+      *CREATE*)
-                clvm "${vt}" "${cn}" "${ln}"
+        new_list=$(echo "" | DMENU 0 "Enter the name of the new list")
-        done
+        [ -n "$new_list" ] && touch "$CUSTOM_LIST_DIR/$new_list"
+        ;;
+      *DELETE*)
+        delete_list=$(find "$CUSTOM_LIST_DIR" -maxdepth 1 -type f -exec basename {} \; |
+		DMENU 10 "Choose a list to delete")
+        [ -n "$delete_list" ] && rm "$CUSTOM_LIST_DIR/$delete_list"
+        ;;
+      *)
+        custom_list_video_menu "$list"
+        ;;
+    esac
+  done
 }
-clvm() {
+
-        vt="${1}"
+custom_list_video_menu() {
-        cn="${2}"
+  list_name="$1"
-        ln="${3}"
+  while true; do
-        ac="$(lva)"
+    video_info=$(DMENU 20 "Choose a video" < "$CUSTOM_LIST_DIR/$list_name")
-        case "${ac}" in
+    [ -z "$video_info" ] && return
-                "WATCH") vp "${ac}" "${vt}" "${cn}" ;;
+    channel_name="${video_info%%: *}"
-                "DOWNLOAD") vp "${ac}" "${vt}" "${cn}" && n "Download finished" ;;
+    video_title="${video_info##*: }"
-                "DELETE") sed -i "/${vt}/d" "${cld}/${ln}" ;;
+    custom_list_video_action_menu "$video_title" "$channel_name" "$list_name"
-                *) return ;;
+  done
-        esac
 }
-vp() {
+
-        ac="${1}"
+custom_list_video_action_menu() {
-        vt="${2}"
+  video_title="$1"
-        cn="${3}"
+  channel_name="$2"
-        vu="$(vu "${cn}" "${vt}")"
+  list_name="$3"
-        case "${ac}" in
+
-                "WATCH") mpv "${vu}" ;;
+  action=$(list_video_action)
-                "DOWNLOAD")
+
-                        cdd="${yd}/${cn}"
+  case $action in
-                        mkdir -p "${cdd}"
+    WATCH)
-                        yt-dlp -o "${cdd}/%(title)s.%(ext)s" "${vu}"
+      video_process WATCH "$video_title" "$channel_name"
-                        n "Download finished"
+      ;;
-                        ;;
+    DOWNLOAD)
-        esac
+      video_process DOWNLOAD "$video_title" "$channel_name" && notify-send "Downloading has finished."
+      ;;
+    DELETE)
+      escaped_video_title="$(echo "$video_title" | sed 's/[][\^$.\/|*+?(){}#]/\\&/g')"
+      sed "/$escaped_video_title/d" "$CUSTOM_LIST_DIR/$list_name" > "$CUSTOM_LIST_DIR/${list_name}.tmp" &&
+	  mv "$CUSTOM_LIST_DIR/${list_name}.tmp" "$CUSTOM_LIST_DIR/$list_name"
+      ;;
+    *)
+      return
+      ;;
+  esac
 }
-gav() {
+
-        so="${1}"
+video_process() {
-        avf="${dd}/all_videos.tsv"
+    action="$1"
-        sv "${avf}" "${so}"
+    video_title="$2"
+    channel_name="$3"
+    video_url=$(video_url "$channel_name" "$video_title")
+
+    case "$action" in
+        WATCH)
+            mpv "$video_url"
+            ;;
+        DOWNLOAD)
+            local channel_download_dir="$DOWNLOAD_DIR/$channel_name"
+            mkdir -p "$channel_download_dir"
+            yt-dlp -o "$channel_download_dir/%(title)s.%(ext)s" "$video_url"
+            notify-send "Downloading has finished."
+            ;;
+    esac
 }
-bac() {
+
-        while vt="$(gav | d "Videos | Sort: @@s{v,d}")"; do
+get_all_videos() {
-                [ "${vt}" ] || break
+  sort_option="$1"
-                [ "${vt}" = "@@sv" ] || [ "${vt}" = "@@sd" ] && {
+  all_videos_file="$DATA_DIR/all_videos.tsv"
-                        vt=$(gav "${vt}" | d "Videos")
+
-                        [ "${vt}" ] || continue
+  rm -f "$all_videos_file"
-                }
+  while IFS= read -r line
-                grep -lF "${vt}" "${dd}"/*.tsv | head -n "1" | while read -r "vf"; do
+  do
-                        vam "${vt}" "$(basename "${vf}" .tsv)"
+    channel_name="${line%%=*}"
-                        break
+    cat "$DATA_DIR/$channel_name.tsv" >> "$all_videos_file"
-                done
+  done < "$CHANNEL_LIST"
-        done
+
+  sort_videos "$all_videos_file" "$sort_option"
 }
-cm() {
+
-        while true; do
+browse_all_channels() {
-                c="$(cut -d= -f1 "${cl}" | d "Categories")"
+  while video_title=$(get_all_videos | DMENU 20 "Choose a video or enter @@sv or @@sd"); do
-                [ "${c}" ] || return
+    [ -z "$video_title" ] && break
-                chm "${c}"
+    [ "$video_title" = "@@sv" ] || [ "$video_title" = "@@sd" ] && video_title=$(get_all_videos "$video_title" | DMENU 20 "Choose a video")
-        done
+
+    grep -lF "$video_title" "${DATA_DIR}"/*.tsv | head -n 1 | while read -r video_file; do
+      channel_name=$(basename "$video_file" .tsv)
+      video_action_menu "$video_title" "$channel_name"
+      break
+    done
+  done
 }
-chm() {
+
-        c="${1}"
+category_menu() {
-        IFS="|"
+  while true; do
-        ch="$(sed -n "s/^${c}=\(.*\)$/\1/p" "${cl}")"
+    category="$(cut -d= -f1 "$CATEGORY_LIST" | DMENU 12 "Choose a category")"
-        set -- ${ch}
+    [ -z "$category" ] && return
-        while true; do
+
-                cn="$(pr "${@}" | d "Channels")"
+    channel_menu "$category"
-                [ "${cn}" ] || return
+  done
-                vm "${cn}"
-        done
 }
-vm() {
+
-        cn="${1}"
+channel_menu() {
-        while true; do
+  category="$1"
-                vt=$(gv "${cn}" | d "Videos")
+  IFS="|"
-                [ "${vt}" ] || return
+
-                [ "${vt}" = "@@sv" ] || [ "${vt}" = "@@sd" ] && {
+  channels=$(sed -n "s/^${category}=\(.*\)$/\1/p" "$CATEGORY_LIST")
-                        so="${vt}"
+  set -- $channels
-                        vt="$(gv "${cn}" "${so}" | d "Videos")"
+
-                }
+  while true; do
-                vam "${vt}" "${cn}"
+    channel_name=$(printf '%s\n' "$@" | DMENU 20 "Choose a channel")
-        done
+    [ -z "$channel_name" ] && return
+
+    video_menu "$channel_name"
+  done
 }
-vam() {
+
-        vt="${1}"
+video_menu() {
-        cn="${2}"
+  channel_name="$1"
-        while [ "${vt}" ] && [ "${vt}" != "@@sv" ] && [ "${vt}" != "@@sd" ]; do
+
-                ac="$(ma)"
+  while true; do
-                case "${ac}" in
+    video_title=$(get_videos "$channel_name" | DMENU 20 "Choose a video")
-                        "WATCH") vp "${ac}" "${vt}" "${cn}" ;;
+    [ -z "$video_title" ] && return
-                        "DOWNLOAD") vp "${ac}" "${vt}" "${cn}" && n "Download finished" ;;
+    [ "$video_title" = "@@sv" ] || [ "$video_title" = "@@sd" ] && {
-                        "SEND TO A LIST")
+      sort_option="$video_title"
-                                ln="$(find "${cld}" -mindepth "1" -exec basename {} \; | d "Lists")"
+      video_title=$(get_videos "$channel_name" "$sort_option" | DMENU 20 "Choose a video")
-                                [ "${ln}" ] && atl "${vt}" "${cn}" "${ln}" && n "${vt} > list: ${ln}"
+    }
-                                ;;
+
-                        *) return ;;
+    video_action_menu "$video_title" "$channel_name"
-                esac
+  done
-        done
 }
-mm() {
+
-        pr " ## ALL CHANNELS ##" " ## CATEGORIES ##" " ## CUSTOM LISTS ##" \
+video_action_menu() {
-                "$(cut -d= -f1 "${chl}")" | d "YouTube"
+  video_title="$1"
+  channel_name="$2"
+
+  while [ -n "$video_title" ] && [ "$video_title" != "@@sv" ] && [ "$video_title" != "@@sd" ]; do
+    action=$(rofi_action)
+
+    case $action in
+      WATCH)
+	video_process WATCH "$video_title" "$channel_name"
+        ;;
+      DOWNLOAD)
+        video_process DOWNLOAD "$video_title" "$channel_name" && notify-send "Downloading has finished."
+        ;;
+      "SEND TO A LIST")
+        list_name=$(find "$CUSTOM_LIST_DIR" -maxdepth 1 -type f -exec basename {} \; | DMENU 10 "Choose a list")
+        [ -n "$list_name" ] && add_to_list "$video_title" "$channel_name" "$list_name" &&
+		notify-send "$video_title is added to the list: $list_name"
+        ;;
+      *)
+        return
+        ;;
+    esac
+  done
 }
+
+main_menu() {
+  options=" #### ALL CHANNELS ####
+ #### CATEGORIES ####
+ #### CUSTOM LISTS ####
+$(cut -d= -f1 "$CHANNEL_LIST")"
+
+  echo "$options" | DMENU 20 "Choose an Option or a Category"
+}
+
 while true; do
-        mc="$(mm)"
+
-        case "${mc}" in
+  main_choice=$(main_menu)
-                " ## ALL CHANNELS ##") bac ;;
+
-                " ## CATEGORIES ##") cm ;;
+  [ -z "$main_choice" ] && exit
-                " ## CUSTOM LISTS ##") clm ;;
+
-                "") exit ;;
+  case "$main_choice" in
-                *) vm "${mc}" ;;
+    *ALL\ CHANNELS*) browse_all_channels       ;;
-        esac
+    *CATEGORIES*   ) category_menu             ;;
+    *CUSTOM\ LISTS*) custom_list_menu          ;;
+                  *) video_menu "$main_choice" ;;
+  esac
 done