Initialize commit

2025-03-16 00:21:29 +08:00
commit 2dd4aa94ad
14 changed files with 472 additions and 0 deletions
--- a/operation/operation.go
+++ b/operation/operation.go
@@ -0,0 +1,50 @@
+package operation 
+ 
+import ( 
+"fmt" 
+ 
+"gitea.com/gitea/gitea-mcp/operation/repo" 
+"gitea.com/gitea/gitea-mcp/operation/user" 
+"gitea.com/gitea/gitea-mcp/pkg/log" 
+"github.com/mark3labs/mcp-go/server" 
+) 
+ 
+var ( 
+mcpServer *server.MCPServer 
+) 
+ 
+func RegisterTool(s *server.MCPServer) { 
+// User Tool 
+s.AddTool(user.GetMyUserInfoTool, user.MyUserInfoFn) 
+ 
+// Repo Tool 
+s.AddTool(repo.GetMyReposTool, repo.MyUserReposFn) 
+} 
+ 
+func Run(transport, version string) error { 
+mcpServer = newMCPServer(version) 
+RegisterTool(mcpServer) 
+switch transport { 
+case "stdio": 
+if err := server.ServeStdio(mcpServer); err != nil { 
+return err 
+} 
+case "sse": 
+sseServer := server.NewSSEServer(mcpServer) 
+log.Infof("Gitea MCP SSE server listening on :8080") 
+if err := sseServer.Start(":8080"); err != nil { 
+return err 
+} 
+default: 
+return fmt.Errorf("invalid transport type: %s. Must be 'stdio' or 'sse'", transport) 
+} 
+return nil 
+} 
+ 
+func newMCPServer(version string) *server.MCPServer { 
+return server.NewMCPServer( 
+"Gitea MCP Server", 
+version, 
+server.WithLogging(), 
+) 
+} 
--- a/operation/repo/repo.go
+++ b/operation/repo/repo.go
@@ -0,0 +1,40 @@
+package repo 
+ 
+import ( 
+"context" 
+"encoding/json" 
+ 
+"code.gitea.io/sdk/gitea" 
+giteaPkg "gitea.com/gitea/gitea-mcp/pkg/gitea" 
+"github.com/mark3labs/mcp-go/mcp" 
+) 
+ 
+const ( 
+ListMyReposToolName = "list_my_repos" 
+) 
+ 
+var ( 
+GetMyReposTool = mcp.NewTool( 
+ListMyReposToolName, 
+mcp.WithDescription("List My Repositories"), 
+) 
+) 
+ 
+func MyUserReposFn(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) { 
+opts := gitea.ListReposOptions{ 
+ListOptions: gitea.ListOptions{ 
+Page: 1, 
+PageSize: 100, 
+}, 
+} 
+repos, _, err := giteaPkg.Client().ListMyRepos(opts) 
+if err != nil { 
+return mcp.NewToolResultError("Get My User Info Error"), err 
+} 
+ 
+result, err := json.Marshal(repos) 
+if err != nil { 
+return mcp.NewToolResultError("Get My User Info Error"), err 
+} 
+return mcp.NewToolResultText(string(result)), nil 
+} 
--- a/operation/user/user.go
+++ b/operation/user/user.go
@@ -0,0 +1,34 @@
+package user 
+ 
+import ( 
+"context" 
+"encoding/json" 
+ 
+"gitea.com/gitea/gitea-mcp/pkg/gitea" 
+ 
+"github.com/mark3labs/mcp-go/mcp" 
+) 
+ 
+const ( 
+GetMyUserInfoToolName = "get_my_user_info" 
+) 
+ 
+var ( 
+GetMyUserInfoTool = mcp.NewTool( 
+GetMyUserInfoToolName, 
+mcp.WithDescription("Get My User Info"), 
+) 
+) 
+ 
+func MyUserInfoFn(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) { 
+user, _, err := gitea.Client().GetMyUserInfo() 
+if err != nil { 
+return mcp.NewToolResultError("Get My User Info Error"), err 
+} 
+ 
+result, err := json.Marshal(user) 
+if err != nil { 
+return mcp.NewToolResultError("Get My User Info Error"), err 
+} 
+return mcp.NewToolResultText(string(result)), nil 
+}