path: root/rofi-pass

#!/usr/bin/env bash

# rofi-pass
# (c) 2015 Rasmus Steinke <rasi@xssn.at>
basecommand="$0"

# set default settings
_rofi () {
    rofi -no-auto-select -i "$@"
}

# We expect to find these fields in pass(1)'s output
URL_field='url'
USERNAME_field='user'
AUTOTYPE_field='autotype'

default_autotype="user :tab pass"
delay=2
default_do='menu' # menu, copyPass, typeUser, typePass, copyUser, copyUrl, viewEntry, typeMenu, actionMenu, copyMenu, openUrl
auto_enter='false'
notify='false'
help_color=""
clip=primary
default_user="$(whoami)"
default_user2=john_doe
password_length=12

# default shortcuts
autotype="Alt+1"
type_user="Alt+2"
type_pass="Alt+3"
open_url="Alt+4"
copy_name="Alt+u"
copy_url="Alt+l"
copy_pass="Alt+p"
show="Alt+o"
copy_entry="Alt+2"
copy_menu="Alt+c"
action_menu="Alt+a"
type_menu="Alt+t"
help="Alt+h"
switch="Alt+x"
insert_pass="Alt+n"

# Safe permissions
umask 077

# get all password files and create an array
list_passwords() {
  cd "${root}" || exit

  find -L . -iname '*.gpg' -printf '%P\n' | sort -n | \
    while read -r filename; do
      printf '%s\n' "${filename%.gpg}"
  done
}
doClip () {
  case "$clip" in
    "primary") xclip ;;
    "clipboard") xclip -selection clipboard;;
    "both") xclip; xclip -o | xclip -selection clipboard;;
  esac
}

checkIfPass () {
    rm -f "$HOME/.cache/rofi-pass/last_used"
    printf '%s\n' "${root}: $selected_password" > "$HOME/.cache/rofi-pass/last_used"
}


autopass () {
  x_repeat_enabled=$(xset q | awk '/auto repeat:/ {print $3}')
  xset r off

  rm -f "$HOME/.cache/rofi-pass/last_used"
  printf '%s\n' "${root}: $selected_password" > "$HOME/.cache/rofi-pass/last_used"
  for word in ${stuff["$AUTOTYPE_field"]}; do
  case "$word" in
    ":tab") xdotool key Tab;;
    ":space") xdotool key space;;
    ":delay") sleep "${delay}";;
    ":enter") xdotool key Return;;
    "pass") printf '%s' "${password}" | xdotool type --clearmodifiers --file -;;
    *) printf '%s' "${stuff[${word}]}" | xdotool type --clearmodifiers --file -;;
  esac
  done
  if [[ ${auto_enter} == "true" ]]; then
    xdotool key Return
  fi
  xset r "$x_repeat_enabled"
  unset x_repeat_enabled
  clearUp
}

openURL () {
  checkIfPass
  $BROWSER "$(pass "$selected_password" | grep "${URL_field}: " | gawk '{sub(/:/,"")}{print $2}1' | head -1)"; exit;
  clearUp
}

typeUser () {
  checkIfPass

  x_repeat_enabled=$(xset q | awk '/auto repeat:/ {print $3}')
  xset r off

  printf '%s' "${stuff[${USERNAME_field}]}" | xdotool type --clearmodifiers --file -

  xset r "$x_repeat_enabled"
  unset x_repeat_enabled

  clearUp
}

typePass () {
  checkIfPass

  x_repeat_enabled=$(xset q | awk '/auto repeat:/ {print $3}')
  xset r off

  printf '%s' "${password}" | xdotool type --clearmodifiers --file -
  if [[ $notify == "true" ]]; then
    if [[ "${stuff[notify]}" == "false" ]]; then
      :
    else
      notify-send "rofi-pass" "finished typing password";
    fi
  elif [[ $notify == "false" ]]; then
    if [[ "${stuff[notify]}" == "true" ]]; then
      notify-send "rofi-pass" "finished typing password";
    else
      :
    fi
  fi

  xset r "$x_repeat_enabled"
  unset x_repeat_enabled

  clearUp
}

typeField () {
  checkIfPass

  x_repeat_enabled=$(xset q | awk '/auto repeat:/ {print $3}')
  xset r off

  printf '%s' "${stuff[${typefield}]}" | xdotool type --clearmodifiers --file -

  xset r "$x_repeat_enabled"
  unset x_repeat_enabled

  clearUp
}

copyUser () {
  checkIfPass
  printf '%s' "${stuff[${USERNAME_field}]}" | doClip
  clearUp
}

copyField () {
  checkIfPass
  printf '%s' "${stuff[${copyfield}]}" | doClip
  clearUp
}

copyURL () {
  checkIfPass
  printf '%s' "${stuff[${URL_field}]}" | doClip
  clearUp
}

copyPass () {
  checkIfPass
  printf '%s' "$password" | doClip
  if [[ $notify == "true" ]]
  then
    notify-send "rofi-pass" "Copied Password\nClearing in 45 seconds"
  fi

  if [[ $notify == "true" ]]
  then
    (sleep 45; printf '%s' "" | xclip; printf '%s' "" | xclip -selection clipboard | notify-send "rofi-pass" "Clipboard cleared") &
  elif [[ $notify == "false" ]]
  then
    (sleep 45; printf '%s' "" | xclip; printf '%s' "" | xclip -selection clipboard) &
  fi
}

viewEntry () {
  checkIfPass
  showEntry "${selected_password}"
}

generatePass () {
  askmenu_content=(
  "Yes"
  "No")

  askGenMenu=$(printf '%s\n' "${askmenu_content[@]}" | _rofi -dmenu -p "Generate new Password for ${selected_password}? > ")
  askgen_exit=$?
  if [[ $askgen_exit -eq 1 ]]
  then
  	  exit
  fi
  if [[ $askGenMenu == "Yes" ]]; then
    true
  elif [[ $askGenMenu == "No" ]]; then
    actionMenu
  fi

  checkIfPass

  symbols_content=(
  "0  Cancel"
  "1  Yes"
  "2  No")

  symbols=$(printf '%s\n' "${symbols_content[@]}" | _rofi -dmenu -p "Use Symbols? > ")
  symbols_val=$?
  if [[ $symbols_val -eq 1 ]]
  then
  	  exit
  fi
  if [[ $symbols == "0  Cancel" ]]; then
    mainMenu;
  elif [[ $symbols == "1  Yes" ]]; then
    symbols="";
  elif [[ $symbols == "2  No" ]]; then
    symbols="-n";
  fi
  HELP="<span color='$help_color'>Enter Number or hit Enter to use default length</span>"
  length=$(printf '%s' "" | _rofi -dmenu -mesg "${HELP}" -p "Password length? (Default: ${password_length}) > ")
  length_exit=$?
  if [[ $length_exit -eq 1 ]]
  then
    exit
  fi
  if [[ $length == "" ]]; then
    pass generate ${symbols} -i "$selected_password" "${password_length}" > /dev/null;
  else
    pass generate ${symbols} -i "$selected_password" "${length}" > /dev/null;
  fi
}

# main Menu
mainMenu () {
  if [[ $1 == "--bmarks" ]]; then
    selected_password="$(list_passwords 2>/dev/null \
      | _rofi -mesg "Bookmarks Mode. ${switch} to switch" \
      -dmenu \
      -kb-custom-10 "${switch}" \
      -select "$entry" \
      -p "rofi-pass > ")"

    rofi_exit=$?

    if [[ $rofi_exit -eq 1 ]]; then
      exit
    elif [[ $rofi_exit -eq 19 ]]; then
      ${basecommand}
    elif [[ $rofi_exit -eq 0 ]]; then
      openURL
    fi
  else
    unset selected_password

    HELP="Welcome to rofi-pass. Use <span color='$help_color'>${insert_pass}</span> to create a new pass entry.
Run ${default_do} with <span color='$help_color'>Enter</span>. For more help hit <span color='$help_color'>${help}</span>."

    args=( -dmenu
    -kb-custom-1 "${autotype}"
    -kb-custom-2 "${type_user}"
    -kb-custom-3 "${type_pass}"
    -kb-custom-4 "${open_url}"
    -kb-custom-5 "${copy_name}"
    -kb-custom-6 "${copy_pass}"
    -kb-custom-7 "${show}"
    -kb-custom-8 "${copy_url}"
    -kb-custom-9 "${type_menu}"
    -kb-custom-14 "${action_menu}"
    -kb-custom-15 "${copy_menu}"
    -kb-custom-16 "${help}"
    -kb-custom-17 "${switch}"
    -kb-custom-18 "${insert_pass}"
    -select $entry
    -p "rofi-pass > ")

    if [[ $help_header == "true" ]]
    then
      args+=( -mesg "${HELP}" )
    fi
    selected_password="$(list_passwords 2>/dev/null | rofi "${args[@]}")"

    rofi_exit=$?
    if [[ $rofi_exit -eq 1 ]]; then
      exit
    fi
    mapfile -t password_temp < <(PASSWORD_STORE_DIR="${root}" pass "$selected_password")
    password=${password_temp[0]}
    if [[ ${password} == "#FILE="* ]]; then
      pass_file="${password#*=}"
      mapfile -t password_temp < <(PASSWORD_STORE_DIR="${root}" pass "${pass_file}")
      password=${password_temp[0]}
    fi
    fields=$(printf '%s\n' "${password_temp[@]:1}" | awk '$1 ~ /:$/{$1=$1;print}')

    declare -A stuff
    stuff["pass"]=${password}
    if [[ -n $fields ]]; then
      while read -r LINE; do
        _id="${LINE%%: *}"
        _val="${LINE#* }"
        stuff["${_id}"]=${_val}
      done < <(printf '%s\n' "${fields}")
      if test "${stuff['autotype']+autotype}"
        then
            :
        else
          stuff["autotype"]="${USERNAME_field} :tab pass"
        fi
      fi
  fi
  if [[ -z "${stuff["${AUTOTYPE_field}"]}" ]]
  then
    if [[ -n $default_autotype ]]
    then
      stuff["${AUTOTYPE_field}"]="${default_autotype}"
    fi
  fi
  if [[ -z "${stuff["${USERNAME_field}"]}" ]]
  then
    if [[ -n $default_user ]]
    then
      stuff["${USERNAME_field}"]="${default_user}"
    fi
  fi
  pass_content="$(for key in "${!stuff[@]}"; do printf '%s\n' "${key}: ${stuff[$key]}"; done)"

  # actions based on keypresses
  case "${rofi_exit}" in
    0) typeMenu;;
    10) sleep 0.2; autopass;;
    11) sleep 0.2; typeUser;;
    12) sleep 0.2; typePass;;
    13) openURL;;
    14) copyMenu;;
    15) copyPass;;
    16) viewEntry;;
    17) copyURL;;
    18) export default_do="menu"; typeMenu;;
    23) actionMenu;;
    24) copyMenu;;
    25) unset selected_password; helpMenu;;
    26) ${basecommand} --bmarks;;
    27) insertPass;;
  esac
  clearUp
}


clearUp () {
  password=''
  selected_password=''
  unset stuff
  unset password
  unset selected_password
  unset password_temp
  unset stuff
}

helpMenu () {
  helptext=$(printf '%s' "${autotype}: Autotype
${type_user}: Type Username
${type_pass}: Type Password
---
${copy_name}: Copy Username
${copy_pass}: Copy Password
${copy_url}: Copy URL
${open_url}: Open URL
${copy_menu}: Copy Custom Field
---
${action_menu}: Edit, Move, Delete, Re-generate Submenu
${show}: Show Password File
${insert_pass}: Insert new Pass Entry
${switch}: Switch Pass/Bookmark Mode" | _rofi -dmenu -mesg "Hint: All hotkeys are configurable in config file" -p "Help > ")
help_val=$?
if [[ $help_val -eq 1 ]]; then exit;
else unset helptext; mainMenu; fi
}


typeMenu () {
  if [[ -n $default_do ]]; then
    if [[ $default_do == "menu" ]]; then
      checkIfPass
      typefield=$(printf '%s\n' "${!stuff[@]}" | sort | _rofi -dmenu  -p "Choose Field to type > ")
      val=$?
      if [[ $val -eq 1 ]]; then
        exit
      fi
      case "$typefield" in
        "") exit;;
        "password") typePass;;
        "${AUTOTYPE_field}") autopass;;
        *) typeField
      esac
      clearUp
    elif [[ $default_do == "${AUTOTYPE_field}" ]]; then
      autopass
    else
      ${default_do}
    fi
  fi
}

copyMenu () {
  checkIfPass
  copyfield=$(printf '%s\n' "${!stuff[@]}" | sort | _rofi -dmenu  -p "Choose Field to copy > ")
  val=$?
  if [[ $val -eq 1 ]]; then
    exit;
  fi
  if [[ $copyfield == "pass" ]]; then
    copyPass;
  else
    copyField
  fi
  clearUp
}

actionMenu () {
  checkIfPass
  action_content=(
  "< Return"
  "---"
  "1 Move Password File"
  "2 Copy Password File"
  "3 Delete Password File"
  "4 Edit Password File"
  "5 Generate New Password")

  action=$(printf '%s\n' "${action_content[@]}" | _rofi -dmenu -p "Choose Action > ")
  if [[ ${action} == "1 Move Password File" ]]; then
    manageEntry move;
  elif [[ ${action} == "3 Delete Password File" ]]; then
    manageEntry delete;
  elif [[ ${action} == "2 Copy Password File" ]]; then
    manageEntry copy;
  elif [[ ${action} == "4 Edit Password File" ]]; then
    manageEntry edit;
  elif [[ ${action} == "5 Generate New Password" ]]; then
    generatePass;
  elif [[ ${action} == "< Return" ]]; then
    mainMenu;
  elif [[ ${action} == "" ]]; then
    exit
  fi
}

showEntry () {
  if [[ -z $pass_content ]]; then
    password_temp=$(PASSWORD_STORE_DIR="${root}" pass "$selected_password")
    password="${password_temp%%$'\n'*}"
    pass_key_value=$(printf '%s' "${password_temp}" | tail -n+2 | grep ': ')
    declare -A stuff
    while read -r LINE; do
      _id="${LINE%%: *}"
      _val="${LINE#* }"
      stuff["${_id}"]=${_val}
    done < <(printf '%s' "${pass_key_value}")
    stuff["pass"]=${password}
    if test "${stuff['autotype']+autotype}"
      then
          :
      else
          stuff["autotype"]="${USERNAME_field} :tab pass"
    fi
    pass_content="$(for key in "${!stuff[@]}"; do printf '%s' "${key}: ${stuff[$key]}"; done)"
  fi
  HELP="<span color='${help_color}'>${copy_entry}: Copy Entry</span>"
  bla_content=(
  "< Return"
  "${pass_content}")

  bla=$(printf '%s\n' "${bla_content[@]}" | _rofi -dmenu -mesg "Enter: Copy entry to clipboard" -p "> ")
  rofi_exit=$?

  word=$(printf '%s' "$bla" | gawk -F': ' '{print $1}')
  if [[ ${rofi_exit} -eq 1 ]]; then
    exit
  elif [[ ${rofi_exit} -eq 0 ]]; then
    if [[ ${bla} == "< Return" ]]; then
      mainMenu
    else
      if [[ -z $(printf '%s' "${stuff[${word}]}") ]]; then
        printf '%s' "$word" | doClip
      else
        printf '%s' "${stuff[${word}]}" | doClip
      fi
      if [[ $notify == "true" ]]
      then
        notify-send "rofi-pass" "Copied Password\nClearing in 45 seconds"
      fi
      if [[ $notify == "true" ]]
      then
        (sleep 45; printf '%s' "" | xclip; printf '%s' "" | xclip -selection clipboard | notify-send "rofi-pass" "Clipboard cleared") &
      elif [[ $notify == "false" ]]
      then
        (sleep 45; printf '%s' "" | xclip; printf '%s' "" | xclip -selection clipboard) &
      fi
      exit
    fi
  fi
  exit
  unset stuff
  unset password
  unset selected_password
  unset password_temp
  unset stuff
  exit
}

manageEntry () {
  if [[ "$1" == "edit" ]]; then
    EDITOR=$EDITOR PASSWORD_STORE_DIR="${root}" pass edit "${selected_password}"
    mainMenu
  elif [[ $1 == "move" ]]; then
    cd "${root}" || exit
    group=$(find . -type d -not -iwholename '*.git*' -printf '%d\t%P\n' | sort -r -nk1 | cut -f2- | _rofi -dmenu -p "Choose Group > ")
    if [[ $group == "" ]]; then
      exit
    fi
    PASSWORD_STORE_DIR="${root}" pass mv "$selected_password" "${group}"
      mainMenu
  elif [[ $1 == "copy" ]]; then
    cd "${root}" || exit
    group=$(find . -type d -not -iwholename '*.git*' -printf '%d\t%P\n' | sort -r -nk1 | cut -f2- | _rofi -dmenu -p "Choose Group > ")
    if [[ $group == "" ]]; then
      exit
    else
      new_name="$(listgpg | rofi -dmenu -format 'f' -mesg "Copying to same Group. Please enter a name for the new entry" -p "> ")"
    fi
    PASSWORD_STORE_DIR="${root}" pass cp "$selected_password" "${group}/${new_name}"
      mainMenu
  elif [[ "$1" == "delete" ]]; then
    HELP="<span color='$help_color'>Selected entry: ${selected_password}</span>"
    ask_content=(
    "Yes"
    "No")
    ask=$(printf '%s\n' "${ask_content[@]}" | _rofi -mesg "${HELP}" -dmenu -p "Are You Sure? > ")
    if [[ "$ask" == "Yes" ]]; then
      PASSWORD_STORE_DIR="${root}" pass rm --force "${selected_password}"
    elif [[ "$ask" == "No" ]]; then
      mainMenu
    elif [[ -z "$ask" ]]; then
      exit
    fi
  else
    mainMenu
  fi
}

listgpg () {
  find . -type f -not -path '*/\.*' | cut -c 3-
}

insertPass () {
  url=$(xclip --selection clipboard -o)
  if [[ "${url:0:4}" == "http" ]]
  then
    domain_name="$(printf '%s\n' "${url}" | awk -F \/ '{l=split($3,a,"."); print (a[l-1]=="com"?a[l-2] OFS:X) a[l-1] OFS a[l]}' OFS=".")"
	help_content="Domain: ${domain_name}
Type name, make sure it is unique"
  else
  	help_content="Hint: Copy URL to clipboard before calling this menu.
Type name, make sure it is unique"
  fi
  cd "${root}" || exit
  grouplist="$(find . -type d -not -iwholename '*.git*' -printf '%d\t%P\n' | sort -r -nk1 | cut -f2-)"
  name="$(listgpg | rofi -dmenu -format 'f' -filter "${domain_name}" -mesg "${help_content}" -p "> ")"
  val=$?
  if [[ $val -eq 1 ]]; then
    exit
  fi
  user_content=(
  "${default_user2}"
  "${USER}"
  "${default_user}")
  user=$(printf '%s\n' "${user_content[@]}" | rofi -dmenu -mesg "Chose Username or type" -p "> ")
  val=$?
  if [[ $val -eq 1 ]]; then
    exit
  fi
  group_content=(
  "No Group"
  "---"
  "${grouplist}")
  group=$(printf '%s\n' "${group_content[@]}" | rofi -dmenu -p "Choose Group > ")
  val=$?
  if [[ $val -eq 1 ]]; then
    exit
  fi

  pw=$(printf '%s' "Generate" | rofi -dmenu -p "Password > " -mesg "Type Password or hit Enter to generate one")
  if [[ $pw == "Generate" ]]; then
    pw=$(pwgen "${password_length}")
  fi
  clear

  if [[ "$group" == "No Group" ]]; then
    if [[ $url == http* ]]; then
      pass_content=(
      "${pw}"
      "---"
      "${USERNAME_field}: ${user}"
      "${URL_field}: ${url}")
      printf '%s\n' "${pass_content[@]}" | PASSWORD_STORE_DIR="${root}" pass insert -m "${name}" > /dev/null && PASSWORD_STORE_DIR="${root}" pass edit "${name}"
    else
      pass_content=(
      "${pw}"
      "---"
      "${USERNAME_field}: ${user}")
       printf '%s\n' "${pass_content[@]}" | PASSWORD_STORE_DIR="${root}" pass insert -m "${name}" > /dev/null && PASSWORD_STORE_DIR="${root}" pass edit "${name}"
    fi
  else
    if [[ $url == http* ]]; then
      pass_content=(
      "${pw}"
      "---"
      "${USERNAME_field}: ${user}"
      "${URL_field}: ${url}")
      printf '%s\n' "${pass_content[@]}" | PASSWORD_STORE_DIR="${root}" pass insert -m "${group}/${name}" > /dev/null && PASSWORD_STORE_DIR="${root}" pass edit "${group}/${name}"
    else
      pass_content=(
      "${pw}"
      "---"
      "${USERNAME_field}: ${user}")
      printf '%s\n' "${pass_content[@]}" | PASSWORD_STORE_DIR="${root}" pass insert -m "${group}/${name}" > /dev/null && PASSWORD_STORE_DIR="${root}" pass edit "${group}/${name}"
    fi
  fi
}

help_msg () {
  cat <<'EOF' 
rofi-pass (Version: 1.6-git)

Usage:
  --insert         insert new entry to password store
  --manage         edit/move/delete entries
  --root           set custom root directory
  --last-used      highlight last used item
  --show-last      show details of last used Entry
  --bmarks         run bookmarks Mode 
EOF
}

main () {
  # enable extended globbing
  shopt -s nullglob globstar

  # check if global config exists and load it
  if [[ -f /etc/rofi-pass.conf ]]; then
    source /etc/rofi-pass.conf
  fi

  # check if local config exists and load it
  if [[ -f "$HOME/.config/rofi-pass/config" ]]; then
    source "$HOME/.config/rofi-pass/config"
  fi

  # create tmp dir
  if [[ ! -d "$HOME/.cache/rofi-pass" ]]; then
    mkdir "$HOME/.cache/rofi-pass"
  fi

  # fix keyboard layout if enabled in config
  if [[ $fix_layout == "true" ]]
  then
    layout_cmd
  fi

  # set help color
  if [[ $help_color == "" ]]; then
    help_color=$(rofi -dump-xresources | grep 'rofi.color.normal' | gawk -F ',' '/,/{gsub(/ /, "", $2); print $2}')
  fi

  # check for BROWSER variable, use xdg-open as fallback
  if [[ -z $BROWSER ]]; then
    export BROWSER=xdg-open
  fi

  # check if alternative root directory was given on commandline
  if [[ -r "$HOME/.cache/rofi-pass/last_used" ]] && [[ $1 == "--last-used" || $1 == "--show-last" ]]; then
    export root; root=$(awk -F ': ' '{ print $1 }' "$HOME/.cache/rofi-pass/last_used")
  elif [[ -n "$2" && "$1" == "--root" ]]; then
    export root="${2}"
  elif [[ -n $root ]]; then
    export root="${root}"
  elif [[ -n ${PASSWORD_STORE_DIR} ]]; then
    export root=${PASSWORD_STORE_DIR}
  else
    export root="$HOME/.password-store"
  fi
  export PASSWORD_STORE_DIR="${root}"
    case $1 in
      --insert)
        insertPass
        ;;
      --root)
        mainMenu
        ;;
      --manage)
        manageEntry
        ;;
      --help)
        help_msg
        ;;
      --last-used)
        if [[ -r "$HOME/.cache/rofi-pass/last_used" ]]; then
          entry="$(awk -F ': ' '{ print $2 }' "$HOME/.cache/rofi-pass/last_used")"
        fi
        mainMenu
        ;;
      --show-last)
        if [[ -r "$HOME/.cache/rofi-pass/last_used" ]]; then
          selected_password="$(awk -F ': ' '{ print $2 }' "$HOME/.cache/rofi-pass/last_used")" viewEntry
        else
          mainMenu
        fi
        ;;
      --bmarks)
        mainMenu --bmarks;
        ;;
      *)
        mainMenu
        ;;
  esac
}

main "$@"