@kevisual/cnb 0.0.10 → 0.0.13

package/agent/app.ts

@@ -2,7 +2,6 @@ import { QueryRouterServer as App } from '@kevisual/router'
 import { useContextKey } from '@kevisual/context'
 import { useConfig, useKey } from '@kevisual/use-config'
 import { CNB } from '../src/index.ts';
-import { nanoid } from 'nanoid';
 
 export const config = useConfig()
 export const cnb = useContextKey<CNB>('cnb', () => {
@@ -13,9 +12,6 @@ export const cnb = useContextKey<CNB>('cnb', () => {
   const cookie = useKey('CNB_COOKIE') as string
   return new CNB({ token: token, cookie: cookie });
 })
-export const appId = nanoid();
 export const app = useContextKey<App>('app', () => {
-  return new App({
-    appId
-  })
+  return new App({})
 })

package/agent/opencode.ts

@@ -1,4 +1,4 @@
-import { app} from './index.ts';
+import { app } from './index.ts';
 import { createRouterAgentPluginFn } from '@kevisual/router/opencode'
 
 export const CnbPlugin = createRouterAgentPluginFn({

package/agent/routes/call/index.ts

@@ -2,31 +2,33 @@ import { createSkill } from '@kevisual/router'
 import { app } from '../../app.ts'
 import { tool } from '@opencode-ai/plugin/tool'
 
-// "调用 path: cnb key: list-repos"
-app.route({
-  path: 'call',
-  key: '',
-  description: '调用',
-  middleware: ['auth'],
-  metadata: {
-    tags: ['opencode'],
-    ...createSkill({
-      skill: 'call-app',
-      title: '调用app应用',
-      summary: '调用router的应用, 参数path, key, payload',
-      args: {
-        path: tool.schema.string().describe('应用路径，例如 cnb'),
-        key: tool.schema.string().optional().describe('应用key，例如 list-repos'),
-        payload: tool.schema.object({}).optional().describe('调用参数'),
-      }
-    })
-  },
-}).define(async (ctx) => {
-  const { path, key } = ctx.query;
-  console.log('call app', ctx.query);
-  if (!path) {
-    ctx.throw('路径path不能为空');
-  }
-  const res = await ctx.run({ path, key, payload: ctx.query.payload || {} });
-  ctx.forward(res);
-}).addTo(app)
+if (!app.hasRoute('call')) {
+  // "调用 path: cnb key: list-repos"
+  app.route({
+    path: 'call',
+    key: '',
+    description: '调用',
+    middleware: ['admin-auth'],
+    metadata: {
+      tags: ['opencode'],
+      ...createSkill({
+        skill: 'call-app',
+        title: '调用app应用',
+        summary: '调用router的应用, 参数path, key, payload',
+        args: {
+          path: tool.schema.string().describe('应用路径，例如 cnb'),
+          key: tool.schema.string().optional().describe('应用key，例如 list-repos'),
+          payload: tool.schema.object({}).optional().describe('调用参数'),
+        }
+      })
+    },
+  }).define(async (ctx) => {
+    const { path, key } = ctx.query;
+    console.log('call app', ctx.query);
+    if (!path) {
+      ctx.throw('路径path不能为空');
+    }
+    const res = await ctx.run({ path, key, payload: ctx.query.payload || {} });
+    ctx.forward(res);
+  }).addTo(app)
+}

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

@@ -7,7 +7,7 @@ app.route({
   path: 'cnb',
   key: 'user-check',
   description: '检查用户登录状态，参数checkToken,default true; checkCookie, default false',
-  middleware: ['auth'],
+  middleware: ['admin-auth'],
   metadata: {
     tags: ['opencode'],
     ...createSkill({

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

@@ -6,7 +6,7 @@ app.route({
   path: 'cnb',
   key: 'set-cnb-cookie',
   description: '设置当前cnb工作空间的cookie环境变量',
-  middleware: ['auth'],
+  middleware: ['admin-auth'],
   metadata: {
     tags: ['opencode'],
     ...createSkill({
@@ -33,7 +33,7 @@ app.route({
   path: 'cnb',
   key: 'get-cnb-cookie',
   description: '获取当前cnb工作空间的cookie环境变量',
-  middleware: ['auth'],
+  middleware: ['admin-auth'],
   metadata: {
     tags: ['opencode'],
     ...createSkill({

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

@@ -11,7 +11,7 @@ app.route({
   path: 'cnb',
   key: 'get-cnb-port-uri',
   description: '获取当前cnb工作空间的port代理uri',
-  middleware: ['auth'],
+  middleware: ['admin-auth'],
   metadata: {
     tags: ['opencode'],
     ...createSkill({
@@ -40,7 +40,7 @@ app.route({
   path: 'cnb',
   key: 'get-cnb-vscode-uri',
   description: '获取当前cnb工作空间的vscode代理uri, 包括多种访问方式, 如web、vscode、codebuddy、cursor、ssh',
-  middleware: ['auth'],
+  middleware: ['admin-auth'],
   metadata: {
     tags: ['opencode'],
     ...createSkill({

package/agent/routes/index.ts

@@ -1,4 +1,4 @@
-import { app, appId } from '../app.ts';
+import { app } from '../app.ts';
 import './cnb-env/check.ts'
 import './repo/index.ts'
 import './workspace/index.ts'
@@ -31,7 +31,7 @@ if (!app.hasRoute('auth')) {
     path: 'auth',
   }).define(async (ctx) => {
     // ctx.body = 'Auth Route';
-    if (checkAppId(ctx, appId)) {
+    if (checkAppId(ctx, app.appId)) {
       return;
     }
   }).addTo(app);
@@ -42,7 +42,7 @@ if (!app.hasRoute('auth')) {
     middleware: ['auth'],
   }).define(async (ctx) => {
     // ctx.body = 'Admin Auth Route';
-    if (checkAppId(ctx, appId)) {
+    if (checkAppId(ctx, app.appId)) {
       return;
     }
   }).addTo(app);

package/agent/routes/issues/issue.ts

@@ -7,7 +7,7 @@ app.route({
   path: 'cnb',
   key: 'create-issue',
   description: '创建 Issue, 参数 repo, title, body, assignees, labels, priority',
-  middleware: ['auth'],
+  middleware: ['admin-auth'],
   metadata: {
     tags: ['opencode'],
     ...createSkill({
@@ -51,7 +51,7 @@ app.route({
   path: 'cnb',
   key: 'complete-issue',
   description: '完成 Issue, 参数 repo, issueNumber',
-  middleware: ['auth'],
+  middleware: ['admin-auth'],
   metadata: {
     tags: ['opencode'],
     ...createSkill({

package/agent/routes/issues/list.ts

@@ -6,7 +6,7 @@ app.route({
   path: 'cnb',
   key: 'list-issues',
   description: '查询 Issue 列表, 参数 repo, state, keyword, labels, page, page_size 等',
-  middleware: ['auth'],
+  middleware: ['admin-auth'],
   metadata: {
     tags: ['opencode'],
     ...createSkill({

package/agent/routes/knowledge/ai.ts

@@ -12,7 +12,7 @@ app.route({
   path: 'cnb',
   key: 'cnb-ai-chat',
   description: '调用cnb的知识库ai对话功能进行聊天',
-  middleware: ['auth'],
+  middleware: ['admin-auth'],
   metadata: {
     tags: ['opencode'],
     ...createSkill({
@@ -88,7 +88,7 @@ app.route({
   path: 'cnb',
   key: 'cnb-rag-query',
   description: '调用cnb的知识库RAG查询功能进行问答',
-  middleware: ['auth'],
+  middleware: ['admin-auth'],
   metadata: {
     tags: ['opencode'],
     ...createSkill({

package/agent/routes/repo/list.ts

@@ -8,7 +8,7 @@ app.route({
   path: 'cnb',
   key: 'list-repos',
   description: '列出我的代码仓库',
-  middleware: ['auth'],
+  middleware: ['admin-auth'],
   metadata: {
     tags: ['opencode'],
     ...createSkill({

package/agent/routes/repo/repo.ts

@@ -7,7 +7,7 @@ app.route({
   path: 'cnb',
   key: 'create-repo',
   description: '创建代码仓库, 参数name, visibility, description',
-  middleware: ['auth'],
+  middleware: ['admin-auth'],
   metadata: {
     tags: ['opencode'],
     ...createSkill({
@@ -47,7 +47,7 @@ app.route({
   path: 'cnb',
   key: 'create-repo-file',
   description: '在代码仓库中创建文件, repoName, filePath, content, encoding',
-  middleware: ['auth'],
+  middleware: ['admin-auth'],
   metadata: {
     tags: ['opencode'],
     ...createSkill({
@@ -86,7 +86,7 @@ app.route({
   path: 'cnb',
   key: 'delete-repo',
   description: '删除代码仓库, 参数name',
-  middleware: ['auth'],
+  middleware: ['admin-auth'],
   metadata: {
     tags: ['opencode'],
     ...createSkill({

package/agent/routes/workspace/index.ts

@@ -2,13 +2,14 @@ import { createSkill, tool } from '@kevisual/router';
 import { app, cnb } from '../../app.ts';
 import z from 'zod';
 import './skills.ts';
+import './keep.ts';
 
 // 启动工作空间
 app.route({
   path: 'cnb',
   key: 'start-workspace',
   description: '启动开发工作空间, 参数 repo',
-  middleware: ['auth'],
+  middleware: ['admin-auth'],
   metadata: {
     tags: ['opencode'],
     ...createSkill({
@@ -41,7 +42,7 @@ app.route({
   path: 'cnb',
   key: 'list-workspace',
   description: '获取cnb开发工作空间列表，可选参数 status=running 获取运行中的环境',
-  middleware: ['auth'],
+  middleware: ['admin-auth'],
   metadata: {
     tags: ['opencode'],
     ...createSkill({
@@ -72,7 +73,7 @@ app.route({
   path: 'cnb',
   key: 'get-workspace',
   description: '获取工作空间详情，通过 repo 和 sn 获取',
-  middleware: ['auth'],
+  middleware: ['admin-auth'],
   metadata: {
     tags: ['opencode'],
     ...createSkill({
@@ -103,7 +104,7 @@ app.route({
   path: 'cnb',
   key: 'delete-workspace',
   description: '删除工作空间，通过 pipelineId 或 sn',
-  middleware: ['auth'],
+  middleware: ['admin-auth'],
   metadata: {
     tags: ['opencode'],
     ...createSkill({
@@ -113,7 +114,7 @@ app.route({
       args: {
         pipelineId: tool.schema.string().optional().describe('流水线 ID，优先使用'),
         sn: tool.schema.string().optional().describe('流水线构建号'),
-        sns: tool.schema.array(z.string()).optional().describe('流水线构建号'),
+        sns: tool.schema.array(z.string()).optional().describe('批量流水线构建号'),
       },
     })
   }
@@ -142,7 +143,7 @@ app.route({
   path: 'cnb',
   key: 'stop-workspace',
   description: '停止工作空间，通过 pipelineId 或 sn',
-  middleware: ['auth'],
+  middleware: ['admin-auth'],
   metadata: {
     tags: ['opencode'],
     ...createSkill({

package/agent/routes/workspace/keep.ts

@@ -0,0 +1,214 @@
+import { createSkill, tool } from '@kevisual/router';
+import { app, cnb } from '../../app.ts';
+import { nanoid } from 'nanoid';
+import dayjs from 'dayjs';
+import { createKeepAlive } from '../../../src/keep.ts';
+
+type AliveInfo = {
+  startTime: number;
+  updatedTime?: number;
+  KeepAlive: ReturnType<typeof createKeepAlive>;
+  id: string;// 6位唯一标识符
+}
+
+const keepAliveMap = new Map<string, AliveInfo>();
+
+// 保持工作空间存活技能
+app.route({
+  path: 'cnb',
+  key: 'keep-workspace-alive',
+  description: '保持工作空间存活技能,参数wsUrl:工作空间访问URL，cookie:访问工作空间所需的cookie',
+  middleware: ['admin-auth'],
+  metadata: {
+    tags: [],
+    ...({
+      args: {
+        wsUrl: tool.schema.string().describe('工作空间的访问URL'),
+        cookie: tool.schema.string().describe('访问工作空间所需的cookie')
+      }
+    })
+  }
+}).define(async (ctx) => {
+  const wsUrl = ctx.query?.wsUrl as string;
+  const cookie = ctx.query?.cookie as string;
+  if (!wsUrl) {
+    ctx.throw(400, '缺少工作空间访问URL参数');
+  }
+  if (!cookie) {
+    ctx.throw(400, '缺少访问工作空间所需的cookie参数');
+  }
+
+  // 检测是否已在运行（通过 wsUrl 遍历检查）
+  const existing = Array.from(keepAliveMap.values()).find(info => (info as AliveInfo).id && (info as any).KeepAlive?.wsUrl === wsUrl);
+  if (existing) {
+    ctx.body = { message: `工作空间 ${wsUrl} 的保持存活任务已在运行中`, id: (existing as AliveInfo).id };
+    return;
+  }
+
+  console.log(`启动保持工作空间 ${wsUrl} 存活的任务`);
+  const keep = createKeepAlive({
+    wsUrl,
+    cookie,
+    onConnect: () => {
+      console.log(`工作空间 ${wsUrl} 保持存活任务已连接`);
+    },
+    onMessage: (data) => {
+      // 可选：处理收到的消息
+      // console.log(`工作空间 ${wsUrl} 收到消息: ${data}`);
+      // 通过 wsUrl 找到对应的 id 并更新时间
+      for (const info of keepAliveMap.values()) {
+        if ((info as any).KeepAlive?.wsUrl === wsUrl) {
+          info.updatedTime = Date.now();
+          break;
+        }
+      }
+    },
+    debug: true,
+    onExit: (code) => {
+      console.log(`工作空间 ${wsUrl} 保持存活任务已退出，退出码: ${code}`);
+      // 通过 wsUrl 找到对应的 id 并删除
+      for (const [id, info] of keepAliveMap.entries()) {
+        if ((info as any).KeepAlive?.wsUrl === wsUrl) {
+          keepAliveMap.delete(id);
+          break;
+        }
+      }
+    }
+  });
+
+  const id = nanoid(6).toLowerCase();
+  keepAliveMap.set(id, { startTime: Date.now(), updatedTime: Date.now(), KeepAlive: keep, id });
+
+  ctx.body = { content: `已启动保持工作空间 ${wsUrl} 存活的任务`, id };
+}).addTo(app);
+
+// 获取保持工作空间存活任务列表技能
+app.route({
+  path: 'cnb',
+  key: 'list-keep-alive-tasks',
+  description: '获取保持工作空间存活任务列表技能',
+  middleware: ['admin-auth'],
+  metadata: {
+    tags: [],
+  }
+}).define(async (ctx) => {
+  const list = Array.from(keepAliveMap.entries()).map(([id, info]) => {
+    const now = Date.now();
+    const duration = Math.floor((now - info.startTime) / 60000); // 分钟
+    return {
+      id,
+      wsUrl: (info as any).KeepAlive?.wsUrl,
+      startTime: info.startTime,
+      startTimeStr: dayjs(info.startTime).format('YYYY-MM-DD HH:mm'),
+      updatedTime: info.updatedTime,
+      updatedTimeStr: dayjs(info.updatedTime).format('YYYY-MM-DD HH:mm'),
+      duration,
+    }
+  });
+  ctx.body = { list };
+}).addTo(app);
+
+// 停止保持工作空间存活技能
+app.route({
+  path: 'cnb',
+  key: 'stop-keep-workspace-alive',
+  description: '停止保持工作空间存活技能, 参数wsUrl:工作空间访问URL或者id',
+  middleware: ['admin-auth'],
+  metadata: {
+    tags: [],
+    ...({
+      args: {
+        wsUrl: tool.schema.string().optional().describe('工作空间的访问URL'),
+        id: tool.schema.string().optional().describe('保持存活任务的唯一标识符'),
+      }
+    })
+  }
+}).define(async (ctx) => {
+  const wsUrl = ctx.query?.wsUrl as string;
+  const id = ctx.query?.id as string;
+  if (!wsUrl && !id) {
+    ctx.throw(400, '缺少工作空间访问URL参数或唯一标识符');
+  }
+
+  let targetId: string | undefined;
+  let wsUrlFound: string | undefined;
+
+  if (id) {
+    const info = keepAliveMap.get(id);
+    if (info) {
+      targetId = id;
+      wsUrlFound = (info as any).KeepAlive?.wsUrl;
+    }
+  } else if (wsUrl) {
+    for (const [key, info] of keepAliveMap.entries()) {
+      if ((info as any).KeepAlive?.wsUrl === wsUrl) {
+        targetId = key;
+        wsUrlFound = wsUrl;
+        break;
+      }
+    }
+  }
+
+  if (targetId) {
+    const keepAlive = keepAliveMap.get(targetId);
+    const endTime = Date.now();
+    const duration = endTime - keepAlive!.startTime;
+    keepAlive?.KeepAlive?.disconnect();
+    keepAliveMap.delete(targetId);
+    ctx.body = { content: `已停止保持工作空间 ${wsUrlFound} 存活的任务，持续时间: ${duration}ms`, id: targetId };
+  } else {
+    ctx.body = { content: `没有找到对应的工作空间保持存活任务` };
+  }
+}).addTo(app);
+
+app.route({
+  path: 'cnb',
+  key: 'reset-keep-workspace-alive',
+  description: '对存活的工作空间，startTime进行重置',
+  middleware: ['admin-auth'],
+  metadata: {
+    tags: [],
+  }
+}).define(async (ctx) => {
+  const now = Date.now();
+  for (const info of keepAliveMap.values()) {
+    info.startTime = now;
+  }
+  ctx.body = { content: `已重置所有存活工作空间的开始时间` };
+}).addTo(app);
+
+app.route({
+  path: 'cnb',
+  key: 'clear-keep-workspace-alive',
+  description: '对存活的工作空间，超过5小时的进行清理',
+  middleware: ['admin-auth'],
+  metadata: {
+    tags: [],
+  }
+}).define(async (ctx) => {
+  const res = clearKeepAlive();
+  ctx.body = {
+    content: `已清理所有存活工作空间中超过5小时的任务` + (res.length ? `，清理项：${res.map(i => i.wsUrl).join(', ')}` : ''),
+    list: res
+  };
+}).addTo(app);
+
+const clearKeepAlive = () => {
+  const now = Date.now();
+  let clearedArr: { id: string; wsUrl: string }[] = [];
+  for (const [id, info] of keepAliveMap.entries()) {
+    if (now - info.startTime > FIVE_HOURS) {
+      console.log(`工作空间 ${(info as any).KeepAlive?.wsUrl} 超过5小时，自动停止`);
+      info.KeepAlive?.disconnect?.();
+      keepAliveMap.delete(id);
+      clearedArr.push({ id, wsUrl: (info as any).KeepAlive?.wsUrl });
+    }
+  }
+  return clearedArr;
+}
+
+// 每5小时自动清理超时的keepAlive任务
+const FIVE_HOURS = 5 * 60 * 60 * 1000;
+setInterval(() => {
+  clearKeepAlive();
+}, FIVE_HOURS);

package/agent/routes/workspace/skills.ts

@@ -35,7 +35,7 @@ app.route({
   path: 'cnb',
   key: 'clean-closed-workspace',
   description: '批量删除已停止的cnb工作空间',
-  middleware: ['auth'],
+  middleware: ['admin-auth'],
   metadata: {
     tags: ['opencode'],
     ...createSkill({

package/dist/keep.d.ts

@@ -13,6 +13,7 @@ interface KeepAliveConfig {
         data: string;
         signedData: string;
     }) => void;
+    onExit?: (code?: number) => void;
     debug?: boolean;
 }
 interface ParsedMessage {

package/dist/keep.js

@@ -20,6 +20,7 @@ class WSKeepAlive {
       onDisconnect: config.onDisconnect ?? (() => {}),
       onError: config.onError ?? (() => {}),
       onSign: config.onSign ?? (() => {}),
+      onExit: config.onExit ?? (() => {}),
       debug: config.debug ?? false
     };
     this.url = new URL(this.config.wsUrl);
@@ -119,6 +120,7 @@ class WSKeepAlive {
   handleReconnect() {
     if (this.reconnectAttempts >= this.config.maxReconnectAttempts) {
       this.log(`Max reconnect attempts (${this.config.maxReconnectAttempts}) reached. Giving up.`);
+      this.config.onExit(1);
       return;
     }
     this.reconnectAttempts++;
@@ -133,6 +135,7 @@ class WSKeepAlive {
     this.stopPing();
     if (this.ws) {
       this.ws.close();
+      this.config.onExit(0);
       this.ws = null;
     }
   }