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
86
operation/organization/activity.go
Normal file
86
operation/organization/activity.go
Normal file
@@ -0,0 +1,86 @@
|
||||
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 (
|
||||
ListOrgActivityFeedsToolName = "list_org_activity_feeds"
|
||||
ListTeamActivityFeedsToolName = "list_team_activity_feeds"
|
||||
)
|
||||
|
||||
var (
|
||||
ListOrgActivityFeedsTool = mcp.NewTool(
|
||||
ListOrgActivityFeedsToolName,
|
||||
mcp.WithDescription("List an organization's activity feeds"),
|
||||
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)),
|
||||
)
|
||||
|
||||
ListTeamActivityFeedsTool = mcp.NewTool(
|
||||
ListTeamActivityFeedsToolName,
|
||||
mcp.WithDescription("List a team's activity feeds"),
|
||||
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)),
|
||||
)
|
||||
)
|
||||
|
||||
func init() {
|
||||
Tool.RegisterRead(server.ServerTool{Tool: ListOrgActivityFeedsTool, Handler: ListOrgActivityFeedsFn})
|
||||
Tool.RegisterRead(server.ServerTool{Tool: ListTeamActivityFeedsTool, Handler: ListTeamActivityFeedsFn})
|
||||
}
|
||||
|
||||
func ListOrgActivityFeedsFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called ListOrgActivityFeedsFn")
|
||||
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))
|
||||
}
|
||||
feeds, _, err := client.ListOrgActivityFeeds(org, gitea_sdk.ListOrgActivityFeedsOptions{
|
||||
ListOptions: gitea_sdk.ListOptions{Page: int(page), PageSize: int(pageSize)},
|
||||
})
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("list org %s activity feeds err: %v", org, err))
|
||||
}
|
||||
return to.TextResult(feeds)
|
||||
}
|
||||
|
||||
func ListTeamActivityFeedsFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called ListTeamActivityFeedsFn")
|
||||
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))
|
||||
}
|
||||
feeds, _, err := client.ListTeamActivityFeeds(id, gitea_sdk.ListTeamActivityFeedsOptions{
|
||||
ListOptions: gitea_sdk.ListOptions{Page: int(page), PageSize: int(pageSize)},
|
||||
})
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("list team %d activity feeds err: %v", id, err))
|
||||
}
|
||||
return to.TextResult(feeds)
|
||||
}
|
||||
145
operation/organization/blocks.go
Normal file
145
operation/organization/blocks.go
Normal file
@@ -0,0 +1,145 @@
|
||||
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 (
|
||||
ListOrgBlocksToolName = "list_org_blocks"
|
||||
CheckOrgBlockToolName = "check_org_block"
|
||||
BlockOrgUserToolName = "block_org_user"
|
||||
UnblockOrgUserToolName = "unblock_org_user"
|
||||
)
|
||||
|
||||
var (
|
||||
ListOrgBlocksTool = mcp.NewTool(
|
||||
ListOrgBlocksToolName,
|
||||
mcp.WithDescription("List users blocked by an organization"),
|
||||
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)),
|
||||
)
|
||||
|
||||
CheckOrgBlockTool = mcp.NewTool(
|
||||
CheckOrgBlockToolName,
|
||||
mcp.WithDescription("Check if a user is blocked by an organization"),
|
||||
mcp.WithString("org", mcp.Required(), mcp.Description("organization name")),
|
||||
mcp.WithString("username", mcp.Required(), mcp.Description("username to check")),
|
||||
)
|
||||
|
||||
BlockOrgUserTool = mcp.NewTool(
|
||||
BlockOrgUserToolName,
|
||||
mcp.WithDescription("Block a user from an organization"),
|
||||
mcp.WithString("org", mcp.Required(), mcp.Description("organization name")),
|
||||
mcp.WithString("username", mcp.Required(), mcp.Description("username to block")),
|
||||
)
|
||||
|
||||
UnblockOrgUserTool = mcp.NewTool(
|
||||
UnblockOrgUserToolName,
|
||||
mcp.WithDescription("Unblock a user from an organization"),
|
||||
mcp.WithString("org", mcp.Required(), mcp.Description("organization name")),
|
||||
mcp.WithString("username", mcp.Required(), mcp.Description("username to unblock")),
|
||||
)
|
||||
)
|
||||
|
||||
func init() {
|
||||
Tool.RegisterRead(server.ServerTool{Tool: ListOrgBlocksTool, Handler: ListOrgBlocksFn})
|
||||
Tool.RegisterRead(server.ServerTool{Tool: CheckOrgBlockTool, Handler: CheckOrgBlockFn})
|
||||
Tool.RegisterWrite(server.ServerTool{Tool: BlockOrgUserTool, Handler: BlockOrgUserFn})
|
||||
Tool.RegisterWrite(server.ServerTool{Tool: UnblockOrgUserTool, Handler: UnblockOrgUserFn})
|
||||
}
|
||||
|
||||
func ListOrgBlocksFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called ListOrgBlocksFn")
|
||||
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))
|
||||
}
|
||||
users, _, err := client.ListOrgBlocks(org, gitea_sdk.ListOrgBlocksOptions{
|
||||
ListOptions: gitea_sdk.ListOptions{Page: int(page), PageSize: int(pageSize)},
|
||||
})
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("list org %s blocks err: %v", org, err))
|
||||
}
|
||||
return to.TextResult(users)
|
||||
}
|
||||
|
||||
func CheckOrgBlockFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called CheckOrgBlockFn")
|
||||
org, ok := req.GetArguments()["org"].(string)
|
||||
if !ok {
|
||||
return to.ErrorResult(errors.New("org is required"))
|
||||
}
|
||||
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))
|
||||
}
|
||||
isBlocked, _, err := client.CheckOrgBlock(org, username)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("check org %s block for %s err: %v", org, username, err))
|
||||
}
|
||||
return to.TextResult(map[string]any{"org": org, "username": username, "is_blocked": isBlocked})
|
||||
}
|
||||
|
||||
func BlockOrgUserFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called BlockOrgUserFn")
|
||||
org, ok := req.GetArguments()["org"].(string)
|
||||
if !ok {
|
||||
return to.ErrorResult(errors.New("org is required"))
|
||||
}
|
||||
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.BlockOrgUser(org, username)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("block user %s from org %s err: %v", username, org, err))
|
||||
}
|
||||
return to.TextResult(map[string]string{"status": "blocked", "org": org, "username": username})
|
||||
}
|
||||
|
||||
func UnblockOrgUserFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called UnblockOrgUserFn")
|
||||
org, ok := req.GetArguments()["org"].(string)
|
||||
if !ok {
|
||||
return to.ErrorResult(errors.New("org is required"))
|
||||
}
|
||||
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.UnblockOrgUser(org, username)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("unblock user %s from org %s err: %v", username, org, err))
|
||||
}
|
||||
return to.TextResult(map[string]string{"status": "unblocked", "org": org, "username": username})
|
||||
}
|
||||
251
operation/organization/hooks.go
Normal file
251
operation/organization/hooks.go
Normal file
@@ -0,0 +1,251 @@
|
||||
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 (
|
||||
ListOrgHooksToolName = "list_org_hooks"
|
||||
GetOrgHookToolName = "get_org_hook"
|
||||
CreateOrgHookToolName = "create_org_hook"
|
||||
EditOrgHookToolName = "edit_org_hook"
|
||||
DeleteOrgHookToolName = "delete_org_hook"
|
||||
)
|
||||
|
||||
var (
|
||||
ListOrgHooksTool = mcp.NewTool(
|
||||
ListOrgHooksToolName,
|
||||
mcp.WithDescription("List an organization's webhooks"),
|
||||
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)),
|
||||
)
|
||||
|
||||
GetOrgHookTool = mcp.NewTool(
|
||||
GetOrgHookToolName,
|
||||
mcp.WithDescription("Get an organization webhook by ID"),
|
||||
mcp.WithString("org", mcp.Required(), mcp.Description("organization name")),
|
||||
mcp.WithNumber("id", mcp.Required(), mcp.Description("webhook ID")),
|
||||
)
|
||||
|
||||
CreateOrgHookTool = mcp.NewTool(
|
||||
CreateOrgHookToolName,
|
||||
mcp.WithDescription("Create an organization webhook"),
|
||||
mcp.WithString("org", mcp.Required(), mcp.Description("organization name")),
|
||||
mcp.WithString("type", mcp.Required(), mcp.Description("hook type: gitea, slack, discord, dingtalk, telegram, msteams, feishu, matrix, wechatwork, packagist")),
|
||||
mcp.WithString("url", mcp.Required(), mcp.Description("target URL for the webhook")),
|
||||
mcp.WithString("content_type", mcp.Description("content type: json or form"), mcp.DefaultString("json")),
|
||||
mcp.WithString("secret", mcp.Description("webhook secret")),
|
||||
mcp.WithBoolean("active", mcp.Description("whether the webhook is active (default: true)")),
|
||||
mcp.WithArray("events", mcp.Description("list of events to trigger on"), mcp.Items(map[string]any{"type": "string"})),
|
||||
)
|
||||
|
||||
EditOrgHookTool = mcp.NewTool(
|
||||
EditOrgHookToolName,
|
||||
mcp.WithDescription("Edit an organization webhook"),
|
||||
mcp.WithString("org", mcp.Required(), mcp.Description("organization name")),
|
||||
mcp.WithNumber("id", mcp.Required(), mcp.Description("webhook ID")),
|
||||
mcp.WithString("url", mcp.Description("target URL")),
|
||||
mcp.WithString("content_type", mcp.Description("content type: json or form")),
|
||||
mcp.WithString("secret", mcp.Description("webhook secret")),
|
||||
mcp.WithBoolean("active", mcp.Description("whether the webhook is active")),
|
||||
mcp.WithArray("events", mcp.Description("list of events to trigger on"), mcp.Items(map[string]any{"type": "string"})),
|
||||
)
|
||||
|
||||
DeleteOrgHookTool = mcp.NewTool(
|
||||
DeleteOrgHookToolName,
|
||||
mcp.WithDescription("Delete an organization webhook"),
|
||||
mcp.WithString("org", mcp.Required(), mcp.Description("organization name")),
|
||||
mcp.WithNumber("id", mcp.Required(), mcp.Description("webhook ID")),
|
||||
)
|
||||
)
|
||||
|
||||
func init() {
|
||||
Tool.RegisterRead(server.ServerTool{Tool: ListOrgHooksTool, Handler: ListOrgHooksFn})
|
||||
Tool.RegisterRead(server.ServerTool{Tool: GetOrgHookTool, Handler: GetOrgHookFn})
|
||||
Tool.RegisterWrite(server.ServerTool{Tool: CreateOrgHookTool, Handler: CreateOrgHookFn})
|
||||
Tool.RegisterWrite(server.ServerTool{Tool: EditOrgHookTool, Handler: EditOrgHookFn})
|
||||
Tool.RegisterWrite(server.ServerTool{Tool: DeleteOrgHookTool, Handler: DeleteOrgHookFn})
|
||||
}
|
||||
|
||||
func ListOrgHooksFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called ListOrgHooksFn")
|
||||
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))
|
||||
}
|
||||
hooks, _, err := client.ListOrgHooks(org, gitea_sdk.ListHooksOptions{
|
||||
ListOptions: gitea_sdk.ListOptions{Page: int(page), PageSize: int(pageSize)},
|
||||
})
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("list org %s hooks err: %v", org, err))
|
||||
}
|
||||
return to.TextResult(hooks)
|
||||
}
|
||||
|
||||
func GetOrgHookFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called GetOrgHookFn")
|
||||
org, ok := req.GetArguments()["org"].(string)
|
||||
if !ok {
|
||||
return to.ErrorResult(errors.New("org is required"))
|
||||
}
|
||||
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))
|
||||
}
|
||||
hook, _, err := client.GetOrgHook(org, id)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("get org %s hook %d err: %v", org, id, err))
|
||||
}
|
||||
return to.TextResult(hook)
|
||||
}
|
||||
|
||||
func CreateOrgHookFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called CreateOrgHookFn")
|
||||
org, ok := req.GetArguments()["org"].(string)
|
||||
if !ok {
|
||||
return to.ErrorResult(errors.New("org is required"))
|
||||
}
|
||||
hookType, ok := req.GetArguments()["type"].(string)
|
||||
if !ok {
|
||||
return to.ErrorResult(errors.New("type is required"))
|
||||
}
|
||||
hookURL, ok := req.GetArguments()["url"].(string)
|
||||
if !ok {
|
||||
return to.ErrorResult(errors.New("url is required"))
|
||||
}
|
||||
contentType := "json"
|
||||
if v, ok := req.GetArguments()["content_type"].(string); ok {
|
||||
contentType = v
|
||||
}
|
||||
config := map[string]string{
|
||||
"url": hookURL,
|
||||
"content_type": contentType,
|
||||
}
|
||||
if v, ok := req.GetArguments()["secret"].(string); ok {
|
||||
config["secret"] = v
|
||||
}
|
||||
opt := gitea_sdk.CreateHookOption{
|
||||
Type: gitea_sdk.HookType(hookType),
|
||||
Config: config,
|
||||
Active: true,
|
||||
}
|
||||
if v, ok := req.GetArguments()["active"].(bool); ok {
|
||||
opt.Active = v
|
||||
}
|
||||
if eventsArg, exists := req.GetArguments()["events"]; exists {
|
||||
if eventsSlice, ok := eventsArg.([]any); ok {
|
||||
events := make([]string, 0, len(eventsSlice))
|
||||
for _, e := range eventsSlice {
|
||||
if s, ok := e.(string); ok {
|
||||
events = append(events, s)
|
||||
}
|
||||
}
|
||||
opt.Events = events
|
||||
}
|
||||
}
|
||||
client, err := gitea.ClientFromContext(ctx)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("get gitea client err: %v", err))
|
||||
}
|
||||
hook, _, err := client.CreateOrgHook(org, opt)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("create org %s hook err: %v", org, err))
|
||||
}
|
||||
return to.TextResult(hook)
|
||||
}
|
||||
|
||||
func EditOrgHookFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called EditOrgHookFn")
|
||||
org, ok := req.GetArguments()["org"].(string)
|
||||
if !ok {
|
||||
return to.ErrorResult(errors.New("org is required"))
|
||||
}
|
||||
id, err := params.GetIndex(req.GetArguments(), "id")
|
||||
if err != nil {
|
||||
return to.ErrorResult(err)
|
||||
}
|
||||
opt := gitea_sdk.EditHookOption{}
|
||||
config := map[string]string{}
|
||||
if v, ok := req.GetArguments()["url"].(string); ok {
|
||||
config["url"] = v
|
||||
}
|
||||
if v, ok := req.GetArguments()["content_type"].(string); ok {
|
||||
config["content_type"] = v
|
||||
}
|
||||
if v, ok := req.GetArguments()["secret"].(string); ok {
|
||||
config["secret"] = v
|
||||
}
|
||||
if len(config) > 0 {
|
||||
opt.Config = config
|
||||
}
|
||||
if v, ok := req.GetArguments()["active"].(bool); ok {
|
||||
opt.Active = &v
|
||||
}
|
||||
if eventsArg, exists := req.GetArguments()["events"]; exists {
|
||||
if eventsSlice, ok := eventsArg.([]any); ok {
|
||||
events := make([]string, 0, len(eventsSlice))
|
||||
for _, e := range eventsSlice {
|
||||
if s, ok := e.(string); ok {
|
||||
events = append(events, s)
|
||||
}
|
||||
}
|
||||
opt.Events = events
|
||||
}
|
||||
}
|
||||
client, err := gitea.ClientFromContext(ctx)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("get gitea client err: %v", err))
|
||||
}
|
||||
_, err = client.EditOrgHook(org, id, opt)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("edit org %s hook %d err: %v", org, id, err))
|
||||
}
|
||||
hook, _, err := client.GetOrgHook(org, id)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("get org %s hook %d after edit err: %v", org, id, err))
|
||||
}
|
||||
return to.TextResult(hook)
|
||||
}
|
||||
|
||||
func DeleteOrgHookFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called DeleteOrgHookFn")
|
||||
org, ok := req.GetArguments()["org"].(string)
|
||||
if !ok {
|
||||
return to.ErrorResult(errors.New("org is required"))
|
||||
}
|
||||
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.DeleteOrgHook(org, id)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("delete org %s hook %d err: %v", org, id, err))
|
||||
}
|
||||
return to.TextResult(map[string]any{"status": "deleted", "org": org, "hook_id": id})
|
||||
}
|
||||
241
operation/organization/members.go
Normal file
241
operation/organization/members.go
Normal file
@@ -0,0 +1,241 @@
|
||||
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 (
|
||||
ListOrgMembersToolName = "list_org_members"
|
||||
CheckOrgMembershipToolName = "check_org_membership"
|
||||
RemoveOrgMemberToolName = "remove_org_member"
|
||||
ListOrgPublicMembersToolName = "list_org_public_members"
|
||||
CheckOrgPublicMemberToolName = "check_org_public_member"
|
||||
SetOrgPublicMemberToolName = "set_org_public_member"
|
||||
GetOrgPermissionsToolName = "get_org_permissions"
|
||||
)
|
||||
|
||||
var (
|
||||
ListOrgMembersTool = mcp.NewTool(
|
||||
ListOrgMembersToolName,
|
||||
mcp.WithDescription("List an organization's members"),
|
||||
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)),
|
||||
)
|
||||
|
||||
CheckOrgMembershipTool = mcp.NewTool(
|
||||
CheckOrgMembershipToolName,
|
||||
mcp.WithDescription("Check if a user is a member of an organization"),
|
||||
mcp.WithString("org", mcp.Required(), mcp.Description("organization name")),
|
||||
mcp.WithString("username", mcp.Required(), mcp.Description("username to check")),
|
||||
)
|
||||
|
||||
RemoveOrgMemberTool = mcp.NewTool(
|
||||
RemoveOrgMemberToolName,
|
||||
mcp.WithDescription("Remove a member from an organization"),
|
||||
mcp.WithString("org", mcp.Required(), mcp.Description("organization name")),
|
||||
mcp.WithString("username", mcp.Required(), mcp.Description("username to remove")),
|
||||
)
|
||||
|
||||
ListOrgPublicMembersTool = mcp.NewTool(
|
||||
ListOrgPublicMembersToolName,
|
||||
mcp.WithDescription("List an organization's public members"),
|
||||
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)),
|
||||
)
|
||||
|
||||
CheckOrgPublicMemberTool = mcp.NewTool(
|
||||
CheckOrgPublicMemberToolName,
|
||||
mcp.WithDescription("Check if a user is a public member of an organization"),
|
||||
mcp.WithString("org", mcp.Required(), mcp.Description("organization name")),
|
||||
mcp.WithString("username", mcp.Required(), mcp.Description("username to check")),
|
||||
)
|
||||
|
||||
SetOrgPublicMemberTool = mcp.NewTool(
|
||||
SetOrgPublicMemberToolName,
|
||||
mcp.WithDescription("Set or remove a user's public membership in an organization"),
|
||||
mcp.WithString("org", mcp.Required(), mcp.Description("organization name")),
|
||||
mcp.WithString("username", mcp.Required(), mcp.Description("username")),
|
||||
mcp.WithBoolean("visible", mcp.Required(), mcp.Description("true to make public, false to make private")),
|
||||
)
|
||||
|
||||
GetOrgPermissionsTool = mcp.NewTool(
|
||||
GetOrgPermissionsToolName,
|
||||
mcp.WithDescription("Get a user's permissions in an organization"),
|
||||
mcp.WithString("org", mcp.Required(), mcp.Description("organization name")),
|
||||
mcp.WithString("username", mcp.Required(), mcp.Description("username")),
|
||||
)
|
||||
)
|
||||
|
||||
func init() {
|
||||
Tool.RegisterRead(server.ServerTool{Tool: ListOrgMembersTool, Handler: ListOrgMembersFn})
|
||||
Tool.RegisterRead(server.ServerTool{Tool: CheckOrgMembershipTool, Handler: CheckOrgMembershipFn})
|
||||
Tool.RegisterRead(server.ServerTool{Tool: ListOrgPublicMembersTool, Handler: ListOrgPublicMembersFn})
|
||||
Tool.RegisterRead(server.ServerTool{Tool: CheckOrgPublicMemberTool, Handler: CheckOrgPublicMemberFn})
|
||||
Tool.RegisterRead(server.ServerTool{Tool: GetOrgPermissionsTool, Handler: GetOrgPermissionsFn})
|
||||
Tool.RegisterWrite(server.ServerTool{Tool: RemoveOrgMemberTool, Handler: RemoveOrgMemberFn})
|
||||
Tool.RegisterWrite(server.ServerTool{Tool: SetOrgPublicMemberTool, Handler: SetOrgPublicMemberFn})
|
||||
}
|
||||
|
||||
func ListOrgMembersFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called ListOrgMembersFn")
|
||||
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))
|
||||
}
|
||||
members, _, err := client.ListOrgMembership(org, gitea_sdk.ListOrgMembershipOption{
|
||||
ListOptions: gitea_sdk.ListOptions{Page: int(page), PageSize: int(pageSize)},
|
||||
})
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("list org %s members err: %v", org, err))
|
||||
}
|
||||
return to.TextResult(members)
|
||||
}
|
||||
|
||||
func CheckOrgMembershipFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called CheckOrgMembershipFn")
|
||||
org, ok := req.GetArguments()["org"].(string)
|
||||
if !ok {
|
||||
return to.ErrorResult(errors.New("org is required"))
|
||||
}
|
||||
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))
|
||||
}
|
||||
isMember, _, err := client.CheckOrgMembership(org, username)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("check org %s membership for %s err: %v", org, username, err))
|
||||
}
|
||||
return to.TextResult(map[string]any{"org": org, "username": username, "is_member": isMember})
|
||||
}
|
||||
|
||||
func RemoveOrgMemberFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called RemoveOrgMemberFn")
|
||||
org, ok := req.GetArguments()["org"].(string)
|
||||
if !ok {
|
||||
return to.ErrorResult(errors.New("org is required"))
|
||||
}
|
||||
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.DeleteOrgMembership(org, username)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("remove member %s from org %s err: %v", username, org, err))
|
||||
}
|
||||
return to.TextResult(map[string]string{"status": "removed", "org": org, "username": username})
|
||||
}
|
||||
|
||||
func ListOrgPublicMembersFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called ListOrgPublicMembersFn")
|
||||
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))
|
||||
}
|
||||
members, _, err := client.ListPublicOrgMembership(org, gitea_sdk.ListOrgMembershipOption{
|
||||
ListOptions: gitea_sdk.ListOptions{Page: int(page), PageSize: int(pageSize)},
|
||||
})
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("list org %s public members err: %v", org, err))
|
||||
}
|
||||
return to.TextResult(members)
|
||||
}
|
||||
|
||||
func CheckOrgPublicMemberFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called CheckOrgPublicMemberFn")
|
||||
org, ok := req.GetArguments()["org"].(string)
|
||||
if !ok {
|
||||
return to.ErrorResult(errors.New("org is required"))
|
||||
}
|
||||
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))
|
||||
}
|
||||
isPublic, _, err := client.CheckPublicOrgMembership(org, username)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("check org %s public membership for %s err: %v", org, username, err))
|
||||
}
|
||||
return to.TextResult(map[string]any{"org": org, "username": username, "is_public_member": isPublic})
|
||||
}
|
||||
|
||||
func SetOrgPublicMemberFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called SetOrgPublicMemberFn")
|
||||
org, ok := req.GetArguments()["org"].(string)
|
||||
if !ok {
|
||||
return to.ErrorResult(errors.New("org is required"))
|
||||
}
|
||||
username, ok := req.GetArguments()["username"].(string)
|
||||
if !ok {
|
||||
return to.ErrorResult(errors.New("username is required"))
|
||||
}
|
||||
visible, ok := req.GetArguments()["visible"].(bool)
|
||||
if !ok {
|
||||
return to.ErrorResult(errors.New("visible is required"))
|
||||
}
|
||||
client, err := gitea.ClientFromContext(ctx)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("get gitea client err: %v", err))
|
||||
}
|
||||
_, err = client.SetPublicOrgMembership(org, username, visible)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("set public membership for %s in org %s err: %v", username, org, err))
|
||||
}
|
||||
return to.TextResult(map[string]any{"org": org, "username": username, "visible": visible})
|
||||
}
|
||||
|
||||
func GetOrgPermissionsFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called GetOrgPermissionsFn")
|
||||
org, ok := req.GetArguments()["org"].(string)
|
||||
if !ok {
|
||||
return to.ErrorResult(errors.New("org is required"))
|
||||
}
|
||||
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))
|
||||
}
|
||||
perms, _, err := client.GetOrgPermissions(org, username)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("get permissions for %s in org %s err: %v", username, org, err))
|
||||
}
|
||||
return to.TextResult(perms)
|
||||
}
|
||||
319
operation/organization/org.go
Normal file
319
operation/organization/org.go
Normal file
@@ -0,0 +1,319 @@
|
||||
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"
|
||||
"git.lethalbits.com/lethalbits/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 (
|
||||
ListOrgsToolName = "list_orgs"
|
||||
GetOrgToolName = "get_org"
|
||||
CreateOrgToolName = "create_org"
|
||||
EditOrgToolName = "edit_org"
|
||||
DeleteOrgToolName = "delete_org"
|
||||
RenameOrgToolName = "rename_org"
|
||||
ListOrgReposToolName = "list_org_repos"
|
||||
CreateOrgRepoToolName = "create_org_repo"
|
||||
)
|
||||
|
||||
var (
|
||||
ListOrgsTool = mcp.NewTool(
|
||||
ListOrgsToolName,
|
||||
mcp.WithDescription("List all visible organizations"),
|
||||
mcp.WithNumber("page", mcp.Description("page number"), mcp.DefaultNumber(1)),
|
||||
mcp.WithNumber("pageSize", mcp.Description("page size"), mcp.DefaultNumber(100)),
|
||||
)
|
||||
|
||||
GetOrgTool = mcp.NewTool(
|
||||
GetOrgToolName,
|
||||
mcp.WithDescription("Get an organization by name"),
|
||||
mcp.WithString("org", mcp.Required(), mcp.Description("organization name")),
|
||||
)
|
||||
|
||||
CreateOrgTool = mcp.NewTool(
|
||||
CreateOrgToolName,
|
||||
mcp.WithDescription("Create an organization"),
|
||||
mcp.WithString("username", mcp.Required(), mcp.Description("organization username")),
|
||||
mcp.WithString("full_name", mcp.Description("organization full name")),
|
||||
mcp.WithString("description", mcp.Description("organization description")),
|
||||
mcp.WithString("website", mcp.Description("organization website")),
|
||||
mcp.WithString("location", mcp.Description("organization location")),
|
||||
mcp.WithString("visibility", mcp.Description("visibility: public, limited, or private"), mcp.DefaultString("public")),
|
||||
)
|
||||
|
||||
EditOrgTool = mcp.NewTool(
|
||||
EditOrgToolName,
|
||||
mcp.WithDescription("Edit an organization"),
|
||||
mcp.WithString("org", mcp.Required(), mcp.Description("organization name")),
|
||||
mcp.WithString("full_name", mcp.Description("organization full name")),
|
||||
mcp.WithString("description", mcp.Description("organization description")),
|
||||
mcp.WithString("website", mcp.Description("organization website")),
|
||||
mcp.WithString("location", mcp.Description("organization location")),
|
||||
mcp.WithString("visibility", mcp.Description("visibility: public, limited, or private")),
|
||||
)
|
||||
|
||||
DeleteOrgTool = mcp.NewTool(
|
||||
DeleteOrgToolName,
|
||||
mcp.WithDescription("Delete an organization"),
|
||||
mcp.WithString("org", mcp.Required(), mcp.Description("organization name")),
|
||||
)
|
||||
|
||||
RenameOrgTool = mcp.NewTool(
|
||||
RenameOrgToolName,
|
||||
mcp.WithDescription("Rename an organization"),
|
||||
mcp.WithString("org", mcp.Required(), mcp.Description("current organization name")),
|
||||
mcp.WithString("new_name", mcp.Required(), mcp.Description("new organization name")),
|
||||
)
|
||||
|
||||
ListOrgReposTool = mcp.NewTool(
|
||||
ListOrgReposToolName,
|
||||
mcp.WithDescription("List an organization's repositories"),
|
||||
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)),
|
||||
)
|
||||
|
||||
CreateOrgRepoTool = mcp.NewTool(
|
||||
CreateOrgRepoToolName,
|
||||
mcp.WithDescription("Create a repository in an organization"),
|
||||
mcp.WithString("org", mcp.Required(), mcp.Description("organization name")),
|
||||
mcp.WithString("name", mcp.Required(), mcp.Description("repository name")),
|
||||
mcp.WithString("description", mcp.Description("repository description")),
|
||||
mcp.WithBoolean("private", mcp.Description("whether the repository is private")),
|
||||
mcp.WithBoolean("auto_init", mcp.Description("whether to auto-initialize with README")),
|
||||
mcp.WithString("default_branch", mcp.Description("default branch name")),
|
||||
mcp.WithString("gitignores", mcp.Description("gitignore template")),
|
||||
mcp.WithString("license", mcp.Description("license template")),
|
||||
)
|
||||
)
|
||||
|
||||
func init() {
|
||||
Tool.RegisterRead(server.ServerTool{Tool: ListOrgsTool, Handler: ListOrgsFn})
|
||||
Tool.RegisterRead(server.ServerTool{Tool: GetOrgTool, Handler: GetOrgFn})
|
||||
Tool.RegisterRead(server.ServerTool{Tool: ListOrgReposTool, Handler: ListOrgReposFn})
|
||||
Tool.RegisterWrite(server.ServerTool{Tool: CreateOrgTool, Handler: CreateOrgFn})
|
||||
Tool.RegisterWrite(server.ServerTool{Tool: EditOrgTool, Handler: EditOrgFn})
|
||||
Tool.RegisterWrite(server.ServerTool{Tool: DeleteOrgTool, Handler: DeleteOrgFn})
|
||||
Tool.RegisterWrite(server.ServerTool{Tool: RenameOrgTool, Handler: RenameOrgFn})
|
||||
Tool.RegisterWrite(server.ServerTool{Tool: CreateOrgRepoTool, Handler: CreateOrgRepoFn})
|
||||
}
|
||||
|
||||
func ListOrgsFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called ListOrgsFn")
|
||||
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))
|
||||
}
|
||||
orgs, _, err := client.ListOrgs(gitea_sdk.ListOrgsOptions{
|
||||
ListOptions: gitea_sdk.ListOptions{Page: int(page), PageSize: int(pageSize)},
|
||||
})
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("list orgs err: %v", err))
|
||||
}
|
||||
return to.TextResult(orgs)
|
||||
}
|
||||
|
||||
func GetOrgFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called GetOrgFn")
|
||||
org, ok := req.GetArguments()["org"].(string)
|
||||
if !ok {
|
||||
return to.ErrorResult(errors.New("org is required"))
|
||||
}
|
||||
client, err := gitea.ClientFromContext(ctx)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("get gitea client err: %v", err))
|
||||
}
|
||||
organization, _, err := client.GetOrg(org)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("get org %s err: %v", org, err))
|
||||
}
|
||||
return to.TextResult(organization)
|
||||
}
|
||||
|
||||
func CreateOrgFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called CreateOrgFn")
|
||||
username, ok := req.GetArguments()["username"].(string)
|
||||
if !ok {
|
||||
return to.ErrorResult(errors.New("username is required"))
|
||||
}
|
||||
opt := gitea_sdk.CreateOrgOption{
|
||||
Name: username,
|
||||
}
|
||||
if v, ok := req.GetArguments()["full_name"].(string); ok {
|
||||
opt.FullName = v
|
||||
}
|
||||
if v, ok := req.GetArguments()["description"].(string); ok {
|
||||
opt.Description = v
|
||||
}
|
||||
if v, ok := req.GetArguments()["website"].(string); ok {
|
||||
opt.Website = v
|
||||
}
|
||||
if v, ok := req.GetArguments()["location"].(string); ok {
|
||||
opt.Location = v
|
||||
}
|
||||
if v, ok := req.GetArguments()["visibility"].(string); ok {
|
||||
opt.Visibility = gitea_sdk.VisibleType(v)
|
||||
}
|
||||
client, err := gitea.ClientFromContext(ctx)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("get gitea client err: %v", err))
|
||||
}
|
||||
organization, _, err := client.CreateOrg(opt)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("create org %s err: %v", username, err))
|
||||
}
|
||||
return to.TextResult(organization)
|
||||
}
|
||||
|
||||
func EditOrgFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called EditOrgFn")
|
||||
org, ok := req.GetArguments()["org"].(string)
|
||||
if !ok {
|
||||
return to.ErrorResult(errors.New("org is required"))
|
||||
}
|
||||
opt := gitea_sdk.EditOrgOption{}
|
||||
if v, ok := req.GetArguments()["full_name"].(string); ok {
|
||||
opt.FullName = v
|
||||
}
|
||||
if v, ok := req.GetArguments()["description"].(string); ok {
|
||||
opt.Description = v
|
||||
}
|
||||
if v, ok := req.GetArguments()["website"].(string); ok {
|
||||
opt.Website = v
|
||||
}
|
||||
if v, ok := req.GetArguments()["location"].(string); ok {
|
||||
opt.Location = v
|
||||
}
|
||||
if v, ok := req.GetArguments()["visibility"].(string); ok {
|
||||
opt.Visibility = gitea_sdk.VisibleType(v)
|
||||
}
|
||||
client, err := gitea.ClientFromContext(ctx)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("get gitea client err: %v", err))
|
||||
}
|
||||
_, err = client.EditOrg(org, opt)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("edit org %s err: %v", org, err))
|
||||
}
|
||||
updated, _, err := client.GetOrg(org)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("get org %s after edit err: %v", org, err))
|
||||
}
|
||||
return to.TextResult(updated)
|
||||
}
|
||||
|
||||
func DeleteOrgFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called DeleteOrgFn")
|
||||
org, ok := req.GetArguments()["org"].(string)
|
||||
if !ok {
|
||||
return to.ErrorResult(errors.New("org is required"))
|
||||
}
|
||||
client, err := gitea.ClientFromContext(ctx)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("get gitea client err: %v", err))
|
||||
}
|
||||
_, err = client.DeleteOrg(org)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("delete org %s err: %v", org, err))
|
||||
}
|
||||
return to.TextResult(map[string]string{"status": "deleted", "org": org})
|
||||
}
|
||||
|
||||
func RenameOrgFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called RenameOrgFn")
|
||||
org, ok := req.GetArguments()["org"].(string)
|
||||
if !ok {
|
||||
return to.ErrorResult(errors.New("org is required"))
|
||||
}
|
||||
newName, ok := req.GetArguments()["new_name"].(string)
|
||||
if !ok {
|
||||
return to.ErrorResult(errors.New("new_name is required"))
|
||||
}
|
||||
client, err := gitea.ClientFromContext(ctx)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("get gitea client err: %v", err))
|
||||
}
|
||||
_, err = client.RenameOrg(org, gitea_sdk.RenameOrgOption{NewName: newName})
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("rename org %s to %s err: %v", org, newName, err))
|
||||
}
|
||||
return to.TextResult(map[string]string{"status": "renamed", "old_name": org, "new_name": newName})
|
||||
}
|
||||
|
||||
func ListOrgReposFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called ListOrgReposFn")
|
||||
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))
|
||||
}
|
||||
repos, _, err := client.ListOrgRepos(org, gitea_sdk.ListOrgReposOptions{
|
||||
ListOptions: gitea_sdk.ListOptions{Page: int(page), PageSize: int(pageSize)},
|
||||
})
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("list org %s repos err: %v", org, err))
|
||||
}
|
||||
return to.TextResult(repos)
|
||||
}
|
||||
|
||||
func CreateOrgRepoFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||
log.Debugf("Called CreateOrgRepoFn")
|
||||
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.CreateRepoOption{
|
||||
Name: name,
|
||||
}
|
||||
if v, ok := req.GetArguments()["description"].(string); ok {
|
||||
opt.Description = v
|
||||
}
|
||||
if v, ok := req.GetArguments()["private"].(bool); ok {
|
||||
opt.Private = v
|
||||
}
|
||||
if v, ok := req.GetArguments()["auto_init"].(bool); ok {
|
||||
opt.AutoInit = v
|
||||
}
|
||||
if v, ok := req.GetArguments()["default_branch"].(string); ok {
|
||||
opt.DefaultBranch = v
|
||||
}
|
||||
if v, ok := req.GetArguments()["gitignores"].(string); ok {
|
||||
opt.Gitignores = v
|
||||
}
|
||||
if v, ok := req.GetArguments()["license"].(string); ok {
|
||||
opt.License = v
|
||||
}
|
||||
client, err := gitea.ClientFromContext(ctx)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("get gitea client err: %v", err))
|
||||
}
|
||||
repo, _, err := client.CreateOrgRepo(org, opt)
|
||||
if err != nil {
|
||||
return to.ErrorResult(fmt.Errorf("create repo %s/%s err: %v", org, name, err))
|
||||
}
|
||||
return to.TextResult(repo)
|
||||
}
|
||||
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