legit/git/tree.go

67 lines
1.1 KiB
Go
Raw Normal View History

2022-12-11 15:46:42 +00:00
package git
import (
"fmt"
"github.com/go-git/go-git/v5/plumbing/object"
)
func (g *GitRepo) FileTree(path string) ([]NiceTree, error) {
c, err := g.r.CommitObject(g.h)
if err != nil {
return nil, fmt.Errorf("commit object: %w", err)
}
files := []NiceTree{}
tree, err := c.Tree()
if err != nil {
return nil, fmt.Errorf("file tree: %w", err)
}
if path == "" {
files = makeNiceTree(tree)
2022-12-11 15:46:42 +00:00
} else {
o, err := tree.FindEntry(path)
if err != nil {
return nil, err
}
if !o.Mode.IsFile() {
subtree, err := tree.Tree(path)
if err != nil {
return nil, err
}
files = makeNiceTree(subtree)
2022-12-11 15:46:42 +00:00
}
}
return files, nil
}
// A nicer git tree representation.
type NiceTree struct {
Name string
Mode string
Size int64
IsFile bool
IsSubtree bool
}
func makeNiceTree(t *object.Tree) []NiceTree {
2022-12-11 15:46:42 +00:00
nts := []NiceTree{}
for _, e := range t.Entries {
2022-12-11 15:46:42 +00:00
mode, _ := e.Mode.ToOSFileMode()
sz, _ := t.Size(e.Name)
2022-12-11 15:46:42 +00:00
nts = append(nts, NiceTree{
Name: e.Name,
Mode: mode.String(),
IsFile: e.Mode.IsFile(),
Size: sz,
2022-12-11 15:46:42 +00:00
})
}
return nts
}