@cnbcool/cnb-api-generate 2.14.0 → 2.14.2

package/client/lib/git-credential.ts

@@ -1,5 +1,6 @@
 import { Command } from 'commander';
 import { resolveToken } from '../utils/resolve-token';
+import { getGitRemoteHosts } from '../utils/resolve-platform-host';
 
 /**
  * 用作 git credential helper。
@@ -11,7 +12,7 @@ import { resolveToken } from '../utils/resolve-token';
  *
  * git 在执行需要凭据的操作时，会以 `get`、`store`、`erase` 之一作为 action 调用本命令，
  * 并通过 stdin 传入若干 `key=value` 形式的字段（protocol/host/path 等）。
- * 我们仅处理 `get`：根据 host 白名单校验后，从后端获取凭据，按 git-credential
+ * 我们仅处理 `get`：校验 host 属于当前仓库 git remote 的域名后，从后端获取凭据，按 git-credential
  * 协议把 `username=...` / `password=...` 写到 stdout；`store` / `erase` 直接忽略。
  */
 
@@ -27,8 +28,6 @@ interface Credential {
   password: string;
 }
 
-const ALLOWED_HOSTS = ['cnb.cool', 'cnb.woa.com'];
-
 /** git-credential get 等待 stdin 的最长时间，避免 git 无限挂起。 */
 const GET_INPUT_TIMEOUT_MS = 8000;
 
@@ -73,7 +72,7 @@ export function registerGitCredentialCommand(program: Command): void {
 }
 
 /**
- * 校验白名单并向后端获取凭据，按照 git-credential 协议输出 username/password。
+ * 校验 remote 域名并向后端获取凭据，按照 git-credential 协议输出 username/password。
  */
 async function auth(data: GitCredentialInput): Promise<void> {
   const { host } = data;
@@ -81,8 +80,11 @@ async function auth(data: GitCredentialInput): Promise<void> {
     `[git-credential]: for ${data.protocol}://${data.host}/${data.path ?? ''}`,
   );
 
-  // 必须校验域名，否则传入第三方域名就会被盗密码
-  if (!host || !ALLOWED_HOSTS.includes(host)) {
+  // 只向当前仓库 git remote 的域名提供凭据；remote 由用户配置（含子域名、api 子域），
+  // git-credential 调用链天然与仓库绑定，非本仓库 remote 的域名一律拒绝，
+  // 避免第三方域名盗取密码
+  const remoteHosts = getGitRemoteHosts();
+  if (!host || remoteHosts.length === 0 || !remoteHosts.includes(host)) {
     throw new Error(`unknown host: ${host}`);
   }
 

package/client/lib/login.ts

@@ -9,19 +9,20 @@ import {
   type TokenStore,
 } from '../utils/device-auth';
 import { tryOpenBrowser } from '../utils/open-browser';
+import { resolvePlatformUrl } from '../utils/resolve-platform-host';
 
 export function registerLoginCommand(program: Command): void {
   program
     .command('login')
     .description('通过 OAuth2 设备授权流登录 CNB，获取并保存 access_token')
     .option('--client-id <string>', 'OAuth2 client_id', process.env.OAUTH2_CLIENT_ID || 'cnb_cli')
-    .option('--woa', '使用内网环境 (https://cnb.woa.com)', false)
     .option('--debug', '打印调试信息', false)
     .helpOption('-h, --help', '显示帮助文档')
     .action(async (opts) => {
       const cfg: LoginConfig = {
         clientID: opts.clientId,
-        platformURL: opts.woa ? 'https://cnb.woa.com' : 'https://cnb.cool',
+        // 从 git remote -v 动态识别平台域名，兼容不同部署（公开 / 内网 / 自定义域名）
+        platformURL: resolvePlatformUrl(),
         debug: opts.debug,
       };
 

package/client/lib/skills-list.ts

@@ -90,22 +90,24 @@ export interface SkillsListOptions {
   global?: boolean;
   /** 仅列出项目 scope（对应 skills list -p） */
   project?: boolean;
-  /** 按 agent 过滤：仅保留 agents 中包含指定 agent 的条目 */
+  /** 按 agent 过滤：透传 -a 给底层 skills 命令（值会归一化为小写 key） */
   agent?: string;
 }
 
 /**
- * 运行原始 `skills list [scope] --json` 并返回解析后的 JSON 数组。
+ * 运行原始 `skills list [scope] [--agent] --json` 并返回解析后的 JSON 数组。
  *
- * scope 参数规则与 `skills list` 一致：
+ * 参数规则与 `skills list` 一致：
  * - 指定 `-g` → `-g`
  * - 指定 `-p` → `-p`
  * - 都不指定 → 不传 scope（由 skills 默认：项目优先，否则全局）
+ * - 指定 `-a/--agent` → 归一化为小写后透传（底层按 agent 目录 key 匹配）
  */
 export function runSkillsListJson(opts: SkillsListOptions): SkillEntry[] {
   const args = ['list'];
   if (opts.global) args.push('-g');
   if (opts.project) args.push('-p');
+  if (opts.agent) args.push('-a', opts.agent.toLowerCase());
   args.push('--json');
 
   let stdout: string;
@@ -132,24 +134,17 @@ export function runSkillsListJson(opts: SkillsListOptions): SkillEntry[] {
 
 /**
  * 对 `skills list --json` 输出做后处理：
- * 1. 从 SKILL.md frontmatter 回填缺失的 description；
- * 2. 按 agent 过滤（若指定 -a）。
+ * 仅从 SKILL.md frontmatter 回填缺失的 description。
  *
+ * agent 过滤已下沉到底层 `skills list -a`（大小写归一化后透传），此处不再过滤。
  * 返回的是全新数组/对象，不修改原始输入。
  */
 export function decorateSkillEntries(
   entries: SkillEntry[],
-  opts: SkillsListOptions,
 ): SkillEntry[] {
   const result: SkillEntry[] = [];
 
   for (const entry of entries) {
-    // agent 过滤：保留 agents 中包含指定 agent 的条目
-    if (opts.agent) {
-      const agents = Array.isArray(entry.agents) ? entry.agents : [];
-      if (!agents.includes(opts.agent)) continue;
-    }
-
     const decorated: SkillEntry = { ...entry };
 
     // 回填 description：缺失或为空时从 path 下的 SKILL.md frontmatter 解析
@@ -171,7 +166,7 @@ export function decorateSkillEntries(
  */
 export function skillsList(opts: SkillsListOptions & { json?: boolean }): string {
   const raw = runSkillsListJson(opts);
-  const decorated = decorateSkillEntries(raw, opts);
+  const decorated = decorateSkillEntries(raw);
 
   // 与 `skills list --json` 保持一致的 JSON 输出（默认即 JSON，字段结构不变）
   return JSON.stringify(decorated, null, 2);

package/client/utils/resolve-platform-host.ts

@@ -0,0 +1,64 @@
+import { execSync } from 'node:child_process';
+
+/**
+ * 从当前仓库的 `git remote -v` 动态识别平台域名。
+ *
+ * 统一以 git remote 解析为准，不再维护静态域名白名单，
+ * 天然兼容不同平台部署（公开 / 内网 / 自定义域名）。
+ */
+
+/**
+ * 执行 `git remote -v` 并解析出所有 remote 的 host（去重）。
+ * 仅在当前目录是 git 仓库时返回非空数组；解析失败或非 git 仓库时返回空数组。
+ * @returns 去重后的 host 列表，如 ['cnb.cool']
+ */
+export function getGitRemoteHosts(): string[] {
+  try {
+    const out = execSync('git remote -v', {
+      encoding: 'utf8',
+      stdio: ['ignore', 'pipe', 'ignore'],
+    });
+    const hosts = new Set<string>();
+    for (const line of out.split('\n')) {
+      // 每行形如：origin\thttps://cnb.cool/cnb/skills/cnb-skill.git (fetch)
+      const url = line.split(/\s+/)[1];
+      if (!url) continue;
+      const host = parseUrlHost(url);
+      if (host) hosts.add(host);
+    }
+    return [...hosts];
+  } catch {
+    // 非 git 仓库或 git 不可用等场景，静默返回空数组
+    return [];
+  }
+}
+
+/**
+ * 从 remote URL 中解析出 host。
+ * 兼容 https://、http://、ssh:// 以及 scp-like 形式（git@host:path）。
+ * @param url
+ * @returns host，解析失败返回 null
+ */
+function parseUrlHost(url: string): string | null {
+  try {
+    // scp-like: git@github.com:org/repo.git
+    if (/^[^@/]+@[^:/]+:/.test(url) && !/^[a-z]+:\/\//i.test(url)) {
+      const host = url.slice(url.indexOf('@') + 1, url.indexOf(':'));
+      return host || null;
+    }
+    return new URL(url).hostname || null;
+  } catch {
+    return null;
+  }
+}
+
+/**
+ * 解析当前平台的 base URL（形如 https://cnb.cool）。
+ * 优先取 git remote 中的 host 拼成 https://{host}；
+ * 无 remote 或解析失败时兜底返回 https://cnb.cool。
+ * @returns 平台 base URL
+ */
+export function resolvePlatformUrl(): string {
+  const host = getGitRemoteHosts()[0];
+  return host ? `https://${host}` : 'https://cnb.cool';
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cnbcool/cnb-api-generate",
-  "version": "2.14.0",
+  "version": "2.14.2",
   "main": "./built/index.js",
   "module": "./src/index.ts",
   "types": "./src/index.ts",