ChatLab/apps/cli/src/mcp.ts

/**
 * CLI MCP Server entry point
 *
 * Thin wrapper: initializes Node.js runtime, then delegates to chatlab-mcp.
 */

import fs from 'fs'
import { loadConfig } from '@openchatlab/config'
import {
  NodePathProvider,
  DatabaseManager,
  hasPendingElectronDataWarning,
  verifyCliDataPath,
} from '@openchatlab/node-runtime'
import { startMcpServer } from 'chatlab-mcp'
import { getVersion } from './version'
import { resolveCliPath } from './paths'
import { assertCliDataDirCompatible } from './runtime-compat'

function resolveNativeBinding(): string | undefined {
  if (process.versions.electron) return undefined
  const nativePath = resolveCliPath('native/better_sqlite3.node')
  if (fs.existsSync(nativePath)) return nativePath
  return undefined
}

export function initMcpRuntime(): DatabaseManager {
  const config = loadConfig()
  const userDataDir = config.data.user_data_dir || undefined
  const pathProvider = new NodePathProvider(userDataDir)
  pathProvider.ensureAllDirs()
  const runtime = assertCliDataDirCompatible(pathProvider, 'mcp')

  if (hasPendingElectronDataWarning() || !verifyCliDataPath(pathProvider.getDatabaseDir())) {
    console.error('[MCP] Electron desktop data detected but databases not found.')
    console.error('[MCP] Set CHATLAB_DATA_DIR or edit ~/.chatlab/config.toml to point to your data directory.')
    process.exit(1)
  }

  const nativeBinding = resolveNativeBinding()
  return new DatabaseManager(pathProvider, { nativeBinding, runtime })
}

export async function startCliMcpServer(): Promise<void> {
  const dbManager = initMcpRuntime()
  await startMcpServer({ version: getVersion(), dbManager })
}