aboutsummaryrefslogtreecommitdiff
path: root/internal/testgit/tree.go
blob: 501c7949ce3c10239f12afd98dcb314e03ca4205 (about) (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
package testgit

import (
	"bytes"
	"fmt"
	"strings"
	"testing"

	"lindenii.org/go/furgit/object/id"
	"lindenii.org/go/furgit/object/typ"
)

// TreeEntry is one entry of a tree built by [Repo.MkTree]
// or reported by [Repo.LsTree].
type TreeEntry struct {
	Mode string
	Type typ.Type
	OID  id.ObjectID
	Name string
}

// MkTree builds a tree object from entries and returns its object ID.
func (repo *Repo) MkTree(tb testing.TB, entries []TreeEntry) (id.ObjectID, error) {
	tb.Helper()

	var stdin bytes.Buffer
	for _, entry := range entries {
		fmt.Fprintf(&stdin, "%s %s %s\t%s\n", entry.Mode, entry.Type.Name(), entry.OID.String(), entry.Name)
	}

	stdout, err := repo.run(tb, &stdin, "git", "mktree")
	if err != nil {
		return id.ObjectID{}, fmt.Errorf("mktree: %w", err)
	}

	treeID, err := repo.objectFormat.FromString(strings.TrimSuffix(string(stdout), "\n"))
	if err != nil {
		return id.ObjectID{}, fmt.Errorf("parse git mktree output %q: %w", string(stdout), err)
	}

	return treeID, nil
}

// LsTree lists the top-level entries of a tree object in Git tree order.
func (repo *Repo) LsTree(tb testing.TB, oid id.ObjectID) ([]TreeEntry, error) {
	tb.Helper()

	stdout, err := repo.run(tb, nil, "git", "ls-tree", "-z", "--full-tree", "--end-of-options", oid.String())
	if err != nil {
		return nil, fmt.Errorf("ls-tree: %w", err)
	}

	var entries []TreeEntry

	for record := range bytes.SplitSeq(stdout, []byte{0}) {
		if len(record) == 0 {
			continue
		}

		meta, name, found := bytes.Cut(record, []byte{'\t'})
		if !found {
			return nil, fmt.Errorf("ls-tree: record %q has no tab separator", record)
		}

		fields := bytes.SplitN(meta, []byte{' '}, 3)
		if len(fields) != 3 {
			return nil, fmt.Errorf("ls-tree: record %q has malformed metadata", record)
		}

		ty, err := typ.Parse(string(fields[1]))
		if err != nil {
			return nil, fmt.Errorf("ls-tree: record %q: parse type: %w", record, err)
		}

		entryOID, err := repo.objectFormat.FromString(string(fields[2]))
		if err != nil {
			return nil, fmt.Errorf("ls-tree: record %q: parse oid: %w", record, err)
		}

		entries = append(entries, TreeEntry{
			Mode: string(fields[0]),
			Type: ty,
			OID:  entryOID,
			Name: string(name),
		})
	}

	return entries, nil
}