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

Allow non-symbolic-ref HEAD
This commit is contained in:
Casey Lee 2019-01-17 14:52:20 -08:00 committed by GitHub
commit 19d1d0cfda
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 8 deletions

View File

@ -72,19 +72,26 @@ func findGitHead(file string) (string, error) {
}() }()
headBuffer := new(bytes.Buffer) headBuffer := new(bytes.Buffer)
_, err = headBuffer.ReadFrom(bufio.NewReader(headFile)) length, err := headBuffer.ReadFrom(bufio.NewReader(headFile))
if err != nil { if err != nil {
log.Error(err) log.Error(err)
} }
var ref string
if length <= 42 {
ref = string(headBuffer.Bytes()[:40])
} else {
head := make(map[string]string) head := make(map[string]string)
err = yaml.Unmarshal(headBuffer.Bytes(), head) err = yaml.Unmarshal(headBuffer.Bytes(), head)
if err != nil { if err != nil {
log.Error(err) 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 // FindGithubRepo get the repo