#!/usr/bin/env node
import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';

import { parseGlobalReviewCliArgs, parseWorkstreamCloseoutCliArgs } from './core/cli.js';
import { cleanupHistoryArtifacts, parseCliCleanupArgs } from './tools/cleanup.js';
import { buildWorkstreamCloseout } from './tools/closeout.js';
import { buildGlobalCodeReview } from './tools/review.js';
import { registerTools } from './registerTools.js';
import { runSelfTest } from './selfTest.js';

if (process.argv.includes('--cleanup-history')) {
  console.log(JSON.stringify(cleanupHistoryArtifacts(parseCliCleanupArgs(process.argv)), null, 2));
} else if (process.argv.includes('--global-review')) {
  console.log(
    JSON.stringify(
      await buildGlobalCodeReview(parseGlobalReviewCliArgs(process.argv)),
      null,
      2,
    ),
  );
} else if (process.argv.includes('--self-test')) {
  await runSelfTest();
} else if (process.argv.includes('--workstream-closeout')) {
  console.log(JSON.stringify(buildWorkstreamCloseout(parseWorkstreamCloseoutCliArgs(process.argv)), null, 2));
} else {
  const server = new McpServer({
    name: 'kt-workflow',
    version: '0.6.2',
  });
  registerTools(server);
  await server.connect(new StdioServerTransport());
}