60f05e5f1e
Rename the fork from gitea-mcp to gitea-mcp-extended to reflect the significantly expanded tool coverage (299 vs upstream's 93 tools). - Rename Go module path and all import references - Rename binary to gitea-mcp-extended in Makefile, Dockerfile, .gitignore - Point .goreleaser.yaml gitea_urls to git.lethalbits.com - Replace release-tag workflow with goreleaser + Generic Package Registry publishing - Replace release-nightly workflow with cross-platform build + nightly package publishing - Update CLAUDE.md project description and tool count
242 lines
9.5 KiB
Go
242 lines
9.5 KiB
Go
package organization
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
|
|
"git.lethalbits.com/lethalbits/gitea-mcp-extended/pkg/gitea"
|
|
"git.lethalbits.com/lethalbits/gitea-mcp-extended/pkg/log"
|
|
"git.lethalbits.com/lethalbits/gitea-mcp-extended/pkg/params"
|
|
"git.lethalbits.com/lethalbits/gitea-mcp-extended/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)
|
|
}
|