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
|
package ref
import (
"bufio"
"context"
"errors"
"fmt"
"strings"
"gitlab.com/gitlab-org/gitaly/v14/internal/git"
"gitlab.com/gitlab-org/gitaly/v14/internal/helper"
"gitlab.com/gitlab-org/gitaly/v14/proto/go/gitalypb"
)
// FindRefName returns a ref that starts with the given prefix, if one exists.
// If there is more than one such ref there is no guarantee which one is
// returned or that the same one is returned on each call.
func (s *server) FindRefName(ctx context.Context, in *gitalypb.FindRefNameRequest) (*gitalypb.FindRefNameResponse, error) {
if in.CommitId == "" {
return nil, helper.ErrInvalidArgument(fmt.Errorf("empty commit sha"))
}
ref, err := s.findRefName(ctx, in.Repository, in.CommitId, string(in.Prefix))
if err != nil {
return nil, helper.ErrInternal(err)
}
return &gitalypb.FindRefNameResponse{Name: []byte(ref)}, nil
}
// We assume `repo` and `commitID` and `prefix` are non-empty
func (s *server) findRefName(ctx context.Context, repo *gitalypb.Repository, commitID, prefix string) (string, error) {
flags := []git.Option{
git.Flag{Name: "--format=%(refname)"},
git.Flag{Name: "--count=1"},
}
subCmd := ForEachRefCmd{PostArgFlags: []git.Option{
git.ValueFlag{Name: "--contains", Value: commitID},
}}
subCmd.Name = "for-each-ref"
subCmd.Flags = flags
subCmd.Args = []string{prefix}
cmd, err := s.gitCmdFactory.New(ctx, repo, subCmd)
if err != nil {
return "", err
}
scanner := bufio.NewScanner(cmd)
scanner.Scan()
if err := scanner.Err(); err != nil {
return "", err
}
refName := scanner.Text()
if err := cmd.Wait(); err != nil {
// We're suppressing the error since invalid commits isn't an error
// according to Rails
return "", nil
}
// Trailing spaces are not allowed per the documentation
// https://www.kernel.org/pub/software/scm/git/docs/git-check-ref-format.html
return strings.TrimSpace(refName), nil
}
// ForEachRefCmd is a command specialized for for-each-ref
type ForEachRefCmd struct {
git.SubCmd
PostArgFlags []git.Option
}
var (
// ErrOnlyForEachRefAllowed indicates a command other than for-each-ref is being used with ForEachRefCmd
ErrOnlyForEachRefAllowed = errors.New("only for-each-ref allowed")
// ErrNoPostSeparatorArgsAllowed indicates post separator args exist when none are allowed
ErrNoPostSeparatorArgsAllowed = errors.New("post separator args not allowed")
)
// CommandArgs validates and returns the flags and arguments for the for-each-ref command
func (f ForEachRefCmd) CommandArgs() ([]string, error) {
if f.Name != "for-each-ref" {
return nil, ErrOnlyForEachRefAllowed
}
args, err := f.SubCmd.CommandArgs()
if err != nil {
return nil, err
}
var postArgFlags []string
for _, o := range f.PostArgFlags {
args, err := o.OptionArgs()
if err != nil {
return nil, err
}
postArgFlags = append(postArgFlags, args...)
}
if len(f.SubCmd.PostSepArgs) > 0 {
return nil, ErrNoPostSeparatorArgsAllowed
}
return append(args, postArgFlags...), nil
}
|
