ai-agent-router 0.1.21 → 0.2.1

package/dist/src/app/logs/page.js

@@ -10,6 +10,20 @@ const react_1 = require("react");
 const Nav_1 = __importDefault(require("../components/Nav"));
 const ToastProvider_1 = require("../components/ToastProvider");
 const ConfirmDialog_1 = __importDefault(require("../components/ConfirmDialog"));
+/** 将服务端返回的 created_at 格式化为本地显示。数据库存储的是本地时间，无需追加 Z 后缀。 */
+function formatCreatedAt(createdAt) {
+    if (!createdAt || typeof createdAt !== 'string')
+        return '';
+    const s = createdAt.trim();
+    if (!s)
+        return '';
+    // 数据库使用 datetime('now','localtime') 存储的已经是本地时间，
+    // 不带时区后缀时直接按本地时间解析，不要追加 'Z'（否则会多偏移 8 小时）
+    const hasTz = /[Z+-]\d{2}:?\d{2}$/.test(s);
+    const iso = hasTz ? s : s.replace(' ', 'T');
+    const date = new Date(iso);
+    return Number.isNaN(date.getTime()) ? s : date.toLocaleString('zh-CN');
+}
 function LogsPage() {
     const [logs, setLogs] = (0, react_1.useState)([]);
     const [selectedLog, setSelectedLog] = (0, react_1.useState)(null);
@@ -244,7 +258,7 @@ function LogsPage() {
                                                             }, className: "mt-2 text-xs text-emerald-600 hover:text-emerald-700", children: "\u6E05\u9664\u641C\u7D22" }))] }) }) })) : (logs.map((log) => {
                                             const parsedModel = parseModelFromRequest(log.request_query, log.request_body);
                                             const displayModel = parsedModel || log.model_name || 'Gateway';
-                                            return ((0, jsx_runtime_1.jsxs)("tr", { className: "hover:bg-emerald-50/20 transition-colors duration-300", children: [(0, jsx_runtime_1.jsx)("td", { className: "px-4 py-3", children: (0, jsx_runtime_1.jsx)("input", { type: "checkbox", checked: selectedIds.includes(log.id), onChange: (e) => handleSelectLog(log.id, e.target.checked), className: "w-4 h-4 rounded border-slate-300 text-emerald-600 focus:ring-emerald-500", onClick: (e) => e.stopPropagation() }) }), (0, jsx_runtime_1.jsx)("td", { className: "px-4 py-3 whitespace-nowrap", children: (0, jsx_runtime_1.jsx)("div", { className: "text-xs text-slate-600", children: new Date(log.created_at + 'Z').toLocaleString('zh-CN') }) }), (0, jsx_runtime_1.jsx)("td", { className: "px-4 py-3 whitespace-nowrap", children: (0, jsx_runtime_1.jsxs)("div", { className: "text-xs", children: [(0, jsx_runtime_1.jsx)("div", { className: "font-medium text-slate-800", children: displayModel }), log.provider_name && ((0, jsx_runtime_1.jsx)("div", { className: "text-[10px] text-slate-500", children: log.provider_name }))] }) }), (0, jsx_runtime_1.jsx)("td", { className: "px-4 py-3 whitespace-nowrap", children: (0, jsx_runtime_1.jsx)("span", { className: `px-2 py-0.5 inline-flex text-xs leading-4 font-semibold rounded-lg border ${log.request_method === 'GET' ? 'bg-sky-50 text-sky-700 border-sky-200/50' :
+                                            return ((0, jsx_runtime_1.jsxs)("tr", { className: "hover:bg-emerald-50/20 transition-colors duration-300", children: [(0, jsx_runtime_1.jsx)("td", { className: "px-4 py-3", children: (0, jsx_runtime_1.jsx)("input", { type: "checkbox", checked: selectedIds.includes(log.id), onChange: (e) => handleSelectLog(log.id, e.target.checked), className: "w-4 h-4 rounded border-slate-300 text-emerald-600 focus:ring-emerald-500", onClick: (e) => e.stopPropagation() }) }), (0, jsx_runtime_1.jsx)("td", { className: "px-4 py-3 whitespace-nowrap", children: (0, jsx_runtime_1.jsx)("div", { className: "text-xs text-slate-600", children: formatCreatedAt(log.created_at) }) }), (0, jsx_runtime_1.jsx)("td", { className: "px-4 py-3 whitespace-nowrap", children: (0, jsx_runtime_1.jsxs)("div", { className: "text-xs", children: [(0, jsx_runtime_1.jsx)("div", { className: "font-medium text-slate-800", children: displayModel }), log.provider_name && ((0, jsx_runtime_1.jsx)("div", { className: "text-[10px] text-slate-500", children: log.provider_name }))] }) }), (0, jsx_runtime_1.jsx)("td", { className: "px-4 py-3 whitespace-nowrap", children: (0, jsx_runtime_1.jsx)("span", { className: `px-2 py-0.5 inline-flex text-xs leading-4 font-semibold rounded-lg border ${log.request_method === 'GET' ? 'bg-sky-50 text-sky-700 border-sky-200/50' :
                                                                 log.request_method === 'POST' ? 'bg-emerald-50 text-emerald-700 border-emerald-200/50' :
                                                                     'bg-slate-50 text-slate-600 border-slate-200/50'}`, children: log.request_method }) }), (0, jsx_runtime_1.jsx)("td", { className: "px-4 py-3", children: (0, jsx_runtime_1.jsx)("div", { className: "text-xs text-slate-600 font-mono max-w-xs truncate", children: log.request_path }) }), (0, jsx_runtime_1.jsx)("td", { className: "px-4 py-3 whitespace-nowrap", children: (0, jsx_runtime_1.jsx)("span", { className: `px-2 py-0.5 inline-flex text-xs leading-4 font-semibold rounded-full border ${log.response_status >= 200 && log.response_status < 300 ? 'bg-emerald-100/80 text-emerald-700 border-emerald-200/50' :
                                                                 log.response_status >= 400 && log.response_status < 500 ? 'bg-amber-100/80 text-amber-700 border-amber-200/50' :

package/dist/src/cli/index.js

@@ -7,51 +7,249 @@ Object.defineProperty(exports, "__esModule", { value: true });
 const commander_1 = require("commander");
 const child_process_1 = require("child_process");
 const path_1 = __importDefault(require("path"));
+const fs_1 = __importDefault(require("fs"));
+const os_1 = __importDefault(require("os"));
+const http_1 = __importDefault(require("http"));
 const database_1 = require("../db/database");
 const queries_1 = require("../db/queries");
 const program = new commander_1.Command();
 const packageJsonPath = require.resolve('../../../package.json');
 const packageJson = require(packageJsonPath);
+const AAR_DIR = path_1.default.join(os_1.default.homedir(), '.aar');
+const UI_PID_FILE = path_1.default.join(AAR_DIR, 'ui.pid');
+const GATEWAY_PID_FILE = path_1.default.join(AAR_DIR, 'gateway.pid');
+const GATEWAY_PORT_FILE = path_1.default.join(AAR_DIR, 'gateway.port');
+function ensureAarDir() {
+    if (!fs_1.default.existsSync(AAR_DIR)) {
+        fs_1.default.mkdirSync(AAR_DIR, { recursive: true });
+    }
+}
+function getPackageRoot() {
+    return path_1.default.dirname(path_1.default.dirname(path_1.default.dirname(__dirname)));
+}
+/** Wait for HTTP server to respond (e.g. UI or gateway ready) */
+function waitForReady(baseUrl, maxWaitMs = 30000) {
+    const url = new URL(baseUrl);
+    return new Promise((resolve) => {
+        const start = Date.now();
+        const tryOnce = () => {
+            const req = http_1.default.request({
+                hostname: url.hostname,
+                port: url.port || '80',
+                path: url.pathname || '/',
+                method: 'GET',
+                timeout: 2000,
+            }, (res) => {
+                resolve(res.statusCode !== undefined && res.statusCode < 500);
+            });
+            req.on('error', () => {
+                if (Date.now() - start >= maxWaitMs) {
+                    resolve(false);
+                    return;
+                }
+                setTimeout(tryOnce, 800);
+            });
+            req.on('timeout', () => {
+                req.destroy();
+                if (Date.now() - start >= maxWaitMs)
+                    resolve(false);
+                else
+                    setTimeout(tryOnce, 800);
+            });
+            req.end();
+        };
+        tryOnce();
+    });
+}
+function printStatus(uiPort, gatewayPort, gatewayRunning, background) {
+    const uiUrl = `http://localhost:${uiPort}`;
+    const gatewayUrl = `http://localhost:${gatewayPort}`;
+    console.log('');
+    console.log('-------------------------------------------');
+    console.log('  AI Agent Router');
+    console.log('-------------------------------------------');
+    console.log(`  前台 UI:     ${uiUrl}`);
+    console.log(`  网关地址:   ${gatewayUrl}`);
+    console.log(`  网关状态:   ${gatewayRunning ? '运行中' : '未启动'}`);
+    if (background) {
+        console.log('  运行模式:   后台运行（关闭终端不影响）');
+        console.log('  停止服务:   aar stop');
+    }
+    console.log('-------------------------------------------');
+    console.log('');
+}
 program
     .name('aar')
     .description('AI Agent Router - Web UI for managing the API gateway')
     .version(packageJson.version);
 program
     .command('start')
-    .description('Start the Web UI management interface')
+    .description('Start the Web UI and gateway (default: both; use --no-gateway for UI only)')
     .option('-p, --port <port>', 'Port for Web UI', '9527')
+    .option('-g, --gateway-port <port>', 'Port for gateway (default: from config or 1357)')
+    .option('--no-gateway', 'Only start Web UI, do not start gateway')
+    .option('--no-background', 'Run in foreground (attach to terminal)')
     .action(async (options) => {
-    const port = parseInt(options.port || '9527');
-    console.log(`Starting AI Agent Router Web UI`);
-    console.log(`  Port: ${port}`);
-    console.log(`  Access the UI at: http://localhost:${port}`);
-    console.log('');
-    // Get the package root directory (where package.json is located)
-    const packageRoot = path_1.default.dirname(path_1.default.dirname(path_1.default.dirname(__dirname)));
-    // Start Web UI using Next.js
-    // Always use production mode for globally installed package
+    const uiPort = parseInt(options.port || '9527', 10);
+    const startGateway = options.gateway !== false;
+    const background = options.background !== false;
+    const packageRoot = getPackageRoot();
+    let gatewayPort = 1357;
+    if (startGateway) {
+        try {
+            await (0, database_1.getDatabase)();
+            const portConfig = await (0, queries_1.getConfig)('port');
+            gatewayPort = options.gatewayPort
+                ? parseInt(options.gatewayPort, 10)
+                : parseInt(portConfig?.value || '1357', 10);
+        }
+        catch (e) {
+            gatewayPort = options.gatewayPort ? parseInt(options.gatewayPort, 10) : 1357;
+        }
+    }
     const serverPath = path_1.default.join(packageRoot, 'node_modules', 'next', 'dist', 'bin', 'next');
-    const uiProcess = (0, child_process_1.spawn)(process.execPath, [serverPath, 'start', '-p', port.toString()], {
+    const gatewayScriptPath = path_1.default.join(packageRoot, 'dist', 'src', 'cli', 'gateway-server.js');
+    if (background) {
+        ensureAarDir();
+        const envBase = { ...process.env, NODE_ENV: 'production' };
+        const uiProc = (0, child_process_1.spawn)(process.execPath, [serverPath, 'start', '-p', uiPort.toString()], {
+            cwd: packageRoot,
+            env: { ...envBase, PORT: uiPort.toString() },
+            detached: true,
+            stdio: 'ignore',
+        });
+        uiProc.unref();
+        if (uiProc.pid) {
+            fs_1.default.writeFileSync(UI_PID_FILE, String(uiProc.pid));
+        }
+        let gatewayRunning = false;
+        if (startGateway && fs_1.default.existsSync(gatewayScriptPath)) {
+            const gwProc = (0, child_process_1.spawn)(process.execPath, [gatewayScriptPath, '--port', String(gatewayPort)], {
+                cwd: packageRoot,
+                env: envBase,
+                detached: true,
+                stdio: 'ignore',
+            });
+            gwProc.unref();
+            if (gwProc.pid) {
+                fs_1.default.writeFileSync(GATEWAY_PID_FILE, String(gwProc.pid));
+                fs_1.default.writeFileSync(GATEWAY_PORT_FILE, String(gatewayPort));
+                gatewayRunning = true;
+            }
+            try {
+                await (0, database_1.getDatabase)();
+                await (0, queries_1.setServiceStatus)({
+                    status: 'running',
+                    port: gatewayPort,
+                    pid: gwProc.pid ?? null,
+                    started_at: new Date().toISOString(),
+                });
+            }
+            catch {
+                // ignore db errors
+            }
+        }
+        printStatus(uiPort, gatewayPort, gatewayRunning, true);
+        process.exit(0);
+    }
+    // Foreground: start UI first
+    console.log('Starting AI Agent Router...');
+    const uiProcess = (0, child_process_1.spawn)(process.execPath, [serverPath, 'start', '-p', uiPort.toString()], {
         cwd: packageRoot,
         stdio: ['ignore', 'inherit', 'inherit'],
-        env: { ...process.env, PORT: port.toString(), NODE_ENV: 'production' },
-    });
-    // Handle UI process exit
-    uiProcess.on('exit', (code) => {
-        console.log(`Web UI process exited with code ${code}`);
-        process.exit(code || 0);
+        env: { ...process.env, PORT: uiPort.toString(), NODE_ENV: 'production' },
     });
     uiProcess.on('error', (error) => {
         console.error(`Failed to start Web UI: ${error.message}`);
         process.exit(1);
     });
-    // Keep the process alive
-    process.on('SIGINT', () => {
-        console.log('\nShutting down...');
+    const gatewayProcess = startGateway && fs_1.default.existsSync(gatewayScriptPath)
+        ? (0, child_process_1.spawn)(process.execPath, [gatewayScriptPath, '--port', String(gatewayPort)], {
+            cwd: packageRoot,
+            stdio: ['ignore', 'pipe', 'pipe'],
+            env: { ...process.env, NODE_ENV: 'production' },
+        })
+        : null;
+    if (gatewayProcess?.stdout) {
+        gatewayProcess.stdout.on('data', (d) => process.stdout.write(d));
+    }
+    if (gatewayProcess?.stderr) {
+        gatewayProcess.stderr.on('data', (d) => process.stderr.write(d));
+    }
+    const ready = await waitForReady(`http://localhost:${uiPort}`);
+    if (ready) {
+        printStatus(uiPort, gatewayPort, !!gatewayProcess, false);
+    }
+    const shutdown = (signal) => {
+        console.log(`\nShutting down (${signal})...`);
+        if (gatewayProcess?.pid) {
+            try {
+                process.kill(gatewayProcess.pid, 'SIGTERM');
+            }
+            catch {
+                // ignore
+            }
+        }
         uiProcess.kill('SIGTERM');
         process.exit(0);
+    };
+    process.on('SIGINT', () => shutdown('SIGINT'));
+    process.on('SIGTERM', () => shutdown('SIGTERM'));
+    uiProcess.on('exit', (code) => {
+        if (gatewayProcess?.pid) {
+            try {
+                process.kill(gatewayProcess.pid, 'SIGTERM');
+            }
+            catch {
+                // ignore
+            }
+        }
+        process.exit(code ?? 0);
     });
 });
+program
+    .command('stop')
+    .description('Stop AI Agent Router when running in background')
+    .action(async () => {
+    ensureAarDir();
+    let stopped = 0;
+    const killPidFile = (file, name) => {
+        if (!fs_1.default.existsSync(file))
+            return;
+        try {
+            const pid = parseInt(fs_1.default.readFileSync(file, 'utf8').trim(), 10);
+            if (!isNaN(pid)) {
+                try {
+                    process.kill(pid, 'SIGTERM');
+                    console.log(`Stopped ${name} (PID ${pid})`);
+                    stopped++;
+                }
+                catch (e) {
+                    if (e?.code !== 'ESRCH')
+                        console.warn(`${name} (PID ${pid}): ${e.message}`);
+                }
+            }
+            fs_1.default.unlinkSync(file);
+        }
+        catch (e) {
+            // ignore
+        }
+    };
+    killPidFile(GATEWAY_PID_FILE, 'gateway');
+    killPidFile(UI_PID_FILE, 'Web UI');
+    if (fs_1.default.existsSync(GATEWAY_PORT_FILE))
+        fs_1.default.unlinkSync(GATEWAY_PORT_FILE);
+    try {
+        await (0, database_1.getDatabase)();
+        await (0, queries_1.updateServiceStatus)({ status: 'stopped', pid: null });
+    }
+    catch {
+        // ignore
+    }
+    if (stopped === 0) {
+        console.log('No background processes found (or already stopped).');
+    }
+});
 program
     .command('config')
     .description('Manage gateway configuration')

package/dist/src/db/database.js

@@ -15,14 +15,55 @@ const path_1 = __importDefault(require("path"));
 const fs_1 = __importDefault(require("fs"));
 const os_1 = __importDefault(require("os"));
 const DB_PATH = process.env.DB_PATH || path_1.default.join(os_1.default.homedir(), '.aar', 'gateway.db');
+// Use hardcoded path for the built/distributed WASM file
+// Next.js copies sql-wasm.wasm to its chunks directory
+function findSqlJsWasmPath() {
+    // Try Next.js chunks directory first (works in dev and production)
+    const nextChunksPath = path_1.default.join(process.cwd(), '.next', 'dev', 'server', 'chunks', 'sql-wasm.wasm');
+    if (fs_1.default.existsSync(nextChunksPath)) {
+        return nextChunksPath;
+    }
+    // Try production .next directory
+    const prodChunksPath = path_1.default.join(process.cwd(), '.next', 'server', 'chunks', 'sql-wasm.wasm');
+    if (fs_1.default.existsSync(prodChunksPath)) {
+        return prodChunksPath;
+    }
+    // Try node_modules as fallback
+    const nodeModulesPath = path_1.default.join(process.cwd(), 'node_modules', 'sql.js', 'dist', 'sql-wasm.wasm');
+    if (fs_1.default.existsSync(nodeModulesPath)) {
+        return nodeModulesPath;
+    }
+    // Use the most likely location as default
+    return nodeModulesPath;
+}
+const SQLJS_WASM_PATH = findSqlJsWasmPath();
 let dbInstance = null;
 let sqlJsInstance = null;
+let lastLoadMtimeMs = 0;
+function reloadFromFileIfNewer() {
+    if (!dbInstance || !fs_1.default.existsSync(DB_PATH))
+        return;
+    const mtime = fs_1.default.statSync(DB_PATH).mtimeMs;
+    if (mtime <= lastLoadMtimeMs)
+        return;
+    try {
+        const fresh = loadDatabase();
+        if (fresh) {
+            dbInstance.close();
+            dbInstance = fresh;
+            lastLoadMtimeMs = fs_1.default.statSync(DB_PATH).mtimeMs;
+        }
+    }
+    catch (e) {
+        console.warn('[Database] Reload from file failed:', e.message);
+    }
+}
 async function initSqlJsEngine() {
     if (!sqlJsInstance) {
         sqlJsInstance = await (0, sql_js_1.default)({
             locateFile: (file) => {
                 if (file.endsWith('.wasm')) {
-                    return path_1.default.join(process.cwd(), 'node_modules', 'sql.js', 'dist', file);
+                    return SQLJS_WASM_PATH;
                 }
                 return `https://sql.js.org/dist/${file}`;
             }
@@ -39,6 +80,9 @@ function saveDatabase(db) {
             fs_1.default.mkdirSync(dbDir, { recursive: true });
         }
         fs_1.default.writeFileSync(DB_PATH, buffer);
+        if (fs_1.default.existsSync(DB_PATH)) {
+            lastLoadMtimeMs = fs_1.default.statSync(DB_PATH).mtimeMs;
+        }
     }
     catch (error) {
         console.error('Failed to save database:', error);
@@ -59,7 +103,16 @@ function loadDatabase() {
 async function getDatabase() {
     if (!dbInstance) {
         const engine = await initSqlJsEngine();
-        dbInstance = loadDatabase() || new engine.Database();
+        const loaded = loadDatabase();
+        if (loaded) {
+            dbInstance = loaded;
+            if (fs_1.default.existsSync(DB_PATH))
+                lastLoadMtimeMs = fs_1.default.statSync(DB_PATH).mtimeMs;
+        }
+        else {
+            dbInstance = new engine.Database();
+            lastLoadMtimeMs = 0;
+        }
         try {
             dbInstance.run(schema_1.CREATE_TABLES_SQL);
             // Migration: Allow NULL model_id in request_logs table
@@ -113,6 +166,7 @@ async function getDatabase() {
             }
         }
     }
+    reloadFromFileIfNewer();
     return dbInstance;
 }
 async function closeDatabase() {
@@ -125,7 +179,6 @@ async function closeDatabase() {
 function getDbPath() {
     return DB_PATH;
 }
-// Helper to run a query and get single result
 async function queryOne(sql, params = []) {
     const db = await getDatabase();
     const stmt = db.prepare(sql);
@@ -138,7 +191,6 @@ async function queryOne(sql, params = []) {
     stmt.free();
     return null;
 }
-// Helper to run a query and get all results
 async function queryAll(sql, params = []) {
     const db = await getDatabase();
     const results = [];
@@ -150,7 +202,6 @@ async function queryAll(sql, params = []) {
     stmt.free();
     return results;
 }
-// Helper to run INSERT/UPDATE/DELETE
 async function run(sql, params = []) {
     const db = await getDatabase();
     db.run(sql, params);

package/dist/src/db/queries.js

@@ -155,9 +155,9 @@ async function createRequestLog(log) {
     try {
         const result = await (0, database_1.run)(`INSERT INTO request_logs (
         model_id, request_method, request_path, request_headers,
-        request_query, request_body, response_status, response_body, response_time_ms
-      ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)`, [
-            log.model_id || null,
+        request_query, request_body, response_status, response_body, response_time_ms, created_at
+      ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, datetime('now', 'localtime'))`, [
+            log.model_id === undefined ? null : log.model_id,
             log.request_method,
             log.request_path,
             log.request_headers,
@@ -178,9 +178,9 @@ async function createRequestLog(log) {
             console.warn(`[RequestLog] Foreign key constraint failed for model_id ${log.model_id}, retrying with NULL`);
             const result = await (0, database_1.run)(`INSERT INTO request_logs (
           model_id, request_method, request_path, request_headers,
-          request_query, request_body, response_status, response_body, response_time_ms
-        ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)`, [
-                null,
+          request_query, request_body, response_status, response_body, response_time_ms, created_at
+        ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, datetime('now', 'localtime'))`, [
+                log.model_id === undefined ? null : log.model_id,
                 log.request_method,
                 log.request_path,
                 log.request_headers,

package/dist/src/server/logger.js

@@ -3,6 +3,24 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.logRequest = logRequest;
 const queries_1 = require("../db/queries");
 const crypto_1 = require("./crypto");
+/** Safe JSON.stringify that never throws; use for logging arbitrary response/request bodies. */
+function safeStringify(value) {
+    if (value === undefined)
+        return '';
+    if (value === null)
+        return 'null';
+    try {
+        return JSON.stringify(value);
+    }
+    catch {
+        try {
+            return String(value);
+        }
+        catch {
+            return '[Non-serializable]';
+        }
+    }
+}
 async function logRequest(request, response) {
     try {
         // Mask sensitive information
@@ -12,11 +30,11 @@ async function logRequest(request, response) {
             model_id: request.modelId,
             request_method: request.method,
             request_path: request.path,
-            request_headers: JSON.stringify(maskedHeaders),
-            request_query: JSON.stringify(request.query),
-            request_body: JSON.stringify(maskedBody),
+            request_headers: safeStringify(maskedHeaders),
+            request_query: safeStringify(request.query),
+            request_body: safeStringify(maskedBody),
             response_status: response.status,
-            response_body: JSON.stringify(response.body),
+            response_body: safeStringify(response.body),
             response_time_ms: response.responseTimeMs,
         });
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-agent-router",
-  "version": "0.1.21",
+  "version": "0.2.1",
   "description": "A unified API gateway for managing multiple AI model providers (Anthropic, OpenAI, Gemini, etc.)",
   "main": "dist/src/cli/index.js",
   "bin": {
@@ -10,6 +10,7 @@
     "dev": "next dev -p 9527",
     "build": "next build && tsc",
     "start": "node dist/src/cli/index.js start",
+    "stop": "node dist/src/cli/index.js stop",
     "lint": "eslint .",
     "pub": "npm run build && npm publish",
     "type-check": "tsc --noEmit"

package/src/app/api/gateway/[...path]/route.ts

@@ -51,7 +51,7 @@ async function handleRequest(
   method: string
 ) {
   try {
-    getDatabase();
+    await getDatabase();
 
     const path = params.path.join('/');
     const searchParams = request.nextUrl.searchParams;

package/src/app/api/gateway/route.ts

@@ -19,7 +19,7 @@ async function handleGatewayRequestDirect(
   method: string
 ) {
   try {
-    getDatabase();
+    await getDatabase();
     const searchParams = request.nextUrl.searchParams;
     let modelId = searchParams.get('model') || searchParams.get('model_id');
     const providerName = searchParams.get('provider');

package/src/app/api/ide/claude/apply/route.ts

@@ -38,11 +38,9 @@ export async function POST(request: NextRequest) {
     const gatewayAddress = await getGatewayAddress();
     const gatewayApiKey = await getGatewayApiKey();
 
-    // 生成配置对象
     const modelMapping = { haiku, sonnet, opus, default: defaultModel, reasoning: reasoningModel };
-    const config = generateClaudeConfig(gatewayAddress, gatewayApiKey, modelMapping);
+    const aarEnv = buildAarEnv(gatewayAddress, gatewayApiKey, modelMapping);
 
-    // 备份原有配置
     const backupResult = backupOriginalConfig();
     if (!backupResult.success) {
       return NextResponse.json(
@@ -51,13 +49,27 @@ export async function POST(request: NextRequest) {
       );
     }
 
-    // 确保目录存在
     if (!existsSync(CLAUDE_DIR)) {
       mkdirSync(CLAUDE_DIR, { recursive: true });
     }
 
-    // 验证并写入新配置
-    const validationResult = validateAndWriteConfig(config);
+    let configToWrite: any;
+    if (existsSync(SETTINGS_FILE)) {
+      try {
+        const existing = JSON.parse(readFileSync(SETTINGS_FILE, 'utf-8'));
+        configToWrite = {
+          ...existing,
+          env: { ...(existing.env || {}), ...aarEnv },
+          router_provider: 'aar',
+        };
+      } catch {
+        configToWrite = generateClaudeConfig(gatewayAddress, gatewayApiKey, modelMapping);
+      }
+    } else {
+      configToWrite = generateClaudeConfig(gatewayAddress, gatewayApiKey, modelMapping);
+    }
+
+    const validationResult = validateAndWriteConfig(configToWrite);
     if (!validationResult.success) {
       return NextResponse.json(
         { error: validationResult.error },
@@ -68,7 +80,7 @@ export async function POST(request: NextRequest) {
     return NextResponse.json({
       success: true,
       message: 'Configuration applied successfully',
-      config,
+      config: configToWrite,
       backup: backupResult.existed ? 'Created backup' : 'No existing config to backup',
       previousWasFromAar: backupResult.existed && backupResult.isFromAar,
     });
@@ -101,57 +113,60 @@ async function getGatewayApiKey(): Promise<string> {
   return apiKeyConfig?.value || 'your-gateway-api-key';
 }
 
-/**
- * 生成 Claude 配置
- */
+/** 仅生成 AAR 相关的 env 字段，用于合并到现有配置 */
+function buildAarEnv(
+  gatewayAddress: string,
+  gatewayApiKey: string,
+  modelMapping: { haiku: string; sonnet: string; opus: string; default: string; reasoning: string }
+): Record<string, unknown> {
+  return {
+    ANTHROPIC_AUTH_TOKEN: gatewayApiKey,
+    ANTHROPIC_BASE_URL: gatewayAddress,
+    ANTHROPIC_DEFAULT_HAIKU_MODEL: modelMapping.haiku,
+    ANTHROPIC_DEFAULT_OPUS_MODEL: modelMapping.opus,
+    ANTHROPIC_DEFAULT_SONNET_MODEL: modelMapping.sonnet,
+    ANTHROPIC_MODEL: modelMapping.default,
+    ANTHROPIC_REASONING_MODEL: modelMapping.reasoning,
+    API_TIMEOUT_MS: '3000000',
+    CLAUDE_CODE_DISABLE_NONESSENTIAL_TRAFFIC: 1,
+    hasCompletedOnboarding: true,
+  };
+}
+
+/** 生成完整 Claude 配置（仅在无现有文件时使用） */
 function generateClaudeConfig(
   gatewayAddress: string,
   gatewayApiKey: string,
   modelMapping: { haiku: string; sonnet: string; opus: string; default: string; reasoning: string }
 ): any {
   return {
-    // 路由提供者标识，标识配置来自当前工具 aar
     router_provider: 'aar',
-    env: {
-      ANTHROPIC_AUTH_TOKEN: gatewayApiKey,
-      ANTHROPIC_BASE_URL: gatewayAddress,
-      ANTHROPIC_DEFAULT_HAIKU_MODEL: modelMapping.haiku,
-      ANTHROPIC_DEFAULT_OPUS_MODEL: modelMapping.opus,
-      ANTHROPIC_DEFAULT_SONNET_MODEL: modelMapping.sonnet,
-      ANTHROPIC_MODEL: modelMapping.default,
-      ANTHROPIC_REASONING_MODEL: modelMapping.reasoning,
-      API_TIMEOUT_MS: '3000000',
-      CLAUDE_CODE_DISABLE_NONESSENTIAL_TRAFFIC: 1,
-      hasCompletedOnboarding: true,
-    },
+    env: buildAarEnv(gatewayAddress, gatewayApiKey, modelMapping),
   };
 }
 
 /**
  * 备份原有配置
- * 返回原有配置是否来自当前工具（router_provider === 'aar'）
+ * 仅在当前配置不是 AAR 时写入备份，这样「还原」才能恢复到应用 AAR 前的状态
  */
 function backupOriginalConfig(): { success: boolean; existed: boolean; isFromAar: boolean; error?: string } {
   try {
-    // 检查原有配置是否存在
     const existing = existsSync(SETTINGS_FILE);
 
     if (existing) {
-      // 读取原有配置
       const originalConfig = readFileSync(SETTINGS_FILE, 'utf-8');
 
-      // 解析配置检查 router_provider
       let isFromAar = false;
       try {
         const parsedConfig = JSON.parse(originalConfig);
         isFromAar = parsedConfig.router_provider === 'aar';
-      } catch (error) {
-        // 解析失败，不影响备份操作
+      } catch {
         isFromAar = false;
       }
 
-      // 写入备份文件
-      writeFileSync(BACKUP_FILE, originalConfig, 'utf-8');
+      if (!isFromAar) {
+        writeFileSync(BACKUP_FILE, originalConfig, 'utf-8');
+      }
 
       return { success: true, existed: true, isFromAar };
     }