#!/usr/bin/env node
import { spawn, execSync } from 'node:child_process';
import fs from 'node:fs';
import path from 'node:path';
import { fileURLToPath } from 'node:url';
import {
  applyMemindRuntimeProfile,
  describeMemindRuntimeProfile,
  loadMemindEnvFiles,
} from './memind-runtime-profile.mjs';
import {
  deepseekDisableThinkingEnabled,
  resolveDeepseekNoThinkListenPort,
} from '../deepseek-no-think-proxy.mjs';

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

// Load .env before reading ports — otherwise ADMIN_PORT defaults to 8082 and
// split-service MindSpace (also on 8082) gets killed by freePort().
loadMemindEnvFiles(root);
applyMemindRuntimeProfile({ rootDir: root });
console.log(`==> Runtime profile: ${describeMemindRuntimeProfile()}`);

const portalPort = Number(process.env.H5_PORT ?? 8081);
const vitePort = Number(process.env.VITE_PORT ?? 5173);
const plazaPort = Number(process.env.PLAZA_PORT ?? 3001);
const opsPort = Number(process.env.OPS_PORT ?? 3002);
const adminPort = Number(process.env.ADMIN_PORT ?? 8082);
const deepseekNoThinkEnabled = deepseekDisableThinkingEnabled();
const deepseekNoThinkPort = resolveDeepseekNoThinkListenPort();
const deepseekNoThinkUrl = `http://127.0.0.1:${deepseekNoThinkPort}`;
const portalUrl = `http://127.0.0.1:${portalPort}`;
const adminUrl = `http://127.0.0.1:${adminPort}`;
const localMindSpacePublicBase = `http://127.0.0.1:${vitePort}`;
const plazaPublicBase = (
  process.env.PLAZA_PUBLIC_BASE ?? `http://127.0.0.1:${plazaPort}`
).replace(/\/$/, '');

function freePort(port) {
  try {
    execSync(`lsof -ti TCP:${port} -sTCP:LISTEN | xargs kill -9`, { stdio: 'ignore' });
  } catch {
    // port already free
  }
}

function spawnChild(command, args, label, cwd = root, extraEnv = {}) {
  const child = spawn(command, args, {
    cwd,
    env: { ...process.env, ...extraEnv },
    stdio: 'inherit',
  });
  child.on('exit', (code, signal) => {
    if (signal) return;
    if (code && code !== 0) {
      console.error(`[${label}] exited with code ${code}`);
      shutdown(code ?? 1);
    }
  });
  return child;
}

let server;
let admin;
let ops;
let vite;
let deepseekNoThinkProxy;
let stopping = false;

function shutdown(code = 0) {
  if (stopping) return;
  stopping = true;
  server?.kill('SIGTERM');
  admin?.kill('SIGTERM');
  ops?.kill('SIGTERM');
  vite?.kill('SIGTERM');
  deepseekNoThinkProxy?.kill('SIGTERM');
  setTimeout(() => process.exit(code), 300);
}

process.on('SIGINT', () => shutdown(0));
process.on('SIGTERM', () => shutdown(0));

async function waitFor(url, check, label, retries = 60) {
  for (let i = 0; i < retries; i += 1) {
    try {
      if (await check(url)) return;
    } catch {
      // still starting
    }
    await new Promise((resolve) => setTimeout(resolve, 500));
  }
  throw new Error(`${label} 未在 ${url} 启动`);
}

const useConfiguredPublicBase = ['1', 'true', 'yes', 'on'].includes(
  String(process.env.MEMIND_DEV_USE_PROD_PUBLIC_BASE ?? '').trim().toLowerCase(),
);
const mindSpacePublicBase = (
  useConfiguredPublicBase && process.env.H5_PUBLIC_BASE_URL
    ? process.env.H5_PUBLIC_BASE_URL
    : portalUrl
).replace(/\/$/, '');
process.env.H5_PUBLIC_BASE_URL = mindSpacePublicBase;

const viteEnv = {
  VITE_PLAZA_BASE: plazaPublicBase,
  VITE_MINDSPACE_BASE: mindSpacePublicBase,
  H5_DEV_PORTAL: portalUrl,
};

const opsEnv = {
  OPS_API_PROXY: portalUrl,
  OPS_ADMIN_PROXY: adminUrl,
  VITE_PLAZA_BASE: plazaPublicBase,
  VITE_MINDSPACE_BASE: mindSpacePublicBase,
};

const managedPorts = [
  portalPort,
  adminPort,
  vitePort,
  opsPort,
  ...(deepseekNoThinkEnabled ? [deepseekNoThinkPort] : []),
];
console.log(`==> 清理端口 ${managedPorts.join(' / ')}...`);
freePort(portalPort);
freePort(adminPort);
freePort(vitePort);
freePort(opsPort);
if (deepseekNoThinkEnabled) freePort(deepseekNoThinkPort);
await new Promise((resolve) => setTimeout(resolve, 500));

try {
  if (deepseekNoThinkEnabled) {
    console.log('==> 启动 DeepSeek no-thinking proxy...');
    deepseekNoThinkProxy = spawnChild(
      'node',
      [path.join(root, 'deepseek-no-think-proxy.mjs')],
      'deepseek-no-think',
      root,
      { MEMIND_DEEPSEEK_PROXY_ENTRYPOINT: '1' },
    );
    await waitFor(
      deepseekNoThinkUrl,
      async (url) => (await fetch(`${url}/health`)).ok,
      'DeepSeek no-thinking proxy',
    );
    console.log(`==> DeepSeek no-thinking proxy 就绪: ${deepseekNoThinkUrl}`);
  }

  console.log('==> 启动 portal (server.mjs)...');
  server = spawnChild('node', [path.join(root, 'server.mjs')], 'portal');
  await waitFor(portalUrl, async (url) => (await fetch(`${url}/auth/status`)).ok, 'Portal');
  console.log(`==> Portal 就绪: ${portalUrl}`);

  console.log('==> 启动 memind_adm (admin-server.mjs)...');
  admin = spawnChild('node', [path.join(root, 'admin-server.mjs')], 'admin');
  await waitFor(adminUrl, async (url) => (await fetch(`${url}/healthz`)).ok, 'Admin');
  console.log(`==> memind_adm 就绪: ${adminUrl}`);

  console.log(`==> 启动 Ops 后台 @ http://127.0.0.1:${opsPort}/ops/`);
  ops = spawnChild('npm', ['run', 'dev'], 'ops', opsDir, opsEnv);
  await waitFor(
    `http://127.0.0.1:${opsPort}`,
    async (url) => (await fetch(`${url}/ops/`)).ok,
    'Ops',
    40,
  );
  console.log(`==> Ops 就绪: http://127.0.0.1:${opsPort}/ops/`);

  console.log(`==> 启动 Vite @ http://127.0.0.1:${vitePort}`);
  vite = spawnChild('npx', ['vite'], 'vite', root, viteEnv);
  await waitFor(
    `http://127.0.0.1:${vitePort}`,
    async (url) => {
      const [rootRes, mainRes] = await Promise.all([
        fetch(`${url}/`),
        fetch(`${url}/src/main.tsx`),
      ]);
      return rootRes.ok && mainRes.ok;
    },
    'Vite',
  );
  console.log(`==> Vite 就绪: http://127.0.0.1:${vitePort}`);

  console.log('');
  console.log('本地服务：');
  console.log(`  MindSpace UI   http://127.0.0.1:${vitePort}/?preview=mindspace`);
  console.log(`  MindSpace URL  ${mindSpacePublicBase} (Portal 直链，Agent 交付链接)`);
  console.log(`  Ops 审核后台   http://127.0.0.1:${opsPort}/ops/`);
  console.log(`  API / Portal   ${portalUrl}`);
  console.log(`  memind_adm     ${adminUrl}`);
  if (deepseekNoThinkEnabled) {
    console.log(`  DeepSeek Proxy ${deepseekNoThinkUrl}`);
  }
  console.log('');
  console.log('Plaza 已独立出去：请另开终端运行 pnpm dev:plaza 或 pnpm start:plaza');
} catch (err) {
  console.error(err instanceof Error ? err.message : err);
  shutdown(1);
}