@kevisual/cnb 0.0.8 → 0.0.12

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/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/workspace/index.ts

@@ -2,6 +2,7 @@ import { createSkill, tool } from '@kevisual/router';
 import { app, cnb } from '../../app.ts';
 import z from 'zod';
 import './skills.ts';
+import './keep.ts';
 
 // 启动工作空间
 app.route({
@@ -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('批量流水线构建号'),
       },
     })
   }

package/agent/routes/workspace/keep.ts

@@ -0,0 +1,121 @@
+import { createSkill, tool } from '@kevisual/router';
+import { app, cnb } from '../../app.ts';
+
+import { createKeepAlive } from '../../../src/keep.ts';
+
+type AliveInfo = {
+  startTime: number;
+  updatedTime?: number;
+  KeepAlive: ReturnType<typeof createKeepAlive>;
+}
+
+const keepAliveMap = new Map<string, AliveInfo>();
+
+// 保持工作空间存活技能
+app.route({
+  path: 'cnb',
+  key: 'keep-workspace-alive',
+  description: '保持工作空间存活技能,参数wsUrl:工作空间访问URL，cookie:访问工作空间所需的cookie',
+  middleware: ['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参数');
+  }
+
+  // 检测是否已在运行
+  const existing = keepAliveMap.get(wsUrl);
+  if (existing) {
+    ctx.body = { message: `工作空间 ${wsUrl} 的保持存活任务已在运行中` };
+    return;
+  }
+
+  console.log(`启动保持工作空间 ${wsUrl} 存活的任务`);
+  const keep = createKeepAlive({
+    wsUrl,
+    cookie,
+    onConnect: () => {
+      console.log(`工作空间 ${wsUrl} 保持存活任务已连接`);
+    },
+    onMessage: (data) => {
+      // 可选：处理收到的消息
+      // console.log(`工作空间 ${wsUrl} 收到消息: ${data}`);
+      const aliveInfo = keepAliveMap.get(wsUrl);
+      if (aliveInfo) {
+        aliveInfo.updatedTime = Date.now();
+      }
+    },
+    debug: true,
+    onExit: (code) => {
+      console.log(`工作空间 ${wsUrl} 保持存活任务已退出，退出码: ${code}`);
+      keepAliveMap.delete(wsUrl);
+    }
+  });
+
+  keepAliveMap.set(wsUrl, { startTime: Date.now(), updatedTime: Date.now(), KeepAlive: keep });
+
+  ctx.body = { message: `已启动保持工作空间 ${wsUrl} 存活的任务` };
+}).addTo(app);
+
+// 获取保持工作空间存活任务列表技能
+app.route({
+  path: 'cnb',
+  key: 'list-keep-alive-tasks',
+  description: '获取保持工作空间存活任务列表技能',
+  middleware: ['auth'],
+  metadata: {
+    tags: [],
+  }
+}).define(async (ctx) => {
+  const list = Array.from(keepAliveMap.entries()).map(([wsUrl, info]) => ({
+    wsUrl,
+    startTime: info.startTime,
+    updatedTime: info.updatedTime
+  }));
+  ctx.body = { list };
+}).addTo(app);
+
+// 停止保持工作空间存活技能
+app.route({
+  path: 'cnb',
+  key: 'stop-keep-workspace-alive',
+  description: '停止保持工作空间存活技能, 参数wsUrl:工作空间访问URL',
+  middleware: ['auth'],
+  metadata: {
+    tags: [],
+    ...({
+      args: {
+        wsUrl: tool.schema.string().describe('工作空间的访问URL'),
+      }
+    })
+  }
+}).define(async (ctx) => {
+  const wsUrl = ctx.query?.wsUrl as string;
+  if (!wsUrl) {
+    ctx.throw(400, '缺少工作空间访问URL参数');
+  }
+
+  const keepAlive = keepAliveMap.get(wsUrl);
+  if (keepAlive) {
+    const endTime = Date.now();
+    const duration = endTime - keepAlive.startTime;
+    keepAlive?.KeepAlive?.disconnect();
+    keepAliveMap.delete(wsUrl);
+    ctx.body = { message: `已停止保持工作空间 ${wsUrl} 存活的任务，持续时间: ${duration}ms` };
+  } else {
+    ctx.body = { message: `没有找到工作空间 ${wsUrl} 的保持存活任务` };
+  }
+}).addTo(app);

package/dist/keep.d.ts

@@ -2,6 +2,7 @@ interface KeepAliveConfig {
     wsUrl: string;
     cookie: string;
     reconnectInterval?: number;
+    maxReconnectAttempts?: number;
     pingInterval?: number;
     onMessage?: (data: Buffer | string) => void;
     onConnect?: () => void;
@@ -12,6 +13,7 @@ interface KeepAliveConfig {
         data: string;
         signedData: string;
     }) => void;
+    onExit?: (code?: number) => void;
     debug?: boolean;
 }
 interface ParsedMessage {