@cnbcool/cnb-api-generate 2.14.0 → 2.14.1

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/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.1",
   "main": "./built/index.js",
   "module": "./src/index.ts",
   "types": "./src/index.ts",