@kevisual/cnb 0.0.7 → 0.0.10

package/agent/routes/cnb-env/vscode.ts

@@ -5,6 +5,8 @@ import { CNB_ENV } from "@/common/cnb-env.ts";
 
 // 执行技能  get-cnb-port-uri，端口为4096
 // 执行技能  get-cnb-port-uri，端口为51515
+
+// TODO： 获取仓库的
 app.route({
   path: 'cnb',
   key: 'get-cnb-port-uri',

package/dist/keep.d.ts

@@ -0,0 +1,44 @@
+interface KeepAliveConfig {
+    wsUrl: string;
+    cookie: string;
+    reconnectInterval?: number;
+    maxReconnectAttempts?: number;
+    pingInterval?: number;
+    onMessage?: (data: Buffer | string) => void;
+    onConnect?: () => void;
+    onDisconnect?: (code: number) => void;
+    onError?: (error: Error) => void;
+    onSign?: (data: {
+        type: string;
+        data: string;
+        signedData: string;
+    }) => void;
+    debug?: boolean;
+}
+interface ParsedMessage {
+    type: string;
+    raw: Buffer;
+    payload?: any;
+}
+type MessageHandler = (msg: ParsedMessage) => void;
+declare class WSKeepAlive {
+    private ws;
+    private config;
+    private reconnectAttempts;
+    private pingTimer;
+    private messageHandlers;
+    private url;
+    constructor(config: KeepAliveConfig);
+    private log;
+    private parseMessage;
+    connect(): void;
+    private startPing;
+    private stopPing;
+    private handleReconnect;
+    onMessage(handler: MessageHandler): () => boolean;
+    disconnect(): void;
+}
+declare function createKeepAlive(config: KeepAliveConfig): WSKeepAlive;
+
+export { WSKeepAlive, createKeepAlive };
+export type { KeepAliveConfig, ParsedMessage };

package/dist/keep.js

@@ -0,0 +1,148 @@
+// src/workspace/keep-live.ts
+import WebSocket from "ws";
+
+class WSKeepAlive {
+  ws = null;
+  config;
+  reconnectAttempts = 0;
+  pingTimer = null;
+  messageHandlers = new Set;
+  url;
+  constructor(config) {
+    this.config = {
+      wsUrl: config.wsUrl,
+      cookie: config.cookie,
+      reconnectInterval: config.reconnectInterval ?? 5000,
+      maxReconnectAttempts: config.maxReconnectAttempts ?? 3,
+      pingInterval: config.pingInterval ?? 30000,
+      onMessage: config.onMessage ?? (() => {}),
+      onConnect: config.onConnect ?? (() => {}),
+      onDisconnect: config.onDisconnect ?? (() => {}),
+      onError: config.onError ?? (() => {}),
+      onSign: config.onSign ?? (() => {}),
+      debug: config.debug ?? false
+    };
+    this.url = new URL(this.config.wsUrl);
+  }
+  log(message) {
+    if (!this.config.debug)
+      return;
+    const timestamp = new Date().toISOString();
+    const msg = `[${timestamp}] ${message}`;
+    console.log(msg);
+  }
+  parseMessage(data) {
+    const result = { type: "unknown", raw: data };
+    if (data.length < 14) {
+      result.type = "raw";
+      return result;
+    }
+    const prefix = data.slice(0, 13);
+    const msgType = prefix[0];
+    const jsonStart = data.indexOf(113);
+    if (jsonStart !== -1) {
+      try {
+        const jsonStr = data.slice(jsonStart + 1).toString();
+        const payload = JSON.parse(jsonStr);
+        result.type = `binary(0x${msgType.toString(16)})`;
+        result.payload = payload;
+        if (payload.type === "sign" && this.config.onSign) {
+          this.config.onSign(payload);
+        }
+        return result;
+      } catch {
+        result.type = "binary(json-parse-error)";
+        return result;
+      }
+    }
+    result.type = "raw";
+    return result;
+  }
+  connect() {
+    const { wsUrl, cookie, debug } = this.config;
+    this.log(`Connecting to ${wsUrl}...`);
+    this.ws = new WebSocket(wsUrl, {
+      headers: {
+        Origin: this.url.origin,
+        Cookie: cookie,
+        "Cache-Control": "no-cache",
+        "Accept-Language": "zh-CN,zh;q=0.9",
+        Pragma: "no-cache",
+        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36",
+        "Sec-WebSocket-Extensions": "permessage-deflate"
+      }
+    });
+    this.ws.on("open", () => {
+      debug && this.log("Connected!");
+      this.reconnectAttempts = 0;
+      this.config.onConnect();
+      this.startPing();
+    });
+    this.ws.on("message", (data) => {
+      if (Buffer.isBuffer(data)) {
+        const parsed = this.parseMessage(data);
+        this.config.onMessage(parsed?.raw ?? data);
+        this.messageHandlers.forEach((handler) => {
+          if (parsed)
+            handler(parsed);
+        });
+      } else {
+        this.config.onMessage(data);
+      }
+    });
+    this.ws.on("close", (code) => {
+      debug && this.log(`Disconnected (code: ${code})`);
+      this.stopPing();
+      this.config.onDisconnect(code);
+      this.handleReconnect();
+    });
+    this.ws.on("error", (err) => {
+      debug && this.log(`Error: ${err.message}`);
+      this.config.onError(err);
+    });
+  }
+  startPing() {
+    this.stopPing();
+    this.pingTimer = setInterval(() => {
+      if (this.ws && this.ws.readyState === WebSocket.OPEN) {
+        this.ws.ping();
+        this.log("Sent ping");
+      }
+    }, this.config.pingInterval);
+  }
+  stopPing() {
+    if (this.pingTimer) {
+      clearInterval(this.pingTimer);
+      this.pingTimer = null;
+    }
+  }
+  handleReconnect() {
+    if (this.reconnectAttempts >= this.config.maxReconnectAttempts) {
+      this.log(`Max reconnect attempts (${this.config.maxReconnectAttempts}) reached. Giving up.`);
+      return;
+    }
+    this.reconnectAttempts++;
+    this.log(`Reconnecting in ${this.config.reconnectInterval}ms... (attempt ${this.reconnectAttempts}/${this.config.maxReconnectAttempts})`);
+    setTimeout(() => this.connect(), this.config.reconnectInterval);
+  }
+  onMessage(handler) {
+    this.messageHandlers.add(handler);
+    return () => this.messageHandlers.delete(handler);
+  }
+  disconnect() {
+    this.stopPing();
+    if (this.ws) {
+      this.ws.close();
+      this.ws = null;
+    }
+  }
+}
+function createKeepAlive(config) {
+  const client = new WSKeepAlive(config);
+  client.connect();
+  return client;
+}
+export {
+  createKeepAlive,
+  WSKeepAlive
+};

package/dist/opencode.js

@@ -17243,7 +17243,7 @@ function requireStream() {
       if (!duplex.push(data))
         ws.pause();
     });
-    ws.once("error", function error(err) {
+    ws.once("error", function error2(err) {
       if (duplex.destroyed)
         return;
       terminateOnDestroy = false;
@@ -17261,7 +17261,7 @@ function requireStream() {
         return;
       }
       let called = false;
-      ws.once("error", function error(err2) {
+      ws.once("error", function error2(err2) {
         called = true;
         callback(err2);
       });
@@ -32161,7 +32161,7 @@ function date5(params) {
 
 // node_modules/.pnpm/zod@4.1.8/node_modules/zod/v4/classic/external.js
 config3(en_default());
-// node_modules/.pnpm/@opencode-ai+plugin@1.1.39/node_modules/@opencode-ai/plugin/dist/tool.js
+// node_modules/.pnpm/@opencode-ai+plugin@1.1.44/node_modules/@opencode-ai/plugin/dist/tool.js
 function tool2(input) {
   return input;
 }
@@ -52972,7 +52972,7 @@ var init_crypto = __esm(() => {
       } catch (e) {
         errorProto = getProto(getProto(e)), INTRINSICS["%Error.prototype%"] = errorProto;
       }
-    var errorProto, doEval = function doEval(name) {
+    var errorProto, doEval = function doEval2(name) {
       var value;
       if (name === "%AsyncFunction%")
         value = getEvalledConstructor("async function () {}");
@@ -52981,11 +52981,11 @@ var init_crypto = __esm(() => {
       else if (name === "%AsyncGeneratorFunction%")
         value = getEvalledConstructor("async function* () {}");
       else if (name === "%AsyncGenerator%") {
-        var fn = doEval("%AsyncGeneratorFunction%");
+        var fn = doEval2("%AsyncGeneratorFunction%");
         if (fn)
           value = fn.prototype;
       } else if (name === "%AsyncIteratorPrototype%") {
-        var gen = doEval("%AsyncGenerator%");
+        var gen = doEval2("%AsyncGenerator%");
         if (gen && getProto)
           value = getProto(gen.prototype);
       }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kevisual/cnb",
-  "version": "0.0.7",
+  "version": "0.0.10",
   "description": "",
   "main": "index.js",
   "scripts": {
@@ -21,9 +21,10 @@
     "@kevisual/ai": "^0.0.24",
     "@kevisual/context": "^0.0.4",
     "@kevisual/types": "^0.0.12",
-    "@opencode-ai/plugin": "^1.1.39",
-    "@types/bun": "^1.3.7",
+    "@opencode-ai/plugin": "^1.1.44",
+    "@types/bun": "^1.3.8",
     "@types/node": "^25.1.0",
+    "@types/ws": "^8.18.1",
     "dotenv": "^17.2.3"
   },
   "publishConfig": {
@@ -35,11 +36,13 @@
     "@kevisual/use-config": "^1.0.28",
     "es-toolkit": "^1.44.0",
     "nanoid": "^5.1.6",
+    "ws": "npm:@kevisual/ws",
     "zod": "^4.3.6"
   },
   "exports": {
     ".": "./mod.ts",
     "./opencode": "./dist/opencode.js",
+    "./keep": "./dist/keep.js",
     "./src/*": "./src/*",
     "./agent/*": "./agent/*"
   }

package/src/keep.ts

@@ -0,0 +1 @@
+export * from './workspace/keep-live.ts';

package/src/workspace/keep-live.ts

@@ -0,0 +1,187 @@
+// WebSocket Keep-Alive Client Library
+import WebSocket from "ws";
+
+export interface KeepAliveConfig {
+  wsUrl: string;
+  cookie: string;
+  reconnectInterval?: number;
+  maxReconnectAttempts?: number;
+  pingInterval?: number;
+  onMessage?: (data: Buffer | string) => void;
+  onConnect?: () => void;
+  onDisconnect?: (code: number) => void;
+  onError?: (error: Error) => void;
+  onSign?: (data: { type: string; data: string; signedData: string }) => void;
+  debug?: boolean;
+}
+
+export interface ParsedMessage {
+  type: string;
+  raw: Buffer;
+  payload?: any;
+}
+
+type MessageHandler = (msg: ParsedMessage) => void;
+
+export class WSKeepAlive {
+  private ws: WebSocket | null = null;
+  private config: Required<KeepAliveConfig>;
+  private reconnectAttempts = 0;
+  private pingTimer: NodeJS.Timeout | null = null;
+  private messageHandlers: Set<MessageHandler> = new Set();
+  private url: URL;
+
+  constructor(config: KeepAliveConfig) {
+    this.config = {
+      wsUrl: config.wsUrl,
+      cookie: config.cookie,
+      reconnectInterval: config.reconnectInterval ?? 5000,
+      maxReconnectAttempts: config.maxReconnectAttempts ?? 3,
+      pingInterval: config.pingInterval ?? 30000,
+      onMessage: config.onMessage ?? (() => {}),
+      onConnect: config.onConnect ?? (() => {}),
+      onDisconnect: config.onDisconnect ?? (() => {}),
+      onError: config.onError ?? (() => {}),
+      onSign: config.onSign ?? (() => {}),
+      debug: config.debug ?? false,
+    };
+    this.url = new URL(this.config.wsUrl);
+  }
+
+  private log(message: string) {
+    if (!this.config.debug) return;
+    const timestamp = new Date().toISOString();
+    const msg = `[${timestamp}] ${message}`;
+    console.log(msg);
+  }
+
+  private parseMessage(data: Buffer): ParsedMessage | null {
+    const result: ParsedMessage = { type: "unknown", raw: data };
+
+    if (data.length < 14) {
+      result.type = "raw";
+      return result;
+    }
+
+    const prefix = data.slice(0, 13);
+    const msgType = prefix[0];
+    const jsonStart = data.indexOf(0x71); // 0x71 = 'q'
+
+    if (jsonStart !== -1) {
+      try {
+        const jsonStr = data.slice(jsonStart + 1).toString();
+        const payload = JSON.parse(jsonStr);
+        result.type = `binary(0x${msgType.toString(16)})`;
+        result.payload = payload;
+
+        // 特殊处理 sign 类型
+        if (payload.type === "sign" && this.config.onSign) {
+          this.config.onSign(payload);
+        }
+        return result;
+      } catch {
+        result.type = "binary(json-parse-error)";
+        return result;
+      }
+    }
+
+    result.type = "raw";
+    return result;
+  }
+
+  connect() {
+    const { wsUrl, cookie, debug } = this.config;
+    this.log(`Connecting to ${wsUrl}...`);
+
+    this.ws = new WebSocket(wsUrl, {
+      headers: {
+        "Origin": this.url.origin,
+        "Cookie": cookie,
+        "Cache-Control": "no-cache",
+        "Accept-Language": "zh-CN,zh;q=0.9",
+        "Pragma": "no-cache",
+        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36",
+        "Sec-WebSocket-Extensions": "permessage-deflate",
+      }
+    });
+
+    this.ws.on("open", () => {
+      debug && this.log("Connected!");
+      this.reconnectAttempts = 0;
+      this.config.onConnect();
+      this.startPing();
+    });
+
+    this.ws.on("message", (data: any) => {
+      if (Buffer.isBuffer(data)) {
+        const parsed = this.parseMessage(data);
+        this.config.onMessage(parsed?.raw ?? data);
+
+        this.messageHandlers.forEach(handler => {
+          if (parsed) handler(parsed);
+        });
+      } else {
+        this.config.onMessage(data);
+      }
+    });
+
+    this.ws.on("close", (code: number) => {
+      debug && this.log(`Disconnected (code: ${code})`);
+      this.stopPing();
+      this.config.onDisconnect(code);
+      this.handleReconnect();
+    });
+
+    this.ws.on("error", (err: Error) => {
+      debug && this.log(`Error: ${err.message}`);
+      this.config.onError(err);
+    });
+  }
+
+  private startPing() {
+    this.stopPing();
+    this.pingTimer = setInterval(() => {
+      if (this.ws && this.ws.readyState === WebSocket.OPEN) {
+        this.ws.ping();
+        this.log("Sent ping");
+      }
+    }, this.config.pingInterval);
+  }
+
+  private stopPing() {
+    if (this.pingTimer) {
+      clearInterval(this.pingTimer);
+      this.pingTimer = null;
+    }
+  }
+
+  private handleReconnect() {
+    if (this.reconnectAttempts >= this.config.maxReconnectAttempts) {
+      this.log(`Max reconnect attempts (${this.config.maxReconnectAttempts}) reached. Giving up.`);
+      return;
+    }
+    this.reconnectAttempts++;
+    this.log(`Reconnecting in ${this.config.reconnectInterval}ms... (attempt ${this.reconnectAttempts}/${this.config.maxReconnectAttempts})`);
+    setTimeout(() => this.connect(), this.config.reconnectInterval);
+  }
+
+  onMessage(handler: MessageHandler) {
+    this.messageHandlers.add(handler);
+    return () => this.messageHandlers.delete(handler);
+  }
+
+  disconnect() {
+    this.stopPing();
+    if (this.ws) {
+      this.ws.close();
+      this.ws = null;
+    }
+  }
+}
+
+// 便捷函数：快速创建并启动
+export function createKeepAlive(config: KeepAliveConfig): WSKeepAlive {
+  const client = new WSKeepAlive(config);
+  client.connect();
+  return client;
+}