Function registerNonCodemodeTools

src/tools/non-codemode.ts:24–69

(server: McpServer, props: AuthProps)

 * `tools/call` validates and dispatches only the requested operation.
 */
export async function registerNonCodemodeTools(server: McpServer, props: AuthProps): Promise<void> {
  const tools = await getNonCodemodeTools()
  const toolsByName = await getNonCodemodeToolMap()
  const resolvedAccountId = autoResolvedAccountId(props)

  server.server.registerCapabilities({ tools: { listChanged: false } })

  server.server.setRequestHandler('tools/list', () => ({
    tools: [
      DOCS_TOOL,
      ...tools.map((tool) => toWireTool(toolForAccountAccess(tool, resolvedAccountId, props)))
    ]
  }))

  server.server.setRequestHandler('tools/call', async (request) => {
    const name = request.params.name
    let result: CallToolResult

    try {
      if (name === DOCS_TOOL.name) {
        const parsed = z.object({ query: z.string() }).safeParse(request.params.arguments ?? {})
        result = parsed.success
          ? await runDocsTool(parsed.data.query)
          : validationError(name, parsed.error)
      } else {
        const baseTool = toolsByName.get(name)
        if (!baseTool) {
          result = toolError(`Tool ${name} not found`)
        } else {
          const tool = toolForAccountAccess(baseTool, resolvedAccountId, props)
          const parsed = z
            .object(zodInputSchemaFromJson(tool.inputSchema))
            .safeParse(request.params.arguments ?? {})
          result = parsed.success
            ? await callNonCodemodeTool(baseTool, parsed.data, resolvedAccountId, props.accessToken)
            : validationError(name, parsed.error)
        }
      }
    } catch (error) {
      result = toolError(error instanceof Error ? error.message : String(error))
    }

    recordToolCall(props, name, result.isError === true)
    return result
  })
}

async function callNonCodemodeTool(
  tool: NonCodemodeTool,