MCP/operation/repo/branch.go

package repo

import (
	"context"
	"fmt"

	"gitea.com/gitea/gitea-mcp/pkg/annotation"
	"gitea.com/gitea/gitea-mcp/pkg/gitea"
	"gitea.com/gitea/gitea-mcp/pkg/params"
	"gitea.com/gitea/gitea-mcp/pkg/to"
	"gitea.com/gitea/gitea-mcp/pkg/tool"

	gitea_sdk "gitea.dev/sdk"
	"github.com/modelcontextprotocol/go-sdk/mcp"
)

// BranchTool holds the branch-related tools (scope "branch").
var BranchTool = tool.New("branch")

const (
	CreateBranchToolName = "create_branch"
	DeleteBranchToolName = "delete_branch"
	ListBranchesToolName = "list_branches"
)

var (
	CreateBranchTool = tool.NewDefinition(
		CreateBranchToolName,
		"Create a new branch in a repository, optionally from a specific source branch (defaults to the repository's default branch).",
		annotation.Write("Create a new branch"),
		tool.String("owner", tool.Required(), tool.Description(params.OwnerDesc)),
		tool.String("repo", tool.Required(), tool.Description(params.RepoDesc)),
		tool.String("branch", tool.Required()),
		tool.String("old_branch", tool.Description("source branch (default: repo default)")),
	)

	DeleteBranchTool = tool.NewDefinition(
		DeleteBranchToolName,
		"Permanently delete a branch from a repository. This action is destructive and cannot be undone.",
		annotation.Destructive("Delete a branch"),
		tool.String("owner", tool.Required(), tool.Description(params.OwnerDesc)),
		tool.String("repo", tool.Required(), tool.Description(params.RepoDesc)),
		tool.String("branch", tool.Required()),
	)

	ListBranchesTool = tool.NewDefinition(
		ListBranchesToolName,
		"List all branches in a repository, paginated.",
		annotation.ReadOnly("List repository branches"),
		tool.String("owner", tool.Required(), tool.Description(params.OwnerDesc)),
		tool.String("repo", tool.Required(), tool.Description(params.RepoDesc)),
		tool.Number("page", tool.Description(params.PageDesc), tool.Default(1)),
		tool.Number("per_page", tool.Description(params.PaginationDesc), tool.Default(30)),
	)
)

func init() {
	BranchTool.RegisterWrite(tool.ServerTool{
		Tool:    CreateBranchTool,
		Handler: CreateBranchFn,
	})
	BranchTool.RegisterWrite(tool.ServerTool{
		Tool:    DeleteBranchTool,
		Handler: DeleteBranchFn,
	})
	BranchTool.RegisterRead(tool.ServerTool{
		Tool:    ListBranchesTool,
		Handler: ListBranchesFn,
	})
}

type createBranchArgs struct {
	Owner     string `json:"owner" required:"true"`
	Repo      string `json:"repo" required:"true"`
	Branch    string `json:"branch" required:"true"`
	OldBranch string `json:"old_branch"`
}

func CreateBranchFn(ctx context.Context, args map[string]any) (*mcp.CallToolResult, error) {
	var in createBranchArgs
	if err := params.Bind(args, &in); err != nil {
		return to.ErrorResult(err)
	}

	client, err := gitea.ClientFromContext(ctx)
	if err != nil {
		return to.ErrorResult(fmt.Errorf("get gitea client err: %v", err))
	}
	_, _, err = client.Repositories.CreateBranch(ctx, in.Owner, in.Repo, gitea_sdk.CreateBranchOption{
		BranchName:    in.Branch,
		OldBranchName: in.OldBranch,
	})
	if err != nil {
		return to.ErrorResult(fmt.Errorf("create branch error: %v", err))
	}

	return to.TextResult("Branch Created")
}

type deleteBranchArgs struct {
	Owner  string `json:"owner" required:"true"`
	Repo   string `json:"repo" required:"true"`
	Branch string `json:"branch" required:"true"`
}

func DeleteBranchFn(ctx context.Context, args map[string]any) (*mcp.CallToolResult, error) {
	var in deleteBranchArgs
	if err := params.Bind(args, &in); err != nil {
		return to.ErrorResult(err)
	}
	client, err := gitea.ClientFromContext(ctx)
	if err != nil {
		return to.ErrorResult(fmt.Errorf("get gitea client err: %v", err))
	}
	_, _, err = client.Repositories.DeleteRepoBranch(ctx, in.Owner, in.Repo, in.Branch)
	if err != nil {
		return to.ErrorResult(fmt.Errorf("delete branch error: %v", err))
	}

	return to.TextResult("Branch Deleted")
}

type listBranchesArgs struct {
	Owner string `json:"owner" required:"true"`
	Repo  string `json:"repo" required:"true"`
}

func ListBranchesFn(ctx context.Context, args map[string]any) (*mcp.CallToolResult, error) {
	var in listBranchesArgs
	if err := params.Bind(args, &in); err != nil {
		return to.ErrorResult(err)
	}
	page, pageSize := params.GetPagination(args, 30)
	opt := gitea_sdk.ListRepoBranchesOptions{
		ListOptions: gitea_sdk.ListOptions{
			Page:     page,
			PageSize: pageSize,
		},
	}
	client, err := gitea.ClientFromContext(ctx)
	if err != nil {
		return to.ErrorResult(fmt.Errorf("get gitea client err: %v", err))
	}
	branches, _, err := client.Repositories.ListRepoBranches(ctx, in.Owner, in.Repo, opt)
	if err != nil {
		return to.ErrorResult(fmt.Errorf("list branches error: %v", err))
	}

	return to.TextResult(slimBranches(branches))
}