#! bash oh-my-bash.module
#
# The current version is based on the following upstream version:
# https://github.com/owenthereal/gh/blob/04a7985fa9a1c1d4d63738f4edb7b07d228bdb12/etc/gh.bash_completion.sh
#------------------------------------------------------------------------------
# hub tab-completion script for bash.
# This script complements the completion script that ships with git.

# Check that git tab completion is available
if _omb_util_function_exists _git; then
  # Duplicate and rename the 'list_all_commands' function
  eval "$(declare -f __git_list_all_commands | \
        sed 's/__git_list_all_commands/__git_list_all_commands_without_hub/')"

  # Wrap the 'list_all_commands' function with extra hub commands
  function __git_list_all_commands {
    cat <<-EOF
alias
pull-request
fork
create
browse
compare
ci-status
release
issue
update
EOF
    __git_list_all_commands_without_hub
  }

  # Ensure cached commands are cleared
  __git_all_commands=""

  ##########################
  # hub command completions
  ##########################

  # hub alias [-s] [SHELL]
  function _git_alias {
    local i c=2 s=-s sh shells="bash zsh sh ksh csh fish"
    while [ $c -lt $cword ]; do
      i="${words[c]}"
      case "$i" in
      -s)
        unset s
        ;;
      *)
        for sh in $shells; do
          if [ "$sh" = "$i" ]; then
            unset shells
            break
          fi
        done
        ;;
      esac
      ((c++))
    done
    __gitcomp "$s $shells"
  }

  # hub browse [-u] [--|[USER/]REPOSITORY] [SUBPAGE]
  function _git_browse {
    local i c=2 u=-u repo subpage
    local subpages_="commits issues tree wiki pulls branches stargazers
      contributors network network/ graphs graphs/"
    local subpages_network="members"
    local subpages_graphs="commit-activity code-frequency punch-card"
    while [ $c -lt $cword ]; do
      i="${words[c]}"
      case "$i" in
      -u)
        unset u
        ;;
      *)
        if [ -z "$repo" ]; then
          repo=$i
        else
          subpage=$i
        fi
        ;;
      esac
      ((c++))
    done
    if [ -z "$repo" ]; then
      __gitcomp "$u -- $(__hub_github_repos '\p')"
    elif [ -z "$subpage" ]; then
      case "$cur" in
      */*)
        local pfx="${cur%/*}" cur_="${cur#*/}"
        local subpages_var="subpages_$pfx"
        __gitcomp "${!subpages_var}" "$pfx/" "$cur_"
        ;;
      *)
        __gitcomp "$u ${subpages_}"
        ;;
      esac
    else
      __gitcomp "$u"
    fi
  }

  # hub compare [-u] [USER[/REPOSITORY]] [[START...]END]
  function _git_compare {
    local i c=$((cword - 1)) u=-u user remote owner repo arg_repo rev
    while [ $c -gt 1 ]; do
      i="${words[c]}"
      case "$i" in
      -u)
        unset u
        ;;
      *)
        if [ -z "$rev" ]; then
          # Even though the logic below is able to complete both user/repo
          # and revision in the right place, when there is only one argument
          # (other than -u) in the command, that argument will be taken as
          # revision. For example:
          # $ hub compare -u upstream
          # > https://github.com/USER/REPO/compare/upstream
          if __hub_github_repos '\p' | grep -Eqx "^$i(/[^/]+)?"; then
            arg_repo=$i
          else
            rev=$i
          fi
        elif [ -z "$arg_repo" ]; then
          arg_repo=$i
        fi
        ;;
      esac
      ((c--))
    done

    # Here we want to find out the git remote name of user/repo, in order to
    # generate an appropriate revision list
    if [ -z "$arg_repo" ]; then
      user=$(__hub_github_user)
      if [ -z "$user" ]; then
        for i in $(__hub_github_repos); do
          remote=${i%%:*}
          repo=${i#*:}
          if [ "$remote" = origin ]; then
            break
          fi
        done
      else
        for i in $(__hub_github_repos); do
          remote=${i%%:*}
          repo=${i#*:}
          owner=${repo%%/*}
          if [ "$user" = "$owner" ]; then
            break
          fi
        done
      fi
    else
      for i in $(__hub_github_repos); do
        remote=${i%%:*}
        repo=${i#*:}
        owner=${repo%%/*}
        case "$arg_repo" in
          "$repo"|"$owner")
            break
            ;;
        esac
      done
    fi

    local pfx cur_="$cur"
    case "$cur_" in
    *..*)
      pfx="${cur_%%..*}..."
      cur_="${cur_##*..}"
      __gitcomp_nl "$(__hub_revlist $remote)" "$pfx" "$cur_"
      ;;
    *)
      if [ -z "${arg_repo}${rev}" ]; then
        __gitcomp "$u $(__hub_github_repos '\o\n\p') $(__hub_revlist $remote)"
      elif [ -z "$rev" ]; then
        __gitcomp "$u $(__hub_revlist $remote)"
      else
        __gitcomp "$u"
      fi
      ;;
    esac
  }

  # hub create [NAME] [-p] [-d DESCRIPTION] [-h HOMEPAGE]
  function _git_create {
    local i c=2 name repo flags="-p -d -h"
    while [ $c -lt $cword ]; do
      i="${words[c]}"
      case "$i" in
      -d|-h)
        ((c++))
        flags=${flags/$i/}
        ;;
      -p)
        flags=${flags/$i/}
        ;;
      *)
        name=$i
        ;;
      esac
      ((c++))
    done
    if [ -z "$name" ]; then
      repo=$(basename "$PWD")
    fi
    case "$prev" in
    -d|-h)
      COMPREPLY=()
      ;;
    -p|*)
      __gitcomp "$repo $flags"
      ;;
    esac
  }

  # hub fork [--no-remote]
  function _git_fork {
    local i c=2 remote=yes
    while [ $c -lt $cword ]; do
      i="${words[c]}"
      case "$i" in
      --no-remote)
        unset remote
        ;;
      esac
      ((c++))
    done
    if [ -n "$remote" ]; then
      __gitcomp "--no-remote"
    fi
  }

  # hub pull-request [-f] [-m <MESSAGE>|-F <FILE>|-i <ISSUE>|<ISSUE-URL>] [-b <BASE>] [-h <HEAD>]
  function _git_pull_request {
    local i c=2 flags="-f -m -F -i -b -h"
    while [ $c -lt $cword ]; do
      i="${words[c]}"
      case "$i" in
      -m|-F|-i|-b|-h)
        ((c++))
        flags=${flags/$i/}
        ;;
      -f)
        flags=${flags/$i/}
        ;;
      esac
      ((c++))
    done
    case "$prev" in
    -i)
      COMPREPLY=()
      ;;
    -b|-h)
      # (Doesn't seem to need this...)
      # Uncomment the following line when 'owner/repo:[TAB]' misbehaved
      #_get_comp_words_by_ref -n : cur
      __gitcomp_nl "$(__hub_heads)"
      # __ltrim_colon_completions "$cur"
      ;;
    -F)
      COMPREPLY=( "$cur"* )
      ;;
    -f|*)
      __gitcomp "$flags"
      ;;
    esac
  }

  ###################
  # Helper functions
  ###################

  # __hub_github_user [HOST]
  # Return $GITHUB_USER or the default github user defined in hub config
  # HOST - Host to be looked-up in hub config. Default is "github.com"
  function __hub_github_user {
    if [ -n "$GITHUB_USER" ]; then
      _omb_util_print $GITHUB_USER
      return
    fi
    local line h k v host=${1:-github.com} config=${HUB_CONFIG:-~/.config/gh}
    if [ -f "$config" ]; then
      while read line; do
        if [ "$line" = "---" ]; then
          continue
        fi
        k=${line%%:*}
        v=${line#*:}
        if [ -z "$v" ]; then
          if [ "$h" = "$host" ]; then
            break
          fi
          h=$k
          continue
        fi
        k=${k#* }
        v=${v#* }
        if [ "$h" = "$host" ] && [ "$k" = "user" ]; then
          _omb_util_print "$v"
          break
        fi
      done < "$config"
    fi
  }

  # __hub_github_repos [FORMAT]
  # List all github hosted repository
  # FORMAT - Format string contains multiple of these:
  #   \m  remote
  #   \p  owner/repo
  #   \o  owner
  #   escaped characters (\n, \t ...etc) work
  # If omitted, prints all github repos in the format of "remote:owner/repo"
  function __hub_github_repos {
    local f format=$1
    if [ -z "$(__gitdir)" ]; then
      return
    fi
    if [ -z "$format" ]; then
      format='\1:\2'
    else
      format=${format//\m/\1}
      format=${format//\p/\2}
      format=${format//\o/\3}
    fi
    _omb_prompt_git config --get-regexp 'remote\.[^.]*\.url' |
    grep -E ' ((https?|git)://|git@)github\.com[:/][^:/]+/[^/]+$' |
    sed -E 's#^remote\.([^.]+)\.url +.+[:/](([^/]+)/[^.]+)(\.git)?$#'"$format"'#'
  }

  # __hub_heads
  # List all local "branch", and remote "owner/repo:branch"
  function __hub_heads {
    local i remote repo branch dir=$(__gitdir)
    if [ -d "$dir" ]; then
      _omb_prompt_git --git-dir="$dir" for-each-ref --format='%(refname:short)' \
        "refs/heads/"
      for i in $(__hub_github_repos); do
        remote=${i%%:*}
        repo=${i#*:}
        _omb_prompt_git --git-dir="$dir" for-each-ref --format='%(refname:short)' \
          "refs/remotes/${remote}/" | while read branch; do
          _omb_util_print "${repo}:${branch#${remote}/}"
        done
      done
    fi
  }

  # __hub_revlist [REMOTE]
  # List all tags, and branches under REMOTE, without the "remote/" prefix
  # REMOTE - Remote name to search branches from. Default is "origin"
  function __hub_revlist {
    local i remote=${1:-origin} dir=$(__gitdir)
    if [ -d "$dir" ]; then
      _omb_prompt_git --git-dir="$dir" for-each-ref --format='%(refname:short)' \
        "refs/remotes/${remote}/" | while read i; do
        _omb_util_print "${i#${remote}/}"
      done
      _omb_prompt_git --git-dir="$dir" for-each-ref --format='%(refname:short)' \
        "refs/tags/"
    fi
  }

  # Enable completion for hub even when not using the alias
  complete -o bashdefault -o default -o nospace -F _git gh 2>/dev/null \
    || complete -o default -o nospace -F _git gh
fi