88 lines
3.4 KiB
Go
88 lines
3.4 KiB
Go
package tools
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/modelcontextprotocol/go-sdk/mcp"
|
|
|
|
"isphere-ai-bridge/internal/isphere"
|
|
"isphere-ai-bridge/internal/msglib"
|
|
)
|
|
|
|
const ToolNameSearchGroups = "isphere_search_groups"
|
|
|
|
type SearchGroupsArgs struct {
|
|
Query string `json:"query" jsonschema:"group search query; currently matches bare group JIDs from local readonly message logs"`
|
|
Limit int `json:"limit,omitempty" jsonschema:"maximum number of groups to return; zero or negative returns all matches"`
|
|
Cursor string `json:"cursor,omitempty" jsonschema:"pagination cursor; currently only empty cursor is supported"`
|
|
SourcePreference string `json:"source_preference,omitempty" jsonschema:"source selector; currently supports auto or local_readonly only"`
|
|
IncludeArchived bool `json:"include_archived,omitempty" jsonschema:"accepted for contract compatibility; local readonly message logs do not expose archived state"`
|
|
}
|
|
|
|
func RegisterISphereGroupTools(server *mcp.Server, source ReceiveMessagesSource) {
|
|
RegisterISphereGroupToolsWithDisplayEntities(server, source, nil)
|
|
}
|
|
|
|
func RegisterISphereGroupToolsWithDisplayEntities(server *mcp.Server, source ReceiveMessagesSource, displaySource DisplayEntitySource) {
|
|
if source == nil {
|
|
source = isphere.EncryptedPacketLogSource{}
|
|
}
|
|
|
|
mcp.AddTool[SearchGroupsArgs, map[string]any](server, &mcp.Tool{
|
|
Name: ToolNameSearchGroups,
|
|
Description: "Search iSphere group candidates from the configured read-only message source.",
|
|
}, func(ctx context.Context, _ *mcp.CallToolRequest, input SearchGroupsArgs) (*mcp.CallToolResult, map[string]any, error) {
|
|
if err := validateLocalReadonlySourceAndCursor(ToolNameSearchGroups, input.SourcePreference, input.Cursor); err != nil {
|
|
return nil, nil, err
|
|
}
|
|
started := time.Now().UTC()
|
|
messages, err := source.ReceiveMessages(ctx, isphere.ReceiveMessagesQuery{Limit: 0})
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
logGroups := isphere.SearchGroupsFromMessages(messages.Messages, isphere.SearchGroupsQuery{Query: input.Query, Limit: input.Limit})
|
|
groups := logGroups
|
|
if displaySource != nil {
|
|
entities, err := displaySource.DisplayEntities(ctx, msglib.DisplayEntitiesOptions{
|
|
EntityType: msglib.EntityTypeGroups,
|
|
Query: input.Query,
|
|
Limit: displayEntityQueryLimit(input.Limit),
|
|
})
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
groups = mergeGroups(groupsFromDisplayEntities(entities), logGroups.Groups, input.Limit)
|
|
}
|
|
return nil, searchGroupsResultToMap(groups, started, time.Now().UTC()), nil
|
|
})
|
|
}
|
|
|
|
func searchGroupsResultToMap(result isphere.SearchGroupsResult, started time.Time, finished time.Time) map[string]any {
|
|
groups := make([]map[string]any, 0, len(result.Groups))
|
|
for _, group := range result.Groups {
|
|
groups = append(groups, map[string]any{
|
|
"group_id": group.GroupID,
|
|
"display_name": group.DisplayName,
|
|
"member_count": nil,
|
|
"owner_ref": nil,
|
|
"source": group.Source,
|
|
"confidence": group.Confidence,
|
|
"raw_ref": group.RawRef,
|
|
})
|
|
}
|
|
return map[string]any{
|
|
"ok": true,
|
|
"groups": groups,
|
|
"next_cursor": nil,
|
|
"audit": map[string]any{
|
|
"tool": ToolNameSearchGroups,
|
|
"source": "local_readonly",
|
|
"execution_mode": "read",
|
|
"started_at": started.Format(time.RFC3339Nano),
|
|
"finished_at": finished.Format(time.RFC3339Nano),
|
|
"result": "ok",
|
|
},
|
|
}
|
|
}
|