Add SetDefaultBranch method (#626)

Co-authored-by: Dev Agent <[email protected]>

Author: QinYuuuu

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

@@ -176,3 +176,28 @@ func (c *Client) CreateBranch(ctx context.Context, req gitserver.CreateBranchReq
 	}
 	return nil
 }
+
+func (c *Client) SetDefaultBranch(ctx context.Context, req gitserver.SetDefaultBranchReq) error {
+	repoType := fmt.Sprintf("%ss", string(req.RepoType))
+	relativePath, err := c.BuildRelativePath(ctx, req.RepoType, req.Namespace, req.Name)
+	if err != nil {
+		return err
+	}
+	_, err = c.repoClient.WriteRef(ctx, &gitalypb.WriteRefRequest{
+		Repository: &gitalypb.Repository{
+			StorageName:  c.config.GitalyServer.Storage,
+			RelativePath: relativePath,
+			GlRepository: filepath.Join(repoType, req.Namespace, req.Name),
+		},
+		Ref:      []byte("HEAD"),
+		Revision: []byte("refs/heads/" + req.BranchName),
+	})
+	if err != nil {
+		return errorx.SetDefaultBranchFailed(err, errorx.Ctx().
+			Set("repo_type", req.RepoType).
+			Set("path", relativePath).
+			Set("branch", req.BranchName),
+		)
+	}
+	return nil
+}

builder/git/gitserver/gitaly/branch.go

@@ -66,3 +66,7 @@ func (c *Client) DeleteRepoBranch(ctx context.Context, req gitserver.DeleteBranc
 func (c *Client) CreateBranch(ctx context.Context, req gitserver.CreateBranchReq) error {
 	return nil
 }
+
+func (c *Client) SetDefaultBranch(ctx context.Context, req gitserver.SetDefaultBranchReq) error {
+	return nil
+}

builder/git/gitserver/gitea/branch.go

@@ -30,6 +30,7 @@ type GitServer interface {
 	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
+	SetDefaultBranch(ctx context.Context, req SetDefaultBranchReq) error
 	GetRepoCommits(ctx context.Context, req GetRepoCommitsReq) ([]types.Commit, *types.RepoPageOpts, error)
 	GetRepoLastCommit(ctx context.Context, req GetRepoLastCommitReq) (*types.Commit, error)
 	GetSingleCommit(ctx context.Context, req GetRepoLastCommitReq) (*types.CommitResponse, error)

builder/git/gitserver/interface.go

@@ -108,6 +108,13 @@ type CreateBranchReq struct {
 	RepoType   types.RepositoryType `json:"repo_type"`
 }
 
+type SetDefaultBranchReq struct {
+	Namespace  string               `json:"namespace"`
+	Name       string               `json:"name"`
+	BranchName string               `json:"branch_name"`
+	RepoType   types.RepositoryType `json:"repo_type"`
+}
+
 type GetBranchReq struct {
 	Namespace string               `json:"namespace"`
 	Name      string               `json:"name"`

builder/git/gitserver/types.go

@@ -20,6 +20,7 @@ const (
 	gitBranchNotFound
 	gitDeleteBranchFailed
 	gitCreateBranchFailed
+	gitSetDefaultBranchFailed
 	gitFileNotFound
 	gitUploadFailed
 	gitDownloadFailed
@@ -565,6 +566,15 @@ func CreateBranchFailed(err error, ctx context) error {
 	}
 }
 
+func SetDefaultBranchFailed(err error, ctx context) error {
+	return CustomError{
+		prefix:  errGitPrefix,
+		code:    gitSetDefaultBranchFailed,
+		err:     err,
+		context: ctx,
+	}
+}
+
 func GitFileNotFound(err error, ctx context) error {
 	return CustomError{
 		prefix:  errGitPrefix,