1package main
 2
 3import (
 4	"os"
 5	"path/filepath"
 6
 7	git "github.com/libgit2/git2go/v33"
 8)
 9
10// Recursively traverse up until we find .git
11// and return the git repo path.
12func getGitDir(cwd string) string {
13	for {
14		dirs, _ := os.ReadDir(cwd)
15		for _, d := range dirs {
16			if ".git" == d.Name() {
17				return cwd
18			} else if cwd == "/" {
19				return ""
20			}
21		}
22		cwd = filepath.Dir(cwd)
23	}
24}
25
26// Returns the current git branch or current ref sha.
27func getGitBranch(repo *git.Repository) string {
28	ref, _ := repo.Head()
29	// Quick hack to fix crash when ref is nil;
30	// i.e., new repo with no commits.
31	if ref == nil {
32		return "no commit"
33	}
34	if ref.IsBranch() {
35		name, _ := ref.Branch().Name()
36		return name
37	} else {
38		return ref.Target().String()[:8]
39	}
40}
41
42// Returns • if clean, else ×.
43func getGitStatus(repo *git.Repository, clean, dirty string) string {
44	sl, _ := repo.StatusList(&git.StatusOptions{
45		Show:  git.StatusShowIndexAndWorkdir,
46		Flags: git.StatusOptIncludeUntracked,
47	})
48	n, _ := sl.EntryCount()
49	if n != 0 {
50		return dirty
51	} else {
52		return clean
53	}
54}