takt/src/commands/ensemble/add.ts

/**
 * takt ensemble add — install an ensemble package from GitHub.
 *
 * Usage:
 *   takt ensemble add github:{owner}/{repo}@{ref}
 *   takt ensemble add github:{owner}/{repo}          (uses default branch)
 */

import { mkdirSync, copyFileSync, existsSync, readFileSync, writeFileSync, rmSync } from 'node:fs';
import { join, dirname } from 'node:path';
import { tmpdir } from 'node:os';
import { execFileSync } from 'node:child_process';
import { stringify as stringifyYaml } from 'yaml';
import { getEnsemblePackageDir } from '../../infra/config/paths.js';
import { parseGithubSpec } from '../../features/ensemble/github-spec.js';
import {
  parseTaktPackConfig,
  validateTaktPackPath,
  validateMinVersion,
  isVersionCompatible,
  checkPackageHasContent,
  validateRealpathInsideRoot,
  resolvePackConfigPath,
} from '../../features/ensemble/takt-pack-config.js';
import { collectCopyTargets } from '../../features/ensemble/file-filter.js';
import { parseTarVerboseListing } from '../../features/ensemble/tar-parser.js';
import { resolveRef } from '../../features/ensemble/github-ref-resolver.js';
import { atomicReplace, cleanupResiduals } from '../../features/ensemble/atomic-update.js';
import { generateLockFile, extractCommitSha } from '../../features/ensemble/lock-file.js';
import { TAKT_PACKAGE_MANIFEST_FILENAME } from '../../features/ensemble/constants.js';
import { summarizeFacetsByType, detectEditPieces, formatEditPieceWarnings } from '../../features/ensemble/pack-summary.js';
import { confirm } from '../../shared/prompt/index.js';
import { info, success } from '../../shared/ui/index.js';
import { createLogger, getErrorMessage } from '../../shared/utils/index.js';

// eslint-disable-next-line @typescript-eslint/no-require-imports
const { version: TAKT_VERSION } = require('../../../package.json') as { version: string };

const log = createLogger('ensemble-add');

export async function ensembleAddCommand(spec: string): Promise<void> {
  const { owner, repo, ref: specRef } = parseGithubSpec(spec);

  try {
    execFileSync('gh', ['--version'], { stdio: 'pipe' });
  } catch {
    throw new Error(
      '`gh` CLI がインストールされていません。https://cli.github.com からインストールしてください',
    );
  }

  const execGh = (args: string[]) =>
    execFileSync('gh', args, { encoding: 'utf-8', stdio: 'pipe' });

  const ref = resolveRef(specRef, owner, repo, execGh);

  const tmpBase = join(tmpdir(), `takt-import-${Date.now()}`);
  const tmpTarPath = `${tmpBase}.tar.gz`;
  const tmpExtractDir = `${tmpBase}-extract`;
  const tmpIncludeFile = `${tmpBase}-include.txt`;

  try {
    mkdirSync(tmpExtractDir, { recursive: true });

    info(`📦 ${owner}/${repo} @${ref} をダウンロード中...`);
    execFileSync(
      'gh',
      [
        'api',
        `/repos/${owner}/${repo}/tarball/${ref}`,
        '--header', 'Accept: application/octet-stream',
        '--output', tmpTarPath,
      ],
      { stdio: ['inherit', 'pipe', 'pipe'] },
    );

    const tarVerboseList = execFileSync('tar', ['tvzf', tmpTarPath], {
      encoding: 'utf-8',
      stdio: 'pipe',
    });

    const verboseLines = tarVerboseList.split('\n').filter(l => l.trim());
    const { firstDirEntry, includePaths } = parseTarVerboseListing(verboseLines);

    const commitSha = extractCommitSha(firstDirEntry);

    if (includePaths.length > 0) {
      writeFileSync(tmpIncludeFile, includePaths.join('\n') + '\n');
      execFileSync(
        'tar',
        ['xzf', tmpTarPath, '-C', tmpExtractDir, '--strip-components=1', '-T', tmpIncludeFile],
        { stdio: 'pipe' },
      );
    }

    const packConfigPath = resolvePackConfigPath(tmpExtractDir);

    const packConfigYaml = readFileSync(packConfigPath, 'utf-8');
    const config = parseTaktPackConfig(packConfigYaml);
    validateTaktPackPath(config.path);

    if (config.takt?.min_version) {
      validateMinVersion(config.takt.min_version);
      if (!isVersionCompatible(config.takt.min_version, TAKT_VERSION)) {
        throw new Error(
          `このパッケージは TAKT ${config.takt.min_version} 以降が必要です（現在: ${TAKT_VERSION}）`,
        );
      }
    }

    const packageRoot = config.path === '.' ? tmpExtractDir : join(tmpExtractDir, config.path);

    validateRealpathInsideRoot(packageRoot, tmpExtractDir);

    checkPackageHasContent(packageRoot);

    const targets = collectCopyTargets(packageRoot);
    const facetFiles = targets.filter(t => t.relativePath.startsWith('faceted/'));
    const pieceFiles = targets.filter(t => t.relativePath.startsWith('pieces/'));

    const facetSummary = summarizeFacetsByType(facetFiles.map(t => t.relativePath));

    const pieceYamls: Array<{ name: string; content: string }> = [];
    for (const pf of pieceFiles) {
      try {
        const content = readFileSync(pf.absolutePath, 'utf-8');
        pieceYamls.push({ name: pf.relativePath.replace(/^pieces\//, ''), content });
      } catch (err) {
        log.debug('Failed to parse piece YAML for edit check', { path: pf.absolutePath, error: getErrorMessage(err) });
      }
    }
    const editPieces = detectEditPieces(pieceYamls);

    info(`\n📦 ${owner}/${repo} @${ref}`);
    info(`   faceted: ${facetSummary}`);
    if (pieceFiles.length > 0) {
      const pieceNames = pieceFiles.map(t =>
        t.relativePath.replace(/^pieces\//, '').replace(/\.yaml$/, ''),
      );
      info(`   pieces:  ${pieceFiles.length} (${pieceNames.join(', ')})`);
    } else {
      info('   pieces:  0');
    }
    for (const ep of editPieces) {
      for (const warning of formatEditPieceWarnings(ep)) {
        info(warning);
      }
    }
    info('');

    const confirmed = await confirm('インストールしますか？', false);
    if (!confirmed) {
      info('キャンセルしました');
      return;
    }

    const packageDir = getEnsemblePackageDir(owner, repo);

    if (existsSync(packageDir)) {
      const overwrite = await confirm(
        `${owner}/${repo} は既にインストールされています。上書きしますか？`,
        false,
      );
      if (!overwrite) {
        info('キャンセルしました');
        return;
      }
    }

    cleanupResiduals(packageDir);

    await atomicReplace({
      packageDir,
      install: async () => {
        for (const target of targets) {
          const destFile = join(packageDir, target.relativePath);
          mkdirSync(dirname(destFile), { recursive: true });
          copyFileSync(target.absolutePath, destFile);
        }
        copyFileSync(packConfigPath, join(packageDir, TAKT_PACKAGE_MANIFEST_FILENAME));

        const lock = generateLockFile({
          source: `github:${owner}/${repo}`,
          ref,
          commitSha,
          importedAt: new Date(),
        });
        writeFileSync(join(packageDir, '.takt-pack-lock.yaml'), stringifyYaml(lock));
      },
    });

    success(`✅ ${owner}/${repo} @${ref} をインストールしました`);
  } finally {
    if (existsSync(tmpTarPath)) rmSync(tmpTarPath, { force: true });
    if (existsSync(tmpExtractDir)) rmSync(tmpExtractDir, { recursive: true, force: true });
    if (existsSync(tmpIncludeFile)) rmSync(tmpIncludeFile, { force: true });
  }
}