@axiom-lattice/gateway 2.1.97 → 2.1.98

package/dist/index.mjs

@@ -1,3 +1,9 @@
+import {
+  Logger,
+  getQrCode,
+  getQrCodeStatus
+} from "./chunk-6CUQGDJI.mjs";
+
 // src/index.ts
 import fastify from "fastify";
 import cors from "@fastify/cors";
@@ -3487,7 +3493,7 @@ var WorkspaceController = class {
         }
         const inferredContentType = this.getMimeType(filename2);
         try {
-          let contentType = inferredContentType;
+          const contentType = inferredContentType;
           const isHtml = contentType?.toLowerCase().includes("text/html") || filename2.toLowerCase().endsWith(".html") || filename2.toLowerCase().endsWith(".htm");
           let outputBuf = buf;
           if (isHtml) {
@@ -3599,7 +3605,7 @@ var WorkspaceController = class {
       const filename = data.filename || "file";
       const pathEntry = data.fields?.path;
       const pathValue = pathEntry && typeof pathEntry === "object" && "value" in pathEntry ? String(pathEntry.value) : typeof pathEntry === "string" ? pathEntry : void 0;
-      if (pathValue && !/^[a-zA-Z0-9_./~\-]+$/.test(pathValue)) {
+      if (pathValue && !/^[a-zA-Z0-9_./~-]+$/.test(pathValue)) {
         return reply.status(400).send({ success: false, error: "Invalid path parameter" });
       }
       if (workspace.storageType === "sandbox") {
@@ -6167,144 +6173,6 @@ function normalizeChatType(chatType) {
   return chatType === "p2p" ? "direct" : "group";
 }
 
-// src/logger/Logger.ts
-import pino from "pino";
-import "pino-pretty";
-import "pino-roll";
-var PinoLoggerFactory = class _PinoLoggerFactory {
-  constructor() {
-    const isProd = process.env.NODE_ENV === "production";
-    const loggerConfig = {
-      // 自定义时间戳格式
-      timestamp: () => `,"@timestamp":"${(/* @__PURE__ */ new Date()).toISOString()}"`,
-      // 关闭默认的时间戳键
-      base: {
-        "@version": "1",
-        app_name: "lattice",
-        service_name: "lattice/graph-server",
-        thread_name: "main",
-        logger_name: "lattice-graph-logger"
-      },
-      formatters: {
-        level: (label, number) => {
-          return {
-            level: label.toUpperCase(),
-            level_value: number * 1e3
-          };
-        }
-      }
-    };
-    if (isProd) {
-      try {
-        this.pinoLogger = pino(
-          loggerConfig,
-          pino.transport({
-            target: "pino-roll",
-            options: {
-              file: "./logs/fin_ai_graph_server",
-              frequency: "daily",
-              mkdir: true
-            }
-          })
-        );
-      } catch (error) {
-        console.error(
-          "\u65E0\u6CD5\u521D\u59CB\u5316 pino-roll \u65E5\u5FD7\u8BB0\u5F55\u5668\uFF0C\u56DE\u9000\u5230\u63A7\u5236\u53F0\u65E5\u5FD7",
-          error
-        );
-        this.pinoLogger = pino({
-          ...loggerConfig,
-          transport: {
-            target: "pino-pretty",
-            options: {
-              colorize: true
-            }
-          }
-        });
-      }
-    } else {
-      this.pinoLogger = pino({
-        ...loggerConfig,
-        transport: {
-          target: "pino-pretty",
-          options: {
-            colorize: true
-          }
-        }
-      });
-    }
-  }
-  static getInstance() {
-    if (!_PinoLoggerFactory.instance) {
-      _PinoLoggerFactory.instance = new _PinoLoggerFactory();
-    }
-    return _PinoLoggerFactory.instance;
-  }
-  getPinoLogger() {
-    return this.pinoLogger;
-  }
-};
-var Logger = class _Logger {
-  constructor(options) {
-    this.context = options?.context || {};
-    this.name = options?.name || "lattice-graph-logger";
-    this.serviceName = options?.serviceName || "lattice/graph-server";
-  }
-  /**
-   * 获取合并了上下文的日志对象
-   * @param additionalContext 额外的上下文数据
-   * @returns 带有上下文的pino日志对象
-   */
-  getContextualLogger(additionalContext) {
-    const pinoLogger = PinoLoggerFactory.getInstance().getPinoLogger();
-    const contextObj = {
-      "x-user-id": this.context["x-user-id"] || "",
-      "x-tenant-id": this.context["x-tenant-id"] || "",
-      "x-request-id": this.context["x-request-id"] || "",
-      "x-task-id": this.context["x-task-id"] || "",
-      "x-thread-id": this.context["x-thread-id"] || "",
-      service_name: this.serviceName,
-      logger_name: this.name,
-      ...additionalContext
-    };
-    return pinoLogger.child(contextObj);
-  }
-  info(msg, obj) {
-    this.getContextualLogger(obj).info(msg);
-  }
-  error(msg, obj) {
-    this.getContextualLogger(obj).error(msg);
-  }
-  warn(msg, obj) {
-    this.getContextualLogger(obj).warn(msg);
-  }
-  debug(msg, obj) {
-    this.getContextualLogger(obj).debug(msg);
-  }
-  /**
-   * 更新Logger实例的上下文
-   */
-  updateContext(context) {
-    this.context = {
-      ...this.context,
-      ...context
-    };
-  }
-  /**
-   * 创建一个新的Logger实例，继承当前Logger的上下文
-   */
-  child(options) {
-    return new _Logger({
-      name: options.name || this.name,
-      serviceName: options.serviceName || this.serviceName,
-      context: {
-        ...this.context,
-        ...options.context
-      }
-    });
-  }
-};
-
 // src/channels/lark/LarkChannelAdapter.ts
 import * as Lark from "@larksuiteoapi/node-sdk";
 var logger = new Logger({ serviceName: "lattice/gateway/lark" });
@@ -6437,6 +6305,24 @@ var larkChannelAdapter = {
     await client.start({ eventDispatcher });
     activeConnections.set(installationId, client);
     logger.info("Lark WS client connected", { installationId });
+  },
+  async disconnect(installationId) {
+    const client = activeConnections.get(installationId);
+    if (!client) {
+      logger.warn("Lark WS not connected, nothing to disconnect", { installationId });
+      return;
+    }
+    logger.info("Lark WS client disconnecting", { installationId });
+    try {
+      client.close({ force: true });
+    } catch (err) {
+      logger.error("Lark WS client stop error", {
+        installationId,
+        error: err instanceof Error ? err.message : String(err)
+      });
+    }
+    activeConnections.delete(installationId);
+    logger.info("Lark WS client disconnected", { installationId });
   }
 };
 
@@ -6504,9 +6390,62 @@ function registerLarkChannelRoutes(app2, deps) {
   );
 }
 
+// src/channels/wechat/controller.ts
+var qrSessions = /* @__PURE__ */ new Map();
+var SESSION_TTL_MS = 5 * 60 * 1e3;
+function cleanupExpiredSessions() {
+  const now = Date.now();
+  for (const [key, session] of qrSessions) {
+    if (now - session.createdAt > SESSION_TTL_MS) {
+      qrSessions.delete(key);
+    }
+  }
+}
+async function handleGetQrCode(_request, reply) {
+  try {
+    cleanupExpiredSessions();
+    const { qrcode, qrcodeImgUrl } = await getQrCode();
+    qrSessions.set(qrcode, { qrcode, createdAt: Date.now() });
+    reply.send({ success: true, data: { qrcode, qrcodeImgUrl } });
+  } catch (err) {
+    reply.status(500).send({
+      success: false,
+      message: err instanceof Error ? err.message : "Failed to get QR code"
+    });
+  }
+}
+async function handleGetQrCodeStatus(request, reply) {
+  const { qrcode } = request.query;
+  if (!qrcode) {
+    reply.status(400).send({ success: false, message: "qrcode is required" });
+    return;
+  }
+  const session = qrSessions.get(qrcode);
+  if (!session) {
+    reply.status(404).send({ success: false, message: "Session not found or expired" });
+    return;
+  }
+  try {
+    const result = await getQrCodeStatus(qrcode);
+    reply.send({ success: true, data: result });
+  } catch (err) {
+    reply.status(500).send({
+      success: false,
+      message: err instanceof Error ? err.message : "Failed to check QR status"
+    });
+  }
+}
+
+// src/channels/wechat/routes.ts
+function registerWechatChannelRoutes(app2) {
+  app2.get("/api/channels/wechat/setup/qrcode", handleGetQrCode);
+  app2.get("/api/channels/wechat/setup/status", handleGetQrCodeStatus);
+}
+
 // src/channels/routes.ts
 var channelRouteRegistrars = [
-  (app2, deps) => registerLarkChannelRoutes(app2, deps)
+  (app2, deps) => registerLarkChannelRoutes(app2, deps),
+  (app2, _deps) => registerWechatChannelRoutes(app2)
 ];
 function registerChannelRoutes(app2, dependencies) {
   if (!dependencies) return;
@@ -6517,6 +6456,12 @@ function registerChannelRoutes(app2, dependencies) {
 
 // src/controllers/channel-installations.ts
 import { randomUUID as randomUUID7 } from "crypto";
+var _adapterRegistry;
+var _router;
+function setChannelControllerDeps(deps) {
+  _adapterRegistry = deps.adapterRegistry;
+  _router = deps.router;
+}
 function getTenantId13(request) {
   const userTenantId = request.user?.tenantId;
   if (userTenantId) {
@@ -6638,6 +6583,16 @@ async function createChannelInstallation(request, reply) {
       installationId,
       body
     );
+    if (body.enabled !== false && _adapterRegistry) {
+      const adapter = _adapterRegistry.get(body.channel);
+      if (adapter?.connect) {
+        try {
+          await adapter.connect(installation, { router: _router });
+        } catch (err) {
+          console.error("Failed to start channel connection after creation:", err);
+        }
+      }
+    }
     reply.code(201);
     return {
       success: true,
@@ -6692,6 +6647,14 @@ async function updateChannelInstallation(request, reply) {
         message: "Channel installation not found"
       };
     }
+    if (body.enabled !== void 0 && body.enabled !== existing.enabled) {
+      const adapter = _adapterRegistry?.get(existing.channel);
+      if (body.enabled && adapter?.connect) {
+        await adapter.connect(installation, { router: _router });
+      } else if (!body.enabled && adapter?.disconnect) {
+        await adapter.disconnect(installationId);
+      }
+    }
     return {
       success: true,
       message: "Channel installation updated successfully",
@@ -6725,6 +6688,16 @@ async function deleteChannelInstallation(request, reply) {
         message: "Access denied"
       };
     }
+    if (_adapterRegistry) {
+      const adapter = _adapterRegistry.get(existing.channel);
+      if (adapter?.disconnect) {
+        try {
+          await adapter.disconnect(installationId);
+        } catch (err) {
+          console.error("Failed to disconnect channel adapter:", err);
+        }
+      }
+    }
     const deleted = await store.deleteInstallation(tenantId, installationId);
     if (!deleted) {
       reply.code(404);
@@ -8337,12 +8310,12 @@ var start = async (config) => {
     let channelDeps;
     const adapterRegistry = new ChannelAdapterRegistry();
     adapterRegistry.register(larkChannelAdapter);
+    const { wechatChannelAdapter } = await import("./WechatChannelAdapter-QQYOHZTL.mjs");
+    adapterRegistry.register(wechatChannelAdapter);
     try {
       const { getStoreLattice: getStore2 } = await import("@axiom-lattice/core");
-      let bindingStore;
-      let installationStore;
-      bindingStore = getStore2("default", "channelBinding").store;
-      installationStore = getStore2("default", "channelInstallation").store;
+      const bindingStore = getStore2("default", "channelBinding").store;
+      const installationStore = getStore2("default", "channelInstallation").store;
       setBindingRegistry(bindingStore);
       const router = new MessageRouter({
         middlewares: [
@@ -8374,6 +8347,9 @@ var start = async (config) => {
       logger4.info("Menu registry initialized");
     } catch {
     }
+    if (channelDeps?.router) {
+      setChannelControllerDeps({ adapterRegistry, router: channelDeps.router });
+    }
     registerLatticeRoutes(app, channelDeps);
     if (!sandboxLatticeManager2.hasLattice("default")) {
       sandboxLatticeManager2.registerLattice("default", getConfiguredSandboxProvider());