From 55a98fc06d9ea9d003683dec5aae6cb6b0130ab0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Marc=20Cornell=C3=A0?= <marc.cornella@live.com>
Date: Sat, 11 Jul 2020 20:02:06 +0200
Subject: [PATCH] git: better algorithm to decide what's the main branch

Fixes #9103

Co-authored-by: Yufan You <ouuansteve@gmail.com>
---
 plugins/git/git.plugin.zsh | 27 ++++++++++++++++++++++-----
 1 file changed, 22 insertions(+), 5 deletions(-)

diff --git a/plugins/git/git.plugin.zsh b/plugins/git/git.plugin.zsh
index 6cae3dac2..7284cd121 100644
--- a/plugins/git/git.plugin.zsh
+++ b/plugins/git/git.plugin.zsh
@@ -25,13 +25,30 @@ function work_in_progress() {
   fi
 }
 
-# Check if main exists and use instead of master
+# Get the default 'main' branch
+# Marc(2020-07-11): I hope to be able to remove this someday
 function git_main_branch() {
-  if [[ -n "$(git branch --list master)" ]]; then
-    echo master
-  else
-    echo main
+  # Get default branch from the origin remote
+  local branch
+  branch="${$(command git symbolic-ref refs/remotes/origin/HEAD 2>/dev/null)#refs/remotes/origin/}"
+
+  if [[ -n "$branch" ]]; then
+    echo "$branch"
+    return
   fi
+
+  # Look up list of local branches and return the first one that exists
+  local -a branches
+  branches=(${(@f)"$(command git for-each-ref --format='%(refname:short)' refs/heads 2>/dev/null)"})
+  for branch in master main; do
+    if (( ${branches[(Ie)$branch]} )); then
+      echo "$branch"
+      return
+    fi
+  done
+
+  echo master
+  return 1
 }
 
 #