change most menus to rofi and add history db

all menus (except the last one) use rofi menu additionally all anime's selected in a search or episodes selected to watch will be inserted into a local sqlite3 database [history.sqlite3]
2026-05-28 00:55:11 -07:00 · 2021-10-29 22:49:52 -07:00
parent e5c2a1796e
commit 9254956775
5 changed files with 234 additions and 92 deletions
@@ -0,0 +1,3 @@
+*.sqlite3
+.git
+.vscode
@@ -6,6 +6,7 @@ player_fn="mpv"

 prog="ani-cli"
 logfile="${XDG_CACHE_HOME:-$HOME/.cache}/ani-hsts"
+history_db="history.sqlite3"

 c_red="\033[1;31m"
 c_green="\033[1;32m"
@@ -15,6 +16,7 @@ c_magenta="\033[1;35m"
 c_cyan="\033[1;36m"
 c_reset="\033[0m"

+rofi_cmd="rofi -config ~/SudacodeRice/rofi/rofidmenu.rasi -dmenu -l 10 -i -p 'Enter Choice:'"

 help_text() {
 	while IFS= read line; do
@@ -27,7 +29,6 @@ help_text () {
 	EOF
 }

-
 die() {
 	printf "$c_red%s$c_reset\n" "$*" >&2
 	exit 1
@@ -95,11 +96,69 @@ dep_ch () {
 	done
 }

+check_db() {
+	echo "$1 $2"
+	if [[ "$2" == "search" ]]; then
+		stmt="SELECT DISTINCT COUNT(*) FROM search_history WHERE name = '$1'"
+		res=$(printf "%s\n" "$stmt" | sqlite3 "$history_db")
+		echo "$res"
+		return "$res"
+	else
+		stmt="SELECT DISTINCT COUNT(*) FROM watch_history WHERE anime_name = '$1' AND episode_number = $2"
+		res=$(printf "%s\n" "$stmt" | sqlite3 "$history_db")
+		echo "$res"
+		return "$res"
+	fi
+}
+
+update_date() {
+	if [[ "$2" == "search" ]]; then
+		curdate=$(date +'%Y-%m-%d')
+		stmt="UPDATE search_history SET search_date = '$curdate' WHERE name = '$1'" && return 0 || return 1
+	else
+		curdate=$(date +'%Y-%m-%d')
+		stmt="UPDATE watch_history SET watch_date = '$curdate' WHERE anime_name = '$1' AND episode_number = $2" && return 0 || return 1
+	fi
+}
+
+insert_history() {
+	curdate=$(date +'%Y-%m-%d')
+	check_db "$@"
+	num=$?
+	# echo "$num"
+	if [[ "$num" -gt 0 ]]; then
+		if [[ "$2" == "search" ]]; then
+			printf "%s\n" "Already in search db... Updating search_date"
+		else
+			printf "%s\n" "Already in search db... Updating watch_date"
+		fi
+		update_date "$@"
+	else
+		if [[ "$2" == "search" ]]; then
+			stmt="INSERT INTO search_history(name, search_date) VALUES('$1', '$curdate')"
+			printf "%s\n" "$stmt" | sqlite3 "$history_db"
+		else
+			stmt="INSERT INTO watch_history(anime_name, episode_number, watch_date) VALUES('$1', '$2', '$curdate')"
+			printf "%s\n" "$stmt" | sqlite3 "$history_db"
+		fi
+	fi
+}
+
 # get query
 get_search_query() {
+	stmt="SELECT DISTINCT name FROM search_history"
+	cnt_stmt="SELECT DISTINCT COUNT(*) FROM search_history"
+	hist=$(echo "$stmt" | sqlite3 "$history_db")
+	cnt=$(printf "%s\n" "$cnt_stmt" | sqlite3 "$history_db")
 	if [ -z "$*" ]; then
-		printf "Search Anime: "
-		read -r query
+		query=$(printf "%s\n" "${hist[@]}" | rofi -dmenu -l "$cnt" -p "Search Anime:" -config ~/SudacodeRice/rofi/rofidmenu.rasi)
+		# if [[ "$query" != "" ]]; then
+		# 	insert_history "$query" "search"
+		# else
+		# 	echo 'Query empty... Skipping insert'
+		# fi
+		# printf "Search Anime: "
+		# read -r query
 	else
 		query=$*
 	fi
@@ -119,21 +178,29 @@ anime_selection () {
 	menu_format_string_c2="$c_blue[$c_cyan%d$c_blue] $c_yellow%s$c_reset\n"

 	count=1
+	menu=()
 	while read anime_id; do
 		# alternating colors for menu
 		[ $((count % 2)) -eq 0 ] &&
 			menu_format_string=$menu_format_string_c1 ||
 			menu_format_string=$menu_format_string_c2

-		printf "$menu_format_string" "$count" "$anime_id"
+		# printf "$menu_format_string" "$count" "$anime_id"
+		menu+="$count - $anime_id\n"
 		count=$((count + 1))
 	done <<-EOF
 		$search_results
 	EOF

+	user_input=$(printf "${menu[@]}" | rofi -dmenu -config ~/SudacodeRice/rofi/rofidmenu.rasi -l 10 -i -p "Enter number:")
+
+	choice=$(printf '%s\n' "$user_input" | awk '{print $1}')
+	name=$(printf '%s\n' "$user_input" | awk '{print $NF}')
+	insert_history "$name" "search"
+	echo "Number Chosen: $choice"
 	# User input
-	printf "$c_blue%s$c_green" "Enter number: "
-	read choice
+	# printf "$c_blue%s$c_green" "Enter number: "
+	# read choice
 	printf "$c_reset"

 	# Check if input is a number
@@ -166,8 +233,12 @@ episode_selection () {
 	[ $is_download -eq 1 ] &&
 		printf "Range of episodes can be specified: start_number end_number\n"

-	printf "${c_blue}Choose episode $c_cyan[1-%d]$c_reset:$c_green " $last_ep_number
-	read ep_choice_start ep_choice_end
+	# printf "${c_blue}Choose episode $c_cyan[1-%d]$c_reset:$c_green " $last_ep_number
+	choice=$(rofi -config ~/SudacodeRice/rofi/rofidmenu.rasi -dmenu -l 1 -i -p "Choose episode:")
+	ep_choice_start=$(printf '%s\n' "$choice" | awk '{print $1}')
+	ep_choice_end=$(printf '%s\n' "$choice" | awk '{print $2}')
+	# echo "$ep_choice_start $ep_choice_end"
+	# read ep_choice_start ep_choice_end
 	printf "$c_reset"

 }
@@ -180,13 +251,16 @@ open_episode () {
 	printf '\x1B[2J\x1B[1;1H'
 	if [ $episode -lt 1 ] || [ $episode -gt $last_ep_number ]; then
 		err "Episode out of range"
-		printf "${c_blue}Choose episode $c_cyan[1-%d]$c_reset:$c_green " $last_ep_number
-		read episode
+		# printf "${c_blue}Choose episode $c_cyan[1-%d]$c_reset:$c_green " $last_ep_number
+		# read episode
+		episode=$(rofi -dmenu -l 1 -p "${c_blue}Choose episode")
 		printf "$c_reset"
 	fi

 	printf "Getting data for episode %d\n" $episode

+	insert_history "$anime_id" "$episode"
+
 	dpage_url=$(get_dpage_link "$anime_id" "$episode")
 	video_url=$(get_links "$dpage_url")

@@ -200,7 +274,8 @@ open_episode () {
 				s/^[^"]*"([^"]*)" \+ '\''([^'\'']*).*/https:\1\2\&dl=1/p
 				q
 				}
-			');;
+			')
+		;;
 	esac

 	if [ $is_download -eq 0 ]; then
@@ -235,8 +310,9 @@ dep_ch "$player_fn" "curl" "sed" "grep"

 # option parsing
 is_download=0
+list_history=0
 scrape=query
-while getopts 'hdH' OPT; do
+while getopts 'hdHl' OPT; do
 	case $OPT in
 	h)
 		help_text
@@ -248,6 +324,9 @@ while getopts 'hdH' OPT; do
 	H)
 		scrape=history
 		;;
+	l)
+		list_history=1
+		;;
 	esac
 done
 shift $((OPTIND - 1))
@@ -256,8 +335,18 @@ shift $((OPTIND - 1))
 # main #
 ########

+if [[ "$list_history" -eq 1 ]]; then
+	stmt="SELECT DISTINCT name FROM search_history"
+	cnt_stmt="SELECT DISTINCT COUNT(*) FROM search_history"
+	hist=$(echo "$stmt" | sqlite3 "$history_db")
+	cnt=$(printf "%s\n" "$cnt_stmt" | sqlite3 "$history_db")
+	printf "%s\n" "${hist[@]}" | rofi -config ~/SudacodeRice/rofi/rofidmenu.rasi -dmenu -l "$cnt" -i -p "Search History"
+	exit 0
+fi
+
 case $scrape in
 query)
+
 	get_search_query "$*"
 	search_results=$(search_anime "$query")
 	[ -z "$search_results" ] && die "No search results found"
@@ -272,7 +361,6 @@ case $scrape in
 	;;
 esac

-
 { # checking input
 	[ "$ep_choice_start" -eq "$ep_choice_start" ] 2>/dev/null || die "Invalid number entered"
 	episodes=$ep_choice_start
@@ -288,12 +376,12 @@ esac
 grep -q -w "${selection_id}" "$logfile" ||
 	printf "%s\t%d\n" "$selection_id" $((episode + 1)) >>"$logfile"

-for ep in $episodes
-do
+for ep in $episodes; do
 	open_episode "$selection_id" "$ep"
 done
 episode=${ep_choice_end:-$ep_choice_start}

+choice=''
 while :; do
 	printf "\n${c_green}Currently playing %s episode ${c_cyan}%d/%d\n" "$selection_id" $episode $last_ep_number
 	printf "$c_blue[${c_cyan}%s$c_blue] $c_yellow%s$c_reset\n" "n" "next episode"
@@ -302,7 +390,11 @@ while :; do
 	printf "$c_blue[${c_cyan}%s$c_blue] $c_magenta%s$c_reset\n" "r" "replay current episode"
 	printf "$c_blue[${c_cyan}%s$c_blue] $c_red%s$c_reset\n" "q" "exit"
 	printf "${c_blue}Enter choice:${c_green} "
+	printf "$c_reset"
 	read choice
+
+	# choice=$(printf '%s\n' "${args[@]}" | rofi -dmenu -l 8 -i -p "Enter choice:")
+	# choice=$(printf '%s\n' "$choice" | awk '{print $1}')
 	printf "$c_reset"
 	case $choice in
 	n)
@@ -312,7 +404,8 @@ while :; do
 		episode=$((episode - 1))
 		;;

-		s)	printf "${c_blue}Choose episode $c_cyan[1-%d]$c_reset:$c_green " $last_ep_number
+	s)
+		printf "${c_blue}Choose episode $c_cyan[1-%d]$c_reset:$c_green " $last_ep_number
 		read episode
 		printf "$c_reset"
 		[ "$episode" -eq "$episode" ] 2>/dev/null || die "Invalid number entered"
@@ -321,12 +414,12 @@ while :; do
 	r) ;;

 	q)
-			break;;
+		break
+		;;

 	*)
 		die "invalid choice"
 		;;
 	esac
-
 	open_episode "$selection_id" "$episode"
 done
@@ -0,0 +1,33 @@
+#!/usr/bin/env bash
+
+DB='history.sqlite3'
+
+while getopts 'cdr' OPT; do
+	case "$OPT" in
+	c)
+		printf "%s\n" "Creating database..."
+		sqlite3 "$DB" <sql/anime_search_history.sql
+		sqlite3 "$DB" <sql/watch_history.sql
+		printf "%s\n" "Created database..."
+		;;
+	d)
+		printf "%s\n" "Deleting database..."
+		rm -rf "$DB"
+		printf "%s\n" "Database deleted..."
+		;;
+
+	r)
+		printf "%s\n" "Deleting database..."
+		rm -rf "$DB"
+		printf "%s\n" "Database deleted..."
+		printf "%s\n" "Creating database..."
+		sqlite3 "$DB" <sql/anime_search_history.sql
+		sqlite3 "$DB" <sql/watch_history.sql
+		printf "%s\n" "Created database..."
+		;;
+	*)
+		printf "%s\n" "Don't get here"
+		exit 1
+		;;
+	esac
+done
@@ -0,0 +1,6 @@
+CREATE TABLE search_history (
+    id integer PRIMARY KEY autoincrement,
+    name varchar(200) NOT NULL,
+    search_date date NOT NULL
+);
+
@@ -0,0 +1,7 @@
+CREATE TABLE watch_history (
+    id integer PRIMARY KEY AUTOINCREMENT,
+    anime_name varchar(200) NOT NULL,
+    episode_number integer NOT NULL,
+    watch_date date
+);
+