aniwrapper/db.sh

#!/usr/bin/env bash

[ -z "$XDG_CONFIG_HOME" ] && XDG_CONFIG_HOME="$HOME/.config"

DIR="$XDG_CONFIG_HOME/aniwrapper"
DB='history.sqlite3'

log() {
	printf "%s\n" "$1"
}

sync_search_history() {
	cnt=0
	temp_db="/tmp/aniwrapper_tmp_history.sqlite3"
	while read -r line; do
		anime_name=$(awk -F '|' '{print $2}' <<<"$line")
		res=$(sqlite3 -noheader "$DIR/$DB" <<<"SELECT anime_name FROM search_history WHERE anime_name = '$anime_name'")
		if [[ "${res/ //}" == "" ]]; then
			log "Adding $line to search history..."
			search_date=$(awk -F '|' '{print $3}' <<<"$line")
			sqlite3 "$DIR/$DB" <<<"INSERT INTO search_history(anime_name, search_date) VALUES('$anime_name', '$search_date')"
			if [[ "$?" -ne 0 ]]; then
				log "Error inserting row $line"
			fi
			((++cnt))
		fi
	done <<<"$(sqlite3 -noheader "$temp_db" <<<"SELECT DISTINCT * FROM search_history")"
	log "Inserted $cnt rows into search_history table"
}

sync_watch_history() {
	cnt=0
	temp_db="/tmp/aniwrapper_tmp_history.sqlite3"
	while read -r line; do
		anime_name=$(awk -F '|' '{print $2}' <<<"$line")
		res=$(sqlite3 -noheader "$DIR/$DB" <<<"SELECT anime_name FROM watch_history WHERE anime_name = '$anime_name'")
		if [[ "${res/ //}" == "" ]]; then
			log "Adding $line to watch history..."
			episode_num=$(awk -F '|' '{print $3}' <<<"$line")
			watch_date=$(awk -F '|' '{print $NF}' <<<"$line")
			sqlite3 "$DIR/$DB" <<<"INSERT INTO watch_history(anime_name, episode_number, watch_date) VALUES('$anime_name', '$episode_num', '$watch_date')"
			if [[ "$?" -ne 0 ]]; then
				log "Error inserting row $line"
			fi
			((++cnt))
		fi
	done <<<"$(sqlite3 -noheader "$temp_db" <<<"SELECT DISTINCT * FROM watch_history")"
	log "Inserted $cnt rows into watch_history table"
}

while getopts 'cdrqCs' OPT; do
	case "$OPT" in
		c)
			log "Creating database..."
			sqlite3 "$DIR"/"$DB" <sql/search_history_tbl.sql
			sqlite3 "$DIR"/"$DB" <sql/watch_history_tbl.sql
			log "Created database..."
			;;
		d)
			log "Deleting database..."
			echo "$DIR/$DB"
			rm "$DIR/$DB"
			log "Database deleted..."
			;;

		r)
			log "Deleting database..."
			rm -rf "$DIR"/"$DB"
			mkdir -p "$DIR"
			log "Database deleted..."
			log "Creating database..."
			sqlite3 "$DIR"/"$DB" <sql/search_history_tbl.sql
			sqlite3 "$DIR"/"$DB" <sql/watch_history_tbl.sql
			log "Created database..."
			;;
		q)
			log "Connecting to database..."
			sqlite3 "$DIR/$DB"
			log "Ending connection to databse..."
			;;
		C)
			log "Cleaning up database..."
			stmt="DELETE FROM search_history WHERE anime_name IS NULL or anime_name = ''"
			log "Cleaning up search history..."
			sqlite3 "$DIR/$DB" <<<"$stmt"
			stmt="DELETE FROM watch_history WHERE anime_name IS NULL or anime_name = ''"
			log "Cleaning up watch history..."
			sqlite3 "$DIR/$DB" <<<"$stmt"
			;;
		s)
			printf "%s" "Enter connection string for remote user in the form user@host: "
			read -r connection_str
			printf "%s" "Enter port to connect to remote host with or leave blank for default (22): "
			read -r port
			if [[ ! "${port/ //}" == "" ]]; then
				PORT="$port"
			fi

			if [[ ! "@" != *"$connection_str" ]]; then
				log "Enter in full connection string to remote user in the form: user@host"
				exit 1
			fi

			log "Syncing database with: $connection_str on port $PORT"

			scp -P "$PORT" "$connection_str:$DIR/$DB" "$temp_db"
			if [[ "$?" -ne 0 ]]; then
				printf "%s\n" "Error getting database file from remote host"
				exit 1
			fi
			sync_search_history && sync_watch_history
			;;
		*)
			log "Does not exist or not implemented yet..."
			exit 1
			;;
	esac
done