Add CreateBranch method (#623)

Co-authored-by: Dev Agent <dev-agent@example.com>

Author: QinYuuuu

_mocks/opencsg.com/csghub-server/builder/git/gitserver/mock_GitServer.go

@@ -130,3 +130,49 @@ func (c *Client) DeleteRepoBranch(ctx context.Context, req gitserver.DeleteBranc
 
 	return nil
 }
+
+func (c *Client) CreateBranch(ctx context.Context, req gitserver.CreateBranchReq) error {
+	repoType := fmt.Sprintf("%ss", string(req.RepoType))
+	relativePath, err := c.BuildRelativePath(ctx, req.RepoType, req.Namespace, req.Name)
+	if err != nil {
+		return err
+	}
+	client, err := c.refClient.UpdateReferences(ctx)
+	if err != nil {
+		return err
+	}
+	createBranchReq := &gitalypb.UpdateReferencesRequest{
+		Repository: &gitalypb.Repository{
+			StorageName:  c.config.GitalyServer.Storage,
+			RelativePath: relativePath,
+			GlRepository: filepath.Join(repoType, req.Namespace, req.Name),
+		},
+		Updates: []*gitalypb.UpdateReferencesRequest_Update{
+			{
+				Reference:   []byte("refs/heads/" + req.BranchName),
+				NewObjectId: []byte(req.CommitID),
+			},
+		},
+	}
+
+	err = client.Send(createBranchReq)
+	if err != nil {
+		return errorx.CreateBranchFailed(err, errorx.Ctx().
+			Set("repo_type", req.RepoType).
+			Set("path", relativePath).
+			Set("branch", req.BranchName).
+			Set("commit_id", req.CommitID),
+		)
+	}
+
+	_, err = client.CloseAndRecv()
+	if err != nil {
+		return errorx.CreateBranchFailed(err, errorx.Ctx().
+			Set("repo_type", req.RepoType).
+			Set("path", relativePath).
+			Set("branch", req.BranchName).
+			Set("commit_id", req.CommitID),
+		)
+	}
+	return nil
+}

builder/git/gitserver/gitaly/branch.go

@@ -62,3 +62,7 @@ func (c *Client) GetRepoBranchByName(ctx context.Context, req gitserver.GetBranc
 func (c *Client) DeleteRepoBranch(ctx context.Context, req gitserver.DeleteBranchReq) error {
 	return nil
 }
+
+func (c *Client) CreateBranch(ctx context.Context, req gitserver.CreateBranchReq) error {
+	return nil
+}

builder/git/gitserver/gitea/branch.go

@@ -26,6 +26,7 @@ type GitServer interface {
 	CreateRepo(ctx context.Context, req CreateRepoReq) (*CreateRepoResp, error)
 	UpdateRepo(ctx context.Context, req UpdateRepoReq) (*CreateRepoResp, error)
 	DeleteRepo(ctx context.Context, relativePath string) error
+	CreateBranch(ctx context.Context, req CreateBranchReq) error
 	GetRepoBranches(ctx context.Context, req GetBranchesReq) ([]types.Branch, error)
 	GetRepoBranchByName(ctx context.Context, req GetBranchReq) (*types.Branch, error)
 	DeleteRepoBranch(ctx context.Context, req DeleteBranchReq) error

builder/git/gitserver/interface.go

@@ -100,6 +100,14 @@ type GetBranchesReq struct {
 	RepoType  types.RepositoryType `json:"repo_type"`
 }
 
+type CreateBranchReq struct {
+	Namespace  string               `json:"namespace"`
+	Name       string               `json:"name"`
+	BranchName string               `json:"branch_name"`
+	CommitID   string               `json:"commit_id"`
+	RepoType   types.RepositoryType `json:"repo_type"`
+}
+
 type GetBranchReq struct {
 	Namespace string               `json:"namespace"`
 	Name      string               `json:"name"`

builder/git/gitserver/types.go

@@ -19,6 +19,7 @@ const (
 	gitFindBranchFailed
 	gitBranchNotFound
 	gitDeleteBranchFailed
+	gitCreateBranchFailed
 	gitFileNotFound
 	gitUploadFailed
 	gitDownloadFailed
@@ -555,6 +556,15 @@ func DeleteBranchFailed(err error, ctx context) error {
 	}
 }
 
+func CreateBranchFailed(err error, ctx context) error {
+	return CustomError{
+		prefix:  errGitPrefix,
+		code:    gitCreateBranchFailed,
+		err:     err,
+		context: ctx,
+	}
+}
+
 func GitFileNotFound(err error, ctx context) error {
 	return CustomError{
 		prefix:  errGitPrefix,