refactor code

This commit is contained in:
ksyasuda 2022-01-08 14:53:56 -08:00
parent 3b64d74bf2
commit aaaad7d986

View File

@ -27,6 +27,30 @@ theme=default
############# #############
# Functions # # Functions #
############# #############
help_text() {
while IFS= read -r line; do
printf "%s\n" "$line"
done <<< "
Usage:
aniwrapper [-cdhvq] [-t <theme> or -T <config_path>]
aniwrapper [-cv] [-t <theme> or -T <config_path>] -f <path_to_directory>
aniwrapper -Q <query>
aniwrapper -C
Options:
-h show this help text
-q enable quality selection
-c enable command-line mode (rofi disabled)
-d download episode in command-line mode
-C connect to history database
-t <aniwrapper (default)|dracula|fancy|flamingo|material|onedark> change rofi theme
-T <config_path> specify custom rofi theme
-f <path_to_directory> (no trailing slash) specify starting directory for play for file mode
-Q <query> query the history database
"
exit 0
}
logger() { logger() {
if [[ "$VERBOSE" -eq 1 ]]; then if [[ "$VERBOSE" -eq 1 ]]; then
printf "%s\n" "$*" printf "%s\n" "$*"
@ -58,13 +82,6 @@ run() {
fi fi
} }
create_default_download() {
# make sure download directory exists
if [ ! -d "$DEFAULT_DOWNLOAD" ]; then
mkdir -p "$DEFAULT_DOWNLOAD"
fi
}
get_quality() { get_quality() {
if [ "$IS_ROFI" -eq 1 ]; then if [ "$IS_ROFI" -eq 1 ]; then
selection=$(rofi -dmenu -config "$CFG_FILE" \ selection=$(rofi -dmenu -config "$CFG_FILE" \
@ -87,9 +104,7 @@ generate_span() {
printf "%s\n" "$span" printf "%s\n" "$span"
} }
######## parse_args() {
# Main #
########
while getopts 'vhqcdf:-:t:T:CQ:' OPT; do while getopts 'vhqcdf:-:t:T:CQ:' OPT; do
case "$OPT" in case "$OPT" in
h) h)
@ -168,9 +183,9 @@ while getopts 'vhqcdf:-:t:T:CQ:' OPT; do
esac esac
done done
logger "CONFIG DIR: $CFG_DIR" }
logger "ROFI CFG: $CFG_FILE"
check_flags() {
# Check if command-line flag is set # Check if command-line flag is set
if [[ "$IS_ROFI" -eq 0 ]] && [[ "$IS_DOWNLOAD" -eq 0 ]]; then if [[ "$IS_ROFI" -eq 0 ]] && [[ "$IS_DOWNLOAD" -eq 0 ]]; then
run -c run -c
@ -189,10 +204,12 @@ elif [[ "$IS_ROFI" -eq 1 ]] && [[ "$IS_PLAY_FROM_FILE" -eq 1 ]] && [[ "$IS_CUSTO
run "-f$play_path" run "-f$play_path"
exit $? exit $?
fi fi
}
# ------------------------------------------------------------------------------- ########
# Main # Main #
# ------------------------------------------------------------------------------- ########
main() {
choice=$(echo "${options[@]}" | rofi -dmenu -only-match -sep '|' \ choice=$(echo "${options[@]}" | rofi -dmenu -only-match -sep '|' \
-config "$CFG_FILE" -l 6 -i -p "Aniwrapper") -config "$CFG_FILE" -l 6 -i -p "Aniwrapper")
@ -217,7 +234,6 @@ case "$selection" in
-l 1 -p "Enter download dir:") -l 1 -p "Enter download dir:")
# if dl_dir is none set to current directory # if dl_dir is none set to current directory
[ "$dl_dir" == "" ] && dl_dir="$DEFAULT_DOWNLOAD" [ "$dl_dir" == "" ] && dl_dir="$DEFAULT_DOWNLOAD"
create_default_download
run -d "$dl_dir" run -d "$dl_dir"
;; ;;
3.) 3.)
@ -234,47 +250,28 @@ case "$selection" in
# --------------------------------------------------------------------------- # ---------------------------------------------------------------------------
logger "Play from file selected" logger "Play from file selected"
IS_PLAY_FROM_FILE=1 IS_PLAY_FROM_FILE=1
msg="Provide a valid path to a directory or leave blank to go with the default: $HOME/Videos/sauce/" span=$(printf '%s\n%s\n' "$(generate_span "Provide a valid path to a directory or leave blank to go with the default: $HOME/Videos/sauce/")" "$(generate_span "The program will begin searching for media files from the supplied directory")")
msg="$(printf '%s\n%s\n' "$msg" "The program will begin searching for media files from the supplied directory")"
span=$(generate_span "$msg")
play_dir=$(rofi -dmenu -config "$CFG_FILE" \ play_dir=$(rofi -dmenu -config "$CFG_FILE" \
-l 1 -mesg "$span" -p "Enter path to starting directory:") -l 1 -mesg "$span" -p "Enter path to starting directory:")
if [ -z "$play_dir" ]; then if [ -z "$play_dir" ]; then
create_default_download mkdir -p "$DEFAULT_DOWNLOAD" || seppuku "error creating default download directory"
run -f"$DEFAULT_DOWNLOAD" run -f"$DEFAULT_DOWNLOAD"
else else
run -f"$play_dir" run -f"$play_dir"
fi fi
exit $?
;; ;;
5.) 5.)
logger "Sync history database" logger "Sync history database"
username=$(rofi -dmenu -config "$CFG_FILE" \ roficmd="rofi -dmenu -config $CFG_FILE -l 0 -p"
-l 1 -p "Enter the username of the remote user:") username=$($roficmd "Enter the username of the remote user:")
if [[ -z "$username" ]] || [[ "$username" == "" ]]; then [ -z "$username" ] && seppuku "No username provided... exiting"
logger "No username provided... exiting" host=$($roficmd "Enter the host for the remote machine (eg 192.168.1.99):")
exit 1 [ -z "$host" ] && seppuku "No host provided... exiting"
fi port=$($roficmd "Enter in the ssh port for remote machine or leave blank for default [22]:")
host=$(rofi -dmenu -config "$CFG_FILE" \ [ -z "$port" ] && port=22
-l 1 -p "Enter the host for the remote machine (eg 192.168.1.99):") keypath=$($roficmd "Enter path to private key (leave blank if not needed or if unsure):")
if [[ -z "$host" ]] || [[ "$host" == "" ]]; then if ! printf "%s\n%s\n%d\n%s\n" "$username" "$host" "$port" "$keypath" | run -s; then
logger "No host provided... exiting"
exit 1
fi
port=$(rofi -dmenu -config "$CFG_FILE" \
-l 1 -p "Enter in the ssh port for remote machine or leave blank for default [22]:")
if [[ -z "$port" ]] || [[ "$port" == "" ]]; then
port=22
fi
keypath=$(rofi -dmenu -config "$CFG_FILE" \
-l 1 -p "Enter path to private key (leave blank if not needed or if unsure):")
if [[ -z "$keypath" ]]; then
printf "%s\n%s\n%d\n%s\n" "$username" "$host" "$port" "" | run -s
else
printf "%s\n%s\n%d\n%s\n" "$username" "$host" "$port" "$keypath" | run -s
fi
if [[ "$?" -ne 0 ]]; then
logger "Aniwrapper was unable to sync the databases..." logger "Aniwrapper was unable to sync the databases..."
exit 1 exit 1
else else
@ -293,3 +290,10 @@ case "$selection" in
exit 1 exit 1
;; ;;
esac esac
}
parse_args "$@"
logger "CONFIG DIR: $CFG_DIR"
logger "ROFI CFG: $CFG_FILE"
check_flags
main