Function main

packages/server/src/cli.ts:171–238

()

}

async function main() {
  const { port, host, dataDir, webDir, disableAccessPassword } = parseArgs();
  currentDataDir = dataDir;
  installProcessErrorLogging(dataDir);

  // 确保数据目录存在
  if (!existsSync(dataDir)) {
    mkdirSync(dataDir, { recursive: true });
    console.log(`Created data directory: ${dataDir}`);
  }

  // 设置环境变量（必须在 import app 之前，因为 Prisma client 在模块加载时初始化）
  const dbPath = path.join(dataDir, 'data.db');
  process.env.AGENT_TOWER_DATABASE_URL = `file:${dbPath}`;
  process.env.AGENT_TOWER_DATA_DIR = dataDir;
  process.env.AGENT_TOWER_WEB_DIR = webDir;
  process.env.AGENT_TOWER_HOST = host;
  process.env.AGENT_TOWER_PORT = String(port);
  process.env.AGENT_TOWER_URL = `http://127.0.0.1:${port}`;

  // 初始化数据库
  const schemaPath = path.resolve(__dirname, '../prisma/schema.prisma');
  ensureDatabase(dataDir, dbPath, schemaPath);
  process.env[INTERNAL_API_TOKEN_ENV] = getOrCreateInternalApiToken(dataDir);

  // 启动服务器
  const { buildApp } = await import('./app.js');
  if (disableAccessPassword) {
    const { AccessAuthService } = await import('./services/access-auth.service.js');
    await AccessAuthService.disableForRecovery();
    console.log('Access password disabled.');
  }
  const app = await buildApp();

  let shuttingDown = false;
  const shutdown = async (signal: string) => {
    if (shuttingDown) {
      console.log('\nForce exit.');
      process.exit(1);
    }
    shuttingDown = true;
    console.log(`\n${signal} received, shutting down...`);
    try {
      await app.close();
      process.exit(0);
    } catch (err) {
      console.error('Shutdown error:', err);
      process.exit(1);
    }
  };

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

  await app.listen({ port, host });

  // 写入端口文件，供 MCP 等外部进程发现
  const portFile = path.join(dataDir, 'port');