#!/usr/bin/env node
// Interactive (and --yes non-interactive) setup wizard. Generates a minimal
// config.yaml (+ .env PORT) for a single LLM connection. Run before building:
//   npm run setup
import { createInterface } from 'node:readline/promises';
import { stdin as input, stdout as output } from 'node:process';
import {
  existsSync, readFileSync, openSync, writeFileSync, fsyncSync, closeSync,
  renameSync, copyFileSync, chmodSync, unlinkSync,
} from 'node:fs';
import { join } from 'node:path';
import {
  parseEndpoint, buildWorkerEntry, renderConfigYaml, renderDotenv,
  parseAnswersFromEnv, probeModels,
} from './setup-lib.mjs';

const ROOT = process.cwd();
const CONFIG_PATH = join(ROOT, 'config.yaml');
const ENV_PATH = join(ROOT, '.env');

const maskKey = (k) => (k ? `${String(k).slice(0, 6)}****` : '');

function atomicWrite(path, content, mode) {
  const tmp = `${path}.${process.pid}.tmp`;
  const fd = openSync(tmp, 'w', mode);
  try { writeFileSync(fd, content); fsyncSync(fd); } finally { closeSync(fd); }
  try { chmodSync(tmp, mode); } catch (e) { console.warn(`⚠ chmod ${tmp} failed (${e.message}); continuing`); }
  try { renameSync(tmp, path); } catch (e) { try { unlinkSync(tmp); } catch {} throw e; }
}

function backupExistingConfig() {
  const bak = `${CONFIG_PATH}.bak`;
  copyFileSync(CONFIG_PATH, bak);                 // copy (not rename) so original survives a later failure
  try { chmodSync(bak, 0o600); } catch (e) { console.warn(`⚠ chmod ${bak} failed (${e.message})`); }
  console.log(`  既存 config.yaml を ${bak} (0600) に退避しました。`);
}

function writeOutputs({ worker, port }) {
  // config.yaml is always 0600 (may hold an api_key).
  atomicWrite(CONFIG_PATH, renderConfigYaml(worker), 0o600);
  const existingEnv = existsSync(ENV_PATH) ? readFileSync(ENV_PATH, 'utf-8') : '';
  atomicWrite(ENV_PATH, renderDotenv(existingEnv, { port }), 0o644);
}

function printDone({ port, worker }) {
  console.log('');
  console.log('✅ config.yaml を生成しました（詳細設定は config.yaml.example か Settings UI を参照）。');
  if (worker.api_key) console.log(`   gateway api_key: ${maskKey(worker.api_key)}（config.yaml に保存、権限 0600）`);
  console.log('次の手順:');
  console.log('  scripts/build-all.sh');
  console.log('  scripts/server.sh start');
  console.log(`  ブラウザで http://localhost:${port}`);
}

async function runNonInteractive(force) {
  const { answers, error } = parseAnswersFromEnv(process.env);
  if (error) { console.error(`setup --yes: ${error}`); process.exit(1); }
  if (existsSync(CONFIG_PATH) && !force) {
    console.error('setup --yes: config.yaml already exists; pass --force to overwrite.');
    process.exit(1);
  }
  const probe = await probeModels(answers);   // non-fatal: model already provided via env
  if (!probe.ok) console.warn(`⚠ LLM probe failed (${probe.error}); continuing (SETUP_MODEL is set).`);
  const worker = buildWorkerEntry({
    connectionType: answers.connectionType, endpoint: answers.endpoint, model: answers.model, apiKey: answers.apiKey,
  });
  if (existsSync(CONFIG_PATH)) backupExistingConfig();
  writeOutputs({ worker, port: answers.port });
  printDone({ port: answers.port, worker });
}

async function ask(rl, question, def) {
  const suffix = def !== undefined && def !== '' ? ` [${def}]` : '';
  const a = (await rl.question(`${question}${suffix}: `)).trim();
  return a || (def ?? '');
}

async function runInteractive() {
  const rl = createInterface({ input, output });
  try {
    if (existsSync(CONFIG_PATH)) {
      const yn = (await ask(rl, '既存の config.yaml を上書きしますか? 既存は config.yaml.bak に退避します (y/N)', 'N'));
      if (!/^y(es)?$/i.test(yn)) { console.log('中止しました（変更なし）。'); return; }
    }

    // connection type
    let connectionType = 'direct';
    const ct = await ask(rl, '接続タイプ  1) direct (Ollama/vLLM 等)  2) aao_gateway (別 MAESTRO Gateway)', '1');
    connectionType = ct === '2' || ct === 'aao_gateway' ? 'aao_gateway' : 'direct';

    // endpoint
    let endpoint, base;
    const defEndpoint = connectionType === 'direct' ? 'http://localhost:11434/v1' : '';
    const hint = connectionType === 'aao_gateway' ? '（例: http://gateway-host:9876/v1）' : '';
    for (;;) {
      const raw = await ask(rl, `LLM endpoint URL${hint}`, defEndpoint);
      const r = parseEndpoint(raw);
      if (r.error) { console.log(`  ⚠ ${r.error}`); continue; }
      ({ endpoint, base } = r);
      break;
    }

    // api key (gateway only)
    let apiKey;
    if (connectionType === 'aao_gateway') {
      for (;;) {
        apiKey = await ask(rl, 'gateway API キー (sk-aao-...)', '');
        if (apiKey) break;
        console.log('  ⚠ aao_gateway では API キーが必須です。');
      }
    }

    // probe + model
    console.log('  接続を確認しています...');
    const probe = await probeModels({ endpoint, base, apiKey });
    let model;
    if (probe.ok && probe.models.length > 0) {
      console.log('  利用可能なモデル:');
      probe.models.forEach((m, i) => console.log(`    ${i + 1}) ${m}`));
      for (;;) {
        const sel = await ask(rl, `モデルを選択 (1-${probe.models.length}, または名前を直接入力)`, '1');
        const idx = Number(sel);
        if (Number.isInteger(idx) && idx >= 1 && idx <= probe.models.length) { model = probe.models[idx - 1]; break; }
        if (sel) { model = sel; break; }
      }
    } else {
      if (!probe.ok) console.log(`  ⚠ 接続できませんでした (${probe.error})。後で config を直せます。`);
      else console.log('  ⚠ モデル一覧が空でした。');
      for (;;) { model = await ask(rl, 'モデル名を手入力', ''); if (model) break; }
    }

    // server port
    let port;
    for (;;) {
      const p = Number(await ask(rl, 'MAESTRO サーバーの listen port', '9876'));
      if (Number.isInteger(p) && p >= 1 && p <= 65535) { port = p; break; }
      console.log('  ⚠ port は 1-65535 の整数で入力してください。');
    }

    const worker = buildWorkerEntry({ connectionType, endpoint, model, apiKey });
    if (existsSync(CONFIG_PATH)) backupExistingConfig();
    writeOutputs({ worker, port });
    printDone({ port, worker });
  } finally {
    rl.close();
  }
}

async function main() {
  const argv = process.argv.slice(2);
  const yes = argv.includes('--yes');
  const force = argv.includes('--force');
  if (yes) await runNonInteractive(force);
  else await runInteractive();
}

main().catch((e) => { console.error(`setup failed: ${e && e.message ? e.message : e}`); process.exit(1); });