Browse Source

feat(git): support trunk branches in `git_main_branch` (#9417)

Alisson B. Santos 4 years ago
parent
commit
51772732f5
1 changed files with 8 additions and 5 deletions
  1. 8 5
      plugins/git/git.plugin.zsh

+ 8 - 5
plugins/git/git.plugin.zsh

@@ -31,11 +31,14 @@ function work_in_progress() {
 
 
 # Check if main exists and use instead of master
 # Check if main exists and use instead of master
 function git_main_branch() {
 function git_main_branch() {
-  if [[ -n "$(git branch --list main)" ]]; then
-    echo main
-  else
-    echo master
-  fi
+  local branch
+  for branch in main trunk; do
+    if command git show-ref -q --verify refs/heads/$branch; then
+      echo $branch
+      return
+    fi
+  done
+  echo master
 }
 }
 
 
 #
 #