@kevisual/cnb 0.0.29 → 0.0.31

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@kevisual/cnb",
-  "version": "0.0.29",
+  "version": "0.0.31",
   "description": "",
   "main": "index.js",
   "scripts": {
@@ -52,6 +52,7 @@
     "./opencode": "./dist/opencode.js",
     "./keep": "./dist/keep.js",
     "./keep.ts": "./src/keep.ts",
+    "./keep-file-live.ts": "./src/workspace/keep-file-live.ts",
     "./routes": "./dist/routes.js",
     "./src/*": "./src/*",
     "./agent/*": "./agent/*"

package/src/workspace/keep-file-live.ts

@@ -0,0 +1,137 @@
+import path from 'node:path';
+import fs from 'node:fs';
+import os from 'node:os';
+import { execSync } from 'node:child_process';
+
+export type KeepAliveData = {
+  wsUrl: string;
+  cookie: string;
+  repo: string;
+  pipelineId: string;
+  createdTime: number;
+  filePath: string;
+  pm2Name: string;
+}
+
+type KeepAliveCache = {
+  data: KeepAliveData[];
+}
+
+const keepAliveFilePath = path.join(os.homedir(), '.cnb/keepAliveCache.json');
+
+export const runLive = (filePath: string, pm2Name: string) => {
+  // 使用 npx 运行命令
+  const cmdArgs = `cnb live -c ${filePath}`;
+
+  // 先停止已存在的同名 pm2 进程
+  const stopCmd = `pm2 delete ${pm2Name} 2>/dev/null || true`;
+  console.log('停止已存在的进程:', stopCmd);
+  try {
+    execSync(stopCmd, { stdio: 'inherit' });
+  } catch (error) {
+    console.log('停止进程失败或进程不存在:', error);
+  }
+
+  // 使用pm2启动
+  const pm2Cmd = `pm2 start ev --name ${pm2Name} --no-autorestart -- ${cmdArgs}`;
+  console.log('执行命令:', pm2Cmd);
+  try {
+    const result = execSync(pm2Cmd, { stdio: 'pipe', encoding: 'utf8' });
+    console.log(result);
+  } catch (error) {
+    console.error("状态码:", error.status);
+    console.error("错误详情:", error.stderr.toString()); // 这里会显示 ev 命令报的具体错误
+  }
+}
+
+export const stopLive = (pm2Name: string): boolean => {
+  const stopCmd = `pm2 delete ${pm2Name} 2>/dev/null || true`;
+  console.log('停止进程:', stopCmd);
+  try {
+    execSync(stopCmd, { stdio: 'inherit' });
+    console.log(`已停止 ${pm2Name} 的保持存活任务`);
+    return true;
+  } catch (error) {
+    console.error('停止进程失败:', error);
+  }
+  return false;
+}
+
+export function getKeepAliveCache(): KeepAliveCache {
+  try {
+    if (fs.existsSync(keepAliveFilePath)) {
+      const data = fs.readFileSync(keepAliveFilePath, 'utf-8');
+      const cache = JSON.parse(data) as KeepAliveCache;
+      return cache;
+    } else {
+      return { data: [] };
+    }
+  } catch (error) {
+    console.error('读取保持存活缓存文件失败:', error);
+    return { data: [] };
+  }
+}
+
+export function addKeepAliveData(data: KeepAliveData): KeepAliveCache {
+  const cache = getKeepAliveCache();
+  cache.data.push(data);
+  runLive(data.filePath, data.pm2Name);
+  try {
+    if (!fs.existsSync(path.dirname(keepAliveFilePath))) {
+      fs.mkdirSync(path.dirname(keepAliveFilePath), { recursive: true });
+    }
+    fs.writeFileSync(keepAliveFilePath, JSON.stringify(cache, null, 2), 'utf-8');
+    return cache;
+  } catch (error) {
+    console.error('写入保持存活缓存文件失败:', error);
+    return { data: [] };
+  }
+}
+
+export function removeKeepAliveData(repo: string, pipelineId: string): KeepAliveCache {
+  const cache = getKeepAliveCache();
+  const item = cache.data.find(item => item.repo === repo && item.pipelineId === pipelineId);
+  if (item) {
+    stopLive(item.pm2Name);
+  }
+  cache.data = cache.data.filter(item => item.repo !== repo || item.pipelineId !== pipelineId);
+  try {
+    fs.writeFileSync(keepAliveFilePath, JSON.stringify(cache, null, 2), 'utf-8');
+    return cache;
+  } catch (error) {
+    console.error('写入保持存活缓存文件失败:', error);
+    return { data: [] };
+  }
+}
+
+export const createLiveData = (data: { cookie: string, repo: string, pipelineId: string }): KeepAliveData => {
+  const { cookie, repo, pipelineId } = data;
+  const createdTime = Date.now();
+  const wsUrl = `wss://${pipelineId}.cnb.space:443?skipWebSocketFrames=false`;
+  const pm2Name = `${repo}__${pipelineId}`.replace(/\//g, '__');
+  const filePath = path.join(os.homedir(), '.cnb', `${pm2Name}.json`);
+  const _newData = { wss: wsUrl, wsUrl, cookie, repo, pipelineId, createdTime, filePath, pm2Name };
+  if (!fs.existsSync(path.dirname(filePath))) {
+    fs.mkdirSync(path.dirname(filePath), { recursive: true });
+  }
+  fs.writeFileSync(filePath, JSON.stringify(_newData, null, 2), 'utf-8');
+  return _newData;
+}
+
+export class KeepAliveManager {
+  static getCache() {
+    return getKeepAliveCache();
+  }
+
+  static add(data: KeepAliveData) {
+    return addKeepAliveData(data);
+  }
+
+  static createLiveData(data: { cookie: string, repo: string, pipelineId: string }): KeepAliveData {
+    return createLiveData(data);
+  }
+
+  static remove(repo: string, pipelineId: string) {
+    return removeKeepAliveData(repo, pipelineId);
+  }
+}