code-search.ts

import { Command } from 'commander';

import type { CliContext } from '../context.js';
import { formatCodeSearchResults } from '../formatters/search-results.js';
import { resolveEndpoint } from '../utils/options.js';
import { ensureTrailingNewline } from '../utils/text.js';

interface CodeSearchCommandOptions {
  language?: string;
  json?: boolean;
}

export function registerCodeSearchCommand(program: Command, context: CliContext): void {
  program
    .command('code-search')
    .description('Search Microsoft Learn code samples through the Learn MCP server.')
    .argument('<query>', 'Search query.')
    .option('--language <name>', 'Preferred language filter to pass to Learn.')
    .option('--json', 'Output raw JSON instead of formatted text.')
    .action(async (query: string, options: CodeSearchCommandOptions) => {
      const endpoint = resolveEndpoint(program.opts<{ endpoint?: string }>().endpoint, context.env);
      const client = context.createClient({
        endpoint,
        clientName: 'mslearn',
        clientVersion: context.version,
      });

      try {
        const payload = await client.searchCodeSamples(query, options.language);
        const output = options.json ? payload : formatCodeSearchResults(payload);
        context.writeOut(ensureTrailingNewline(output));
      } finally {
        await client.close();
      }
    });
}