@kevisual/cnb 0.0.40 → 0.0.43

package/agent/app.ts

@@ -3,6 +3,7 @@ import { useContextKey } from '@kevisual/context'
 import { useKey } from '@kevisual/use-config'
 import { CNB } from '../src/index.ts';
 import { CNBManager } from './modules/cnb-manager.ts'
+
 export const cnbManager = new CNBManager()
 
 // CNB_TOKEN是降级兼容变量，推荐使用CNB_API_KEY
@@ -18,9 +19,9 @@ try {
     cnb: new CNB({ token: token, cookie: cookie })
   })
 } catch (error) {
-
+  process.exit(1)
 }
-export const cnb = (await cnbManager.getCNB({ username: 'default' })).cnb
+await new Promise(resolve => setTimeout(resolve, 1000))
 export const app = await useContextKey<App>('app', () => {
   return new App({})
 })
@@ -28,8 +29,7 @@ export const app = await useContextKey<App>('app', () => {
 export const notCNBCheck = (ctx: any) => {
   const isCNB = useKey('CNB');
   if (!isCNB) {
-    ctx.throw(400, '当前环境非 cnb-board 环境，无法获取 live 内容');
-    return true;
+    ctx.throw(400, '当前环境非 cnb-board 环境，无法获取内容');
   }
   return false;
-}
+}

package/agent/modules/cnb-manager.ts

@@ -1,6 +1,7 @@
 import { Result } from '@kevisual/query';
 import { CNB } from '../../src/index.ts';
 import { useKey } from '@kevisual/context';
+import { createOpenAICompatible } from '@ai-sdk/openai-compatible';
 export const getConfig = async (opts: { token?: string }) => {
   const kevisualEnv = useKey('KEVISUAL_ENV')
   const baseUrl = kevisualEnv === 'production' ? 'https://kevisual.cn/api/router' : 'https://kevisual.xiongxiao.me/api/router';
@@ -32,7 +33,14 @@ type CNBItem = {
   runAt?: number
   owner?: boolean
   cnb: CNB
+  cnbAi: ReturnType<typeof createOpenAICompatible>
 }
+// const repo = useKey('CNB_REPO_SLUG_LOWERCASE') as string || 'kevision/kevision';
+// export const cnbAi = createOpenAICompatible({
+//   baseURL: `https://api.cnb.cool/${repo}/-/ai/`,
+//   name: 'custom-cnb',
+//   apiKey: token,
+// });
 export class CNBManager {
   cnbMap: Map<string, CNBItem> = new Map()
   constructor() {
@@ -71,20 +79,24 @@ export class CNBManager {
    * @returns CNB 实例
    */
   async getContext(ctx: any) {
+    const item = await this.getCNBItem(ctx)
+    return item.cnb
+  }
+  async getCNBItem(ctx: any) {
     const tokenUser = ctx?.state?.tokenUser
     const username = tokenUser?.username
     if (!username) {
       ctx.throw(403, 'Unauthorized')
     }
     if (username === 'default') {
-      return this.getDefaultCNB().cnb
+      return this.getDefaultCNB()
     }
     const kevisualToken = ctx.query?.token;
     const item = await this.getCNB({ username, kevisualToken });
     if (!item) {
       ctx.throw(400, '不存在的 CNB 配置项，请检查 登录 Token 是否正确，或添加 CNB 配置')
     }
-    return item.cnb
+    return item;
   }
   addCNB(opts: Partial<CNBItem>) {
     if (!opts.username || !opts.token) {
@@ -98,6 +110,12 @@ export class CNBManager {
     const cnb = opts?.cnb || new CNB({ token: opts.token, cookie: opts.cookie });
     opts.cnb = cnb;
     opts.runAt = Date.now()
+    const repoSlug = useKey('CNB_REPO_SLUG_LOWERCASE') as string || 'kevision/kevision';
+    opts.cnbAi = createOpenAICompatible({
+      baseURL: `https://api.cnb.cool/${repoSlug}/-/ai/`,
+      name: `custom-cnb-${opts.username}`,
+      apiKey: opts.token,
+    })
     this.cnbMap.set(opts.username, opts as CNBItem)
     return opts as CNBItem
   }

package/agent/npc.ts

@@ -0,0 +1,94 @@
+import { app } from './index.ts';
+
+import { useIssueEnv, useCommentEnv, useRepoInfoEnv, IssueLabel } from '../src/index.ts'
+import { pick } from 'es-toolkit';
+
+const writeToProcess = (message: string) => {
+  if (process.send) {
+    process.send(message);
+  } else {
+    console.log(message);
+  }
+}
+const getIssuesLabels = async () => {
+  const issueEnv = useIssueEnv();
+  const repoInfoEnv = useRepoInfoEnv();
+  const issueId = issueEnv.issueId;
+  const repoSlug = repoInfoEnv.repoSlug;
+  if (!issueId || !repoSlug) {
+    return [];
+  }
+  const res = await app.run({
+    path: 'cnb',
+    key: 'getIssue',
+    payload: {
+      repo: repoSlug,
+      issueNumber: issueId
+    }
+  });
+  if (res.code === 200) {
+    const issueData = res.data as any;
+    const labels = issueData.labels || [];
+    return labels as IssueLabel[];
+  }
+  console.error('获取 Issue 详情失败', res);
+  return []
+
+}
+
+const main = async () => {
+  const repoInfoEnv = useRepoInfoEnv();
+  const commentEnv = useCommentEnv();
+  const issueEnv = useIssueEnv();
+  const pickCommentEnv = pick(commentEnv, ['commentId', 'commentIdLabel']);
+  const pickIssueEnv = pick(issueEnv, ['issueId', 'issueIdLabel', 'issueIid', 'issueIidLabel', 'issueTitle', 'issueTitleLabel', 'issueDescription', 'issueDescriptionLabel']);
+  const pickRepoInfoEnv = pick(repoInfoEnv, ['repoId', 'repoIdLabel', 'repoName', 'repoNameLabel', 'repoSlug', 'repoSlugLabel']);
+  // const issueLabels = issueEnv.issueLabels || [];
+  const isComment = !!commentEnv.commentId;
+  const envList = [
+    ...Object.entries(pickRepoInfoEnv).map(([key, value]) => `${key}: ${value}`),
+    ...Object.entries(issueEnv).map(([key, value]) => `${key}: ${value}`),
+    ...Object.entries(pickCommentEnv).map(([key, value]) => `${key}: ${value}`),
+  ]
+  writeToProcess('当前环境变量:');
+  const issueLabels = await getIssuesLabels();
+  const issueLabelsNames = issueLabels.map(label => label.name) || [];
+  envList.forEach(item => writeToProcess(item));
+  if (!isComment && !issueLabelsNames.includes('Run')) {
+    writeToProcess('当前 Issue 不包含 Run 标签，跳过执行');
+    process.exit(0);
+  }
+  const messages = [
+    {
+      role: 'system',
+      content: `你是一个智能的代码助手, 根据用户提供的上下文信息，提供有用的建议和帮助, 如果用户的要求和执行工具不一致，请说出你不能这么做。并把最后的结果提交一个评论到对应的issue中,提交的内容必须不能包含 @ 提及。用户提供的上下文信息如下：`
+    },
+    {
+      role: 'system',
+      content: `相关变量：${JSON.stringify({ ...pickCommentEnv, ...pickIssueEnv, ...pickRepoInfoEnv })}`
+    }, {
+      role: 'user',
+      content: commentEnv.commentBody || pickIssueEnv.issueDescription || '无'
+    }
+  ]
+  writeToProcess('输入消息:');
+  writeToProcess(JSON.stringify(messages, null, 2));
+  const result = await app.run({
+    path: 'cnb',
+    key: 'chat',
+    payload: {
+      messages
+    }
+  }, { appId: app.appId })
+  if (result.code === 200) {
+    let _message = result.data.message || []
+    writeToProcess('执行完成')
+    writeToProcess(JSON.stringify(_message, null, 2))
+    process.exit(0)
+  } else {
+    writeToProcess(result.message || '执行错误')
+    process.exit(1)
+  }
+}
+
+main();

package/agent/routes/build/docker.ts

@@ -0,0 +1,66 @@
+import { execSync } from 'child_process';
+import { app, notCNBCheck } from '../../app.ts'
+import { z } from 'zod'
+import { title } from 'process';
+
+app.route({
+  path: 'cnb',
+  key: 'docker-sync',
+  middleware: ['auth'],
+  metadata: {
+    tag: ['opencode'],
+    skill: 'cnb-docker-sync',
+    title: 'CNB Docker 镜像同步',
+    args: {
+      image: z.string().describe('Docker 同步的具体的镜像名称.'),
+      toVersion: z.string().optional().describe('修改后的版本号.'),
+    }
+  }
+}).define(async (ctx) => {
+  const { image, toVersion } = ctx.args;
+  notCNBCheck(ctx);
+  if (!image) {
+    ctx.body = {
+      message: '请提供 Docker 镜像名称.',
+      data: null,
+    }
+    return;
+  }
+  const config = {
+    registry: 'docker.cnb.cool/kevisual/dev-env',
+    dockers: [{ image, toVersion }]
+  }
+  // docker tag ghcr.io/esm-dev/esm.sh:v137 docker.cnb.cool/kevisual/dev-env/esm.sh:v137
+  // docker push docker.cnb.cool/kevisual/dev-env/esm.sh:v137
+  const run = async () => {
+    const dockers = config.dockers;
+    for (const { image, toVersion } of dockers) {
+      const imageName = image.split(':')[0].split('/').slice(-1)[0]
+      const tag = image.split(':')[1]
+      const newImage = `${config.registry}/${imageName}:${toVersion || tag}`
+      // console.log(`docker tag ${image} ${newImage}`)
+      // console.log(`docker push ${newImage}`)
+      const shell = `docker pull ${image} && docker tag ${image} ${newImage} && docker push ${newImage}`
+      console.log(shell)
+
+      console.log('\n-------------new---------------------------------\n')
+      console.log(`${newImage}`)
+      console.log('\n--------------------------------------------------\n')
+      try {
+        execSync(shell, { stdio: 'inherit' })
+      } catch (error) {
+        console.error(`Error: ${error}`)
+      }
+    }
+  }
+  run().then(() => {
+    // TODO: 通知用户同步完成
+  });;
+  ctx.body = {
+    message: 'Docker 镜像同步任务中，请稍后在目标仓库查看.',
+    data: {
+      registry: config.registry,
+      dockers: config.dockers,
+    }
+  }
+}).addTo(app);

package/agent/routes/build/index.ts

@@ -0,0 +1 @@
+import './docker.ts'

package/agent/routes/chat/chat.ts

@@ -0,0 +1,41 @@
+import { runAgent } from '@kevisual/ai/agent'
+import { app, cnbManager } from '../../app.ts';
+import z from 'zod';
+
+app.route({
+  path: 'cnb',
+  key: 'chat',
+  description: 'cnb智能对话接口',
+  middleware: ['auth'],
+  metadata: {
+    args: {
+      question: z.string().describe('用户输入的问题'),
+      messages: z.array(z.object({
+        role: z.enum(['user', 'assistant']).describe('消息角色，user表示用户输入，assistant表示助手回复'),
+        content: z.string().describe('消息内容')
+      })).describe('对话消息列表，按照时间顺序排列，包含用户和助手的历史消息'),
+      model: z.string().optional().describe('默认auto')
+    }
+  }
+}).define(async (ctx) => {
+  // notCNBCheck(ctx);
+  if (!ctx.args.question && !ctx.args.messages) {
+    ctx.throw(400, '缺少必要参数，必须提供question或messages');
+    return;
+  }
+  const model = ctx.args?.model || 'auto'
+  const item = await cnbManager.getCNBItem(ctx);
+  const cnbAi = item.cnbAi;
+  const messages = ctx.args.messages || [{
+    role: 'user',
+    content: ctx.args.question
+  }]
+  const result = await runAgent({
+    app,
+    messages: messages,
+    languageModel: cnbAi(model),
+    token: '',
+    // token: ctx.query.token as string,
+  });
+  ctx.body = result;
+}).addTo(app);

package/agent/routes/index.ts

@@ -9,6 +9,8 @@ import './issues/index.ts'
 import './cnb-board/index.ts';
 import './share/index.ts';
 import './cnb-manager/index.ts';
+import './build/index.ts';
+import './chat/chat.ts';
 
 /**
  * 验证上下文中的 App ID 是否与指定的 App ID 匹配

package/agent/routes/issues/comments.ts

@@ -0,0 +1,177 @@
+import { createSkill, tool } from '@kevisual/router';
+import { app, cnbManager } from '../../app.ts';
+import { useKey } from '@kevisual/context';
+
+// 查询 Issue 评论列表
+app.route({
+  path: 'cnb',
+  key: 'list-issue-comments',
+  description: '查询 Issue 评论列表, 参数 repo, issueNumber, page, page_size',
+  middleware: ['auth'],
+  metadata: {
+    tags: ['opencode'],
+    ...createSkill({
+      skill: 'list-issue-comments',
+      title: '查询 Issue 评论列表',
+      args: {
+        repo: tool.schema.string().optional().describe('代码仓库名称, 如 my-user/my-repo'),
+        issueNumber: tool.schema.number().describe('Issue 编号'),
+        page: tool.schema.number().optional().describe('分页页码，默认: 1'),
+        page_size: tool.schema.number().optional().describe('分页每页大小，默认: 30'),
+      },
+      summary: '查询 Issue 评论列表',
+    })
+  }
+}).define(async (ctx) => {
+  const cnb = await cnbManager.getContext(ctx);
+  let repo = ctx.query?.repo || useKey('CNB_REPO_SLUG_LOWERCASE');
+  const issueNumber = ctx.query?.issueNumber;
+  const page = ctx.query?.page ? Number(ctx.query.page) : undefined;
+  const page_size = ctx.query?.page_size ? Number(ctx.query.page_size) : undefined;
+
+  if (!repo) {
+    ctx.throw(400, '缺少参数 repo');
+  }
+  if (!issueNumber) {
+    ctx.throw(400, '缺少参数 issueNumber');
+  }
+
+  const params: Record<string, any> = {};
+  if (page) params.page = page;
+  if (page_size) params.page_size = page_size;
+
+  const res = await cnb.issue.getCommentList(repo, issueNumber, params);
+  ctx.forward(res);
+}).addTo(app);
+
+// 创建 Issue 评论
+app.route({
+  path: 'cnb',
+  key: 'create-issue-comment',
+  description: '创建 Issue 评论, 参数 repo, issueNumber, body',
+  middleware: ['auth'],
+  metadata: {
+    tags: ['opencode'],
+    ...createSkill({
+      skill: 'create-issue-comment',
+      title: '创建 Issue 评论',
+      args: {
+        repo: tool.schema.string().optional().describe('代码仓库名称, 如 my-user/my-repo'),
+        issueNumber: tool.schema.number().describe('Issue 编号'),
+        body: tool.schema.string().describe('评论内容'),
+        clearAt: tool.schema.boolean().optional().describe('是否清除评论内容中的 @ 提及，默认: true'),
+      },
+      summary: '创建 Issue 评论',
+    })
+  }
+}).define(async (ctx) => {
+  const cnb = await cnbManager.getContext(ctx);
+  let repo = ctx.query?.repo || useKey('CNB_REPO_SLUG_LOWERCASE');
+  const issueNumber = ctx.query?.issueNumber;
+  let body = ctx.query?.body;
+  const clearAt = ctx.query?.clearAt ?? true;
+
+  if (!repo) {
+    ctx.throw(400, '缺少参数 repo');
+  }
+  if (!issueNumber) {
+    ctx.throw(400, '缺少参数 issueNumber');
+  }
+  if (!body) {
+    ctx.throw(400, '缺少参数 body');
+  }
+  if (clearAt && body) {
+    // 清除评论内容中的 @ 提及
+    body = body.replace(/@/g, '');
+  }
+
+  const res = await cnb.issue.createComment(repo, issueNumber, body);
+  ctx.forward(res);
+}).addTo(app);
+
+// 获取 Issue 指定评论
+app.route({
+  path: 'cnb',
+  key: 'get-issue-comment',
+  description: '获取 Issue 指定评论, 参数 repo, issueNumber, commentId',
+  middleware: ['auth'],
+  metadata: {
+    tags: ['opencode'],
+    ...createSkill({
+      skill: 'get-issue-comment',
+      title: '获取 Issue 评论',
+      args: {
+        repo: tool.schema.string().optional().describe('代码仓库名称, 如 my-user/my-repo'),
+        issueNumber: tool.schema.number().describe('Issue 编号'),
+        commentId: tool.schema.number().describe('评论 ID'),
+      },
+      summary: '获取 Issue 评论',
+    })
+  }
+}).define(async (ctx) => {
+  const cnb = await cnbManager.getContext(ctx);
+  let repo = ctx.query?.repo || useKey('CNB_REPO_SLUG_LOWERCASE');
+  const issueNumber = ctx.query?.issueNumber;
+  const commentId = ctx.query?.commentId;
+
+  if (!repo) {
+    ctx.throw(400, '缺少参数 repo');
+  }
+  if (!issueNumber) {
+    ctx.throw(400, '缺少参数 issueNumber');
+  }
+  if (!commentId) {
+    ctx.throw(400, '缺少参数 commentId');
+  }
+
+  const res = await cnb.issue.getComment(repo, issueNumber, commentId);
+  ctx.forward(res);
+}).addTo(app);
+
+// 修改 Issue 评论
+app.route({
+  path: 'cnb',
+  key: 'update-issue-comment',
+  description: '修改 Issue 评论, 参数 repo, issueNumber, commentId, body',
+  middleware: ['auth'],
+  metadata: {
+    tags: ['opencode'],
+    ...createSkill({
+      skill: 'update-issue-comment',
+      title: '修改 Issue 评论',
+      args: {
+        repo: tool.schema.string().optional().describe('代码仓库名称, 如 my-user/my-repo'),
+        issueNumber: tool.schema.number().describe('Issue 编号'),
+        commentId: tool.schema.number().describe('评论 ID'),
+        body: tool.schema.string().describe('评论内容'),
+        clearAt: tool.schema.boolean().optional().describe('是否清除评论内容中的 @ 提及，默认: true'),
+      },
+      summary: '修改 Issue 评论',
+    })
+  }
+}).define(async (ctx) => {
+  const cnb = await cnbManager.getContext(ctx);
+  let repo = ctx.query?.repo || useKey('CNB_REPO_SLUG_LOWERCASE');
+  const issueNumber = ctx.query?.issueNumber;
+  const commentId = ctx.query?.commentId;
+  let body = ctx.query?.body;
+  const clearAt = ctx.query?.clearAt ?? true;
+  if (!repo) {
+    ctx.throw(400, '缺少参数 repo');
+  }
+  if (!issueNumber) {
+    ctx.throw(400, '缺少参数 issueNumber');
+  }
+  if (!commentId) {
+    ctx.throw(400, '缺少参数 commentId');
+  }
+  if (!body) {
+    ctx.throw(400, '缺少参数 body');
+  }
+  if (clearAt && body) {
+    // 清除评论内容中的 @ 提及
+    body = body.replace(/@/g, '');
+  }
+  const res = await cnb.issue.updateComment(repo, issueNumber, commentId, body);
+  ctx.forward(res);
+}).addTo(app);

package/agent/routes/issues/index.ts

@@ -1,2 +1,3 @@
 import './list.ts'
-import './issue.ts'
+import './issue.ts'
+import './comments.ts'

package/agent/routes/issues/list.ts

@@ -14,7 +14,7 @@ app.route({
       skill: 'list-issues',
       title: '查询 Issue 列表',
       args: {
-        repo: tool.schema.string().describe('代码仓库名称, 如 my-user/my-repo'),
+        repo: tool.schema.string().optional().describe('代码仓库名称, 如 my-user/my-repo'),
         state: tool.schema.string().optional().describe('Issue 状态：open 或 closed'),
         keyword: tool.schema.string().optional().describe('问题搜索关键词'),
         labels: tool.schema.string().optional().describe('问题标签，多个用逗号分隔'),
@@ -27,7 +27,7 @@ app.route({
   }
 }).define(async (ctx) => {
   const cnb = await cnbManager.getContext(ctx);
-  const repo = ctx.query?.repo || useKey('CNB_REPO_SLUG_LOWERCASE');
+  let repo = ctx.query?.repo || useKey('CNB_REPO_SLUG_LOWERCASE');
   const state = ctx.query?.state;
   const keyword = ctx.query?.keyword;
   const labels = ctx.query?.labels;
@@ -49,4 +49,37 @@ app.route({
 
   const res = await cnb.issue.getList(repo, params);
   ctx.forward(res);
+}).addTo(app);
+
+app.route({
+  path: 'cnb',
+  key: 'getIssue',
+  description: '获取 单个 Issue',
+  middleware: ['auth'],
+  metadata: {
+    tags: ['opencode'],
+    ...createSkill({
+      skill: 'getIssue',
+      title: '获取 单个 Issue',
+      args: {
+        repo: tool.schema.string().optional().describe('代码仓库名称, 如 my-user/my-repo'),
+        issueNumber: tool.schema.union([tool.schema.string(), tool.schema.number()]).describe('Issue 编号'),
+      },
+      summary: '获取 单个 Issue',
+    })
+  }
+}).define(async (ctx) => {
+  const cnb = await cnbManager.getContext(ctx);
+  let repo = ctx.query?.repo || useKey('CNB_REPO_SLUG_LOWERCASE');
+  const issueNumber = ctx.query?.issueNumber;
+
+  if (!repo) {
+    ctx.throw(400, '缺少参数 repo');
+  }
+  if (!issueNumber) {
+    ctx.throw(400, '缺少参数 issueNumber');
+  }
+
+  const res = await cnb.issue.getItem(repo, issueNumber);
+  ctx.forward(res);
 }).addTo(app);

package/agent/routes/opencode/index.ts

@@ -0,0 +1,12 @@
+import { createOpencodeClient } from "@opencode-ai/sdk"
+
+const client = await createOpencodeClient({
+  // baseUrl: "https://yccb64t1z-100.cnb.run",
+  // auth: async () => {
+  //   return 'cm9vdDozR0I2MDg5ZGpYOE5oMDFjM1FteE5DWDd0ZkI='
+  // }
+  baseUrl: "http://localhost:4096",
+})
+
+const sessionList = await client.session.list()
+console.log(sessionList.data)