Minor fix for #6

lunny · Feb 11, 2016 · 3b40eae · 3b40eae
1 parent b924465
commit 3b40eae
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/utlis.go b/utlis.go
@@ -81,14 +81,12 @@ func filepathFromSHA1(rootdir, sha1 string) string {
 }
 
 func RefEndName(refStr string) string {
-	if strings.HasPrefix(refStr, "refs/heads/") {
-		// trim the "refs/heads/"
-		return refStr[len("refs/heads/"):]
+	if strings.HasPrefix(refStr, BRANCH_PREFIX) {
+		return refStr[len(BRANCH_PREFIX):]
 	}
 
-	if strings.HasPrefix(refStr, "refs/tags/") {
-		// trim the "refs/heads/"
-		return refStr[len("refs/tags/"):]
+	if strings.HasPrefix(refStr, TAG_PREFIX) {
+		return refStr[len(TAG_PREFIX):]
 	}
 
 	return refStr