Something went wrong. Try again.
forked from tangled.org/core
Something went wrong. Try again.
6.2 kB · 268 lines
Go
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269package git
import ( "bytes" "fmt" "log" "os" "os/exec" "slices" "strings"
"github.com/bluekeyes/go-gitdiff/gitdiff" "github.com/go-git/go-git/v5/plumbing" "github.com/go-git/go-git/v5/plumbing/object" "tangled.sh/tangled.sh/core/patchutil" "tangled.sh/tangled.sh/core/types")
func (g *GitRepo) Diff() (*types.NiceDiff, error) { c, err := g.r.CommitObject(g.h) if err != nil { return nil, fmt.Errorf("commit object: %w", err) }
patch := &object.Patch{} commitTree, err := c.Tree() parent := &object.Commit{} if err == nil { parentTree := &object.Tree{} if c.NumParents() != 0 { parent, err = c.Parents().Next() if err == nil { parentTree, err = parent.Tree() if err == nil { patch, err = parentTree.Patch(commitTree) if err != nil { return nil, fmt.Errorf("patch: %w", err) } } } } else { patch, err = parentTree.Patch(commitTree) if err != nil { return nil, fmt.Errorf("patch: %w", err) } } }
diffs, _, err := gitdiff.Parse(strings.NewReader(patch.String())) if err != nil { log.Println(err) }
nd := types.NiceDiff{} for _, d := range diffs { ndiff := types.Diff{} ndiff.Name.New = d.NewName ndiff.Name.Old = d.OldName ndiff.IsBinary = d.IsBinary ndiff.IsNew = d.IsNew ndiff.IsDelete = d.IsDelete ndiff.IsCopy = d.IsCopy ndiff.IsRename = d.IsRename
for _, tf := range d.TextFragments { ndiff.TextFragments = append(ndiff.TextFragments, *tf) for _, l := range tf.Lines { switch l.Op { case gitdiff.OpAdd: nd.Stat.Insertions += 1 case gitdiff.OpDelete: nd.Stat.Deletions += 1 } } }
nd.Diff = append(nd.Diff, ndiff) }
nd.Stat.FilesChanged = len(diffs) nd.Commit.This = c.Hash.String()
if parent.Hash.IsZero() { nd.Commit.Parent = "" } else { nd.Commit.Parent = parent.Hash.String() } nd.Commit.Author = c.Author nd.Commit.Message = c.Message
return &nd, nil}
func (g *GitRepo) DiffTree(commit1, commit2 *object.Commit) (*types.DiffTree, error) { tree1, err := commit1.Tree() if err != nil { return nil, err }
tree2, err := commit2.Tree() if err != nil { return nil, err }
diff, err := object.DiffTree(tree1, tree2) if err != nil { return nil, err }
patch, err := diff.Patch() if err != nil { return nil, err }
diffs, _, err := gitdiff.Parse(strings.NewReader(patch.String())) if err != nil { return nil, err }
return &types.DiffTree{ Rev1: commit1.Hash.String(), Rev2: commit2.Hash.String(), Patch: patch.String(), Diff: diffs, }, nil}
// FormatPatch generates a git-format-patch output between two commits,// and returns the raw format-patch series, a parsed FormatPatch and an error.func (g *GitRepo) formatSinglePatch(base, commit2 plumbing.Hash, extraArgs ...string) (string, *types.FormatPatch, error) { var stdout bytes.Buffer
args := []string{ "-C", g.path, "format-patch", fmt.Sprintf("%s..%s", base.String(), commit2.String()), "--stdout", } args = append(args, extraArgs...)
cmd := exec.Command("git", args...) cmd.Stdout = &stdout cmd.Stderr = os.Stderr err := cmd.Run() if err != nil { return "", nil, err }
formatPatch, err := patchutil.ExtractPatches(stdout.String()) if err != nil { return "", nil, err }
if len(formatPatch) > 1 { return "", nil, fmt.Errorf("running format-patch on single commit produced more than on patch") }
return stdout.String(), &formatPatch[0], nil}
func (g *GitRepo) MergeBase(commit1, commit2 *object.Commit) (*object.Commit, error) { isAncestor, err := commit1.IsAncestor(commit2) if err != nil { return nil, err }
if isAncestor { return commit1, nil }
mergeBase, err := commit1.MergeBase(commit2) if err != nil { return nil, err }
if len(mergeBase) == 0 { return nil, fmt.Errorf("failed to find a merge-base") }
return mergeBase[0], nil}
func (g *GitRepo) ResolveRevision(revStr string) (*object.Commit, error) { rev, err := g.r.ResolveRevision(plumbing.Revision(revStr)) if err != nil { return nil, fmt.Errorf("resolving revision %s: %w", revStr, err) }
commit, err := g.r.CommitObject(*rev) if err != nil {
return nil, fmt.Errorf("getting commit for %s: %w", revStr, err) }
return commit, nil}
func (g *GitRepo) commitsBetween(newCommit, oldCommit *object.Commit) ([]*object.Commit, error) { var commits []*object.Commit current := newCommit
for { if current.Hash == oldCommit.Hash { break }
commits = append(commits, current)
if len(current.ParentHashes) == 0 { return nil, fmt.Errorf("old commit %s not found in history of new commit %s", oldCommit.Hash, newCommit.Hash) }
parent, err := current.Parents().Next() if err != nil { return nil, fmt.Errorf("error getting parent: %w", err) }
current = parent }
return commits, nil}
func (g *GitRepo) FormatPatch(base, commit2 *object.Commit) (string, []types.FormatPatch, error) { // get list of commits between commir2 and base commits, err := g.commitsBetween(commit2, base) if err != nil { return "", nil, fmt.Errorf("failed to get commits: %w", err) }
// reverse the list so we start from the oldest one and go up to the most recent one slices.Reverse(commits)
var allPatchesContent strings.Builder var allPatches []types.FormatPatch
for _, commit := range commits { changeId := "" if val, ok := commit.ExtraHeaders["change-id"]; ok { changeId = string(val) }
var parentHash plumbing.Hash if len(commit.ParentHashes) > 0 { parentHash = commit.ParentHashes[0] } else { parentHash = plumbing.NewHash("4b825dc642cb6eb9a060e54bf8d69288fbee4904") // git empty tree hash }
var additionalArgs []string if changeId != "" { additionalArgs = append(additionalArgs, "--add-header", fmt.Sprintf("Change-Id: %s", changeId)) }
stdout, patch, err := g.formatSinglePatch(parentHash, commit.Hash, additionalArgs...) if err != nil { return "", nil, fmt.Errorf("failed to format patch for commit %s: %w", commit.Hash.String(), err) }
allPatchesContent.WriteString(stdout) allPatchesContent.WriteString("\n")
allPatches = append(allPatches, *patch) }
return allPatchesContent.String(), allPatches, nil}