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
|
package repository
import (
"bytes"
"errors"
"google.golang.org/grpc/codes"
"google.golang.org/grpc/status"
"gitlab.com/gitlab-org/gitaly-proto/go/gitalypb"
"gitlab.com/gitlab-org/gitaly/internal/git"
"gitlab.com/gitlab-org/gitaly/internal/helper"
"gitlab.com/gitlab-org/gitaly/internal/helper/lines"
)
const surroundContext = "2"
var contentDelimiter = []byte("--\n")
func (s *server) SearchFilesByContent(req *gitalypb.SearchFilesByContentRequest, stream gitalypb.RepositoryService_SearchFilesByContentServer) error {
if err := validateSearchFilesRequest(req); err != nil {
return helper.DecorateError(codes.InvalidArgument, err)
}
repo := req.GetRepository()
if repo == nil {
return status.Errorf(codes.InvalidArgument, "SearchFilesByContent: empty Repository")
}
ctx := stream.Context()
cmd, err := git.Command(ctx, repo, "grep",
"--ignore-case",
"-I", // Don't match binary, there is no long-name for this one
"--line-number",
"--null",
"--before-context", surroundContext,
"--after-context", surroundContext,
"--extended-regexp",
"-e", // next arg is pattern, keep this last
req.GetQuery(),
string(req.GetRef()),
)
if err != nil {
return status.Errorf(codes.Internal, "SearchFilesByContent: cmd start failed: %v", err)
}
var (
buf []byte
matches [][]byte
)
reader := func(objs [][]byte) error {
for _, obj := range objs {
obj = append(obj, '\n')
if bytes.Equal(obj, contentDelimiter) {
matches = append(matches, buf)
buf = nil
} else {
buf = append(buf, obj...)
}
}
if len(matches) > 0 {
err = stream.Send(&gitalypb.SearchFilesByContentResponse{Matches: matches})
matches = nil
return err
}
return nil
}
err = lines.Send(cmd, reader, []byte{'\n'})
if err != nil {
return helper.DecorateError(codes.Internal, err)
}
if len(buf) > 0 {
matches = append(matches, buf)
}
if len(matches) > 0 {
return stream.Send(&gitalypb.SearchFilesByContentResponse{Matches: matches})
}
return nil
}
func (s *server) SearchFilesByName(req *gitalypb.SearchFilesByNameRequest, stream gitalypb.RepositoryService_SearchFilesByNameServer) error {
if err := validateSearchFilesRequest(req); err != nil {
return helper.DecorateError(codes.InvalidArgument, err)
}
repo := req.GetRepository()
if repo == nil {
return status.Errorf(codes.InvalidArgument, "SearchFilesByName: empty Repository")
}
ctx := stream.Context()
cmd, err := git.Command(ctx, repo, "ls-tree", "--full-tree", "--name-status", "-r", string(req.GetRef()), req.GetQuery())
if err != nil {
return status.Errorf(codes.Internal, "SearchFilesByName: cmd start failed: %v", err)
}
lr := func(objs [][]byte) error {
return stream.Send(&gitalypb.SearchFilesByNameResponse{Files: objs})
}
return lines.Send(cmd, lr, []byte{'\n'})
}
type searchFilesRequest interface {
GetRef() []byte
GetQuery() string
}
func validateSearchFilesRequest(req searchFilesRequest) error {
if len(req.GetQuery()) == 0 {
return errors.New("no query given")
}
if len(req.GetRef()) == 0 {
return errors.New("no ref given")
}
return nil
}
|