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
|
package repository
import (
"context"
"fmt"
"os"
"gitlab.com/gitlab-org/gitaly/internal/git"
"gitlab.com/gitlab-org/gitaly/internal/gitalyssh"
"gitlab.com/gitlab-org/gitaly/proto/go/gitalypb"
"google.golang.org/grpc/codes"
"google.golang.org/grpc/status"
)
func (s *server) CreateFork(ctx context.Context, req *gitalypb.CreateForkRequest) (*gitalypb.CreateForkResponse, error) {
targetRepository := req.Repository
sourceRepository := req.SourceRepository
if sourceRepository == nil {
return nil, status.Errorf(codes.InvalidArgument, "CreateFork: empty SourceRepository")
}
if targetRepository == nil {
return nil, status.Errorf(codes.InvalidArgument, "CreateFork: empty Repository")
}
targetRepositoryFullPath, err := s.locator.GetPath(targetRepository)
if err != nil {
return nil, err
}
if info, err := os.Stat(targetRepositoryFullPath); err != nil {
if !os.IsNotExist(err) {
return nil, status.Errorf(codes.Internal, "CreateFork: check destination path: %v", err)
}
// directory does not exist, proceed
} else {
if !info.IsDir() {
return nil, status.Errorf(codes.InvalidArgument, "CreateFork: destination path exists")
}
if err := os.Remove(targetRepositoryFullPath); err != nil {
return nil, status.Errorf(codes.InvalidArgument, "CreateFork: destination directory is not empty")
}
}
if err := os.MkdirAll(targetRepositoryFullPath, 0770); err != nil {
return nil, status.Errorf(codes.Internal, "CreateFork: create dest dir: %v", err)
}
env, err := gitalyssh.UploadPackEnv(ctx, s.cfg, &gitalypb.SSHUploadPackRequest{Repository: sourceRepository})
if err != nil {
return nil, err
}
cmd, err := s.gitCmdFactory.NewWithoutRepo(ctx,
git.SubCmd{
Name: "clone",
Flags: []git.Option{
git.Flag{Name: "--bare"},
git.Flag{Name: "--no-local"},
},
PostSepArgs: []string{
fmt.Sprintf("%s:%s", gitalyssh.GitalyInternalURL, sourceRepository.RelativePath),
targetRepositoryFullPath,
},
},
git.WithEnv(env...),
git.WithRefTxHook(ctx, req.Repository, s.cfg),
)
if err != nil {
return nil, status.Errorf(codes.Internal, "CreateFork: clone cmd start: %v", err)
}
if err := cmd.Wait(); err != nil {
return nil, status.Errorf(codes.Internal, "CreateFork: clone cmd wait: %v", err)
}
if err := s.removeOriginInRepo(ctx, targetRepository); err != nil {
return nil, status.Errorf(codes.Internal, "CreateFork: %v", err)
}
// CreateRepository is harmless on existing repositories with the side effect that it creates the hook symlink.
if _, err := s.CreateRepository(ctx, &gitalypb.CreateRepositoryRequest{Repository: targetRepository}); err != nil {
return nil, status.Errorf(codes.Internal, "CreateFork: create hooks failed: %v", err)
}
return &gitalypb.CreateForkResponse{}, nil
}
|