completion

#!/usr/bin/env bash
# shellcheck disable=SC2034
give_me_hostname="$HOME/.dotfiles/bin/give_me_hostname"
fzf_default_options="-i --select-1 --exit-0 --tac --tiebreak=length,begin,index --reverse --inline-info"
fzf_height="--height ${FZF_TMUX_HEIGHT:-50%} --min-height 25"

export COMP_WORDBREAKS=${COMP_WORDBREAKS/\:/}
export FZF_COMPLETION_TRIGGER='**'

# Tools to complete
net_tools=(
  ping
  host
  mtr
  mosh
  telnet
  ftp
  lftp
  dig
  drill
  nslookup
  ansible
  nmap
)

ssh_tools=(
  ssh
  s
  n
  rsync
  scp
  sftp
  proo
  curl
  ssh-audit
  update-services
  connect_to_proo
  mysql_get_root_password
)

# Use fd (https://github.com/sharkdp/fd) instead of the default find
# command for listing path candidates.
# - The first argument to the function ($1) is the base path to start traversal
# - See the source code (completion.{bash,zsh}) for the details.
_fzf_compgen_path() {
  fd --max-depth 5 --follow --exclude ".git" . "$1"
}

# Use fd to generate the list for directory completion
_fzf_compgen_dir() {
  fd --type d --max-depth 3 --follow --exclude ".git" . "$1"
}

# To redraw line after fzf closes (printf '\e[5n')
bind '"\e[0n": redraw-current-line'

# fzf default completion
#[[ -f /usr/share/fzf/completion.bash ]] && source /usr/share/fzf/completion.bash

_complete() {
  local cur="${COMP_WORDS[COMP_CWORD]}"

  if [[ ${cur} == *@*  ]] ; then
    local opts="-P ${cur/@*/}@ -- ${cur/*@/}"
  else
    local opts=" -- ${cur}"
  fi
  # shellcheck disable=SC2086
  COMPREPLY=( $(compgen -W "$(</dev/stdin)" ${opts}) )
  return 0
}

_fzf_privkey() {
  find "$HOME/.ssh" -type f -name 'id_*' | grep -v '.pub'
}

_fzf_playbook() {
  find playbooks -maxdepth 1 -type f -name '*.yml'
}

_fzf_ansible_groups() {
  local inventory
  inventory=$(awk '/inventory/ {print $3}' "$HOME"/.ansible.cfg)
  inventory=${inventory/\~/$HOME}
  if [ -d "$inventory" ]; then
    awk '/\[/' "${inventory}"* | tr -d '[]'
  else
    awk '/\[/' "$inventory" | tr -d '[]'
  fi
}

_fzf_dsh_groups() {
  local inventory
  inventory="$HOME/.dsh/group"
  if [ -d "$inventory" ]; then
    ls -1 "$inventory"
  fi
}

_fzf_archlinux_installed_packages() {
   pacman -Qq
}

_fzf_log() {
  local cmd log_file
  cmd="command find /var/log/ -type f -name '*log' 2>/dev/null"
  log_file=$(eval "$cmd" | fzf "$fzf_height" --tac --tiebreak=length,begin,index --reverse --inline-info) && less "$log_file"
}

_fzf_complete() {
  local selected fzf post caller
  local current="${COMP_WORDS[COMP_CWORD]}"
  local user_prefix=''
  local options=${1:-'--multi'}
  local separator=${2:- }
  local preview="fzf_preview {}"
  caller=$(caller 0 | awk '{print $2}')
  local delete="fzf_delete $caller {}"

  post="${caller}_post"
  type -t "$post" >/dev/null 2>&1 || post=cat
  fzf="fzf $fzf_default_options $options $fzf_height --preview-window right:75%:hidden --bind ?:toggle-preview"

  if [[ ${current} == *@*  ]] ; then
    user_prefix="${current/@*/}@"
    current="${current/*@/}"
  fi
  selected=$(cat | $fzf --preview "$preview" --bind "alt-bspace:execute:$delete" --query "$current" | $post | tr '\n' "$separator")
  selected=${selected%$separator} # Strip trailing space not to repeat "-o nospace"
  printf '\e[5n'

  if [ -n "$selected" ]; then
    COMPREPLY=("${user_prefix}${selected}")
    return 0
  fi
}

__fzf_history__() {
  local line
  shopt -u nocaseglob nocasematch
  line=$(
    # shellcheck disable=SC1007
    HISTTIMEFORMAT= history |
    FZF_DEFAULT_OPTS="$fzf_height $fzf_default_options --tac -n2..,.. --tiebreak=index --bind=ctrl-r:toggle-sort $FZF_CTRL_R_OPTS +m" fzf |
    command grep '^ *[0-9]') &&
    if [[ $- =~ H ]]; then
      sed 's/^ *\([0-9]*\)\** .*/!\1/' <<< "$line"
    else
      sed 's/^ *\([0-9]*\)\** *//' <<< "$line"
    fi
}

_fzf_complete_kill() {
  [ -n "${COMP_WORDS[COMP_CWORD]}" ] && return 1

  local selected fzf
  fzf="fzf"
  selected=$(ps -ef | sed 1d | FZF_DEFAULT_OPTS="$fzf_height \
      --reverse $FZF_DEFAULT_OPTS --preview 'echo {}' --preview-window down:3:wrap \
      $FZF_COMPLETION_OPTS" $fzf --multi | awk '{print $2}' | tr '\n' ' ')
  printf '\e[5n'

  if [ -n "$selected" ]; then
    COMPREPLY=( "$selected" )
    return 0
  fi
}

__fzf_select__() {
  local cmd="${FZF_CTRL_T_COMMAND:-"command find -L . -mindepth 1 \\( -path '*/\\.*' -o -fstype 'devfs' -o -fstype 'devtmpfs' -o -fstype 'proc' \\) -prune \
    -o -type f -print \
    -o -type d -print \
    -o -type l -print 2> /dev/null | cut -b3-"}"

  eval "$cmd" | FZF_DEFAULT_OPTS="$fzf_height --reverse $FZF_DEFAULT_OPTS $FZF_CTRL_T_OPTS --preview='fzf_preview {}'" fzf -m "$@" | while read -r item; do
    printf '%q ' "$item"
  done
  echo
}

_fzf_file_widget() {
  local selected
  selected="$(__fzf_select__ "$@")"
  READLINE_LINE="${READLINE_LINE:0:$READLINE_POINT}$selected${READLINE_LINE:$READLINE_POINT}"
  READLINE_POINT=$(( READLINE_POINT + ${#selected} ))
}

__fzf_cd__() {
  local cmd dir
  cmd="${FZF_ALT_C_COMMAND:-"command find -L . \\( -path '*/\\.*' -o -fstype 'devfs' -o -fstype 'devtmpfs' -o -fstype 'proc' \\) -prune \
    -o -type d -print 2> /dev/null | sed 1d | cut -b3-"}"
  dir=$(eval "$cmd" | FZF_DEFAULT_OPTS="$fzf_height --reverse $FZF_DEFAULT_OPTS $FZF_ALT_C_OPTS" fzf +m) && printf 'cd %q' "$dir"
}

# v - open files in ~/.viminfo
v() {
  local files
  files=$(grep '^>' ~/.viminfo | cut -c3- |
    while read -r line; do
      [ -f "${line/\~/$HOME}" ] && echo "${line/\~/$HOME}"
    done | fzf-tmux --select-1 --reverse --inline-info +s \
      --tac --multi --preview 'highlight --force -O ansi -l {} 2> /dev/null | head -200' --query "$*" -1) \
      && vim "${files//\~/$HOME}"
}

fmpc() {
  local song_position
  song_position=$(mpc -f "%position%) %artist% - %title%" playlist | \
    fzf-tmux --query="$1" "$fzf_height" --reverse --inline-info +s --select-1 --exit-0 | \
    sed -ne 's/^\([0-9].*\)).*/\1/p') || return 1
  [ -n "$song_position" ] && mpc -q play "$song_position"
}

# c - browse chrome history
ch() {
  local cols sep google_history open
  cols=$(( COLUMNS / 3 ))
  sep='{::}'

  if [ "$(uname)" = "Darwin" ]; then
    google_history="$HOME/Library/Application Support/Google/Chrome/Default/History"
    open=open
  else
    google_history="$HOME/.config/google-chrome/Profile 1/History"
    open=xdg-open
  fi
  cp -f "$google_history" /tmp/h
  # shellcheck disable=2086
  sqlite3 -separator $sep /tmp/h \
    "select substr(title, 1, $cols), url
     from urls order by last_visit_time desc" |
  awk -F $sep '{printf "%-'$cols's  \x1b[36m%s\x1b[m\n", $1, $2}' |
  fzf $fzf_default_options $fzf_height --ansi --multi | sed 's#.*\(https*://\)#\1#' | xargs $open > /dev/null 2> /dev/null
}

# fco - checkout git branch/tag
fco() {
  local tags branches target
  tags=$(
    git tag | awk '{print "\x1b[31;1mtag\x1b[m\t" $1}') || return
  branches=$(
    git branch --all | grep -v HEAD             |
    sed "s/.* //"    | sed "s#remotes/[^/]*/##" |
    sort -u          | awk '{print "\x1b[34;1mbranch\x1b[m\t" $1}') || return
  # shellcheck disable=2086
  target=$(
    (echo "$tags"; echo "$branches") |
    fzf-tmux $fzf_default_options $fzf_height -l30 -- --no-hscroll --ansi +m -d "\t" -n 2) || return
  git checkout "$(echo "$target" | awk '{print $2}')"
}

# fshow - git commit browser
fshow() {
  git log --graph --color=always \
      --format="%C(auto)%h%d %s %C(black)%C(bold)%cr" "$@" |
  fzf --ansi --no-sort --reverse --tiebreak=index --bind=ctrl-s:toggle-sort \
      --bind "ctrl-m:execute:
                (grep -o '[a-f0-9]\{7\}' | head -1 |
                xargs -I % sh -c 'git show --color=always % | less -R') << 'FZF-EOF'
                {}
FZF-EOF"
}

# fcs - get git commit sha
# example usage: git rebase -i `fcs`
fcs() {
  local commits commit
  commits=$(git log --color=always --pretty=oneline --abbrev-commit --reverse) &&
  commit=$(echo "$commits" | fzf --tac +s +m -e --ansi --reverse) &&
  echo -n "$(echo "$commit" | sed "s/ .*//")"
}

_complete_ssh() {
  _complete "$@" < <(give_me_hostname ssh)
}

_complete_net() {
  _complete "$@" < <(give_me_hostname)
}

_fzf_complete_ssh() {
  local cur="${COMP_WORDS[COMP_CWORD]}"
  local prev="${COMP_WORDS[COMP_CWORD-1]}"
  case $prev in
    -i)
      _fzf_complete '--no-multi' ' ' '' "$@" < <(_fzf_privkey)
      return 0
  ;;
  esac
  _fzf_complete '--no-multi' ' ' '' "$@" < <(give_me_hostname ssh)
}

_fzf_complete_ansible() {
  local prev="${COMP_WORDS[COMP_CWORD-1]}"
  case $prev in
    -i)
      _fzf_complete '--multi' ',' "$@" < <(give_me_hostname ssh)
      return 0
    ;;
    -l)
      _fzf_complete '--multi' ',' "$@" < <(_fzf_ansible_groups)
      return 0
    ;;
    -u)
      return 0
    ;;
  esac

  _fzf_complete '--multi' ' ' "$@" < <(_fzf_playbook)
}

_fzf_complete_dsh() {
  local prev="${COMP_WORDS[COMP_CWORD-1]}"
  case $prev in
    -g)
      _fzf_complete '--multi' ',' "$@" < <(_fzf_dsh_groups)
      return 0
    ;;
  esac
}

_fzf_complete_pacman() {
  local prev="${COMP_WORDS[COMP_CWORD-1]}"
  case $prev in
    -Rcs|-Qo|-Qi|-Ql|-Sii)
      _fzf_complete '--multi' ' ' "$@" < <(_fzf_archlinux_installed_packages)
      return 0
  ;;
  esac
}

_fzf_complete_archlinux_installed_packages() {
  _fzf_complete '--multi' ' ' "$@" < <(_fzf_archlinux_installed_packages)
}

_fzf_complete_pssh() {
  local prev="${COMP_WORDS[COMP_CWORD-1]}"
  case $prev in
    --hosts|-h)
      _fzf_complete '--no-multi' ' ' "$@" < <(_fzf_ansible_groups)
      return 0
  ;;
  esac
}

_fzf_complete_pssh_post() {
  # shellcheck disable=2088
  echo "~/.dsh/group/$(cat -)"
}

_fzf_complete_net() {
  _fzf_complete '--no-multi' ' ' "$@" < <(give_me_hostname)
}

_fzf_complete_ips() {
  _fzf_complete '--no-multi' ' ' "$@" < <(give_me_hostname ips)
}

if hash fzf 2>/dev/null; then
  complete -o default -o bashdefault -o nospace -F _fzf_complete_net "${net_tools[@]}"
  complete -o default -o bashdefault -o nospace -F _fzf_complete_ssh "${ssh_tools[@]}"
  complete -o default -o bashdefault -o nospace -F _fzf_complete_ips whois
  complete -o default -o bashdefault -o nospace -F _fzf_complete_ansible ansible-playbook ap
  complete -o default -o bashdefault -o nospace -F _fzf_complete_pacman yaourt pacman yay
  complete -o default -o bashdefault -o nospace -F _fzf_complete_archlinux_installed_packages pactree
  complete -o default -o bashdefault -o nospace -F _fzf_complete_dsh dsh
  complete -o default -o bashdefault -o nospace -F _fzf_complete_pssh pssh psshscp
else
  complete -o default -o bashdefault -o nospace -F _complete_net "${net_tools[@]}"
  complete -o default -o bashdefault -o nospace -F _complete_ssh "${ssh_tools[@]}"
fi

# Kill completion
complete -F _fzf_complete_kill -o nospace -o default -o bashdefault kill

# Required to refresh the prompt after fzf
bind '"\er": redraw-current-line'
bind '"\e^": history-expand-line'
# CTRL-R - Paste the selected command from history into the command line
# shellcheck disable=SC2016
bind '"\C-r": " \C-e\C-u`__fzf_history__`\e\C-e\e^\er"'
# CTRL-f - Paste the selected file path into the command line
bind -x '"\C-f": "_fzf_file_widget"'
# ALT-S - cd into the selected directory
# shellcheck disable=SC2016
bind '"\es": " \C-e\C-u`__fzf_cd__`\e\C-e\er\C-m"'