Compare commits

..

1 Commits

View File

@ -8,13 +8,9 @@ CUSTOM_LIST_DIR="$DATA_DIR/custom_lists"
mkdir -p "$DATA_DIR" "$DOWNLOAD_DIR" "$CUSTOM_LIST_DIR" mkdir -p "$DATA_DIR" "$DOWNLOAD_DIR" "$CUSTOM_LIST_DIR"
DMENU() {
dmenu -i -l $1 -p "$2"
}
sort_videos() { sort_videos() {
data_file="$1" local data_file="$1"
sort_option="$2" local sort_option="$2"
case $sort_option in case $sort_option in
"@@sv") sort -nr -t" " -k3 "$data_file" ;; "@@sv") sort -nr -t" " -k3 "$data_file" ;;
@ -24,52 +20,49 @@ sort_videos() {
} }
get_videos() { get_videos() {
channel_name="$1" local channel_name="$1"
sort_option="$2" local sort_option="$2"
data_file="$DATA_DIR/$channel_name.tsv" local data_file="$DATA_DIR/$channel_name.tsv"
sort_videos "$data_file" "$sort_option" sort_videos "$data_file" "$sort_option"
} }
video_url() { video_url() {
channel_name="$1" local channel_name="$1"
video_title="$2" local video_title="$2"
data_file="$DATA_DIR/$channel_name.tsv" local data_file="$DATA_DIR/$channel_name.tsv"
sed -n "s/$video_title\t\([^\t]*\)\t.*$/\1/p" "$data_file" sed -n "s/$video_title\t\([^\t]*\)\t.*$/\1/p" "$data_file"
} }
rofi_action() { rofi_action() {
echo "WATCH\nDOWNLOAD\nSEND TO A LIST" | DMENU 3 "Choose an action for the video" echo "Watch\nDownload\nSend To a List" | dmenu -i -l 3 -p "Choose an action for the video"
} }
rofi_custom_list_action() { rofi_custom_list_action() {
echo "$(find "$CUSTOM_LIST_DIR" -maxdepth 1 -type f -exec basename {} \;) echo "$(ls "$CUSTOM_LIST_DIR")\nCreate a List\nDelete a List" | dmenu -i -l 10 -p "Choose an action or list"
#### CREATE A LIST ####
#### DELETE A LIST ####" | DMENU 10 "Choose an action or list"
} }
list_video_action() { list_video_action() {
echo "WATCH\nDOWNLOAD\nDELETE" | DMENU 3 "Choose an action for the video" echo "Watch\nDownload\nDelete" | dmenu -i -l 3 -p "Choose an action for the video"
} }
add_to_list() { add_to_list() {
video_title="$1" local video_title="$1"
channel_name="$2" local channel_name="$2"
list_name="$3" local list_name="$3"
echo "$channel_name: $video_title" >> "$CUSTOM_LIST_DIR/$list_name" echo "$channel_name: $video_title" >> "$CUSTOM_LIST_DIR/$list_name"
} }
custom_list_menu() { custom_list_menu() {
while true; do while true; do
list="$(rofi_custom_list_action)" local list="$(rofi_custom_list_action)"
[ -z "$list" ] && return [ -z "$list" ] && return
case "$list" in case "$list" in
*CREATE*) "Create a List")
new_list=$(DMENU 0 "Enter the name of the new list") local new_list=$(dmenu -i -l 0 -p "Enter the name of the new list")
[ -n "$new_list" ] && touch "$CUSTOM_LIST_DIR/$new_list" [ -n "$new_list" ] && touch "$CUSTOM_LIST_DIR/$new_list"
;; ;;
*DELETE*) "Delete a List")
delete_list=$(find "$CUSTOM_LIST_DIR" -maxdepth 1 -type f -exec basename {} \; | local delete_list=$(ls "$CUSTOM_LIST_DIR" | dmenu -i -l 10 -p "Choose a list to delete")
DMENU 10 "Choose a list to delete")
[ -n "$delete_list" ] && rm "$CUSTOM_LIST_DIR/$delete_list" [ -n "$delete_list" ] && rm "$CUSTOM_LIST_DIR/$delete_list"
;; ;;
*) *)
@ -80,9 +73,12 @@ custom_list_menu() {
} }
custom_list_video_menu() { custom_list_video_menu() {
list_name="$1" local list_name="$1"
local video_info
local video_title
local channel_name
while true; do while true; do
video_info=$(cat "$CUSTOM_LIST_DIR/$list_name" | DMENU 20 "Choose a video") video_info=$(cat "$CUSTOM_LIST_DIR/$list_name" | dmenu -i -l 20 -p "Choose a video")
[ -z "$video_info" ] && return [ -z "$video_info" ] && return
channel_name="${video_info%%: *}" channel_name="${video_info%%: *}"
video_title="${video_info##*: }" video_title="${video_info##*: }"
@ -91,23 +87,23 @@ custom_list_video_menu() {
} }
custom_list_video_action_menu() { custom_list_video_action_menu() {
video_title="$1" local video_title="$1"
channel_name="$2" local channel_name="$2"
list_name="$3" local list_name="$3"
local action
action=$(list_video_action) action=$(list_video_action)
case $action in case $action in
WATCH) Watch)
video_process WATCH "$video_title" "$channel_name" play_video "$video_title" "$channel_name"
;; ;;
DOWNLOAD) Download)
video_process DOWNLOAD "$video_title" "$channel_name" && notify-send "Downloading has finished." download_video "$video_title" "$channel_name" && notify-send "Downloading has finished."
;; ;;
DELETE) Delete)
escaped_video_title="$(echo "$video_title" | sed 's/[][\^$.\/|*+?(){}#]/\\&/g')" escaped_video_title="$(echo "$video_title" | tr -d '|')"
sed "/$escaped_video_title/d" "$CUSTOM_LIST_DIR/$list_name" > "$CUSTOM_LIST_DIR/${list_name}.tmp" && sed -i "/$escaped_video_title/d" "$CUSTOM_LIST_DIR/$list_name"
mv "$CUSTOM_LIST_DIR/${list_name}.tmp" "$CUSTOM_LIST_DIR/$list_name"
;; ;;
*) *)
return return
@ -115,33 +111,30 @@ custom_list_video_action_menu() {
esac esac
} }
video_process() { play_video() {
action="$1" local video_title="$1"
video_title="$2" local channel_name="$2"
channel_name="$3" local video_url="$(video_url "$channel_name" "$video_title")"
video_url=$(video_url "$channel_name" "$video_title") mpv "$video_url"
}
case "$action" in download_video() {
WATCH) local video_title="$1"
mpv "$video_url" local channel_name="$2"
;; local video_url=$(video_url "$channel_name" "$video_title")
DOWNLOAD) local channel_download_dir="$DOWNLOAD_DIR/$channel_name"
local channel_download_dir="$DOWNLOAD_DIR/$channel_name" mkdir -p "$channel_download_dir"
mkdir -p "$channel_download_dir" yt-dlp -o "$channel_download_dir/%(title)s.%(ext)s" "$video_url"
yt-dlp -o "$channel_download_dir/%(title)s.%(ext)s" "$video_url"
notify-send "Downloading has finished."
;;
esac
} }
get_all_videos() { get_all_videos() {
sort_option="$1" local sort_option="$1"
all_videos_file="$DATA_DIR/all_videos.tsv" local all_videos_file="$DATA_DIR/all_videos.tsv"
rm -f "$all_videos_file" rm -f "$all_videos_file"
while IFS= read -r line while IFS= read -r line
do do
channel_name="${line%%=*}" local channel_name="${line%%=*}"
cat "$DATA_DIR/$channel_name.tsv" >> "$all_videos_file" cat "$DATA_DIR/$channel_name.tsv" >> "$all_videos_file"
done < "$CHANNEL_LIST" done < "$CHANNEL_LIST"
@ -150,11 +143,10 @@ get_all_videos() {
browse_all_channels() { browse_all_channels() {
while true; do while true; do
video_title=$(get_all_videos | DMENU 20 "Choose a video or enter @@sv or @@sd") video_title=$(get_all_videos | dmenu -i -l 20 -p "Choose a video or enter @@sv or @@sd")
[ -z "$video_title" ] && break || { [ -z "$video_title" ] && break || {
[ "$video_title" = "@@sv" -o "$video_title" = "@@sd" ] && [ "$video_title" = "@@sv" -o "$video_title" = "@@sd" ] && sort_option="$video_title" && video_title=$(get_all_videos "$sort_option" | i -l 20 -dmenu -i -p "Choose a video")
sort_option="$video_title" && video_title=$(get_all_videos "$sort_option" | DMENU 20 "Choose a video")
} }
[ -n "$video_title" -a "$video_title" != "@@sv" -a "$video_title" != "@@sd" ] && { [ -n "$video_title" -a "$video_title" != "@@sv" -a "$video_title" != "@@sd" ] && {
@ -170,7 +162,7 @@ browse_all_channels() {
category_menu() { category_menu() {
while true; do while true; do
category=$(echo "$(cut -d= -f1 "$CATEGORY_LIST")" | DMENU 12 "Choose a category") local category=$(echo "$(cut -d= -f1 "$CATEGORY_LIST")" | dmenu -i -l 12 -p "Choose a category")
[ -z "$category" ] && return [ -z "$category" ] && return
channel_menu "$category" channel_menu "$category"
@ -178,14 +170,16 @@ category_menu() {
} }
channel_menu() { channel_menu() {
category="$1" local category="$1"
IFS="|" local channels
local channel_name
local IFS="|"
channels=$(sed -n "s/^${category}=\(.*\)$/\1/p" "$CATEGORY_LIST") channels=$(sed -n "s/^${category}=\(.*\)$/\1/p" "$CATEGORY_LIST")
set -- $channels set -- $channels
while true; do while true; do
channel_name=$(printf '%s\n' "$@" | DMENU 20 "Choose a channel") channel_name=$(printf '%s\n' "$@" | dmenu -i -l 20 -p "Choose a channel")
[ -z "$channel_name" ] && return [ -z "$channel_name" ] && return
video_menu "$channel_name" video_menu "$channel_name"
@ -193,14 +187,16 @@ channel_menu() {
} }
video_menu() { video_menu() {
channel_name="$1" local channel_name="$1"
local video_title
local sort_option
while true; do while true; do
video_title=$(get_videos "$channel_name" | DMENU 20 "Choose a video") video_title=$(get_videos "$channel_name" | dmenu -i -l 20 -p "Choose a video")
[ -z "$video_title" ] && return [ -z "$video_title" ] && return
[ "$video_title" = "@@sv" -o "$video_title" = "@@sd" ] && { [ "$video_title" = "@@sv" -o "$video_title" = "@@sd" ] && {
sort_option="$video_title" sort_option="$video_title"
video_title=$(get_videos "$channel_name" "$sort_option" | DMENU 20 "Choose a video") video_title=$(get_videos "$channel_name" "$sort_option" | dmenu -i -l 20 -p "Choose a video")
} }
video_action_menu "$video_title" "$channel_name" video_action_menu "$video_title" "$channel_name"
@ -208,23 +204,23 @@ video_menu() {
} }
video_action_menu() { video_action_menu() {
video_title="$1" local video_title="$1"
channel_name="$2" local channel_name="$2"
while [ -n "$video_title" ] && [ "$video_title" != "@@sv" ] && [ "$video_title" != "@@sd" ]; do while [ -n "$video_title" ] && [ "$video_title" != "@@sv" ] && [ "$video_title" != "@@sd" ]; do
local action
action=$(rofi_action) action=$(rofi_action)
case $action in case $action in
WATCH) Watch)
video_process WATCH "$video_title" "$channel_name" play_video "$video_title" "$channel_name"
;; ;;
DOWNLOAD) Download)
video_process DOWNLOAD "$video_title" "$channel_name" && notify-send "Downloading has finished." download_video "$video_title" "$channel_name" && notify-send "Downloading has finished."
;; ;;
"SEND TO A LIST") "Send To a List")
list_name=$(find "$CUSTOM_LIST_DIR" -maxdepth 1 -type f -exec basename {} \; | DMENU 10 "Choose a list") list_name=$(ls "$CUSTOM_LIST_DIR" | dmenu -i -l 10 -p "Choose a list")
[ -n "$list_name" ] && add_to_list "$video_title" "$channel_name" "$list_name" && [ -n "$list_name" ] && add_to_list "$video_title" "$channel_name" "$list_name"
notify-send ""$video_title" is added to the list: "$list_name""
;; ;;
*) *)
return return
@ -233,25 +229,15 @@ video_action_menu() {
done 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 while true; do
main_choice=$(echo "All Channels\nCategories\nCustom Lists\n$(cut -d= -f1 "$CHANNEL_LIST")" | dmenu -i -l 20 -p "Choose an Option or a Category")
main_choice=$(main_menu)
[ -z "$main_choice" ] && exit [ -z "$main_choice" ] && exit
case "$main_choice" in case "$main_choice" in
*ALL\ CHANNELS*) browse_all_channels ;; "All Channels") browse_all_channels ;;
*CATEGORIES* ) category_menu ;; "Categories") category_menu ;;
*CUSTOM\ LISTS*) custom_list_menu ;; "Custom Lists") custom_list_menu ;;
*) video_menu "$main_choice" ;; *) video_menu "$main_choice" ;;
esac esac
done done