takt/src/infra/task/summarize.ts

/**
 * Task name summarization using AI or romanization
 *
 * Generates concise English/romaji summaries for use in branch names and clone paths.
 */

import * as wanakana from 'wanakana';
import { loadGlobalConfig } from '../config/global/globalConfig.js';
import { getProvider, type ProviderType } from '../providers/index.js';
import { createLogger } from '../../shared/utils/index.js';
import { loadTemplate } from '../../shared/prompts/index.js';
import type { SummarizeOptions } from './types.js';

export type { SummarizeOptions };

const log = createLogger('summarize');

/**
 * Sanitize a string for use as git branch name and directory name.
 * Allows only: a-z, 0-9, hyphen.
 */
function sanitizeSlug(input: string, maxLength = 30): string {
  return input
    .trim()
    .toLowerCase()
    .replace(/[^a-z0-9-]/g, '-')
    .replace(/-+/g, '-')
    .replace(/^-+/, '')
    .slice(0, maxLength)
    .replace(/-+$/, '');
}

/**
 * Convert Japanese text to romaji slug.
 */
function toRomajiSlug(text: string): string {
  const romaji = wanakana.toRomaji(text, { customRomajiMapping: {} });
  return sanitizeSlug(romaji);
}

/**
 * Summarizes task names into concise slugs using AI or romanization.
 */
export class TaskSummarizer {
  /**
   * Summarize a task name into a concise slug.
   *
   * @param taskName - Original task name (can be in any language)
   * @param options - Summarization options
   * @returns Slug suitable for branch names (English if LLM, romaji if not)
   */
  async summarize(
    taskName: string,
    options: SummarizeOptions,
  ): Promise<string> {
    const globalConfig = loadGlobalConfig();
    const useLLM = options.useLLM ?? (globalConfig.branchNameStrategy === 'ai');
    log.info('Summarizing task name', { taskName, useLLM });

    if (!useLLM) {
      const slug = toRomajiSlug(taskName);
      log.info('Task name romanized', { original: taskName, slug });
      return slug || 'task';
    }
    const providerType = (globalConfig.provider as ProviderType) ?? 'claude';
    const model = options.model ?? globalConfig.model;

    const provider = getProvider(providerType);
    const agent = provider.setup({
      name: 'summarizer',
      systemPrompt: loadTemplate('score_slug_system_prompt', 'en'),
    });
    const response = await agent.call(taskName, {
      cwd: options.cwd,
      model,
      allowedTools: [],
    });

    const slug = sanitizeSlug(response.content);
    log.info('Task name summarized', { original: taskName, slug });

    return slug || 'task';
  }
}

// ---- Module-level function ----

const defaultSummarizer = new TaskSummarizer();

export async function summarizeTaskName(
  taskName: string,
  options: SummarizeOptions,
): Promise<string> {
  return defaultSummarizer.summarize(taskName, options);
}