plumbing: Reference, support slash separated branch (#302)

1 files changed, 21 insertions, 2 deletions

diff --git a/plumbing/reference.go b/plumbing/reference.go
index 798c3dd..8fa103e 100644
--- a/plumbing/reference.go
+++ b/plumbing/reference.go
@@ -16,6 +16,16 @@ const (
 )
 
 var (
+	refPrefixes = []string{
+		refHeadPrefix,
+		refTagPrefix,
+		refRemotePrefix,
+		refNotePrefix,
+		refPrefix,
+	}
+)
+
+var (
 	ErrReferenceNotFound = errors.New("reference not found")
 )
 
@@ -50,8 +60,17 @@ func (r ReferenceName) String() string {
 
 // Short returns the short name of a ReferenceName
 func (r ReferenceName) Short() string {
-	parts := strings.Split(string(r), "/")
-	return parts[len(parts)-1]
+	return r.removeRefPrefix()
+}
+
+// Instead of hardcoding a number of components, we should remove the prefixes
+// refHeadPrefix, refTagPrefix, refRemotePrefix, refNotePrefix and refPrefix
+func (r ReferenceName) removeRefPrefix() string {
+	s := string(r)
+	for _, prefix := range refPrefixes {
+		s = strings.TrimPrefix(s, prefix)
+	}
+	return s
 }
 
 const (