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 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127
|
//go:build static && system_libgit2
package main
import (
"context"
"encoding/gob"
"errors"
"flag"
"fmt"
git "github.com/libgit2/git2go/v34"
"gitlab.com/gitlab-org/gitaly/v16/cmd/gitaly-git2go/git2goutil"
"gitlab.com/gitlab-org/gitaly/v16/internal/git2go"
)
type cherryPickSubcommand struct{}
func (cmd *cherryPickSubcommand) Flags() *flag.FlagSet {
return flag.NewFlagSet("cherry-pick", flag.ExitOnError)
}
func (cmd *cherryPickSubcommand) Run(ctx context.Context, decoder *gob.Decoder, encoder *gob.Encoder) error {
var request git2go.CherryPickCommand
if err := decoder.Decode(&request); err != nil {
return err
}
commitID, err := cmd.cherryPick(ctx, &request)
return encoder.Encode(git2go.Result{
CommitID: commitID,
Err: git2go.SerializableError(err),
})
}
func (cmd *cherryPickSubcommand) verify(ctx context.Context, r *git2go.CherryPickCommand) error {
if r.Repository == "" {
return errors.New("missing repository")
}
if r.CommitterName == "" {
return errors.New("missing committer name")
}
if r.CommitterMail == "" {
return errors.New("missing committer mail")
}
if r.CommitterDate.IsZero() {
return errors.New("missing committer date")
}
if r.Message == "" {
return errors.New("missing message")
}
if r.Ours == "" {
return errors.New("missing ours")
}
if r.Commit == "" {
return errors.New("missing commit")
}
return nil
}
func (cmd *cherryPickSubcommand) cherryPick(ctx context.Context, r *git2go.CherryPickCommand) (string, error) {
if err := cmd.verify(ctx, r); err != nil {
return "", err
}
repo, err := git2goutil.OpenRepository(r.Repository)
if err != nil {
return "", fmt.Errorf("could not open repository: %w", err)
}
defer repo.Free()
ours, err := lookupCommit(repo, r.Ours)
if err != nil {
return "", fmt.Errorf("ours commit lookup: %w", err)
}
pick, err := lookupCommit(repo, r.Commit)
if err != nil {
return "", fmt.Errorf("commit lookup: %w", err)
}
opts, err := git.DefaultCherrypickOptions()
if err != nil {
return "", fmt.Errorf("could not get default cherry-pick options: %w", err)
}
opts.Mainline = r.Mainline
index, err := repo.CherrypickCommit(pick, ours, opts)
if err != nil {
return "", fmt.Errorf("could not cherry-pick commit: %w", err)
}
if index.HasConflicts() {
conflictingFiles, err := getConflictingFiles(index)
if err != nil {
return "", fmt.Errorf("getting conflicting files: %w", err)
}
return "", git2go.ConflictingFilesError{
ConflictingFiles: conflictingFiles,
}
}
treeOID, err := index.WriteTreeTo(repo)
if err != nil {
return "", fmt.Errorf("could not write tree: %w", err)
}
tree, err := repo.LookupTree(treeOID)
if err != nil {
return "", fmt.Errorf("lookup tree: %w", err)
}
if treeOID.Equal(ours.TreeId()) {
return "", git2go.EmptyError{}
}
committer := git.Signature(git2go.NewSignature(r.CommitterName, r.CommitterMail, r.CommitterDate))
commitID, err := git2goutil.NewCommitSubmitter(repo, r.SigningKey).
Commit(pick.Author(), &committer, git.MessageEncodingUTF8, r.Message, tree, ours)
if err != nil {
return "", fmt.Errorf("create not create cherry-pick commit: %w", err)
}
return commitID.String(), nil
}
|