@mayijuntuan/cnb-chat 1.1.0 → 1.2.0

package/lib/theme-agent.mjs

@@ -1,10 +1,13 @@
 // CodeBuddy 代理开发：编辑主题代码 + 流式回传 + git push
 // 在 CNB 流水线容器内运行。工作目录为 CNB 检出的仓库代码。
+// 保活会话：处理完当前消息后轮询 pending 接口等待后续消息，
+// 空闲超过 KEEPALIVE_IDLE 秒或总存活超过 KEEPALIVE_MAX 秒自动退出。
 // 详见 README.md 的环境变量表
 
 import { execFileSync } from 'node:child_process';
 
 const CALLBACK = process.env.CHAT_CALLBACK || '';
+const PENDING_URL = process.env.CHAT_PENDING_URL || '';
 const CHAT_ID = process.env.CHAT_ID || '';
 const SECRET = process.env.CHAT_SECRET || '';
 const MODEL = process.env.CHAT_MODEL || '';
@@ -12,50 +15,51 @@ const MESSAGES = process.env.CHAT_MESSAGES || '[]';
 const CB_API_KEY = process.env.CHAT_CB_API_KEY || '';
 const BRANCH = process.env.CHAT_BRANCH || 'main';
 
+const POLL_INTERVAL = 3 * 1000;      // 轮询间隔
+const KEEPALIVE_IDLE = 180 * 1000;   // 空闲超时后退出
+const KEEPALIVE_MAX = 1800 * 1000;   // 总存活上限（防失控计费）
+
+const sh = (cmd, args) => execFileSync(cmd, args, { encoding: 'utf8', stdio: ['ignore', 'pipe', 'pipe'] });
+const text = (e) => String((e && e.message) || e).slice(0, 300);
+const sleep = (ms) => new Promise((r) => setTimeout(r, ms));
+
 const post = (payload) => fetch(CALLBACK, {
   method: 'POST',
   headers: { 'Content-Type': 'application/json', 'X-Chat-Secret': SECRET },
   body: JSON.stringify({ chat_id: CHAT_ID, ...payload })
 }).catch(() => {});
 
-const sh = (cmd, args) => execFileSync(cmd, args, { encoding: 'utf8', stdio: ['ignore', 'pipe', 'pipe'] });
-const text = (e) => String((e && e.message) || e).slice(0, 300);
+// 轮询服务器取待处理消息（含心跳）；无待处理返回 null
+const pollPending = async () => {
+  if (!PENDING_URL) return null;
+  try {
+    const res = await fetch(PENDING_URL + '?chat_id=' + encodeURIComponent(CHAT_ID), {
+      headers: { 'X-Chat-Secret': SECRET },
+    });
+    if (!res.ok) return null;
+    const data = await res.json();
+    if (data?.has_pending && Array.isArray(data.messages) && data.messages.length) {
+      return data.messages;
+    }
+  } catch (e) { /* 网络抖动忽略，下轮再试 */ }
+  return null;
+};
 
-(async () => {
-  let messages = [];
-  try { messages = JSON.parse(MESSAGES); } catch (e) { messages = []; }
+// 执行一轮 CodeBuddy 任务：messages 为完整 prompt（system + 历史）
+const runOnce = async (messages, query, cliPath) => {
   const turns = messages.filter((m) => m.role === 'user' || m.role === 'assistant');
   const lastUser = [...turns].reverse().find((m) => m.role === 'user');
-  if (!CALLBACK || !lastUser) {
-    await post({ type: 'error', content: '流水线参数缺失' });
-    process.exit(1);
+  if (!lastUser) {
+    await post({ type: 'error', content: '消息序列无用户内容' });
+    return false;
   }
 
   // 多轮上下文 + 最新需求
   const prompt = turns.map((m) => (m.role === 'user' ? '【用户】' : '【AI】') + m.content).join('\n\n')
     + '\n\n请以上面的对话为上下文（以最后的用户消息为准），直接在当前仓库中修改代码并保存。';
 
-  // CodeBuddy 认证（SDK 要求；中国版需 internal 环境标识）
-  if (CB_API_KEY) {
-    process.env.CODEBUDDY_API_KEY = CB_API_KEY;
-    process.env.CODEBUDDY_INTERNET_ENVIRONMENT = process.env.CODEBUDDY_INTERNET_ENVIRONMENT || 'internal';
-  }
-
-  let query;
-  try {
-    ({ query } = await import('@tencent-ai/agent-sdk'));
-  } catch (e) {
-    await post({ type: 'error', content: 'agent-sdk 加载失败: ' + text(e) });
-    process.exit(1);
-  }
-
   const full = [];
 
-  // 优先使用环境预装的 codebuddy CLI（如 cnbcool/default-dev-env，已由 CNB 注入认证），
-  // 未预装时 SDK 自动回退到包内捆绑的 CLI
-  let cliPath = '';
-  try { cliPath = sh('which', ['codebuddy']).trim(); } catch (e) { /* 未预装 */ }
-
   try {
     const q = query({
       prompt,
@@ -92,7 +96,13 @@ const text = (e) => String((e && e.message) || e).slice(0, 300);
       sh('git', ['config', 'user.name', 'CodeBuddy']);
       sh('git', ['add', '-A']);
       try { sh('git', ['commit', '-m', 'AI: ' + String(lastUser.content).slice(0, 80)]); } catch (e) { /* 无变更 */ }
-      sh('git', ['push', 'https://' + process.env.CNB_TOKEN + '@cnb.cool/' + process.env.CNB_REPO_SLUG + '.git', 'HEAD:' + BRANCH]);
+      try {
+        // 优先用容器内已配置凭据的 origin（Prepare 阶段克隆时注入）
+        sh('git', ['push', 'origin', 'HEAD:' + BRANCH]);
+      } catch (e) {
+        // 兜底：显式 token URL
+        sh('git', ['push', 'https://' + process.env.CNB_TOKEN + '@cnb.cool/' + process.env.CNB_REPO_SLUG + '.git', 'HEAD:' + BRANCH]);
+      }
       await post({ type: 'progress', content: '✅ 代码已推送到 ' + BRANCH + ' 分支' });
     } catch (e) {
       await post({ type: 'progress', content: '⚠️ git push 失败: ' + text(e) });
@@ -105,7 +115,7 @@ const text = (e) => String((e && e.message) || e).slice(0, 300);
     }
 
     await post({ type: 'done', content: full.join('\n') });
-    process.exit(0);
+    return true;
   } catch (e) {
     const msg = text(e);
     // SDK preview 版在 CLI 立即退出（认证失败/CLI 缺失）时仅抛 Transport closed
@@ -113,6 +123,61 @@ const text = (e) => String((e && e.message) || e).slice(0, 300);
       ? 'CodeBuddy CLI 启动失败，请检查 CODEBUDDY_API_KEY 是否有效、CLI 是否已安装'
       : msg;
     await post({ type: 'error', content: hint });
+    return false;
+  }
+};
+
+(async () => {
+  let initial = [];
+  try { initial = JSON.parse(MESSAGES); } catch (e) { initial = []; }
+  const hasInitial = initial.some((m) => m.role === 'user');
+  if (!CALLBACK || (!hasInitial && !PENDING_URL)) {
+    await post({ type: 'error', content: '流水线参数缺失' });
     process.exit(1);
   }
+
+  // CodeBuddy 认证（SDK 要求；中国版需 internal 环境标识）
+  if (CB_API_KEY) {
+    process.env.CODEBUDDY_API_KEY = CB_API_KEY;
+    process.env.CODEBUDDY_INTERNET_ENVIRONMENT = process.env.CODEBUDDY_INTERNET_ENVIRONMENT || 'internal';
+  }
+
+  let query;
+  try {
+    ({ query } = await import('@tencent-ai/agent-sdk'));
+  } catch (e) {
+    await post({ type: 'error', content: 'agent-sdk 加载失败: ' + text(e) });
+    process.exit(1);
+  }
+
+  // 优先使用环境预装的 codebuddy CLI（如 cnbcool/default-dev-env，已由 CNB 注入认证），
+  // 未预装时 SDK 自动回退到包内捆绑的 CLI
+  let cliPath = '';
+  try { cliPath = sh('which', ['codebuddy']).trim(); } catch (e) { /* 未预装 */ }
+
+  // 第一轮：构建触发时随 env 传入的消息
+  if (hasInitial) {
+    const ok = await runOnce(initial, query, cliPath);
+    if (!ok) process.exit(1);
+  }
+
+  // 保活循环：等待后续消息；空闲超时 / 总时长超限则退出
+  if (PENDING_URL) {
+    const startedAt = Date.now();
+    let idleStart = Date.now();
+    while (Date.now() - startedAt < KEEPALIVE_MAX) {
+      if (Date.now() - idleStart > KEEPALIVE_IDLE) {
+        break; // 空闲超时，正常退出（下次消息会触发新构建）
+      }
+      const messages = await pollPending();
+      if (messages) {
+        idleStart = Date.now();
+        await runOnce(messages, query, cliPath);
+        continue; // 处理完立刻再看有没有新消息
+      }
+      await sleep(POLL_INTERVAL);
+    }
+  }
+
+  process.exit(0);
 })();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mayijuntuan/cnb-chat",
-  "version": "1.1.0",
+  "version": "1.2.0",
   "description": "CNB pipeline for AI chat streaming relay and CodeBuddy code editing",
   "type": "module",
   "bin": {