@rezti/dsh-rez-suite 0.1.0

package/src/tools.ts

@@ -0,0 +1,167 @@
+/**
+ * Agent-facing suite tools: status, configuration view, connectivity test,
+ * and audit ledger. MCP tools themselves are registered by mcp-host.ts.
+ */
+
+import { defineTool as defineToolOriginal } from '@deepseek-ai/dsh-tools'
+
+/** Loosen the typed helper for the suite's small fixed tools. */
+const defineTool = defineToolOriginal as unknown as (options: any) => any
+import type { ContentBlock } from '@deepseek-ai/dsh-llm'
+import type { RezConfig, RezTestResult } from './protocol.ts'
+import type { McpHost } from './mcp-host.ts'
+import type { TokenManager } from './token-manager.ts'
+import { publicConfig } from './store.ts'
+
+function text(value: string): ContentBlock[] {
+  return [{ type: 'text', text: value }]
+}
+
+function renderTest(results: RezTestResult[]): string {
+  if (results.length === 0) return 'no servers enabled'
+  return results.map(result => {
+    const head = result.server + ': ' + (result.ok ? 'ok' : 'failed')
+    const detail = result.ok
+      ? 'tools=' + (result.toolCount ?? 0) + ' latency=' + (result.latencyMs ?? 0) + 'ms'
+      : 'error=' + (result.error ?? 'unknown')
+    return head + ' (' + detail + ')'
+  }).join('\n')
+}
+
+export function rezStatusTool(host: McpHost, config: () => RezConfig) {
+  return defineTool({
+    name: 'rez_status',
+    description: 'List the dsh-rez-suite MCP server statuses and how many tools each server registered for the current role. Triggers: rez, MCP, 连接状态.',
+    parameters: {},
+    output: {
+      schema: {
+        type: 'object',
+        properties: {
+          role: { type: 'string', required: true },
+          servers: {
+            type: 'array',
+            required: true,
+            items: {
+              type: 'object',
+              additionalProperties: false,
+              properties: {
+                server: { type: 'string', required: true },
+                enabled: { type: 'boolean', required: true },
+                state: { type: 'string', required: true },
+                toolCount: { type: 'integer', required: true },
+                registeredTools: { type: 'integer', required: true },
+                lastError: { type: 'string' },
+                startedAt: { type: 'integer' },
+              },
+            },
+          },
+        },
+        additionalProperties: false,
+      },
+      render: (_args: any, value: { role?: string; servers?: Array<Record<string, unknown>> }) => {
+        const rows = (value.servers ?? []).map(server => [
+          server.server,
+          server.state,
+          server.toolCount,
+          server.registeredTools,
+          server.lastError ?? '',
+        ].join(' | '))
+        return text('role: ' + (value.role ?? '') + '\n' + ['server | state | discovered | registered | error', '--- | --- | --- | --- | ---', ...rows].join('\n'))
+      },
+    },
+    async execute() {
+      const servers = host.status().map(server => {
+        const clean: Record<string, unknown> = { ...server }
+        if (clean.lastError === undefined) delete clean.lastError
+        if (clean.startedAt === undefined) delete clean.startedAt
+        return clean
+      })
+      return { role: config().role, servers }
+    },
+  })
+}
+
+export function rezConfigTool(config: () => RezConfig) {
+  return defineTool({
+    name: 'rez_config',
+    description: 'Show the current dsh-rez-suite configuration with secrets masked. Use the web GUI Rez panel to change credentials.',
+    parameters: {},
+    output: {
+      schema: {
+        type: 'object',
+        properties: {
+          config: { type: 'json', required: true },
+        },
+        additionalProperties: false,
+      },
+      render: (_args: any, value: { config?: unknown }) => text(JSON.stringify(value.config ?? {}, null, 2)),
+    },
+    async execute() {
+      return { config: publicConfig(config()) }
+    },
+  })
+}
+
+export function rezTestTool(host: McpHost, config: () => RezConfig) {
+  return defineTool({
+    name: 'rez_test_connections',
+    description: 'Test connectivity to every enabled dsh-rez-suite MCP server with a throwaway connection. Triggers: 测试 MCP 连接.',
+    parameters: {},
+    output: {
+      schema: {
+        type: 'object',
+        properties: {
+          results: { type: 'array', required: true, items: { type: 'json' } },
+        },
+        additionalProperties: false,
+      },
+      render: (_args: any, value: { results?: RezTestResult[] }) => text(renderTest(value.results ?? [])),
+    },
+    async execute() {
+      return { results: await host.test(config()) }
+    },
+  })
+}
+
+export function rezAuditTool(tokens: TokenManager, config: () => RezConfig) {
+  return defineTool({
+    name: 'rez_audit',
+    description: 'Read the dsh-rez-suite token/audit ledger: total calls, tokens, cost, per-server breakdown, and recent records.',
+    parameters: {},
+    output: {
+      schema: {
+        type: 'object',
+        properties: {
+          audit: { type: 'json', required: true },
+        },
+        additionalProperties: false,
+      },
+      render: (_args: any, value: { audit?: unknown }) => text(JSON.stringify(value.audit ?? {}, null, 2)),
+    },
+    async execute() {
+      return { audit: tokens.summary(config().billing.monthlyBudget) }
+    },
+  })
+}
+
+export function rezAuditResetTool(tokens: TokenManager) {
+  return defineTool({
+    name: 'rez_audit_reset',
+    description: 'Clear the dsh-rez-suite audit ledger. Destructive; use only when the user asks to reset usage records.',
+    parameters: {},
+    output: {
+      schema: {
+        type: 'object',
+        properties: {
+          ok: { type: 'boolean', required: true },
+        },
+        additionalProperties: false,
+      },
+      render: (_args: any, value: { ok?: boolean }) => text(value.ok === true ? 'audit ledger reset' : 'audit reset failed'),
+    },
+    async execute() {
+      tokens.reset()
+      return { ok: true }
+    },
+  })
+}