#!/usr/bin/env node
import fs from 'node:fs';
import path from 'node:path';
import { fileURLToPath } from 'node:url';
import { createDbPool } from '../db.mjs';
import { createLlmProviderService } from '../llm-providers.mjs';

const root = path.join(path.dirname(fileURLToPath(import.meta.url)), '..');

function loadEnvFile(filePath) {
  if (!fs.existsSync(filePath)) return;
  for (const line of fs.readFileSync(filePath, 'utf8').split('\n')) {
    const trimmed = line.trim();
    if (!trimmed || trimmed.startsWith('#')) continue;
    const eq = trimmed.indexOf('=');
    if (eq < 0) continue;
    const key = trimmed.slice(0, eq).trim();
    const value = trimmed.slice(eq + 1).trim();
    if (!process.env[key]) process.env[key] = value;
  }
}

function shellQuote(value) {
  return `'${String(value).replace(/'/g, `'\\''`)}'`;
}

function parseArgs(argv) {
  const out = {
    executor: String(argv[0] ?? '').trim(),
    purpose: 'default',
    shell: false,
  };
  for (let i = 1; i < argv.length; i += 1) {
    const item = argv[i];
    if (item === '--purpose') out.purpose = String(argv[++i] ?? out.purpose);
    else if (item === '--shell') out.shell = true;
  }
  return out;
}

loadEnvFile(path.join(root, '.env'));
loadEnvFile(path.join(root, '.env.local'));

const parsed = parseArgs(process.argv.slice(2));
if (!parsed.executor) {
  console.error('Usage: node scripts/executor-env.mjs <executor> [--purpose default] [--shell]');
  process.exit(1);
}

const pool = createDbPool();
try {
  const svc = createLlmProviderService(pool, {
    apiTarget: process.env.TKMIND_API_TARGET ?? 'https://127.0.0.1:18006',
    apiSecret: process.env.TKMIND_SERVER__SECRET_KEY ?? 'local-dev-secret',
  });
  const runtime = await svc.getExecutorRuntimeConfig(parsed.executor, {
    purpose: parsed.purpose,
    includeSecret: true,
  });
  if (!runtime?.ok) {
    console.error(runtime?.message ?? 'executor runtime unavailable');
    process.exit(2);
  }
  const env = runtime.env ?? {};
  if (parsed.shell) {
    for (const [key, value] of Object.entries(env)) {
      process.stdout.write(`export ${key}=${shellQuote(value)}\n`);
    }
  } else {
    process.stdout.write(`${JSON.stringify({ executor: parsed.executor, env })}\n`);
  }
} finally {
  await pool.end().catch(() => {});
}