ai-agent-router 0.1.21 → 0.2.1

package/src/app/api/ide/opencode/status/route.ts

@@ -0,0 +1,128 @@
+import { NextRequest, NextResponse } from 'next/server';
+import { existsSync, readFileSync, statSync } from 'fs';
+import { join } from 'path';
+import { homedir } from 'os';
+import { getDatabase } from '@/db/database';
+import { getConfig } from '@/db/queries';
+
+// OpenCode 配置文件路径（XDG）
+const OPENCODE_DIR =
+  process.platform === 'win32'
+    ? join(process.env.APPDATA || homedir(), 'opencode')
+    : join(homedir(), '.config', 'opencode');
+const OPENCODE_FILE = join(OPENCODE_DIR, 'opencode.json');
+const BACKUP_FILE = join(OPENCODE_DIR, 'opencode.json.aar.bak');
+const AAR_DIR = join(homedir(), '.aar');
+const TEMP_OPENCODE_FILE = join(AAR_DIR, 'opencode.tmp.json');
+
+type OpenCodeStatus = {
+  applied: boolean;
+  defaultModel?: string;
+  gatewayAddress?: string;
+  apiKey?: string;
+  lastUpdated?: string | null;
+  backupExists: boolean;
+  matchCurrentGateway?: boolean;
+  routerProvider?: string;
+  tempDefaultModel?: string;
+  providerName?: string;
+  modelsCount?: number;
+  /** 当前磁盘上 opencode.json 的完整内容，用于「生效中」展示；还原后即显示还原后的内容 */
+  config?: Record<string, unknown> | null;
+};
+
+function maskApiKey(apiKey: string): string {
+  if (!apiKey || apiKey === 'your-gateway-api-key') return apiKey;
+  if (apiKey.length <= 8) return apiKey.slice(0, 2) + '***' + apiKey.slice(-2);
+  return apiKey.slice(0, 4) + '****' + apiKey.slice(-4);
+}
+
+export async function GET(request: NextRequest) {
+  try {
+    const fileExists = existsSync(OPENCODE_FILE);
+    const backupExists = existsSync(BACKUP_FILE);
+
+    await getDatabase();
+    const gatewayApiKeyConfig = await getConfig('api_key');
+    const portConfig = await getConfig('port');
+    const gatewayApiKey = gatewayApiKeyConfig?.value || 'your-gateway-api-key';
+    const gatewayAddress = `http://localhost:${portConfig?.value || '1357'}/v1`;
+
+    if (!fileExists) {
+      let tempDefaultModel: string | undefined;
+      if (existsSync(TEMP_OPENCODE_FILE)) {
+        try {
+          const temp = JSON.parse(readFileSync(TEMP_OPENCODE_FILE, 'utf-8'));
+          tempDefaultModel = temp.defaultModel;
+        } catch {}
+      }
+      return NextResponse.json({
+        applied: false,
+        defaultModel: undefined,
+        gatewayAddress,
+        apiKey: maskApiKey(gatewayApiKey),
+        backupExists: false,
+        lastUpdated: undefined,
+        matchCurrentGateway: false,
+        routerProvider: undefined,
+        tempDefaultModel,
+      } as OpenCodeStatus);
+    }
+
+    const content = readFileSync(OPENCODE_FILE, 'utf-8');
+    let config: any;
+    try {
+      config = JSON.parse(content);
+    } catch {
+      return NextResponse.json({ error: 'Invalid opencode.json format' }, { status: 500 });
+    }
+
+    const providerMap = config.provider && typeof config.provider === 'object' ? config.provider : {};
+    const aarProvider = providerMap['aar-gateway'];
+    const provider = aarProvider;
+    const appliedBaseUrl = provider?.options?.baseURL;
+    const appliedApiKey = provider?.options?.apiKey;
+    const matchCurrentGateway = appliedBaseUrl === gatewayAddress && appliedApiKey === gatewayApiKey;
+    const defaultModel =
+      typeof config.model === 'string' && config.model.startsWith('aar-gateway/')
+        ? config.model.slice('aar-gateway/'.length)
+        : config.default_model;
+    const modelsCount = provider?.models ? Object.keys(provider.models).length : 0;
+
+    let tempDefaultModel: string | undefined;
+    if (!matchCurrentGateway && existsSync(TEMP_OPENCODE_FILE)) {
+      try {
+        const temp = JSON.parse(readFileSync(TEMP_OPENCODE_FILE, 'utf-8'));
+        tempDefaultModel = temp.defaultModel;
+      } catch {}
+    }
+
+    let lastUpdated: string | null = null;
+    try {
+      lastUpdated = new Date(statSync(OPENCODE_FILE).mtime).toISOString();
+    } catch {}
+
+    const status: OpenCodeStatus = {
+      applied: true,
+      defaultModel,
+      gatewayAddress: appliedBaseUrl,
+      apiKey: maskApiKey(appliedApiKey || gatewayApiKey),
+      backupExists,
+      lastUpdated,
+      matchCurrentGateway,
+      routerProvider: aarProvider ? 'aar' : undefined,
+      tempDefaultModel,
+      providerName: provider?.name,
+      modelsCount,
+      config,
+    };
+
+    return NextResponse.json(status);
+  } catch (error: any) {
+    console.error('OpenCode Status API error:', error);
+    return NextResponse.json(
+      { error: error.message || 'Internal server error', stack: process.env.NODE_ENV === 'development' ? error.stack : undefined },
+      { status: 500 }
+    );
+  }
+}

package/src/app/api/ide/opencode/test/route.ts

@@ -0,0 +1,145 @@
+import { NextRequest, NextResponse } from 'next/server';
+import { existsSync, readFileSync } from 'fs';
+import { join } from 'path';
+import { homedir } from 'os';
+import { spawn } from 'child_process';
+
+const OPENCODE_FILE =
+  process.platform === 'win32'
+    ? join(process.env.APPDATA || homedir(), 'opencode', 'opencode.json')
+    : join(homedir(), '.config', 'opencode', 'opencode.json');
+
+const TEST_PROMPT = '严格只输出这个 JSON，不要多任何字：{"test":"ok"}';
+const EXPECTED_JSON = '{"test":"ok"}';
+const RUN_TIMEOUT_MS = 90_000;
+
+/**
+ * 在子进程中执行 opencode run，不阻塞主进程；若输出中出现 {"test":"ok"} 则判定通过
+ */
+function runOpencodeTest(model: string): Promise<{ success: boolean; stdout: string; stderr: string; error?: string; latency: number }> {
+  return new Promise((resolve) => {
+    const start = Date.now();
+    let settled = false;
+    let timeoutId: ReturnType<typeof setTimeout>;
+    const done = (success: boolean, error?: string) => {
+      if (settled) return;
+      settled = true;
+      clearTimeout(timeoutId);
+      resolve({
+        success,
+        stdout,
+        stderr,
+        error,
+        latency: Date.now() - start,
+      });
+    };
+
+    const args = ['run', TEST_PROMPT, '--model', model];
+    const child = spawn('opencode', args, {
+      shell: process.platform === 'win32',
+      stdio: ['ignore', 'pipe', 'pipe'],
+    });
+
+    let stdout = '';
+    let stderr = '';
+
+    child.stdout?.on('data', (chunk: Buffer) => {
+      stdout += chunk.toString('utf-8');
+    });
+    child.stderr?.on('data', (chunk: Buffer) => {
+      stderr += chunk.toString('utf-8');
+    });
+
+    child.on('error', (err) => {
+      done(false, err.message || 'Failed to spawn opencode');
+    });
+
+    child.on('close', (code, signal) => {
+      if (settled) return;
+      const combined = stdout + stderr;
+      const hasExpected = combined.includes(EXPECTED_JSON);
+      if (hasExpected) {
+        done(true);
+        return;
+      }
+      if (code !== 0 && code !== null) {
+        done(false, `opencode exited with code ${code}${signal ? ` signal ${signal}` : ''}`);
+        return;
+      }
+      done(false, 'Output did not contain expected JSON');
+    });
+
+    timeoutId = setTimeout(() => {
+      try {
+        child.kill('SIGTERM');
+      } catch {
+        try {
+          child.kill('SIGKILL');
+        } catch {}
+      }
+      done(false, 'Request timed out');
+    }, RUN_TIMEOUT_MS);
+  });
+}
+
+/**
+ * 测试 OpenCode 配置：用子进程执行 opencode run，检查输出是否包含 {"test":"ok"}
+ */
+export async function GET(request: NextRequest) {
+  try {
+    if (!existsSync(OPENCODE_FILE)) {
+      return NextResponse.json({
+        success: false,
+        error: 'Configuration file not found',
+        suggestion: 'Please apply configuration first',
+        gatewayRunning: false,
+      });
+    }
+
+    const content = readFileSync(OPENCODE_FILE, 'utf-8');
+    let config: any;
+    try {
+      config = JSON.parse(content);
+    } catch {
+      return NextResponse.json({
+        success: false,
+        error: 'Invalid configuration format',
+        gatewayRunning: false,
+      });
+    }
+
+    const model = config.model;
+    if (!model || typeof model !== 'string') {
+      return NextResponse.json({
+        success: false,
+        error: 'No model set in config (config.model)',
+        suggestion: 'Apply configuration with a default model',
+        gatewayRunning: false,
+      });
+    }
+
+    const result = await runOpencodeTest(model);
+
+    return NextResponse.json({
+      success: result.success,
+      gatewayRunning: undefined,
+      opencodeTest: {
+        success: result.success,
+        message: result.success ? 'opencode run 输出符合预期' : (result.error || 'Output did not contain expected JSON'),
+        error: result.error || '',
+        latency: result.latency,
+        stdout: result.stdout.slice(-2000),
+        stderr: result.stderr.slice(-1000),
+      },
+      configSummary: {
+        model,
+      },
+    });
+  } catch (error: any) {
+    console.error('OpenCode Test API error:', error);
+    return NextResponse.json(
+      { error: error.message || 'Internal server error', gatewayRunning: false },
+      { status: 500 }
+    );
+  }
+}

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

@@ -34,7 +34,7 @@ function cleanLogData(log: any) {
 
 export async function GET(request: NextRequest) {
   try {
-    getDatabase();
+    await getDatabase();
     const { searchParams } = new URL(request.url);
     const id = searchParams.get('id');
     const limit = parseInt(searchParams.get('limit') || '100');
@@ -79,7 +79,7 @@ export async function GET(request: NextRequest) {
 
 export async function DELETE(request: NextRequest) {
   try {
-    getDatabase();
+    await getDatabase();
     const { searchParams } = new URL(request.url);
     const ids = searchParams.get('ids');
     const clearAll = searchParams.get('clear_all') === 'true';

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

@@ -16,7 +16,7 @@ export const runtime = 'nodejs';
 
 export async function GET(request: NextRequest) {
   try {
-    getDatabase();
+    await getDatabase();
     const { searchParams } = new URL(request.url);
     const providerId = searchParams.get('provider_id');
     const enabledOnly = searchParams.get('enabled');
@@ -47,7 +47,7 @@ export async function GET(request: NextRequest) {
 
 export async function POST(request: NextRequest) {
   try {
-    getDatabase();
+    await getDatabase();
     const body = await request.json();
     const { provider_id, name, model_id, enabled } = body;
 
@@ -77,7 +77,7 @@ export async function POST(request: NextRequest) {
 
 export async function PUT(request: NextRequest) {
   try {
-    getDatabase();
+    await getDatabase();
     const body = await request.json();
     const { id, name, model_id, enabled, provider_id } = body;
 
@@ -114,7 +114,7 @@ export async function PUT(request: NextRequest) {
 
 export async function DELETE(request: NextRequest) {
   try {
-    getDatabase();
+    await getDatabase();
     const { searchParams } = new URL(request.url);
     const id = searchParams.get('id');
 
@@ -146,7 +146,7 @@ export async function DELETE(request: NextRequest) {
 // Fetch models from provider
 export async function PATCH(request: NextRequest) {
   try {
-    getDatabase();
+    await getDatabase();
     const body = await request.json();
     const { provider_id } = body;
 

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

@@ -8,7 +8,7 @@ export const runtime = 'nodejs';
 
 export async function GET(request: NextRequest) {
   try {
-    getDatabase();
+    await getDatabase();
     const { searchParams } = new URL(request.url);
     const id = searchParams.get('id');
     const includeKey = searchParams.get('includeKey') === 'true';
@@ -48,7 +48,7 @@ export async function GET(request: NextRequest) {
 
 export async function POST(request: NextRequest) {
   try {
-    getDatabase();
+    await getDatabase();
     const body = await request.json();
     const { name, protocol, base_url, api_key } = body;
 
@@ -82,7 +82,7 @@ export async function POST(request: NextRequest) {
 
 export async function PUT(request: NextRequest) {
   try {
-    getDatabase();
+    await getDatabase();
     const body = await request.json();
     const { id, name, protocol, base_url, api_key } = body;
 
@@ -132,7 +132,7 @@ export async function PUT(request: NextRequest) {
 
 export async function DELETE(request: NextRequest) {
   try {
-    getDatabase();
+    await getDatabase();
     const { searchParams } = new URL(request.url);
     const id = searchParams.get('id');
 

package/src/app/api/providers/test/route.ts

@@ -9,7 +9,7 @@ export const runtime = 'nodejs';
 
 export async function POST(request: NextRequest) {
   try {
-    getDatabase();
+    await getDatabase();
     const body = await request.json();
     const { provider_id, model_id } = body;
 

package/src/app/api/service/start/route.ts

@@ -10,7 +10,7 @@ export const runtime = 'nodejs';
 export async function POST(request: NextRequest) {
   try {
     // Initialize database
-    getDatabase();
+    await getDatabase();
     
     const body = await request.json().catch(() => ({}));
     const port = body.port ? parseInt(body.port, 10) : null;

package/src/app/api/service/status/route.ts

@@ -8,7 +8,7 @@ export const runtime = 'nodejs';
 export async function GET(request: NextRequest) {
   try {
     // Initialize database
-    getDatabase();
+    await getDatabase();
     
     const status = await serviceManager.getStatus();
     return NextResponse.json(status);

package/src/app/api/service/stop/route.ts

@@ -8,7 +8,7 @@ export const runtime = 'nodejs';
 export async function POST(request: NextRequest) {
   try {
     // Initialize database
-    getDatabase();
+    await getDatabase();
     
     const result = await serviceManager.stop();
     

package/src/app/globals.css

@@ -69,3 +69,20 @@ button:disabled {
 ::-webkit-scrollbar-thumb:hover {
   background: #94a3b8;
 }
+
+/* OpenCode 配置预览：去掉选中高亮与焦点白圈，避免与深色背景冲突 */
+.opencode-preview-textarea::selection {
+  background: transparent !important;
+  color: inherit !important;
+}
+.opencode-preview-textarea::-moz-selection {
+  background: transparent !important;
+  color: inherit !important;
+}
+/* 去掉焦点时的白圈/高亮（覆盖全局 input/textarea:focus 的 ring） */
+.opencode-preview-textarea:focus {
+  outline: none !important;
+  box-shadow: none !important;
+  --tw-ring-shadow: none !important;
+  --tw-ring-offset-shadow: none !important;
+}