#531 add API to get default branch

CAMOBAP · CAMOBAP · commit 87cfbc358005 · 2021-07-15T14:30:16.000+02:00
Signed-off-by: Alex Bobrikovich &lt;abobrikovich@gmail.com&gt;
diff --git a/lib/git/base.rb b/lib/git/base.rb
@@ -592,6 +592,10 @@ def current_branch
       self.lib.branch_current
     end
 
+    # @return [String] name of default branch
+    def branch_default
+      self.lib.branch_default
+    end
   end
 
 end
diff --git a/lib/git/lib.rb b/lib/git/lib.rb
@@ -398,6 +398,20 @@ def branch_current
       branches_all.select { |b| b[1] }.first[0] rescue nil
     end
 
+    def branch_default
+      begin
+        full_name = command("symbolic-ref", "refs/remotes/origin/HEAD")
+        return full_name.gsub(%r{^refs/remotes/origin/}, "")
+      rescue Git::GitExecuteError
+        begin
+          full_name = command("symbolic-ref", "refs/origin/HEAD")
+          return full_name.gsub(%r{^refs/origin/}, "")
+        rescue Git::GitExecuteError
+          return command("symbolic-ref", "--short", "HEAD")
+        end
+      end
+    end
+
     def branch_contains(commit, branch_name="")
       command("branch",  [branch_name, "--contains", commit])
     end
diff --git a/tests/units/test_lib.rb b/tests/units/test_lib.rb
@@ -291,4 +291,9 @@ def test_show
     assert(@lib.show('gitsearch1', 'scott/text.txt') == "hello\nthis is\na file\nthat is\nput here\nto search one\nto search two\nnothing!\n")
   end
 
+  def test_branch_default
+    assert_equal(@lib.branch_default, "git_grep")
+    @lib.change_head_branch("master")
+    assert_equal(@lib.branch_default, "master")
+  end
 end
