73 lines
1.3 KiB
Bash
73 lines
1.3 KiB
Bash
|
#!/usr/bin/env bash
|
||
|
|
|
||
|
|
# rofi-ykman
|
||
|
|
# YubiKey OATH utility
|
||
|
|
|
||
|
|
#2019 nukeop
|
||
|
|
|
||
|
|
print_usage() {
|
||
|
|
echo "Usage: rofi-ykman [command]"
|
||
|
|
echo
|
||
|
|
echo "Commands:"
|
||
|
|
echo " clipboard Copy the code to the clipboard. (default)"
|
||
|
|
echo " type Type the code."
|
||
|
|
echo
|
||
|
|
}
|
||
|
|
|
||
|
|
clipboard () {
|
||
|
|
if [ $XDG_SESSION_TYPE == "wayland" ]
|
||
|
|
then
|
||
|
|
wl-copy
|
||
|
|
else
|
||
|
|
xsel --clipboard
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
type () {
|
||
|
|
if [ $XDG_SESSION_TYPE == "wayland" ]
|
||
|
|
then
|
||
|
|
wtype -
|
||
|
|
else
|
||
|
|
xargs xdotool type
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
# shortcuts
|
||
|
|
type="Alt+1"
|
||
|
|
clipboard="Alt+2"
|
||
|
|
|
||
|
|
main() {
|
||
|
|
accounts=$(ykman oath accounts list)
|
||
|
|
|
||
|
|
shortcuts=(
|
||
|
|
-kb-custom-1 "$type"
|
||
|
|
-kb-custom-2 "$clipboard"
|
||
|
|
)
|
||
|
|
prompt="YubiKey OATH"
|
||
|
|
|
||
|
|
account=$(echo "${accounts/, TOTP/\n}" | awk '{ print $0 "\0icon\x1fyubioath"; }' | rofi -dmenu -i -p "$prompt" ${shortcuts[@]})
|
||
|
|
case $? in
|
||
|
|
1) exit ;;
|
||
|
|
10) command=type ;;
|
||
|
|
11) command=clipboard ;;
|
||
|
|
*) command=$default_command ;;
|
||
|
|
esac
|
||
|
|
|
||
|
|
code=$(ykman oath accounts code "$account")
|
||
|
|
IFS=', ' read -r -a code <<< "$code"
|
||
|
|
|
||
|
|
case $command in
|
||
|
|
clipboard) echo -n "${code[-1]}" | clipboard;;
|
||
|
|
type) echo -n "${code[-1]}" | type;;
|
||
|
|
esac
|
||
|
|
}
|
||
|
|
|
||
|
|
default_command="${1:-type}"
|
||
|
|
|
||
|
|
case $default_command in
|
||
|
|
clipboard|type) ;;
|
||
|
|
*) print_usage; exit 1;;
|
||
|
|
esac
|
||
|
|
|
||
|
|
main
|