#!/bin/bash
. ~/.bash_aliases-common

if [[ -f ~/.bash_tools ]]; then
  . ~/.bash_tools
fi

auto_tmux

function start_branch() {
  (
    # set -o errexit # Can't do this inside a function
    set -o nounset
    set -o pipefail
    branch=${1:?}
    branch=${branch#feature/}
    tmux -2 new-window -a -t "Main Screen:1" -n "$branch" -c ~ -e BRANCH_NAME="$branch" 'bash -ic "_start_branch \"\$BRANCH_NAME\"; exec bash"'
  )
}


function _start_branch() {
  branch=${1:?}
  branch=${branch#feature/}
  (
    # set -o errexit # Can't do this inside a function
    set -o nounset
    set -o pipefail
    if [[ ! -d ~/branches/"$branch" ]]; then
      printf "=== Setting up a branch named ${branch}...\n"
      cd ~/equalizer.git || exit "$?"
      printf "=== Retrieving the latest data from the repository...\n"
      git fetch --all || exit "$?"
      printf "=== Creating the branch 'feature/${branch}' from develop_front and checking it out in a new working tree at ~/branches/${branch}...\n"
      git worktree add -b "feature/${branch}" ~/branches/"${branch}" develop_front || git worktree add ~/branches/"${branch}" "feature/${branch}" || exit "$?"
    else
      printf "=== Accessing an existing branch named ${branch}...\n"
    fi
    cd ~/branches/"${branch}" || exit "$?"
    update_branch || exit "$?"
    printf "\a=== Your new branch ${branch} is ready!\n"
  )
  if [[ -d ~/branches/"$branch" ]]; then
    cd ~/branches/"${branch}"
  else
    cd ~/equalizer.git
  fi
}

function get_current_branch_name() {
  git symbolic-ref --quiet --short HEAD
}

function get_remote_branch_name() {
  git rev-parse --abbrev-ref --symbolic-full-name '@{u}'
}

function update_branch() {
  (
    # set -o errexit # Can't do this inside a function
    set -o nounset
    set -o pipefail
    printf "=== Updating the branch...\n"
    git fetch --all || exit "$?"
    if get_remote_branch_name >&/dev/null; then
      git pull --rebase || exit "$?"
    fi
    git rebase develop_front || exit "$?"
    printf "=== Updating the dependencies...\n"
    install_equalizer_deps
  )
}

function install_equalizer_deps() {
  (
    # set -o errexit # Can't do this inside a function
    set -o nounset
    set -o pipefail
    printf "Installing Ruby dependencies...\n"
    bundle install || exit "$?"
    printf "Installing Javascript dependencies...\n"
    printf "Installing in top level...\n"
    yarn install || exit "$?"
    printf "Installing in widget...\n"
    cd widget || exit "$?"
    yarn install || exit "$?"
    yarn build || exit "$?"
    cd .. || exit "$?"
    printf "Installing in front...\n"
    cd front || exit "$?"
    yarn install
  )
}

function delete_local_branch() {
  (
    # set -o errexit # Can't do this inside a function
    set -o nounset
    set -o pipefail
    branch=${1:+feature/$1}
    branch=${branch:-$(get_current_branch_name)} || exit "$?"
    cd ~/equalizer.git
    worktree="$(git worktree list --porcelain | grep -B2 ${branch} | cut -d' ' -f2 | head -n1)" || exit "$?"
    git worktree remove "$worktree" || exit "$?"
    git branch -d "$branch" || exit "$?"
  ); local lastexit="$?"
  if [[ ! -d "$PWD" ]]; then
    cd ~/equalizer.git
  fi
  return $lastexit
}
function _list_feature_branches() {
  git --git-dir ~/equalizer.git for-each-ref --format '%(refname:short)' refs/heads/ \
    | grep -vF $'develop\ndevelop_front\nmaster' \
    | sed 's#^feature/##'
}

function _branch_completions() {
  if [[ $COMP_CWORD -ne 1 ]]; then
    return
  fi
  COMPREPLY+=($(compgen -W "$(_list_feature_branches)" "${COMP_WORDS[1]}"))
}

complete -F _branch_completions start_branch
complete -F _branch_completions delete_local_branch