git.sh

alias g=git
function git-current-branch() { git rev-parse --abbrev-ref HEAD; }

## Support repos with different main/master branches.
## used to support the `com` quick change to main branch.
function __git_main_branch() {
  local current_main_branch=`git branch --remote --points-at origin/HEAD --format "%(symref:lstrip=3)"|head -n1`

  if [ "$current_main_branch" ]; then
    echo $current_main_branch
  else
    # Test me by running `git remote set-head origin --delete`
    echo "!!! Failed to find remote origin/HEAD main branch, attempting to auto-detect from remote."
    git remote set-head origin --auto
  fi
}

# Status
alias gs='git status'
alias gsf='git status --untracked-files=no'
alias gse='git status --ignored'

# Logs
alias gsl='git log --pretty="format:%Cblue%h%d%Creset %ar %Cgreen%an%Creset %s"'
# Disable the --simplify-merges flag, as it's super slow on monorepo
# alias gsl='git log --simplify-merges --pretty="format:%Cblue%h%d%Creset %ar %Cgreen%an%Creset %s"'
alias gr='gsl -n 10' # Git Recent
alias gra='gr --all'

# Committing
alias git-save="git commit --no-verify -m WIPWIPWIP" ##### WIP Commits Instead of Stashes
alias gcwip=git-save
alias gcm=__git_commit_splat
alias pick='git cherry-pick'

__git_commit_splat()
{
  TEMP_FILE=$(mktemp '/tmp/git.simplecommit.msg.XXXX')
  echo "# $*" >> "$TEMP_FILE"
  git commit --edit --verbose --file "$TEMP_FILE"
  rm -f "$TEMP_FILE"
}

# Diff
alias gd='git diff --color'
alias gds='git diff --color --staged '

# Operate Patchwise (Interactively)
alias gap='git add --patch' # Stage
alias grp='git reset --patch' # Unstage
alias gcp='git checkout --patch' # Discard
alias gvp=gcp
alias gsp='git stash save --patch'

# Remote
alias gup='git up; echo "PRUNE"; __git_prune_automagic'
alias gpushnew='git push --set-upstream origin $(git-current-branch)'
alias __git_prune_merged_branches='git checkout $(__git_main_branch) && git branch --merged |grep -v "\*" | grep -v $(__git_main_branch) |grep -v stable | xargs -I {} git branch -d "{}"'

# Changed files
alias __git_current_branch_revisions='git rev-list $(git-current-branch) ^$(__git_main_branch)'
alias __git_current_branch_changed_files='git diff --name-only $(git merge-base HEAD $(__git_main_branch))..$(git-current-branch)'
alias __git_current_branch_changes='git diff $(git merge-base HEAD $(__git_main_branch))..$(git-current-branch)'
alias gbc=__git_current_branch_changed_files
alias gbd=__git_current_branch_changes

alias __git_modified_files="git diff HEAD --name-only"
alias __git_new_files="git ls-files -o --exclude-standard"
alias gmod=__git_modified_files
alias gnew=__git_new_files

# shellcheck disable=SC1117
alias __git_undo_whitespace_changes="git diff -b --numstat | egrep $'^0\t0\t' | cut -d$'\t' -f3- | xargs git checkout HEAD --"

# Branch management
alias com='git checkout $(__git_main_branch)'
alias cob="__git_checkout_branch_menu '__git_local_branch_list'"
alias cow="__git_checkout_branch_menu '__git_working_branch_list'"
alias cor="__git_checkout_branch_menu '__git_only_remote_branch_list'"


# Submodules are pain
alias git-reinit-submodules='git submodule update --init --recursive --checkout'

# Given the name of a function which returns a list of branch names and an optional string to filter that list,
# present the user a menu, allowing fast branch selection.
__git_checkout_branch_menu()
{
  local branch_fetch_list_command="$1"
  local filter_string="$2"

  PS3="Select a branch or Ctrl+C to cancel: "
  if [ "$filter_string" ]; then
    select opt in $($branch_fetch_list_command |grep --color=never "$filter_string"); do git checkout "${opt}"; break; done
  else
    select opt in $($branch_fetch_list_command); do git checkout "${opt}"; break; done
  fi

  echo ""
}

function __git_local_branch_list()   { git for-each-ref --sort=-committerdate refs/heads/ --format='%(refname:short)' ; }
function __git_working_branch_list() { git branch --list --color=never --no-merged |tr -d ' *' | grep -v '(no branch)'; }
function __git_remote_branch_list()  { git branch --list --color=never --remotes | grep -v "origin/$(__git_main_branch)" | cut -d/ -f 2-; }
function __git_only_remote_branch_list() {
  cat  <(__git_local_branch_list) <(__git_local_branch_list) <(__git_remote_branch_list) |sort |uniq -u
}

alias gbl=git_branches_cool_list
function git_branches_cool_list()
{
  git for-each-ref --sort=committerdate refs/heads/ --format='%(HEAD) %(color:yellow)%(refname:short)%(color:reset) - %(color:red)%(objectname:short)%(color:reset) - %(contents:subject) - %(authorname) (%(color:green)%(committerdate:relative)%(color:reset))'
}

function __git_prune_automagic()
{
  if [[ "$(git-current-branch)"  == "$(__git_main_branch)" ]]; then
    echo "🪄 Auto-pruning Git merged branches"
  else
    echo "⏭️ Not pruning, not on the main branch"
    return
  fi
  __git_prune_merged_branches
  print_info "Orphan Branches"
  if [ command -v git-branch-delete-orphans &> /dev/null ]; then
  git-branch-delete-orphans
  else
    print_warning 'Missing command `git-branch-delete-orphans`'
  fi
}

# Requires
# a git `origin` you have write access to
# a git remote `upstream` you want to keep in sync
function git-sync-upstream() {
  source "${base_dotfiles_path}/shared/core/task_functions.sh"

  local upstream='upstream'
  local origin='origin'
  local main_branch=$(__git_main_branch)

  print_info "Using Upstream"
  git remote get-url "${upstream}"
  r=$?
  if [ $r -ne 0 ]; then
    print_warning "Failed"
    return $r
  fi

  start_task "Fetching ${upstream}/${main_branch}"
  git fetch "${upstream}"
  print "✅"

  start_task "Merge (fast-forward only) ${upstream}/${main_branch} ${main_branch}"
  git merge --ff-only "${upstream}/${main_branch}" "${main_branch}"
  end_task

  start_task "Push ${origin}"
  git push  --verbose "${origin}" "${main_branch}"
  end_task
}