Add functions for checking which tags/branches contain a ref

Change-Id: Ic6e613c21ed26c528ad7c75f41af01d7552729fd

1 files changed, 26 insertions, 0 deletions

diff --git a/morphlib/gitdir.py b/morphlib/gitdir.py
index 03640a22..1c286720 100644
--- a/morphlib/gitdir.py
+++ b/morphlib/gitdir.py
@@ -681,6 +681,32 @@ class GitDirectory(object):
         if not morphlib.git.is_valid_sha1(string):
             raise ExpectedSha1Error(string)
 
+    def _check_ref_exists(self, ref):
+        self._rev_parse('%s^{commit}' % ref)
+
+    def _gitcmd_output_list(self, *args):
+        output = morphlib.git.gitcmd(self._runcmd, *args)
+        separated = [l.strip() for l in output.splitlines()]
+        prefix = '* '
+        for i, l in enumerate(separated):
+            if l.startswith(prefix):
+                separated[i] = l[len(prefix):]
+        return separated
+
+    def tags_containing_sha1(self, ref): # pragma: no cover
+        self._check_is_sha1(ref)
+        self._check_ref_exists(ref)
+
+        args = ['tag', '--contains', ref]
+        return self._gitcmd_output_list(*args)
+
+    def branches_containing_sha1(self, ref):
+        self._check_is_sha1(ref)
+        self._check_ref_exists(ref)
+
+        args = ['branch', '--contains', ref]
+        return self._gitcmd_output_list(*args)
+
     def _update_ref(self, ref_args, message):
         args = ['update-ref']
         # No test coverage, since while this functionality is useful,