Function registerTools

packages/mcp-server/src/server.ts:33–86

(server: McpServer, dbManager: McpDatabaseManager)

}

function registerTools(server: McpServer, dbManager: McpDatabaseManager): void {
  for (const tool of MCP_TOOL_REGISTRY) {
    const mcpName = `${MCP_TOOL_PREFIX}${tool.name}`

    if (tool.name === 'list_sessions') {
      const zodShape = {
        ...jsonSchemaToZod(tool.inputSchema.properties, tool.inputSchema.required),
        format: FORMAT_PARAM,
      }

      server.tool(mcpName, tool.description, zodShape, async (params) => {
        const format = params.format as string | undefined
        const context: SessionListContext = {
          db: null as any,
          sessionId: '',
          listSessionIds: () => dbManager.listSessionIds(),
          openDb: (id) => dbManager.open(id),
        }
        const toolParams = { ...params } as Record<string, unknown>
        delete toolParams.format
        const result = await tool.handler(toolParams, context)
        const text = applyFormat(result.content, format, tool.name)
        return { content: [{ type: 'text' as const, text }] }
      })
      continue
    }

    const zodShape = {
      session_id: z.string().describe('Session ID'),
      ...jsonSchemaToZod(tool.inputSchema.properties, tool.inputSchema.required),
      format: FORMAT_PARAM,
    }

    server.tool(mcpName, tool.description, zodShape, async (params) => {
      const sessionId = params.session_id as string
      const format = params.format as string | undefined
      const db = dbManager.open(sessionId)
      if (!db) {
        return {
          content: [{ type: 'text' as const, text: `Session not found: ${sessionId}` }],
          isError: true,
        }
      }

      const toolParams = { ...params } as Record<string, unknown>
      delete toolParams.session_id
      delete toolParams.format

      const result = await tool.handler(toolParams, { db, sessionId, dataProvider: new CoreDataProvider(db) })
      const text = applyFormat(result.content, format, tool.name)
      return { content: [{ type: 'text' as const, text }] }
    })
  }
}

function registerResources(server: McpServer, dbManager: McpDatabaseManager): void {
  server.resource('sessions-list', 'chatlab://sessions', { description: '所有已导入的聊天会话列表' }, async () => {