#!/bin/sh

shopt -s nullglob globstar

totp_tmp="${XDG_CACHE_HOME:-$HOME/.cache}/totp-tmp.png"

passwordList() { \
	prefix=${PASSWORD_STORE_DIR-~/.password-store}

	password_files=("$prefix"/**/*.gpg)
	password_files=("${password_files[@]#"$prefix"/}")
	password_files=("${password_files[@]%.gpg}")
	password_files=$(printf '%s\n' "${password_files[@]}")
}

getPasswords() { \
	choice="$(echo "$password_files" | dmenu -i -p "Passwords")"

	[ -z "$choice" ] && exit

	pass -c "$choice"
	notify-send "📋 Password for $choice copied!"
}

getTOTP() {
	choice="$(echo "$password_files" | grep "totp" | dmenu -i -p "OTP")"

	[ -z "$choice" ] && exit

	pass otp -c "$choice"
	notify-send "📋 OTP for $choice copied!"
}

addTOTP() { \
	ifinstalled zbarimg || exit

	name=$(printf "" | dmenu -i -p "What is the account name?")

	[ -z "$name" ] && exit

	notify-send "📷 Please screenshot the QR code"
	maim -m 10 -d 0.1 -s "$totp_tmp"
	zbarimg -q --raw "$totp_tmp" | pass otp insert totp-$name &&
		notify-send "🔒 OTP for totp-$name added!" ||
		notify-send "❌ Failed to add QR code!"
	rm "$totp_tmp"
}

addPassword() { \
	name=$(printf "" | dmenu -i -p "What is the account name?")

	[ -z "$name" ] && exit

	case "$(printf "Random password\\nManual password" | dmenu -i -p "Passwords")" in
		"Random password") addPass=$(tr -dc '\40-\176' < /dev/urandom | tr -d ' ' | head -c $(printf "12\\n32\\n128" | dmenu -i -p "How many characters?") ; echo '') ;;
		"Manual password") addPass=$(dmenu -i -P -p "Enter a password") ;;
		* ) exit
	esac

	[ -z "$addPass" ] && exit

	echo "$addPass" | pass insert "$name" -e
	notify-send "🔒 Password for $name added!"
}

removePassword() { \
	name="$(pass | printf '%s\n' "${password_files[@]}" | dmenu -i -p "Passwords")"

	[ -z "$name" ] && exit

	sure="$(printf "No\\nYes" | dmenu -i -sb "#ac3333" -sf "#111111" -nf "#dddddd" -nb "#111111" -p "Are you sure?")"
	if [ $sure = "Yes" ]
	then
		notify-send "❎ Password for $name removed!"
		pass rm "$name" -f
	else
		notify-send "Canceled"
	fi
}

case "$(printf "View password\\nView TOTP\\nAdd password\\nAdd TOTP\\nRemove password" | dmenu -i -p "Passwords")" in
	"View password") passwordList && getPasswords ;;
	"View TOTP") passwordList && getTOTP ;;
	"Add password") addPassword ;;
	"Add TOTP") addTOTP ;;
	"Remove password") passwordList && removePassword ;;
esac