Merge pull request #10 from Code0x58/non-symbolic-ref-HEAD

Allow non-symbolic-ref HEAD
nektos · Jan 17, 2019 · 19d1d0c · 19d1d0c
2 parents 949dc46 + 1ead030
commit 19d1d0c
Showing 1 changed file with 14 additions and 7 deletions.
diff --git a/common/git.go b/common/git.go
@@ -72,19 +72,26 @@ func findGitHead(file string) (string, error) {
 	}()
 
 	headBuffer := new(bytes.Buffer)
-	_, err = headBuffer.ReadFrom(bufio.NewReader(headFile))
+	length, err := headBuffer.ReadFrom(bufio.NewReader(headFile))
 	if err != nil {
 		log.Error(err)
 	}
-	head := make(map[string]string)
-	err = yaml.Unmarshal(headBuffer.Bytes(), head)
-	if err != nil {
-		log.Error(err)
+
+	var ref string
+	if length <= 42 {
+		ref = string(headBuffer.Bytes()[:40])
+	} else {
+		head := make(map[string]string)
+		err = yaml.Unmarshal(headBuffer.Bytes(), head)
+		if err != nil {
+			log.Error(err)
+		}
+		ref = head["ref"]
 	}
 
-	log.Debugf("HEAD points to '%s'", head["ref"])
+	log.Debugf("HEAD points to '%s'", ref)
 
-	return head["ref"], nil
+	return ref, nil
 }
 
 // FindGithubRepo get the repo