Update

2025-03-23 14:53:15 +08:00
1 parent c99f1306e3
commit 24310675ba
@@ -0,0 +1,119 @@
+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/ptr"
+	"gitea.com/gitea/gitea-mcp/pkg/to"
+
+	gitea_sdk "code.gitea.io/sdk/gitea"
+	"github.com/mark3labs/mcp-go/mcp"
+	"github.com/mark3labs/mcp-go/server"
+)
+
+const (
+	SearchUsersToolName    = "search_users"
+	SearchOrgTeamsToolName = "search_org_teams"
+	SearchReposToolName    = "search_repos"
+)
+
+var (
+	SearchUsersTool = mcp.NewTool(
+		SearchUsersToolName,
+		mcp.WithDescription("search users"),
+		mcp.WithString("keyword", mcp.Description("Keyword"), mcp.DefaultString("")),
+		mcp.WithNumber("page", mcp.Description("Page"), mcp.DefaultNumber(1)),
+		mcp.WithNumber("pageSize", mcp.Description("PageSize"), mcp.DefaultNumber(100)),
+	)
+
+	SearOrgTeamsTool = mcp.NewTool(
+		SearchOrgTeamsToolName,
+		mcp.WithDescription("search organization teams"),
+		mcp.WithString("org", mcp.Description("organization name"), mcp.DefaultString("")),
+		mcp.WithString("query", mcp.Description("search organization teams"), mcp.DefaultString("")),
+		mcp.WithBoolean("includeDescription", mcp.Description("include description?"), mcp.DefaultBool(true)),
+		mcp.WithNumber("page", mcp.Description("Page"), mcp.DefaultNumber(1)),
+		mcp.WithNumber("pageSize", mcp.Description("PageSize"), mcp.DefaultNumber(100)),
+	)
+
+	SearchReposTool = mcp.NewTool(
+		SearchReposToolName,
+		mcp.WithDescription("search repos"),
+		mcp.WithString("keyword", mcp.Description("Keyword"), mcp.DefaultString("")),
+		mcp.WithBoolean("keywordIsTopic", mcp.Description("KeywordIsTopic"), mcp.DefaultBool(false)),
+		mcp.WithBoolean("keywordInDescription", mcp.Description("KeywordInDescription"), mcp.DefaultBool(false)),
+		mcp.WithNumber("ownerID", mcp.Description("OwnerID"), mcp.DefaultNumber(0)),
+		mcp.WithBoolean("isPrivate", mcp.Description("IsPrivate"), mcp.DefaultBool(false)),
+		mcp.WithBoolean("isArchived", mcp.Description("IsArchived"), mcp.DefaultBool(false)),
+		mcp.WithString("sort", mcp.Description("Sort"), mcp.DefaultString(""), mcp.Enum("")),
+		mcp.WithString("order", mcp.Description("Order"), mcp.DefaultString(""), mcp.Enum("")),
+		mcp.WithNumber("page", mcp.Description("Page"), mcp.DefaultNumber(1)),
+		mcp.WithNumber("pageSize", mcp.Description("PageSize"), mcp.DefaultNumber(100)),
+	)
+)
+
+func RegisterTool(s *server.MCPServer) {
+	s.AddTool(SearchUsersTool, SearchUsersFn)
+	s.AddTool(SearOrgTeamsTool, SearchOrgTeamsFn)
+	s.AddTool(SearchReposTool, SearchReposFn)
+}
+
+func SearchUsersFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+	log.Debugf("Called SearchUsersFn")
+	opt := gitea_sdk.SearchUsersOption{
+		KeyWord: req.Params.Arguments["keyword"].(string),
+		ListOptions: gitea_sdk.ListOptions{
+			Page:     req.Params.Arguments["page"].(int),
+			PageSize: req.Params.Arguments["pageSize"].(int),
+		},
+	}
+	users, _, err := gitea.Client().SearchUsers(opt)
+	if err != nil {
+		return nil, err
+	}
+	return to.TextResult(users)
+}
+
+func SearchOrgTeamsFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+	log.Debugf("Called SearchOrgTeamsFn")
+	org := req.Params.Arguments["org"].(string)
+	opt := gitea_sdk.SearchTeamsOptions{
+		Query:              req.Params.Arguments["query"].(string),
+		IncludeDescription: req.Params.Arguments["includeDescription"].(bool),
+		ListOptions: gitea_sdk.ListOptions{
+			Page:     req.Params.Arguments["page"].(int),
+			PageSize: req.Params.Arguments["pageSize"].(int),
+		},
+	}
+	teams, _, err := gitea.Client().SearchOrgTeams(org, &opt)
+	if err != nil {
+		return nil, fmt.Errorf("search organization teams error: %v", err)
+	}
+	return to.TextResult(teams)
+}
+
+func SearchReposFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
+	log.Debugf("Called SearchReposFn")
+	opt := gitea_sdk.SearchRepoOptions{
+		Keyword:              req.Params.Arguments["keyword"].(string),
+		KeywordIsTopic:       req.Params.Arguments["keywordIsTopic"].(bool),
+		KeywordInDescription: req.Params.Arguments["keywordInDescription"].(bool),
+		OwnerID:              req.Params.Arguments["ownerID"].(int64),
+		IsPrivate:            ptr.To(req.Params.Arguments["isPrivate"].(bool)),
+		IsArchived:           ptr.To(req.Params.Arguments["isArchived"].(bool)),
+		Sort:                 req.Params.Arguments["sort"].(string),
+		Order:                req.Params.Arguments["order"].(string),
+		ListOptions: gitea_sdk.ListOptions{
+			Page:     req.Params.Arguments["page"].(int),
+			PageSize: req.Params.Arguments["pageSize"].(int),
+		},
+	}
+	repos, _, err := gitea.Client().SearchRepos(opt)
+	if err != nil {
+		return nil, fmt.Errorf("search repos error: %v", err)
+	}
+	return to.TextResult(repos)
+}