@kubb/cli/src/commands/mcp.ts

Command-line interface for Kubb, enabling easy generation of TypeScript, React-Query, Zod, and other code from OpenAPI specifications.

import { styleText } from 'node:util'
import type { ArgsDef } from 'citty'
import { defineCommand, showUsage } from 'citty'
import { createJiti } from 'jiti'

const jiti = createJiti(import.meta.url, {
  sourceMaps: true,
})

const args = {
  help: {
    type: 'boolean',
    description: 'Show help',
    alias: 'h',
    default: false,
  },
} as const satisfies ArgsDef

const command = defineCommand({
  meta: {
    name: 'mcp',
    description: 'Start the server to enable the MCP client to interact with the LLM.',
  },
  args,
  async run(commandContext) {
    const { args } = commandContext

    if (args.help) {
      return showUsage(command)
    }

    let mod: any
    try {
      mod = await jiti.import('@kubb/mcp', { default: true })
    } catch (_e) {
      console.error(`Import of '@kubb/mcp' is required to start the MCP server`)
      process.exit(1)
    }

    const { run } = mod
    try {
      console.log('⏳ Starting MCP server...')
      console.warn(styleText('yellow', 'This feature is still under development — use with caution'))
      await run()
    } catch (error) {
      console.error((error as Error)?.message)
    }
  },
})

export default command