@downcity/city 1.1.4 → 1.1.6

package/src/console/ConsoleGateway.ts

@@ -1,463 +0,0 @@
-/**
- * Console 网关。
- *
- * 关键点（中文）
- * - UI 由 console 进程独立托管，不依赖单个 agent 启动参数。
- * - 提供统一的多 agent 选择能力，并把 `/api/*` 代理到选中 agent。
- */
-
-import { Hono, type Context } from "hono";
-import { cors } from "hono/cors";
-import { logger } from "hono/logger";
-import http from "node:http";
-import fs from "fs-extra";
-import path from "node:path";
-import { fileURLToPath } from "url";
-import { registerConsoleGatewayRoutes } from "@/console/ConsoleGatewayRoutes.js";
-import {
-  buildConsoleAgentsResponse,
-  buildConsoleConfigStatusResponse,
-  buildConsoleModelResponse,
-  inspectConsoleAgentDirectory,
-  listKnownConsoleAgents,
-  readConsoleConfigFileStatus,
-  readRequestedConsoleAgentId,
-  resolveConsoleAgentById,
-  resolveSelectedConsoleAgent,
-} from "@/console/gateway/AgentCatalog.js";
-import {
-  executeConsoleShellCommand,
-  initializeConsoleAgentProject,
-  inspectConsoleAgentRestartSafety,
-  pickConsoleDirectoryPath,
-  restartConsoleAgentByProjectRoot,
-  startConsoleAgentByProjectRoot,
-  stopConsoleAgentByProjectRoot,
-  updateConsoleAgentExecution,
-} from "@/console/gateway/AgentActions.js";
-import { serveConsoleFrontendPath } from "@/console/gateway/FrontendAssets.js";
-import {
-  buildConsoleUpstreamUrl,
-  forwardConsoleRequest,
-} from "@/console/gateway/Proxy.js";
-import type {
-  ConsoleAgentOption,
-  ConsoleAgentsResponse,
-  ConsoleConfigFileStatusItem,
-  ConsoleConfigStatusResponse,
-  ConsoleAgentDirectoryInspection,
-  ConsoleLocalModelsResponse,
-} from "@downcity/agent/shared/types/Console.js";
-import type { AgentProjectInitializationResult } from "@downcity/agent/shared/types/AgentProject.js";
-import { AuthService } from "@/http/auth/AuthService.js";
-import { registerAuthRoutes } from "@/http/auth/AuthRoutes.js";
-import {
-  CONSOLE_AUTH_ROUTE_POLICIES,
-  createRouteAuthGuardMiddleware,
-} from "@/http/auth/RoutePolicy.js";
-import { listBuiltinPluginRuntimeAuthPolicies } from "@downcity/agent/plugin/HttpRoutes.js";
-
-const __filename = fileURLToPath(import.meta.url);
-const __dirname = path.dirname(__filename);
-
-/**
- * 当前 DC 版本号（用于 Console Overview 显示）。
- */
-const DC_VERSION = (() => {
-  try {
-    const pkg = fs.readJsonSync(path.join(__dirname, "../../package.json")) as {
-      version?: string;
-    };
-    const version = String(pkg?.version || "").trim();
-    return version || "unknown";
-  } catch {
-    return "unknown";
-  }
-})();
-
-/**
- * Console 启动参数。
- */
-export interface ConsoleGatewayStartOptions {
-  /**
-   * UI 监听端口。
-   */
-  port: number;
-
-  /**
-   * UI 监听主机。
-   */
-  host: string;
-}
-
-/**
- * Console 网关。
- */
-export class ConsoleGateway {
-  private app: Hono;
-  private server: ReturnType<typeof http.createServer> | null = null;
-  private readonly publicDir: string;
-  private readonly authService: AuthService;
-
-  constructor() {
-    // 关键点（中文）：旧的 ui 构建产物路径都统一回退到 packages/downcity/public。
-    this.publicDir = path.join(__dirname, "../../public");
-    this.app = new Hono();
-    this.authService = new AuthService();
-
-    this.app.use("*", logger());
-    this.app.use(
-      "*",
-      cors({
-        origin: "*",
-        allowMethods: ["GET", "POST", "PUT", "DELETE", "OPTIONS"],
-        allowHeaders: ["Content-Type", "Authorization", "X-DC-Agent"],
-      }),
-    );
-    this.app.use(
-      "*",
-      createRouteAuthGuardMiddleware(
-        this.authService,
-        [
-          ...CONSOLE_AUTH_ROUTE_POLICIES,
-          ...listBuiltinPluginRuntimeAuthPolicies(),
-        ],
-      ),
-    );
-
-    this.setupRoutes();
-  }
-
-  /**
-   * 注册网关路由。
-   */
-  private setupRoutes(): void {
-    registerAuthRoutes({
-      app: this.app,
-      authService: this.authService,
-    });
-    registerConsoleGatewayRoutes({
-      app: this.app,
-      handlers: {
-        readRequestedAgentId: (request) => this.readRequestedAgentId(request),
-        buildAgentsResponse: (requestedAgentId) => this.buildAgentsResponse(requestedAgentId),
-        initializeAgentProject: (projectRoot, initialization) =>
-          this.initializeAgentProject(projectRoot, initialization),
-        startAgentByProjectRoot: (projectRoot, options) => this.startAgentByProjectRoot(projectRoot, options),
-        updateAgentExecution: (projectRoot, input) => this.updateAgentExecution(projectRoot, input),
-        pickDirectoryPath: () => this.pickDirectoryPath(),
-        inspectAgentDirectory: (projectRoot) => this.inspectAgentDirectory(projectRoot),
-        listLocalModels: (projectRoot) => this.listLocalModels(projectRoot),
-        inspectAgentRestartSafety: (projectRoot) => this.inspectAgentRestartSafety(projectRoot),
-        restartAgentByProjectRoot: (projectRoot) => this.restartAgentByProjectRoot(projectRoot),
-        stopAgentByProjectRoot: (projectRoot) => this.stopAgentByProjectRoot(projectRoot),
-        buildConfigStatusResponse: (requestedAgentId) => this.buildConfigStatusResponse(requestedAgentId),
-        resolveAgentById: (requestedAgentId) => this.resolveAgentById(requestedAgentId),
-        executeShellCommand: (args) => this.executeShellCommand(args),
-        buildModelResponse: (requestedAgentId) => this.buildModelResponse(requestedAgentId),
-        resolveSelectedAgent: (requestedAgentId) => this.resolveSelectedAgent(requestedAgentId),
-        buildUpstreamUrl: (requestUrl, baseUrl) => this.buildUpstreamUrl(requestUrl, baseUrl),
-        forwardRequest: (request, upstreamUrl) => this.forwardRequest(request, upstreamUrl),
-        serveFrontendPath: (c, reqPath) => this.serveFrontendPath(c, reqPath),
-      },
-    });
-  }
-
-  private readRequestedAgentId(request: Request): string {
-    return readRequestedConsoleAgentId(request);
-  }
-
-  private async pickDirectoryPath(): Promise<string> {
-    return pickConsoleDirectoryPath();
-  }
-
-  private async listKnownAgents(): Promise<ConsoleAgentOption[]> {
-    return listKnownConsoleAgents();
-  }
-
-  private async buildAgentsResponse(
-    requestedAgentId: string,
-  ): Promise<ConsoleAgentsResponse> {
-    return buildConsoleAgentsResponse({
-      requestedAgentId,
-      cityVersion: DC_VERSION,
-    });
-  }
-
-  /**
-   * 构建 Global Model 面板响应。
-   *
-   * 关键点（中文）
-   * - 模型池来自 console 全局 SQLite，而不是某个 agent。
-   * - `agentPrimaryModelId` 仅用于展示当前选中 agent 的项目绑定。
-   */
-  private async buildModelResponse(requestedAgentId: string): Promise<{
-    success: boolean;
-    model: {
-      primaryModelId: string;
-      primaryModelName: string;
-      providerKey: string;
-      providerType: string;
-      baseUrl: string;
-      agentPrimaryModelId: string;
-      availableModels: Array<{
-        id: string;
-        name: string;
-        providerKey: string;
-        providerType: string;
-        isPaused: boolean;
-      }>;
-    };
-  }> {
-    return buildConsoleModelResponse({
-      requestedAgentId,
-      cityVersion: DC_VERSION,
-    });
-  }
-
-  /**
-   * 读取单个配置文件状态。
-   *
-   * 关键点（中文）
-   * - 不抛出异常，统一返回 `status/reason` 便于 UI 汇总展示。
-   * - 使用最小探测维度：存在性、文件类型、可读性、大小、修改时间。
-   */
-  private async readConfigFileStatus(params: {
-    key: string;
-    scope: "console" | "agent";
-    label: string;
-    filePath: string;
-  }): Promise<ConsoleConfigFileStatusItem> {
-    return readConsoleConfigFileStatus(params);
-  }
-
-  /**
-   * 构建配置文件状态响应。
-   *
-   * 关键点（中文）
-   * - `console` 维度始终返回。
-   * - `agent` 维度仅在存在选中 agent 时返回，避免误导。
-   */
-  private async buildConfigStatusResponse(
-    requestedAgentId: string,
-  ): Promise<ConsoleConfigStatusResponse> {
-    return buildConsoleConfigStatusResponse({
-      requestedAgentId,
-      cityVersion: DC_VERSION,
-    });
-  }
-
-  private async resolveSelectedAgent(
-    requestedAgentId: string,
-  ): Promise<ConsoleAgentOption | null> {
-    return resolveSelectedConsoleAgent(requestedAgentId, DC_VERSION);
-  }
-
-  /**
-   * 根据 id 查找 agent（允许离线 agent，用于 command 页面）。
-   */
-  private async resolveAgentById(
-    requestedAgentId: string,
-  ): Promise<ConsoleAgentOption | null> {
-    return resolveConsoleAgentById(requestedAgentId);
-  }
-
-  /**
-   * 探测目录状态，用于“打开文件夹”流程。
-   */
-  private async inspectAgentDirectory(
-    projectRoot: string,
-  ): Promise<ConsoleAgentDirectoryInspection> {
-    return inspectConsoleAgentDirectory(projectRoot);
-  }
-
-  /**
-   * 列出可直接用于 local executor 的本地 GGUF 模型。
-   */
-  private async listLocalModels(
-    projectRoot?: string,
-  ): Promise<ConsoleLocalModelsResponse> {
-    return { success: true, modelsDir: "~/.models", models: [] };
-  }
-
-  /**
-   * 在 agent 项目目录中执行 shell 命令。
-   *
-   * 关键点（中文）
-   * - 默认 shell 使用 zsh，保持与 CLI 使用习惯一致。
-   * - 输出做大小限制，避免单次命令把 UI 网关进程内存打满。
-   */
-  private async executeShellCommand(params: {
-    command: string;
-    cwd: string;
-    timeoutMs: number;
-  }): Promise<{
-    command: string;
-    cwd: string;
-    exitCode: number | null;
-    signal: string;
-    timedOut: boolean;
-    durationMs: number;
-    stdout: string;
-    stderr: string;
-  }> {
-    return executeConsoleShellCommand(params);
-  }
-
-  private async initializeAgentProject(projectRoot: string, initialization: {
-    agentName?: unknown;
-    modelId?: unknown;
-    forceOverwriteShipJson?: unknown;
-  }): Promise<AgentProjectInitializationResult> {
-    return initializeConsoleAgentProject({
-      projectRoot,
-      agentName: initialization.agentName,
-      modelId: initialization.modelId,
-      forceOverwriteShipJson: initialization.forceOverwriteShipJson,
-    });
-  }
-
-  private async startAgentByProjectRoot(projectRoot: string, options?: {
-    initializeIfNeeded?: boolean;
-    initialization?: {
-      agentName?: unknown;
-      modelId?: unknown;
-      forceOverwriteShipJson?: unknown;
-    };
-  }): Promise<{
-    success: boolean;
-    projectRoot: string;
-    started: boolean;
-    pid?: number;
-    logPath?: string;
-    message?: string;
-  }> {
-    return startConsoleAgentByProjectRoot({
-      projectRoot,
-      cliPath: path.resolve(__dirname, "../cli/Index.js"),
-      initializeIfNeeded: options?.initializeIfNeeded,
-      initialization: options?.initialization,
-    });
-  }
-
-  private async updateAgentExecution(projectRoot: string, input: {
-    modelId?: unknown;
-  }): Promise<{
-    projectRoot: string;
-    modelId: string;
-  }> {
-    return updateConsoleAgentExecution({
-      projectRoot,
-      modelId: input.modelId,
-    });
-  }
-
-
-  private async inspectAgentRestartSafety(projectRoot: string): Promise<{
-    activeContexts: string[];
-    activeTasks: string[];
-  }> {
-    return inspectConsoleAgentRestartSafety({
-      projectRoot,
-      listKnownAgents: () => this.listKnownAgents(),
-    });
-  }
-
-  private async restartAgentByProjectRoot(projectRoot: string): Promise<{
-    success: boolean;
-    projectRoot: string;
-    restarted: boolean;
-    pid?: number;
-    logPath?: string;
-    message?: string;
-  }> {
-    return restartConsoleAgentByProjectRoot({
-      projectRoot,
-      cliPath: path.resolve(__dirname, "../cli/Index.js"),
-    });
-  }
-
-  private async stopAgentByProjectRoot(projectRoot: string): Promise<{
-    success: boolean;
-    projectRoot: string;
-    stopped: boolean;
-    pid?: number;
-    message?: string;
-  }> {
-    return stopConsoleAgentByProjectRoot(projectRoot);
-  }
-
-  private buildUpstreamUrl(requestUrl: URL, baseUrl: string): string {
-    return buildConsoleUpstreamUrl(requestUrl, baseUrl);
-  }
-
-  private async forwardRequest(
-    request: Request,
-    upstreamUrl: string,
-  ): Promise<Response> {
-    return forwardConsoleRequest(request, upstreamUrl);
-  }
-
-  private async serveFrontendPath(c: Context, reqPath: string): Promise<Response> {
-    return serveConsoleFrontendPath({
-      context: c,
-      publicDir: this.publicDir,
-      requestPath: reqPath,
-    });
-  }
-
-  /**
-   * 启动 UI 网关。
-   */
-  async start(options: ConsoleGatewayStartOptions): Promise<void> {
-    const { port, host } = options;
-    await new Promise<void>((resolve) => {
-      const server = http.createServer(async (req, res) => {
-        try {
-          const url = new URL(req.url || "/", `http://${host}:${port}`);
-          const method = req.method || "GET";
-          const bodyBuffer = await new Promise<Buffer>((onDone, onError) => {
-            const chunks: Buffer[] = [];
-            req.on("data", (chunk) => chunks.push(chunk));
-            req.on("end", () => onDone(Buffer.concat(chunks)));
-            req.on("error", onError);
-          });
-          const request = new Request(url.toString(), {
-            method,
-            headers: new Headers(req.headers as Record<string, string>),
-            body: bodyBuffer.length > 0 ? bodyBuffer : undefined,
-          });
-          const response = await this.app.fetch(request);
-          res.statusCode = response.status;
-          for (const [key, value] of response.headers.entries()) {
-            res.setHeader(key, value);
-          }
-          const output = Buffer.from(await response.arrayBuffer());
-          res.end(output);
-        } catch {
-          res.statusCode = 500;
-          res.end("Internal Server Error");
-        }
-      });
-
-      this.server = server;
-      server.listen(port, host, () => resolve());
-    });
-  }
-
-  /**
-   * 停止 UI 网关。
-   */
-  async stop(): Promise<void> {
-    if (!this.server) return;
-    const server = this.server;
-    this.server = null;
-    await new Promise<void>((resolve) => {
-      server.close(() => resolve());
-    });
-  }
-}
-
-export function createConsoleGateway(): ConsoleGateway {
-  return new ConsoleGateway();
-}