#!/bin/dash

DATA_DIR="$HOME/.cache/youtube_channels"
DOWNLOAD_DIR="$HOME/ytvideos"
CATEGORY_LIST="$HOME/.local/share/categories.txt"
CHANNEL_LIST="$HOME/.local/share/channels.txt"
CUSTOM_LIST_DIR="$DATA_DIR/custom_lists"

mkdir -p "$DATA_DIR" "$DOWNLOAD_DIR" "$CUSTOM_LIST_DIR"

[ "$(find "$DATA_DIR" -maxdepth 1 -type f -name "*.tsv" 2>/dev/null)" ] || {
	notify-send "You need to run "channelrefresh" script first."
        exit 1
}

DMENU() {
    dmenu -i -l "$1" -p "$2"
}

sort_videos() {
  data_file="$1"
  sort_option="$2"

  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
}

get_videos() {
  channel_name="$1"
  sort_option="$2"
  data_file="$DATA_DIR/$channel_name.tsv"
  sort_videos "$data_file" "$sort_option"
}

video_url() {
  channel_name="$1"
  video_title="$2"
  data_file="$DATA_DIR/$channel_name.tsv"
  grep -F "$video_title" "$data_file" | cut -f2
}

rofi_action() {
  echo "WATCH\nDOWNLOAD\nSEND TO A LIST" | DMENU 3 "Choose an action for the video"
}

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"
}

list_video_action() {
  echo "WATCH\nDOWNLOAD\nDELETE" | DMENU 3 "Choose an action for the video"
}

add_to_list() {
  video_title="$1"
  channel_name="$2"
  list_name="$3"
  echo "$channel_name: $video_title" >> "$CUSTOM_LIST_DIR/$list_name"
}

custom_list_menu() {
  while true; do
    list="$(rofi_custom_list_action)"
    [ -z "$list" ] && return
    case "$list" in
      *CREATE*)
        new_list=$(echo "" | DMENU 0 "Enter the name of the new list")
        [ -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
}

custom_list_video_menu() {
  list_name="$1"
  while true; do
    video_info=$(DMENU 20 "Choose a video" < "$CUSTOM_LIST_DIR/$list_name")
    [ -z "$video_info" ] && return
    channel_name="${video_info%%: *}"
    video_title="${video_info##*: }"
    custom_list_video_action_menu "$video_title" "$channel_name" "$list_name"
  done
}

custom_list_video_action_menu() {
  video_title="$1"
  channel_name="$2"
  list_name="$3"

  action=$(list_video_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."
      ;;
    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
}

video_process() {
    action="$1"
    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
}

get_all_videos() {
  sort_option="$1"
  all_videos_file="$DATA_DIR/all_videos.tsv"

  rm -f "$all_videos_file"
  while IFS= read -r line
  do
    channel_name="${line%%=*}"
    cat "$DATA_DIR/$channel_name.tsv" >> "$all_videos_file"
  done < "$CHANNEL_LIST"

  sort_videos "$all_videos_file" "$sort_option"
}

browse_all_channels() {
  while video_title=$(get_all_videos | DMENU 20 "Choose a video or enter @@sv or @@sd"); do
    [ -z "$video_title" ] && break
    [ "$video_title" = "@@sv" ] || [ "$video_title" = "@@sd" ] && video_title=$(get_all_videos "$video_title" | DMENU 20 "Choose a video")

    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
}

category_menu() {
  while true; do
    category="$(cut -d= -f1 "$CATEGORY_LIST" | DMENU 12 "Choose a category")"
    [ -z "$category" ] && return

    channel_menu "$category"
  done
}

channel_menu() {
  category="$1"
  IFS="|"

  channels=$(sed -n "s/^${category}=\(.*\)$/\1/p" "$CATEGORY_LIST")
  set -- $channels

  while true; do
    channel_name=$(printf '%s\n' "$@" | DMENU 20 "Choose a channel")
    [ -z "$channel_name" ] && return

    video_menu "$channel_name"
  done
}

video_menu() {
  channel_name="$1"

  while true; do
    video_title=$(get_videos "$channel_name" | DMENU 20 "Choose a video")
    [ -z "$video_title" ] && return
    [ "$video_title" = "@@sv" ] || [ "$video_title" = "@@sd" ] && {
      sort_option="$video_title"
      video_title=$(get_videos "$channel_name" "$sort_option" | DMENU 20 "Choose a video")
    }

    video_action_menu "$video_title" "$channel_name"
  done
}

video_action_menu() {
  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

  main_choice=$(main_menu)

  [ -z "$main_choice" ] && exit

  case "$main_choice" in
    *ALL\ CHANNELS*) browse_all_channels       ;;
    *CATEGORIES*   ) category_menu             ;;
    *CUSTOM\ LISTS*) custom_list_menu          ;;
                  *) video_menu "$main_choice" ;;
  esac
done