clean up code some more

This commit is contained in:
ksyasuda 2021-11-25 10:52:59 -08:00
parent 5d4e0f5a95
commit 0135288231

74
ani-cli
View File

@ -29,7 +29,7 @@ c_cyan="\033[1;36m"
c_reset="\033[0m" c_reset="\033[0m"
help_text() { help_text() {
while IFS= read line; do while IFS= read -r line; do
printf "%s\n" "$line" printf "%s\n" "$line"
done <<-EOF done <<-EOF
USAGE: $prog <query> USAGE: $prog <query>
@ -125,8 +125,8 @@ run_stmt() {
## Database Code ## ## Database Code ##
##################### #####################
check_db() {
# Return number of matches for anime/episode in db # Return number of matches for anime/episode in db
check_db() {
if [[ "$2" == "search" ]]; then if [[ "$2" == "search" ]]; then
stmt="SELECT DISTINCT COUNT(*) \ stmt="SELECT DISTINCT COUNT(*) \
FROM search_history \ FROM search_history \
@ -143,8 +143,8 @@ check_db() {
fi fi
} }
update_date() {
# updates search/watch date for passed in anime # updates search/watch date for passed in anime
update_date() {
datetime=$(date +'%Y-%m-%d %H:%M:%S') datetime=$(date +'%Y-%m-%d %H:%M:%S')
stmt="" stmt=""
if [[ "$2" == "search" ]]; then if [[ "$2" == "search" ]]; then
@ -161,14 +161,15 @@ update_date() {
insert_history() { insert_history() {
# inserts into search/watch history db # inserts into search/watch history db
# check the anime_name/id # check the anime_name/id
check_anime_name "$1"
if [[ $? -ne 0 ]]; then if ! check_anime_name "$1"; then
log "ERROR: Anime name is none... exiting" log "ERROR: Anime name is none... exiting"
return 1 return 1
fi fi
datetime=$(date +'%Y-%m-%d %H:%M:%S') datetime=$(date +'%Y-%m-%d %H:%M:%S')
check_db "$@" check_db "$@"
if [[ $? -gt 0 ]]; then res="$?"
if [[ "$res" -gt 0 ]]; then
if [[ "$2" == "search" ]]; then if [[ "$2" == "search" ]]; then
log "Already in search db... Updating search_date" log "Already in search db... Updating search_date"
else else
@ -197,8 +198,7 @@ sync_search_history() {
if [[ "${res/ //}" == "" ]]; then if [[ "${res/ //}" == "" ]]; then
search_date=$(awk -F '|' '{print $3}' <<<"$line") search_date=$(awk -F '|' '{print $3}' <<<"$line")
log "Adding ($anime_name|$search_date) to search history..." log "Adding ($anime_name|$search_date) to search history..."
sqlite3 "$HISTORY_DB" <<<"INSERT INTO search_history(anime_name, search_date) VALUES('$anime_name', '$search_date')" if ! sqlite3 "$HISTORY_DB" <<<"INSERT INTO search_history(anime_name, search_date) VALUES('$anime_name', '$search_date')"; then
if [[ "$?" -ne 0 ]]; then
err "Error inserting row $line" err "Error inserting row $line"
fi fi
((++cnt)) ((++cnt))
@ -225,8 +225,7 @@ sync_watch_history() {
log "NOT IN DB" log "NOT IN DB"
watch_date=$(awk -F '|' '{print $NF}' <<<"$ep") watch_date=$(awk -F '|' '{print $NF}' <<<"$ep")
log "Adding ($anime_name|$episode_num|$watch_date) to watch history..." log "Adding ($anime_name|$episode_num|$watch_date) to watch history..."
sqlite3 "$HISTORY_DB" <<<"INSERT INTO watch_history(anime_name, episode_number, watch_date) VALUES('$anime_name', '$episode_num', '$watch_date')" if ! sqlite3 "$HISTORY_DB" <<<"INSERT INTO watch_history(anime_name, episode_number, watch_date) VALUES('$anime_name', '$episode_num', '$watch_date')"; then
if [[ "$?" -ne 0 ]]; then
err "Error inserting row $ep" err "Error inserting row $ep"
fi fi
((++cnt)) ((++cnt))
@ -278,18 +277,18 @@ get_search_query() {
anime_selection() { anime_selection() {
# Select anime from query results # Select anime from query results
search_results=$* search_results=$*
menu_format_string='[%d] %s\n' # menu_format_string='[%d] %s\n'
menu_format_string_c1="$c_blue[$c_cyan%d$c_blue] $c_reset%s\n" # menu_format_string_c1="$c_blue[$c_cyan%d$c_blue] $c_reset%s\n"
menu_format_string_c2="$c_blue[$c_cyan%d$c_blue] $c_yellow%s$c_reset\n" # menu_format_string_c2="$c_blue[$c_cyan%d$c_blue] $c_yellow%s$c_reset\n"
count=1 count=1
menu=() menu=()
res=() res=()
while read anime_id; do while read -r anime_id; do
# alternating colors for menu # alternating colors for menu
[ $((count % 2)) -eq 0 ] && # [ $((count % 2)) -eq 0 ] &&
menu_format_string=$menu_format_string_c1 || # menu_format_string=$menu_format_string_c1 ||
menu_format_string=$menu_format_string_c2 # menu_format_string=$menu_format_string_c2
menu+="$count. $anime_id\n" menu+="$count. $anime_id\n"
idx=$((count - 1)) idx=$((count - 1))
@ -322,7 +321,7 @@ anime_selection() {
rofi -dmenu -config "$CFG_DIR/${ROFI_CFG}" \ rofi -dmenu -config "$CFG_DIR/${ROFI_CFG}" \
-a "$searched" \ -a "$searched" \
-l 12 -i -p "Enter number:") -l 12 -i -p "Enter number:")
[ "$?" -ne 0 ] && return 1 [ -z "$user_input" ] && return 1
choice=$(printf '%s\n' "$user_input" | awk '{print $1}') choice=$(printf '%s\n' "$user_input" | awk '{print $1}')
# Remove period after number # Remove period after number
@ -345,8 +344,8 @@ anime_selection() {
# Select respective anime_id # Select respective anime_id
count=1 count=1
while read anime_id; do while read -r anime_id; do
if [ $count -eq $choice ]; then if [ "$count" -eq "$choice" ]; then
selection_id=$anime_id selection_id=$anime_id
break break
fi fi
@ -357,7 +356,7 @@ anime_selection() {
[ -z "$selection_id" ] && die "Invalid number entered" [ -z "$selection_id" ] && die "Invalid number entered"
read last_ep_number <<-EOF read -r last_ep_number <<-EOF
$(search_eps "$selection_id") $(search_eps "$selection_id")
EOF EOF
} }
@ -438,7 +437,7 @@ open_episode() {
fi fi
fi fi
log "Getting data for episode %d\n" $episode log "Getting data for episode $episode"
insert_history "$anime_id" "$episode" insert_history "$anime_id" "$episode"
@ -459,14 +458,14 @@ open_episode() {
;; ;;
esac esac
if [ $is_download -eq 0 ]; then if [ "$is_download" -eq 0 ]; then
# setsid -f $player_fn --http-header-fields="Referer: $dpage_url" "$video_url" >/dev/null 2>&1 # setsid -f $player_fn --http-header-fields="Referer: $dpage_url" "$video_url" >/dev/null 2>&1
nohup $player_fn --http-header-fields="Referer: $dpage_url" "$video_url" >/dev/null 2>&1 & nohup $player_fn --http-header-fields="Referer: $dpage_url" "$video_url" >/dev/null 2>&1 &
else else
log "Downloading episode $episode ..." log "Downloading episode $episode ..."
log "$video_url" log "$video_url"
# add 0 padding to the episode name # add 0 padding to the episode name
episode=$(printf "%03d" $episode) episode=$(printf "%03d" "$episode")
{ {
cd "${ddir/ //}" || die "Could not enter directory $ddir" cd "${ddir/ //}" || die "Could not enter directory $ddir"
mkdir -p "$anime_id" || die "Could not create directory" mkdir -p "$anime_id" || die "Could not create directory"
@ -551,16 +550,18 @@ case $scrape in
get_search_query "$*" get_search_query "$*"
search_results=$(search_anime "$query") search_results=$(search_anime "$query")
[ -z "$search_results" ] && die "No search results found" [ -z "$search_results" ] && die "No search results found"
anime_selection "$search_results" if ! anime_selection "$search_results"; then
[ $? -ne 0 ] && die "No anime selection found" die "No anime selection found"
fi
episode_selection episode_selection
;; ;;
history) history)
stmt="SELECT DISTINCT anime_name FROM watch_history ORDER BY watch_date DESC" stmt="SELECT DISTINCT anime_name FROM watch_history ORDER BY watch_date DESC"
search_results=$(printf "%s\n" "$stmt" | sqlite3 -noheader "$HISTORY_DB") search_results=$(printf "%s\n" "$stmt" | sqlite3 -noheader "$HISTORY_DB")
[ -z "$search_results" ] && die "History is empty" [ -z "$search_results" ] && die "History is empty"
anime_selection "${search_results[@]}" if ! anime_selection "${search_results[@]}"; then
[ $? -ne 0 ] && die "No anime selection found" die "No anime selection found"
fi
log "SELECTION: $selection_id" log "SELECTION: $selection_id"
stmt="SELECT episode_number \ stmt="SELECT episode_number \
@ -572,14 +573,15 @@ case $scrape in
log "Most recently watched episode: $ep_choice_start" log "Most recently watched episode: $ep_choice_start"
;; ;;
playlist) playlist)
lines=$(cat "$playlist_file" | wc -l) lines=$(wc -l < "$playlist_file" )
log "Num lines in playlist: " "$lines" log "Num lines in playlist: " "$lines"
if [[ "$lines" -eq 0 ]]; then if [[ "$lines" -eq 0 ]]; then
get_search_query "$*" get_search_query "$*"
search_results=$(search_anime "$query") search_results=$(search_anime "$query")
[ -z "$search_results" ] && die "No search results found" [ -z "$search_results" ] && die "No search results found"
anime_selection "$search_results" if ! anime_selection "$search_results"; then
[ $? -ne 0 ] && die "No anime selection found" die "No anime selection found"
fi
episode_selection episode_selection
else else
line=($(sed '1q;d' "$playlist_file")) line=($(sed '1q;d' "$playlist_file"))
@ -587,10 +589,10 @@ case $scrape in
die "Something went wrong with the playlist file... exiting" die "Something went wrong with the playlist file... exiting"
fi fi
selection_id="${line[0]}" selection_id="${line[0]}"
episodes=($selection_id) episodes=("$selection_id")
ep_choice_start="${line[1]}" ep_choice_start="${line[1]}"
ep_choice_end="" ep_choice_end=""
read last_ep_number <<-EOF read -r last_ep_number <<-EOF
$(search_eps "$selection_id") $(search_eps "$selection_id")
EOF EOF
[ "$VERBOSE" -eq 1 ] && printf "Anime: %s Episode: %d\n" "$episodes" "$ep_choice_start" [ "$VERBOSE" -eq 1 ] && printf "Anime: %s Episode: %d\n" "$episodes" "$ep_choice_start"
@ -682,7 +684,7 @@ while :; do
printf "$c_blue[${c_cyan}%s$c_blue] $c_red%s$c_reset\n" "q" "exit" 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_blue}Enter choice:${c_green} "
printf "$c_reset" printf "$c_reset"
read choice read -r choice
printf "$c_reset" printf "$c_reset"
case $choice in case $choice in
@ -694,8 +696,8 @@ while :; do
;; ;;
s) s)
printf "${c_blue}Choose episode $c_cyan[1-%d]$c_reset:$c_green " $last_ep_number printf "${c_blue}Choose episode $c_cyan[1-%d]$c_reset:$c_green " "$last_ep_number"
read episode read -r episode
printf "$c_reset" printf "$c_reset"
[ "$episode" -eq "$episode" ] 2>/dev/null || die "Invalid number entered" [ "$episode" -eq "$episode" ] 2>/dev/null || die "Invalid number entered"
;; ;;