Update

2025-08-23 22:33:05 +00:00 · 2025-03-23 14:53:15 +08:00
parent c99f1306e3
commit 24310675ba
9 changed files with 417 additions and 13 deletions
--- a/operation/repo/file.go
+++ b/operation/repo/file.go
@@ -0,0 +1,143 @@
+package repo
+
+import (
+	"context"
+	"fmt"
+
+	"gitea.com/gitea/gitea-mcp/pkg/gitea"
+	"gitea.com/gitea/gitea-mcp/pkg/log"
+	"gitea.com/gitea/gitea-mcp/pkg/to"
+
+	gitea_sdk "code.gitea.io/sdk/gitea"
+	"github.com/mark3labs/mcp-go/mcp"
+)
+
+const (
+	GetFileToolName    = "get_file"
+	CreateFileToolName = "create_file"
+	UpdateFileToolName = "update_file"
+	DeleteFileToolName = "delete_file"
+)
+
+var (
+	GetFileTool = mcp.NewTool(
+		GetFileToolName,
+		mcp.WithDescription("Get file"),
+		mcp.WithString("owner", mcp.Required(), mcp.Description("repository owner"), mcp.DefaultString("")),
+		mcp.WithString("repo", mcp.Required(), mcp.Description("repository name"), mcp.DefaultString("")),
+		mcp.WithString("ref", mcp.Required(), mcp.Description("ref"), mcp.DefaultString("")),
+		mcp.WithString("filePath", mcp.Required(), mcp.Description("file path"), mcp.DefaultString("")),
+	)
+
+	CreateFileTool = mcp.NewTool(
+		CreateFileToolName,
+		mcp.WithDescription("Create file"),
+		mcp.WithString("owner", mcp.Required(), mcp.Description("repository owner"), mcp.DefaultString("")),
+		mcp.WithString("repo", mcp.Required(), mcp.Description("repository name"), mcp.DefaultString("")),
+		mcp.WithString("filePath", mcp.Required(), mcp.Description("file path"), mcp.DefaultString("")),
+		mcp.WithString("content", mcp.Required(), mcp.Description("file content"), mcp.DefaultString("")),
+		mcp.WithString("message", mcp.Required(), mcp.Description("commit message"), mcp.DefaultString("")),
+		mcp.WithString("branch_name", mcp.Required(), mcp.Description("branch name"), mcp.DefaultString("")),
+		mcp.WithString("new_branch_name", mcp.Description("new branch name"), mcp.DefaultString("")),
+	)
+
+	UpdateFileTool = mcp.NewTool(
+		UpdateFileToolName,
+		mcp.WithDescription("Update file"),
+		mcp.WithString("owner", mcp.Required(), mcp.Description("repository owner"), mcp.DefaultString("")),
+		mcp.WithString("repo", mcp.Required(), mcp.Description("repository name"), mcp.DefaultString("")),
+		mcp.WithString("filePath", mcp.Required(), mcp.Description("file path"), mcp.DefaultString("")),
+		mcp.WithString("content", mcp.Required(), mcp.Description("file content"), mcp.DefaultString("")),
+		mcp.WithString("message", mcp.Required(), mcp.Description("commit message"), mcp.DefaultString("")),
+		mcp.WithString("branch_name", mcp.Required(), mcp.Description("branch name"), mcp.DefaultString("")),
+		mcp.WithString("new_branch_name", mcp.Description("new branch name"), mcp.DefaultString("")),
+		mcp.WithString("from_path", mcp.Description("from path"), mcp.DefaultString("")),
+		mcp.WithString("sha", mcp.Description("sha"), mcp.DefaultString("")),
+	)
+
+	DeleteFileTool = mcp.NewTool(
+		DeleteFileToolName,
+		mcp.WithDescription("Delete file"),
+		mcp.WithString("owner", mcp.Required(), mcp.Description("repository owner"), mcp.DefaultString("")),
+		mcp.WithString("repo", mcp.Required(), mcp.Description("repository name"), mcp.DefaultString("")),
+		mcp.WithString("filePath", mcp.Required(), mcp.Description("file path"), mcp.DefaultString("")),
+		mcp.WithString("message", mcp.Required(), mcp.Description("commit message"), mcp.DefaultString("")),
+		mcp.WithString("branch_name", mcp.Required(), mcp.Description("branch name"), mcp.DefaultString("")),
+		mcp.WithString("sha", mcp.Description("sha"), mcp.DefaultString("")),
+	)
+)
+
+func GetFileFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+	log.Debugf("Called GetFileFn")
+	owner := req.Params.Arguments["owner"].(string)
+	repo := req.Params.Arguments["repo"].(string)
+	ref := req.Params.Arguments["ref"].(string)
+	filePath := req.Params.Arguments["filePath"].(string)
+	file, _, err := gitea.Client().GetFile(owner, repo, ref, filePath)
+	if err != nil {
+		return nil, fmt.Errorf("get file err: %v", err)
+	}
+	return to.TextResult(file)
+}
+
+func CreateFileFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+	log.Debugf("Called CreateFileFn")
+	owner := req.Params.Arguments["owner"].(string)
+	repo := req.Params.Arguments["repo"].(string)
+	filePath := req.Params.Arguments["filePath"].(string)
+	opt := gitea_sdk.CreateFileOptions{
+		Content: req.Params.Arguments["content"].(string),
+		FileOptions: gitea_sdk.FileOptions{
+			Message:       req.Params.Arguments["message"].(string),
+			BranchName:    req.Params.Arguments["branch_name"].(string),
+			NewBranchName: req.Params.Arguments["new_branch_name"].(string),
+		},
+	}
+
+	_, _, err := gitea.Client().CreateFile(owner, repo, filePath, opt)
+	if err != nil {
+		return nil, fmt.Errorf("create file err: %v", err)
+	}
+	return to.TextResult("Create file success")
+}
+
+func UpdateFileFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+	log.Debugf("Called UpdateFileFn")
+	owner := req.Params.Arguments["owner"].(string)
+	repo := req.Params.Arguments["repo"].(string)
+	filePath := req.Params.Arguments["filePath"].(string)
+	opt := gitea_sdk.UpdateFileOptions{
+		Content:  req.Params.Arguments["content"].(string),
+		FromPath: req.Params.Arguments["from_path"].(string),
+		SHA:      req.Params.Arguments["sha"].(string),
+		FileOptions: gitea_sdk.FileOptions{
+			Message:       req.Params.Arguments["message"].(string),
+			BranchName:    req.Params.Arguments["branch_name"].(string),
+			NewBranchName: req.Params.Arguments["new_branch_name"].(string),
+		},
+	}
+	_, _, err := gitea.Client().UpdateFile(owner, repo, filePath, opt)
+	if err != nil {
+		return nil, fmt.Errorf("update file err: %v", err)
+	}
+	return to.TextResult("Update file success")
+}
+
+func DeleteFileFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+	log.Debugf("Called DeleteFileFn")
+	owner := req.Params.Arguments["owner"].(string)
+	repo := req.Params.Arguments["repo"].(string)
+	filePath := req.Params.Arguments["filePath"].(string)
+	opt := gitea_sdk.DeleteFileOptions{
+		FileOptions: gitea_sdk.FileOptions{
+			Message:    req.Params.Arguments["message"].(string),
+			BranchName: req.Params.Arguments["branch_name"].(string),
+		},
+		SHA: req.Params.Arguments["sha"].(string),
+	}
+	_, err := gitea.Client().DeleteFile(owner, repo, filePath, opt)
+	if err != nil {
+		return nil, fmt.Errorf("delete file err: %v", err)
+	}
+	return to.TextResult("Delete file success")
+}