feat: expand MCP tool coverage from 93 to 299 tools
Some checks failed
release-nightly / release-image (push) Failing after 2m17s
Some checks failed
release-nightly / release-image (push) Failing after 2m17s
Fork the official gitea-mcp and massively extend API coverage: - Organization: orgs, members, teams, hooks, blocks, activity (34 tools) - User: profile, followers, keys, emails, repos, blocks (30 tools) - Repository: collaborators, webhooks, branch/tag protection, deploy keys, topics, git refs/trees/notes, commit status, stars/watchers, forks, transfers, mirrors, templates (53 tools) - Issue: reactions, pins, subscriptions, timeline, templates (16 tools) - Notifications: list, check, read, repo-scoped (7 tools) - Settings: API, attachment, repo, UI settings (4 tools) - Packages: list, get, delete, files, latest, link/unlink (7 tools) - Miscellaneous: server version, gitignore/label/license templates, markdown/markup rendering, node info, signing keys (12 tools) - Admin: user/org/repo CRUD, system webhooks, cron tasks, unadopted repos, emails, badges (23 tools) Module path updated to git.lethalbits.com/lethalbits/gitea-mcp.
This commit is contained in:
Andrew Miller
427
operation/organization/teams.go
Normal file
427
operation/organization/teams.go
Normal file
@@ -0,0 +1,427 @@
|
||||
package organization
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"fmt"
|
||||
|
||||
"git.lethalbits.com/lethalbits/gitea-mcp/pkg/gitea"
|
||||
"git.lethalbits.com/lethalbits/gitea-mcp/pkg/log"
|
||||
"git.lethalbits.com/lethalbits/gitea-mcp/pkg/params"
|
||||
"git.lethalbits.com/lethalbits/gitea-mcp/pkg/to"
|
||||
|
||||
gitea_sdk "code.gitea.io/sdk/gitea"
|
||||
"github.com/mark3labs/mcp-go/mcp"
|
||||
"github.com/mark3labs/mcp-go/server"
|
||||
)
|
||||
|
||||
const (
|
||||
ListOrgTeamsToolName = "list_org_teams"
|
||||
GetTeamToolName = "get_team"
|
||||
CreateTeamToolName = "create_team"
|
||||
EditTeamToolName = "edit_team"
|
||||
DeleteTeamToolName = "delete_team"
|
||||
ListTeamMembersToolName = "list_team_members"
|
||||
GetTeamMemberToolName = "get_team_member"
|
||||
AddTeamMemberToolName = "add_team_member"
|
||||
RemoveTeamMemberToolName = "remove_team_member"
|
||||
ListTeamReposToolName = "list_team_repos"
|
||||
AddTeamRepoToolName = "add_team_repo"
|
||||
RemoveTeamRepoToolName = "remove_team_repo"
|
||||
)
|
||||
|
||||
var (
|
||||
ListOrgTeamsTool = mcp.NewTool(
|
||||
ListOrgTeamsToolName,
|
||||
mcp.WithDescription("List an organization's teams"),
|
||||
mcp.WithString("org", mcp.Required(), mcp.Description("organization name")),
|
||||
mcp.WithNumber("page", mcp.Description("page number"), mcp.DefaultNumber(1)),
|
||||
mcp.WithNumber("pageSize", mcp.Description("page size"), mcp.DefaultNumber(100)),
|
||||
)
|
||||
|
||||
GetTeamTool = mcp.NewTool(
|
||||
GetTeamToolName,
|
||||
mcp.WithDescription("Get a team by ID"),
|
||||
mcp.WithNumber("id", mcp.Required(), mcp.Description("team ID")),
|
||||
)
|
||||
|
||||
CreateTeamTool = mcp.NewTool(
|
||||
CreateTeamToolName,
|
||||
mcp.WithDescription("Create a team in an organization"),
|
||||
mcp.WithString("org", mcp.Required(), mcp.Description("organization name")),
|
||||
mcp.WithString("name", mcp.Required(), mcp.Description("team name")),
|
||||
mcp.WithString("description", mcp.Description("team description")),
|
||||
mcp.WithString("permission", mcp.Description("team permission: read, write, admin"), mcp.DefaultString("read")),
|
||||
mcp.WithBoolean("can_create_org_repo", mcp.Description("whether team members can create repos in the org")),
|
||||
mcp.WithBoolean("includes_all_repositories", mcp.Description("whether team has access to all repos")),
|
||||
)
|
||||
|
||||
EditTeamTool = mcp.NewTool(
|
||||
EditTeamToolName,
|
||||
mcp.WithDescription("Edit a team"),
|
||||
mcp.WithNumber("id", mcp.Required(), mcp.Description("team ID")),
|
||||
mcp.WithString("name", mcp.Description("team name")),
|
||||
mcp.WithString("description", mcp.Description("team description")),
|
||||
mcp.WithString("permission", mcp.Description("team permission: read, write, admin")),
|
||||
mcp.WithBoolean("can_create_org_repo", mcp.Description("whether team members can create repos in the org")),
|
||||
mcp.WithBoolean("includes_all_repositories", mcp.Description("whether team has access to all repos")),
|
||||
)
|
||||
|
||||
DeleteTeamTool = mcp.NewTool(
|
||||
DeleteTeamToolName,
|
||||
mcp.WithDescription("Delete a team"),
|
||||
mcp.WithNumber("id", mcp.Required(), mcp.Description("team ID")),
|
||||
)
|
||||
|
||||
ListTeamMembersTool = mcp.NewTool(
|
||||
ListTeamMembersToolName,
|
||||
mcp.WithDescription("List a team's members"),
|
||||
mcp.WithNumber("id", mcp.Required(), mcp.Description("team ID")),
|
||||
mcp.WithNumber("page", mcp.Description("page number"), mcp.DefaultNumber(1)),
|
||||
mcp.WithNumber("pageSize", mcp.Description("page size"), mcp.DefaultNumber(100)),
|
||||
)
|
||||
|
||||
GetTeamMemberTool = mcp.NewTool(
|
||||
GetTeamMemberToolName,
|
||||
mcp.WithDescription("Get a specific member of a team"),
|
||||
mcp.WithNumber("id", mcp.Required(), mcp.Description("team ID")),
|
||||
mcp.WithString("username", mcp.Required(), mcp.Description("username")),
|
||||
)
|
||||
|
||||
AddTeamMemberTool = mcp.NewTool(
|
||||
AddTeamMemberToolName,
|
||||
mcp.WithDescription("Add a member to a team"),
|
||||
mcp.WithNumber("id", mcp.Required(), mcp.Description("team ID")),
|
||||
mcp.WithString("username", mcp.Required(), mcp.Description("username to add")),
|
||||
)
|
||||
|
||||
RemoveTeamMemberTool = mcp.NewTool(
|
||||
RemoveTeamMemberToolName,
|
||||
mcp.WithDescription("Remove a member from a team"),
|
||||
mcp.WithNumber("id", mcp.Required(), mcp.Description("team ID")),
|
||||
mcp.WithString("username", mcp.Required(), mcp.Description("username to remove")),
|
||||
)
|
||||
|
||||
ListTeamReposTool = mcp.NewTool(
|
||||
ListTeamReposToolName,
|
||||
mcp.WithDescription("List a team's repositories"),
|
||||
mcp.WithNumber("id", mcp.Required(), mcp.Description("team ID")),
|
||||
mcp.WithNumber("page", mcp.Description("page number"), mcp.DefaultNumber(1)),
|
||||
mcp.WithNumber("pageSize", mcp.Description("page size"), mcp.DefaultNumber(100)),
|
||||
)
|
||||
|
||||
AddTeamRepoTool = mcp.NewTool(
|
||||
AddTeamRepoToolName,
|
||||
mcp.WithDescription("Add a repository to a team"),
|
||||
mcp.WithNumber("id", mcp.Required(), mcp.Description("team ID")),
|
||||
mcp.WithString("org", mcp.Required(), mcp.Description("organization name")),
|
||||
mcp.WithString("repo", mcp.Required(), mcp.Description("repository name")),
|
||||
)
|
||||
|
||||
RemoveTeamRepoTool = mcp.NewTool(
|
||||
RemoveTeamRepoToolName,
|
||||
mcp.WithDescription("Remove a repository from a team"),
|
||||
mcp.WithNumber("id", mcp.Required(), mcp.Description("team ID")),
|
||||
mcp.WithString("org", mcp.Required(), mcp.Description("organization name")),
|
||||
mcp.WithString("repo", mcp.Required(), mcp.Description("repository name")),
|
||||
)
|
||||
)
|
||||
|
||||
func init() {
|
||||
Tool.RegisterRead(server.ServerTool{Tool: ListOrgTeamsTool, Handler: ListOrgTeamsFn})
|
||||
Tool.RegisterRead(server.ServerTool{Tool: GetTeamTool, Handler: GetTeamFn})
|
||||
Tool.RegisterRead(server.ServerTool{Tool: ListTeamMembersTool, Handler: ListTeamMembersFn})
|
||||
Tool.RegisterRead(server.ServerTool{Tool: GetTeamMemberTool, Handler: GetTeamMemberFn})
|
||||
Tool.RegisterRead(server.ServerTool{Tool: ListTeamReposTool, Handler: ListTeamReposFn})
|
||||
Tool.RegisterWrite(server.ServerTool{Tool: CreateTeamTool, Handler: CreateTeamFn})
|
||||
Tool.RegisterWrite(server.ServerTool{Tool: EditTeamTool, Handler: EditTeamFn})
|
||||
Tool.RegisterWrite(server.ServerTool{Tool: DeleteTeamTool, Handler: DeleteTeamFn})
|
||||
Tool.RegisterWrite(server.ServerTool{Tool: AddTeamMemberTool, Handler: AddTeamMemberFn})
|
||||
Tool.RegisterWrite(server.ServerTool{Tool: RemoveTeamMemberTool, Handler: RemoveTeamMemberFn})
|
||||
Tool.RegisterWrite(server.ServerTool{Tool: AddTeamRepoTool, Handler: AddTeamRepoFn})
|
||||
Tool.RegisterWrite(server.ServerTool{Tool: RemoveTeamRepoTool, Handler: RemoveTeamRepoFn})
|
||||
}
|
||||
|
||||
func ListOrgTeamsFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called ListOrgTeamsFn")
|
||||
org, ok := req.GetArguments()["org"].(string)
|
||||
if !ok {
|
||||
return to.ErrorResult(errors.New("org is required"))
|
||||
}
|
||||
page := params.GetOptionalInt(req.GetArguments(), "page", 1)
|
||||
pageSize := params.GetOptionalInt(req.GetArguments(), "pageSize", 100)
|
||||
client, err := gitea.ClientFromContext(ctx)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("get gitea client err: %v", err))
|
||||
}
|
||||
teams, _, err := client.ListOrgTeams(org, gitea_sdk.ListTeamsOptions{
|
||||
ListOptions: gitea_sdk.ListOptions{Page: int(page), PageSize: int(pageSize)},
|
||||
})
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("list org %s teams err: %v", org, err))
|
||||
}
|
||||
return to.TextResult(teams)
|
||||
}
|
||||
|
||||
func GetTeamFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called GetTeamFn")
|
||||
id, err := params.GetIndex(req.GetArguments(), "id")
|
||||
if 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))
|
||||
}
|
||||
team, _, err := client.GetTeam(id)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("get team %d err: %v", id, err))
|
||||
}
|
||||
return to.TextResult(team)
|
||||
}
|
||||
|
||||
func CreateTeamFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called CreateTeamFn")
|
||||
org, ok := req.GetArguments()["org"].(string)
|
||||
if !ok {
|
||||
return to.ErrorResult(errors.New("org is required"))
|
||||
}
|
||||
name, ok := req.GetArguments()["name"].(string)
|
||||
if !ok {
|
||||
return to.ErrorResult(errors.New("name is required"))
|
||||
}
|
||||
opt := gitea_sdk.CreateTeamOption{
|
||||
Name: name,
|
||||
}
|
||||
if v, ok := req.GetArguments()["description"].(string); ok {
|
||||
opt.Description = v
|
||||
}
|
||||
if v, ok := req.GetArguments()["permission"].(string); ok {
|
||||
opt.Permission = gitea_sdk.AccessMode(v)
|
||||
}
|
||||
if v, ok := req.GetArguments()["can_create_org_repo"].(bool); ok {
|
||||
opt.CanCreateOrgRepo = v
|
||||
}
|
||||
if v, ok := req.GetArguments()["includes_all_repositories"].(bool); ok {
|
||||
opt.IncludesAllRepositories = v
|
||||
}
|
||||
client, err := gitea.ClientFromContext(ctx)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("get gitea client err: %v", err))
|
||||
}
|
||||
team, _, err := client.CreateTeam(org, opt)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("create team %s in org %s err: %v", name, org, err))
|
||||
}
|
||||
return to.TextResult(team)
|
||||
}
|
||||
|
||||
func EditTeamFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called EditTeamFn")
|
||||
id, err := params.GetIndex(req.GetArguments(), "id")
|
||||
if err != nil {
|
||||
return to.ErrorResult(err)
|
||||
}
|
||||
opt := gitea_sdk.EditTeamOption{}
|
||||
if v, ok := req.GetArguments()["name"].(string); ok {
|
||||
opt.Name = v
|
||||
}
|
||||
if v, ok := req.GetArguments()["description"].(string); ok {
|
||||
opt.Description = &v
|
||||
}
|
||||
if v, ok := req.GetArguments()["permission"].(string); ok {
|
||||
perm := gitea_sdk.AccessMode(v)
|
||||
opt.Permission = perm
|
||||
}
|
||||
if v, ok := req.GetArguments()["can_create_org_repo"].(bool); ok {
|
||||
opt.CanCreateOrgRepo = &v
|
||||
}
|
||||
if v, ok := req.GetArguments()["includes_all_repositories"].(bool); ok {
|
||||
opt.IncludesAllRepositories = &v
|
||||
}
|
||||
client, err := gitea.ClientFromContext(ctx)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("get gitea client err: %v", err))
|
||||
}
|
||||
_, err = client.EditTeam(id, opt)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("edit team %d err: %v", id, err))
|
||||
}
|
||||
team, _, err := client.GetTeam(id)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("get team %d after edit err: %v", id, err))
|
||||
}
|
||||
return to.TextResult(team)
|
||||
}
|
||||
|
||||
func DeleteTeamFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called DeleteTeamFn")
|
||||
id, err := params.GetIndex(req.GetArguments(), "id")
|
||||
if 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.DeleteTeam(id)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("delete team %d err: %v", id, err))
|
||||
}
|
||||
return to.TextResult(map[string]any{"status": "deleted", "team_id": id})
|
||||
}
|
||||
|
||||
func ListTeamMembersFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called ListTeamMembersFn")
|
||||
id, err := params.GetIndex(req.GetArguments(), "id")
|
||||
if err != nil {
|
||||
return to.ErrorResult(err)
|
||||
}
|
||||
page := params.GetOptionalInt(req.GetArguments(), "page", 1)
|
||||
pageSize := params.GetOptionalInt(req.GetArguments(), "pageSize", 100)
|
||||
client, err := gitea.ClientFromContext(ctx)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("get gitea client err: %v", err))
|
||||
}
|
||||
members, _, err := client.ListTeamMembers(id, gitea_sdk.ListTeamMembersOptions{
|
||||
ListOptions: gitea_sdk.ListOptions{Page: int(page), PageSize: int(pageSize)},
|
||||
})
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("list team %d members err: %v", id, err))
|
||||
}
|
||||
return to.TextResult(members)
|
||||
}
|
||||
|
||||
func GetTeamMemberFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called GetTeamMemberFn")
|
||||
id, err := params.GetIndex(req.GetArguments(), "id")
|
||||
if err != nil {
|
||||
return to.ErrorResult(err)
|
||||
}
|
||||
username, ok := req.GetArguments()["username"].(string)
|
||||
if !ok {
|
||||
return to.ErrorResult(errors.New("username is required"))
|
||||
}
|
||||
client, err := gitea.ClientFromContext(ctx)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("get gitea client err: %v", err))
|
||||
}
|
||||
member, _, err := client.GetTeamMember(id, username)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("get team %d member %s err: %v", id, username, err))
|
||||
}
|
||||
return to.TextResult(member)
|
||||
}
|
||||
|
||||
func AddTeamMemberFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called AddTeamMemberFn")
|
||||
id, err := params.GetIndex(req.GetArguments(), "id")
|
||||
if err != nil {
|
||||
return to.ErrorResult(err)
|
||||
}
|
||||
username, ok := req.GetArguments()["username"].(string)
|
||||
if !ok {
|
||||
return to.ErrorResult(errors.New("username is required"))
|
||||
}
|
||||
client, err := gitea.ClientFromContext(ctx)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("get gitea client err: %v", err))
|
||||
}
|
||||
_, err = client.AddTeamMember(id, username)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("add member %s to team %d err: %v", username, id, err))
|
||||
}
|
||||
return to.TextResult(map[string]any{"status": "added", "team_id": id, "username": username})
|
||||
}
|
||||
|
||||
func RemoveTeamMemberFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called RemoveTeamMemberFn")
|
||||
id, err := params.GetIndex(req.GetArguments(), "id")
|
||||
if err != nil {
|
||||
return to.ErrorResult(err)
|
||||
}
|
||||
username, ok := req.GetArguments()["username"].(string)
|
||||
if !ok {
|
||||
return to.ErrorResult(errors.New("username is required"))
|
||||
}
|
||||
client, err := gitea.ClientFromContext(ctx)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("get gitea client err: %v", err))
|
||||
}
|
||||
_, err = client.RemoveTeamMember(id, username)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("remove member %s from team %d err: %v", username, id, err))
|
||||
}
|
||||
return to.TextResult(map[string]any{"status": "removed", "team_id": id, "username": username})
|
||||
}
|
||||
|
||||
func ListTeamReposFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called ListTeamReposFn")
|
||||
id, err := params.GetIndex(req.GetArguments(), "id")
|
||||
if err != nil {
|
||||
return to.ErrorResult(err)
|
||||
}
|
||||
page := params.GetOptionalInt(req.GetArguments(), "page", 1)
|
||||
pageSize := params.GetOptionalInt(req.GetArguments(), "pageSize", 100)
|
||||
client, err := gitea.ClientFromContext(ctx)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("get gitea client err: %v", err))
|
||||
}
|
||||
repos, _, err := client.ListTeamRepositories(id, gitea_sdk.ListTeamRepositoriesOptions{
|
||||
ListOptions: gitea_sdk.ListOptions{Page: int(page), PageSize: int(pageSize)},
|
||||
})
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("list team %d repos err: %v", id, err))
|
||||
}
|
||||
return to.TextResult(repos)
|
||||
}
|
||||
|
||||
func AddTeamRepoFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called AddTeamRepoFn")
|
||||
id, err := params.GetIndex(req.GetArguments(), "id")
|
||||
if err != nil {
|
||||
return to.ErrorResult(err)
|
||||
}
|
||||
org, ok := req.GetArguments()["org"].(string)
|
||||
if !ok {
|
||||
return to.ErrorResult(errors.New("org is required"))
|
||||
}
|
||||
repo, ok := req.GetArguments()["repo"].(string)
|
||||
if !ok {
|
||||
return to.ErrorResult(errors.New("repo is required"))
|
||||
}
|
||||
client, err := gitea.ClientFromContext(ctx)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("get gitea client err: %v", err))
|
||||
}
|
||||
_, err = client.AddTeamRepository(id, org, repo)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("add repo %s/%s to team %d err: %v", org, repo, id, err))
|
||||
}
|
||||
return to.TextResult(map[string]any{"status": "added", "team_id": id, "org": org, "repo": repo})
|
||||
}
|
||||
|
||||
func RemoveTeamRepoFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called RemoveTeamRepoFn")
|
||||
id, err := params.GetIndex(req.GetArguments(), "id")
|
||||
if err != nil {
|
||||
return to.ErrorResult(err)
|
||||
}
|
||||
org, ok := req.GetArguments()["org"].(string)
|
||||
if !ok {
|
||||
return to.ErrorResult(errors.New("org is required"))
|
||||
}
|
||||
repo, ok := req.GetArguments()["repo"].(string)
|
||||
if !ok {
|
||||
return to.ErrorResult(errors.New("repo is required"))
|
||||
}
|
||||
client, err := gitea.ClientFromContext(ctx)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("get gitea client err: %v", err))
|
||||
}
|
||||
_, err = client.RemoveTeamRepository(id, org, repo)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("remove repo %s/%s from team %d err: %v", org, repo, id, err))
|
||||
}
|
||||
return to.TextResult(map[string]any{"status": "removed", "team_id": id, "org": org, "repo": repo})
|
||||
}
|
||||
Reference in New Issue
Block a user