Genarrative/apps/ai-game-creator-shell/scripts/start-tauri-dev.mjs

import { resolve } from 'node:path';
import { fileURLToPath } from 'node:url';

import { buildLocalRustProcessEnv } from '../../../scripts/dev.mjs';
import {
  defaultEditorFeatures,
  withDefaultCargoFeatures,
} from './cargo-features.mjs';
import {
  readAgcDevEndpoint,
  resolveAgcDevEndpoint,
  withAgcDevEndpointEnv,
} from './dev-port.mjs';
import {
  isAiGameCreatorServer,
  preflightExistingVite,
  readChildFailure,
  readExistingViteServer,
  spawnChild,
  stopChild,
  terminateChildTree,
  waitForChildTermination,
} from './start-dev-stack.mjs';

const appRoot = fileURLToPath(new URL('..', import.meta.url));
const repoRoot = resolve(appRoot, '../..');
const tauriCliPath = resolve(repoRoot, 'node_modules/@tauri-apps/cli/tauri.js');
const AGC_DESIGN_DEBUG_ENV = 'GENARRATIVE_AGC_DESIGN_DEBUG';
const designDebugEnabled =
  process.env[AGC_DESIGN_DEBUG_ENV]?.trim() === '0' ? '0' : '1';

function buildTauriArguments(argv, devUrl = readAgcDevEndpoint().url) {
  const args = [...argv];
  const configOverride = JSON.stringify({
    build: { devUrl, beforeDevCommand: '' },
  });
  const separatorIndex = args.indexOf('--');
  if (separatorIndex < 0) {
    return ['dev', ...args, '--config', configOverride];
  }
  const separatedArguments = args.slice(separatorIndex);
  if (separatedArguments[1] !== '--') {
    separatedArguments.unshift('--');
  }
  return [
    'dev',
    ...args.slice(0, separatorIndex),
    '--config',
    configOverride,
    ...separatedArguments,
  ];
}

// 开发和发行构建使用同一平台编辑器 feature 集合。
// 可用 AGC_DEV_CARGO_FEATURES（逗号分隔）覆盖，传空串即关闭。
function readDevCargoFeatures(env = process.env) {
  const override = env.AGC_DEV_CARGO_FEATURES;
  if (override !== undefined) {
    return override
      .split(',')
      .map((value) => value.trim())
      .filter(Boolean);
  }
  return defaultEditorFeatures(process.platform);
}

function withDevCargoFeatures(argv, features = readDevCargoFeatures()) {
  return withDefaultCargoFeatures(argv, features);
}

function spawnTauriCli(argv, { env = process.env } = {}) {
  return spawnChild(process.execPath, [tauriCliPath, ...argv], {
    cwd: appRoot,
    env,
    shell: false,
  });
}

/// Tauri dev 的 Cargo 直接继承启动器环境，用户级 / 仓库级 Cargo 配置里的
/// `rustc-wrapper`（本地常见为 sccache）会在这里生效。本地 sccache daemon 状态
/// 一旦损坏，`cargo` 的首次 rustc 探测就会失败并阻断整个 AGC 启动；因此这里复用
/// `npm run dev` 的本地 Rust 环境规则，由脚本而不是本机 Cargo 配置决定 wrapper。
function buildTauriDevProcessEnv(endpoint, env = process.env) {
  return buildLocalRustProcessEnv({
    ...withAgcDevEndpointEnv(endpoint, env),
    [AGC_DESIGN_DEBUG_ENV]: designDebugEnabled,
  });
}

async function runTauriDev(
  argv = process.argv.slice(2),
  {
    resolveDevEndpoint = resolveAgcDevEndpoint,
    preflight = preflightExistingVite,
    prepareFrontend = prepareFrontendDev,
    spawnCli = spawnTauriCli,
    waitForCli = waitForChildTermination,
    terminateTree = terminateChildTree,
  } = {},
) {
  const endpoint = await resolveDevEndpoint();
  await preflight({ endpoint });

  let child = null;
  let frontendChild = null;
  const preparationAbort = new AbortController();
  let resolveShutdown;
  let shutdownSignal = '';
  let repeatedSignal = false;
  const shutdownRequested = new Promise((resolveRequest) => {
    resolveShutdown = resolveRequest;
  });
  const signalHandlers = new Map();

  for (const signal of ['SIGINT', 'SIGTERM']) {
    const handler = () => {
      if (!shutdownSignal) {
        shutdownSignal = signal;
        stopChild(child, 'SIGTERM');
        stopChild(frontendChild, 'SIGTERM');
        preparationAbort.abort();
        resolveShutdown(signal);
        return;
      }
      repeatedSignal = true;
      stopChild(child, 'SIGKILL');
      stopChild(frontendChild, 'SIGKILL');
    };
    signalHandlers.set(signal, handler);
    process.on(signal, handler);
  }

  try {
    const preparation = prepareFrontend(endpoint, {
      signal: preparationAbort.signal,
      onChild(frontend) {
        frontendChild = frontend;
      },
    });
    const prepared = await Promise.race([
      preparation.then(() => true),
      shutdownRequested.then(() => false),
    ]);
    if (!prepared || shutdownSignal) return 1;
    const tauriArguments = buildTauriArguments(
      withDevCargoFeatures(argv),
      endpoint.url,
    );
    child = spawnCli(tauriArguments, {
      env: buildTauriDevProcessEnv(endpoint),
    });
    const childResult = waitForCli(child);
    const outcome = await Promise.race([
      childResult.then((failure) => ({ type: 'exit', failure })),
      shutdownRequested.then((signal) => ({ type: 'signal', signal })),
      ...(frontendChild
        ? [
            waitForChildTermination(frontendChild).then((failure) => ({
              type: 'frontend-exit',
              failure,
            })),
          ]
        : []),
    ]);
    const cleanup = await terminateTree(child, {
      gracefulTimeoutMs: repeatedSignal ? 0 : 2500,
    });
    if (!cleanup.stopped) {
      console.error(
        '[ai-game-creator-shell] Tauri dev exited, but its process tree could not be fully stopped.',
      );
      return 1;
    }

    if (outcome.type === 'signal') {
      return 1;
    }
    if (outcome.type === 'frontend-exit') return 1;
    const { failure } = outcome;
    return failure.type === 'error' || failure.signal ? 1 : (failure.code ?? 0);
  } finally {
    for (const [signal, handler] of signalHandlers) {
      process.off(signal, handler);
    }
    preparationAbort.abort();
    if (frontendChild) {
      const cleanup = await terminateTree(frontendChild);
      if (!cleanup.stopped) {
        console.error('[ai-game-creator-shell] 配套开发服务未能完全停止。');
      }
    }
  }
}

async function prepareFrontendDev(endpoint, { onChild, signal }) {
  const frontend = spawnChild(
    process.platform === 'win32' ? 'npm.cmd' : 'npm',
    ['run', 'agc:serve'],
    {
      cwd: repoRoot,
      env: {
        ...withAgcDevEndpointEnv(endpoint),
      },
    },
  );
  onChild(frontend);
  console.log(
    '[ai-game-creator-shell] 正在准备前端与配套后端，完成后启动 Tauri',
  );
  const deadline = Date.now() + 660_000;
  while (Date.now() < deadline) {
    signal.throwIfAborted();
    const failure = readChildFailure(frontend);
    if (failure) {
      throw new Error(
        `配套开发服务退出，前端未就绪：${failure.error?.message ?? failure.signal ?? failure.code}`,
      );
    }
    if (isAiGameCreatorServer(await readExistingViteServer(endpoint))) return;
    await Promise.race([
      new Promise((resolveWait) => setTimeout(resolveWait, 1000)),
      waitForChildTermination(frontend),
    ]);
  }
  throw new Error(`等待前端与配套后端就绪超时：${endpoint.url}`);
}

function isDirectModuleExecution() {
  return Boolean(
    process.argv[1] &&
      resolve(process.argv[1]) === fileURLToPath(import.meta.url),
  );
}

export {
  buildTauriArguments,
  buildTauriDevProcessEnv,
  isDirectModuleExecution,
  runTauriDev,
  spawnTauriCli,
  withDevCargoFeatures,
};

if (isDirectModuleExecution()) {
  try {
    process.exitCode = await runTauriDev();
  } catch (error) {
    console.error(
      `[ai-game-creator-shell] ${error instanceof Error ? error.message : String(error)}`,
    );
    process.exitCode = 1;
  }
}