aicodeswitch 5.1.3 → 5.2.1

package/dist/server/rules-status-service.js

@@ -10,12 +10,14 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.rulesStatusBroadcaster = exports.RulesStatusBroadcaster = void 0;
+const events_1 = require("events");
 /**
  * 规则状态管理服务
  * 负责管理所有规则的状态（使用中、空闲、错误、挂起）
  */
-class RulesStatusBroadcaster {
+class RulesStatusBroadcaster extends events_1.EventEmitter {
     constructor() {
+        super();
         Object.defineProperty(this, "ruleStates", {
             enumerable: true,
             configurable: true,
@@ -70,6 +72,8 @@ class RulesStatusBroadcaster {
             writable: true,
             value: 30000
         }); // error 状态30秒后自动恢复
+        // 允许多标签页场景下的多连接监听
+        this.setMaxListeners(50);
         // 启动定期状态检查定时器
         this.startSyncInterval();
     }
@@ -99,7 +103,7 @@ class RulesStatusBroadcaster {
                     data.timestamp &&
                     now - data.timestamp > this.ERROR_RECOVERY_TIMEOUT) {
                     console.log(`[RulesStatusBroadcaster] 规则 ${ruleId} 错误状态已超时，自动恢复为 idle 状态`);
-                    this.ruleStates.set(ruleId, {
+                    this.updateRuleStatus({
                         ruleId,
                         status: 'idle',
                         routeId: data.routeId,
@@ -130,7 +134,7 @@ class RulesStatusBroadcaster {
                         if (!isBlacklisted) {
                             // 黑名单已过期，恢复为 idle 状态
                             console.log(`[RulesStatusBroadcaster] 规则 ${ruleId} 黑名单已过期，恢复为 idle 状态`);
-                            this.ruleStates.set(ruleId, {
+                            this.updateRuleStatus({
                                 ruleId,
                                 status: 'idle',
                                 routeId,
@@ -172,6 +176,8 @@ class RulesStatusBroadcaster {
     updateRuleStatus(data) {
         // 更新本地状态
         this.ruleStates.set(data.ruleId, data);
+        // 通知 SSE 客户端状态已变更
+        this.emit('statusChanged', data);
     }
     /**
      * 标记规则正在使用
@@ -288,6 +294,12 @@ class RulesStatusBroadcaster {
      */
     clearRuleStatus(ruleId) {
         this.ruleStates.delete(ruleId);
+        // 通知 SSE 客户端状态已清除（恢复为 idle）
+        this.emit('statusChanged', {
+            ruleId,
+            status: 'idle',
+            timestamp: Date.now(),
+        });
     }
     /**
      * 获取当前活动的规则列表
@@ -320,6 +332,7 @@ class RulesStatusBroadcaster {
         this.idleDebounceTimeouts.forEach((timeout) => clearTimeout(timeout));
         this.idleDebounceTimeouts.clear();
         this.ruleStates.clear();
+        this.removeAllListeners();
     }
 }
 exports.RulesStatusBroadcaster = RulesStatusBroadcaster;

package/dist/server/transformers/model-rewrite-transform.js

@@ -0,0 +1,128 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ModelRewriteTransform = void 0;
+exports.rewriteResponseModel = rewriteResponseModel;
+const stream_1 = require("stream");
+const string_decoder_1 = require("string_decoder");
+/**
+ * 流式 SSE 响应中的 model 字段回写 Transform。
+ *
+ * 插入位置：SSESerializerTransform 之后、ChunkCollectorTransform 之前。
+ * 输入/输出均为 text mode（字符串），通过正则替换 SSE data 行中的
+ * "model":"上游模型名" → "model":"客户端原始模型名"。
+ */
+class ModelRewriteTransform extends stream_1.Transform {
+    constructor(originalModel) {
+        super({
+            writableObjectMode: false, // 接收序列化后的文本
+            readableObjectMode: false, // 输出文本
+        });
+        Object.defineProperty(this, "modelRegex", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "replacement", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "modelVersionRegex", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: null
+        });
+        Object.defineProperty(this, "modelVersionReplacement", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: null
+        });
+        Object.defineProperty(this, "stringDecoder", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: new string_decoder_1.StringDecoder('utf8')
+        });
+        Object.defineProperty(this, "buffer", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: ''
+        });
+        // 转义原始模型名中的特殊字符，防止 JSON 注入
+        const escaped = originalModel.replace(/\\/g, '\\\\').replace(/"/g, '\\"');
+        this.replacement = `"model":"${escaped}"`;
+        this.modelRegex = /"model"\s*:\s*"[^"]*"/g;
+        // 处理 Gemini 直通场景中的 modelVersion 字段
+        if (originalModel) {
+            this.modelVersionReplacement = `"modelVersion":"${escaped}"`;
+            this.modelVersionRegex = /"modelVersion"\s*:\s*"[^"]*"/g;
+        }
+    }
+    _transform(chunk, _encoding, callback) {
+        try {
+            this.buffer += this.stringDecoder.write(chunk);
+            // 按行处理（SSE 协议以 \n 分隔）
+            const lines = this.buffer.split('\n');
+            this.buffer = lines.pop() || '';
+            for (const line of lines) {
+                this.push(this.rewriteLine(line) + '\n');
+            }
+            callback();
+        }
+        catch (error) {
+            callback(error);
+        }
+    }
+    _flush(callback) {
+        try {
+            const remaining = this.stringDecoder.end();
+            if (remaining)
+                this.buffer += remaining;
+            if (this.buffer) {
+                this.push(this.rewriteLine(this.buffer));
+            }
+            callback();
+        }
+        catch (error) {
+            callback(error);
+        }
+    }
+    rewriteLine(line) {
+        // 只处理 SSE data 行
+        if (!line.startsWith('data:'))
+            return line;
+        let result = line;
+        // 重置正则 lastIndex（因为带 g 标志）
+        this.modelRegex.lastIndex = 0;
+        result = result.replace(this.modelRegex, this.replacement);
+        if (this.modelVersionRegex && this.modelVersionReplacement) {
+            this.modelVersionRegex.lastIndex = 0;
+            result = result.replace(this.modelVersionRegex, this.modelVersionReplacement);
+        }
+        return result;
+    }
+}
+exports.ModelRewriteTransform = ModelRewriteTransform;
+/**
+ * 非流式响应中的 model 字段回写。
+ *
+ * 将响应对象的 model（以及 Gemini 的 modelVersion）改写为客户端原始模型名。
+ * 直接修改传入对象（原地修改）。
+ */
+function rewriteResponseModel(responseData, originalModel) {
+    if (!responseData || !originalModel || typeof responseData !== 'object')
+        return;
+    // 所有格式在转换后统一使用 "model" 字段
+    if ('model' in responseData) {
+        responseData.model = originalModel;
+    }
+    // Gemini 直通场景可能保留 modelVersion
+    if ('modelVersion' in responseData) {
+        responseData.modelVersion = originalModel;
+    }
+}