feat: align receive messages response contract
This commit is contained in:
@@ -2,6 +2,9 @@ package tools
|
||||
|
||||
import (
|
||||
"context"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/modelcontextprotocol/go-sdk/mcp"
|
||||
|
||||
@@ -27,29 +30,124 @@ func RegisterISphereReadTools(server *mcp.Server, source ReceiveMessagesSource)
|
||||
Name: ToolNameReceiveMessages,
|
||||
Description: "Return iSphere messages from the configured read-only log-backed source.",
|
||||
}, func(ctx context.Context, _ *mcp.CallToolRequest, input ReceiveMessagesArgs) (*mcp.CallToolResult, map[string]any, error) {
|
||||
started := time.Now().UTC()
|
||||
result, err := source.ReceiveMessages(ctx, isphere.ReceiveMessagesQuery{Limit: input.Limit})
|
||||
if err != nil {
|
||||
return nil, nil, err
|
||||
}
|
||||
return nil, receiveMessagesResultToMap(result), nil
|
||||
return nil, receiveMessagesResultToMap(result, started, time.Now().UTC()), nil
|
||||
})
|
||||
}
|
||||
|
||||
func receiveMessagesResultToMap(result isphere.ReceiveMessagesResult) map[string]any {
|
||||
func receiveMessagesResultToMap(result isphere.ReceiveMessagesResult, started time.Time, finished time.Time) map[string]any {
|
||||
messages := make([]map[string]any, 0, len(result.Messages))
|
||||
for _, message := range result.Messages {
|
||||
attachments := receiveMessageAttachments(message)
|
||||
messages = append(messages, map[string]any{
|
||||
"id": message.ID,
|
||||
"message_id": message.ID,
|
||||
"conversation_id": message.ConversationID,
|
||||
"conversation_type": message.ConversationType,
|
||||
"sender_id": message.SenderID,
|
||||
"sender_name": nil,
|
||||
"receiver_id": message.ReceiverID,
|
||||
"text": message.Text,
|
||||
"content_type": receiveMessageContentType(message, attachments),
|
||||
"content_text": message.Text,
|
||||
"attachments": attachments,
|
||||
"timestamp": message.Timestamp,
|
||||
"created_at": packetTimestampToRFC3339(message.Timestamp),
|
||||
"received_at": nil,
|
||||
"subject": message.Subject,
|
||||
"receipt_id": message.ReceiptID,
|
||||
"read": message.Read,
|
||||
"source": "local_readonly",
|
||||
"raw_ref": "packetlog_message",
|
||||
})
|
||||
}
|
||||
return map[string]any{"messages": messages}
|
||||
return map[string]any{
|
||||
"ok": true,
|
||||
"conversation": receiveMessagesConversation(result.Messages),
|
||||
"messages": messages,
|
||||
"next_cursor": nil,
|
||||
"audit": map[string]any{
|
||||
"tool": ToolNameReceiveMessages,
|
||||
"source": "local_readonly",
|
||||
"execution_mode": "read",
|
||||
"started_at": started.Format(time.RFC3339Nano),
|
||||
"finished_at": finished.Format(time.RFC3339Nano),
|
||||
"result": "ok",
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func receiveMessagesConversation(messages []isphere.Message) any {
|
||||
if len(messages) == 0 {
|
||||
return nil
|
||||
}
|
||||
first := messages[0]
|
||||
return map[string]any{
|
||||
"conversation_id": first.ConversationID,
|
||||
"conversation_type": contractConversationType(first.ConversationType),
|
||||
"display_name": first.ConversationID,
|
||||
}
|
||||
}
|
||||
|
||||
func contractConversationType(value string) string {
|
||||
switch strings.ToLower(strings.TrimSpace(value)) {
|
||||
case "groupchat", "group":
|
||||
return "group"
|
||||
case "chat", "direct":
|
||||
return "direct"
|
||||
case "system":
|
||||
return "system"
|
||||
default:
|
||||
return "unknown"
|
||||
}
|
||||
}
|
||||
|
||||
func receiveMessageAttachments(message isphere.Message) []map[string]any {
|
||||
files := isphere.ListFilesFromMessages([]isphere.Message{message}, isphere.ListFilesQuery{})
|
||||
attachments := make([]map[string]any, 0, len(files.Files))
|
||||
for _, file := range files.Files {
|
||||
attachments = append(attachments, map[string]any{
|
||||
"file_id": file.FileID,
|
||||
"file_name": file.FileName,
|
||||
"size_bytes": file.SizeBytes,
|
||||
"download_ref": nullableString(file.DownloadRef),
|
||||
})
|
||||
}
|
||||
return attachments
|
||||
}
|
||||
|
||||
func receiveMessageContentType(message isphere.Message, attachments []map[string]any) string {
|
||||
if len(attachments) > 0 {
|
||||
return "file"
|
||||
}
|
||||
if strings.TrimSpace(message.Text) != "" {
|
||||
return "text"
|
||||
}
|
||||
if strings.TrimSpace(message.Subject) != "" {
|
||||
return "notify"
|
||||
}
|
||||
return "unknown"
|
||||
}
|
||||
|
||||
func packetTimestampToRFC3339(value string) any {
|
||||
trimmed := strings.TrimSpace(value)
|
||||
if trimmed == "" {
|
||||
return nil
|
||||
}
|
||||
epoch, err := strconv.ParseInt(trimmed, 10, 64)
|
||||
if err != nil {
|
||||
return nil
|
||||
}
|
||||
var ts time.Time
|
||||
if epoch > 1_000_000_000_000 {
|
||||
ts = time.UnixMilli(epoch)
|
||||
} else {
|
||||
ts = time.Unix(epoch, 0)
|
||||
}
|
||||
formatted := ts.UTC().Format(time.RFC3339)
|
||||
return formatted
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user