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
|
package remote
import (
"bytes"
"fmt"
"io/ioutil"
"strings"
"golang.org/x/net/context"
"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/rubyserver"
)
// AddRemote adds a remote to the repository
func (s *server) AddRemote(ctx context.Context, req *gitalypb.AddRemoteRequest) (*gitalypb.AddRemoteResponse, error) {
if err := validateAddRemoteRequest(req); err != nil {
return nil, status.Errorf(codes.InvalidArgument, "AddRemote: %v", err)
}
client, err := s.RemoteServiceClient(ctx)
if err != nil {
return nil, err
}
clientCtx, err := rubyserver.SetHeaders(ctx, req.GetRepository())
if err != nil {
return nil, err
}
return client.AddRemote(clientCtx, req)
}
func validateAddRemoteRequest(req *gitalypb.AddRemoteRequest) error {
if strings.TrimSpace(req.GetName()) == "" {
return fmt.Errorf("empty remote name")
}
if req.GetUrl() == "" {
return fmt.Errorf("empty remote url")
}
return nil
}
// RemoveRemote removes the given remote
func (s *server) RemoveRemote(ctx context.Context, req *gitalypb.RemoveRemoteRequest) (*gitalypb.RemoveRemoteResponse, error) {
if err := validateRemoveRemoteRequest(req); err != nil {
return nil, status.Errorf(codes.InvalidArgument, "AddRemote: %v", err)
}
client, err := s.RemoteServiceClient(ctx)
if err != nil {
return nil, err
}
clientCtx, err := rubyserver.SetHeaders(ctx, req.GetRepository())
if err != nil {
return nil, err
}
return client.RemoveRemote(clientCtx, req)
}
func (s *server) FindRemoteRepository(ctx context.Context, req *gitalypb.FindRemoteRepositoryRequest) (*gitalypb.FindRemoteRepositoryResponse, error) {
if req.GetRemote() == "" {
return nil, status.Error(codes.InvalidArgument, "FindRemoteRepository: empty remote can't be checked.")
}
cmd, err := git.CommandWithoutRepo(ctx, "ls-remote", req.GetRemote(), "HEAD")
if err != nil {
return nil, status.Errorf(codes.Internal, "error executing git command: %s", err)
}
output, err := ioutil.ReadAll(cmd)
if err != nil {
return nil, status.Errorf(codes.Internal, "unable to read stdout: %s", err)
}
if err := cmd.Wait(); err != nil {
return &gitalypb.FindRemoteRepositoryResponse{Exists: false}, nil
}
// The output of a successful command is structured like
// Regexp would've read better, but this is faster
// 58fbff2e0d3b620f591a748c158799ead87b51cd HEAD
fields := bytes.Fields(output)
match := len(fields) == 2 && len(fields[0]) == 40 && string(fields[1]) == "HEAD"
return &gitalypb.FindRemoteRepositoryResponse{Exists: match}, nil
}
func validateRemoveRemoteRequest(req *gitalypb.RemoveRemoteRequest) error {
if req.GetName() == "" {
return fmt.Errorf("empty remote name")
}
return nil
}
|