Initial commit: Gitea MCP Server

2026-04-10 21:57:19 -07:00
commit 1f27844dce
119 changed files with 25001 additions and 0 deletions
@@ -0,0 +1,177 @@
+package search
+
+import (
+	"context"
+	"fmt"
+
+	"gitea.com/gitea/gitea-mcp/pkg/gitea"
+	"gitea.com/gitea/gitea-mcp/pkg/log"
+	"gitea.com/gitea/gitea-mcp/pkg/params"
+	"gitea.com/gitea/gitea-mcp/pkg/to"
+	"gitea.com/gitea/gitea-mcp/pkg/tool"
+
+	gitea_sdk "code.gitea.io/sdk/gitea"
+	"github.com/mark3labs/mcp-go/mcp"
+	"github.com/mark3labs/mcp-go/server"
+)
+
+var Tool = tool.New()
+
+const (
+	SearchUsersToolName    = "search_users"
+	SearchOrgTeamsToolName = "search_org_teams"
+	SearchReposToolName    = "search_repos"
+)
+
+var (
+	SearchUsersTool = mcp.NewTool(
+		SearchUsersToolName,
+		mcp.WithDescription("search users"),
+		mcp.WithString("keyword", mcp.Required(), mcp.Description("Keyword")),
+		mcp.WithNumber("page", mcp.Description("Page"), mcp.DefaultNumber(1)),
+		mcp.WithNumber("perPage", mcp.Description("results per page"), mcp.DefaultNumber(30)),
+	)
+
+	SearOrgTeamsTool = mcp.NewTool(
+		SearchOrgTeamsToolName,
+		mcp.WithDescription("search organization teams"),
+		mcp.WithString("org", mcp.Required(), mcp.Description("organization name")),
+		mcp.WithString("query", mcp.Required(), mcp.Description("search organization teams")),
+		mcp.WithBoolean("includeDescription", mcp.Description("include description?")),
+		mcp.WithNumber("page", mcp.Description("Page"), mcp.DefaultNumber(1)),
+		mcp.WithNumber("perPage", mcp.Description("results per page"), mcp.DefaultNumber(30)),
+	)
+
+	SearchReposTool = mcp.NewTool(
+		SearchReposToolName,
+		mcp.WithDescription("search repos"),
+		mcp.WithString("keyword", mcp.Required(), mcp.Description("Keyword")),
+		mcp.WithBoolean("keywordIsTopic", mcp.Description("KeywordIsTopic")),
+		mcp.WithBoolean("keywordInDescription", mcp.Description("KeywordInDescription")),
+		mcp.WithNumber("ownerID", mcp.Description("OwnerID")),
+		mcp.WithBoolean("isPrivate", mcp.Description("IsPrivate")),
+		mcp.WithBoolean("isArchived", mcp.Description("IsArchived")),
+		mcp.WithString("sort", mcp.Description("Sort")),
+		mcp.WithString("order", mcp.Description("Order")),
+		mcp.WithNumber("page", mcp.Description("Page"), mcp.DefaultNumber(1)),
+		mcp.WithNumber("perPage", mcp.Description("results per page"), mcp.DefaultNumber(30)),
+	)
+)
+
+func init() {
+	Tool.RegisterRead(server.ServerTool{
+		Tool:    SearchUsersTool,
+		Handler: UsersFn,
+	})
+	Tool.RegisterRead(server.ServerTool{
+		Tool:    SearOrgTeamsTool,
+		Handler: OrgTeamsFn,
+	})
+	Tool.RegisterRead(server.ServerTool{
+		Tool:    SearchReposTool,
+		Handler: ReposFn,
+	})
+}
+
+func UsersFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+	log.Debugf("Called UsersFn")
+	keyword, err := params.GetString(req.GetArguments(), "keyword")
+	if err != nil {
+		return to.ErrorResult(err)
+	}
+	page, pageSize := params.GetPagination(req.GetArguments(), 30)
+	opt := gitea_sdk.SearchUsersOption{
+		KeyWord: keyword,
+		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))
+	}
+	users, _, err := client.SearchUsers(opt)
+	if err != nil {
+		return to.ErrorResult(fmt.Errorf("search users err: %v", err))
+	}
+	return to.TextResult(slimUserDetails(users))
+}
+
+func OrgTeamsFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+	log.Debugf("Called OrgTeamsFn")
+	org, err := params.GetString(req.GetArguments(), "org")
+	if err != nil {
+		return to.ErrorResult(err)
+	}
+	query, err := params.GetString(req.GetArguments(), "query")
+	if err != nil {
+		return to.ErrorResult(err)
+	}
+	includeDescription, _ := req.GetArguments()["includeDescription"].(bool)
+	page, pageSize := params.GetPagination(req.GetArguments(), 30)
+	opt := gitea_sdk.SearchTeamsOptions{
+		Query:              query,
+		IncludeDescription: includeDescription,
+		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))
+	}
+	teams, _, err := client.SearchOrgTeams(org, &opt)
+	if err != nil {
+		return to.ErrorResult(fmt.Errorf("search organization teams error: %v", err))
+	}
+	return to.TextResult(slimTeams(teams))
+}
+
+func ReposFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+	log.Debugf("Called ReposFn")
+	keyword, err := params.GetString(req.GetArguments(), "keyword")
+	if err != nil {
+		return to.ErrorResult(err)
+	}
+	keywordIsTopic, _ := req.GetArguments()["keywordIsTopic"].(bool)
+	keywordInDescription, _ := req.GetArguments()["keywordInDescription"].(bool)
+	ownerID := params.GetOptionalInt(req.GetArguments(), "ownerID", 0)
+	var pIsPrivate *bool
+	isPrivate, ok := req.GetArguments()["isPrivate"].(bool)
+	if ok {
+		pIsPrivate = new(isPrivate)
+	}
+	var pIsArchived *bool
+	isArchived, ok := req.GetArguments()["isArchived"].(bool)
+	if ok {
+		pIsArchived = new(isArchived)
+	}
+	sort, _ := req.GetArguments()["sort"].(string)
+	order, _ := req.GetArguments()["order"].(string)
+	page, pageSize := params.GetPagination(req.GetArguments(), 30)
+	opt := gitea_sdk.SearchRepoOptions{
+		Keyword:              keyword,
+		KeywordIsTopic:       keywordIsTopic,
+		KeywordInDescription: keywordInDescription,
+		OwnerID:              ownerID,
+		IsPrivate:            pIsPrivate,
+		IsArchived:           pIsArchived,
+		Sort:                 sort,
+		Order:                order,
+		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))
+	}
+	repos, _, err := client.SearchRepos(opt)
+	if err != nil {
+		return to.ErrorResult(fmt.Errorf("search repos error: %v", err))
+	}
+	return to.TextResult(slimRepos(repos))
+}