@cnbcool/cnb-api-generate 2.16.11 → 2.17.0

package/client/lib/git-credential.ts

@@ -88,7 +88,7 @@ async function auth(data: GitCredentialInput): Promise<void> {
     throw new Error(`unknown host: ${host}`);
   }
 
-  const credential = await fetchCredential(data);
+  const credential = await fetchCredential(host);
 
   // git-credential 协议：通过 stdout 输出 key=value
   process.stdout.write(`username=${credential.username}\n`);
@@ -97,9 +97,9 @@ async function auth(data: GitCredentialInput): Promise<void> {
   console.error('[git-credential]: done');
 }
 
-async function fetchCredential(_data: GitCredentialInput): Promise<Credential> {
-  // 使用 resolveToken 获取 access_token 作为 password，username 固定为 cnb
-  const token = await resolveToken();
+async function fetchCredential(host: string): Promise<Credential> {
+  // 按请求 host 取对应域的 access_token 作为 password，username 固定为 cnb
+  const token = await resolveToken(host);
   return {
     username: 'cnb',
     password: token,

package/client/lib/logout.ts

@@ -1,32 +1,46 @@
 import { Command } from 'commander';
 import fs from 'fs';
 import { getTokenPath } from '../utils/token-path';
-import { loadToken } from '../utils/load-token';
+import { readTokenFile } from '../utils/load-token';
+import { removeToken } from '../utils/save-token';
 
 export function registerLogoutCommand(program: Command): void {
   program
     .command('logout')
     .description('退出登录，删除本地保存的 access_token')
+    .option('--host <string>', '只退出指定平台域名的登录态（如 cnb.cool），不传则退出全部域名的登录态')
     .option('--debug', '打印调试信息', false)
     .helpOption('-h, --help', '显示帮助文档')
     .action(async (opts) => {
       const tokenPath = getTokenPath();
-      const token = loadToken();
 
-      if (!token) {
+      if (opts.host) {
+        if (removeToken(opts.host)) {
+          console.log(`已成功退出 ${opts.host} 的登录。`);
+        } else {
+          console.log(`${opts.host} 当前未登录，无需退出。`);
+        }
+        return;
+      }
+
+      const { auths } = readTokenFile();
+      const hosts = Object.keys(auths);
+
+      if (!hosts.length) {
         console.log('当前未登录，无需退出。');
         return;
       }
 
       if (opts.debug) {
         console.log(`Token 文件路径: ${tokenPath}`);
-        console.log(`Platform URL : ${token.platform_url}`);
-        console.log(`Client ID    : ${token.client_id}`);
+        for (const host of hosts) {
+          console.log(`${host}: Platform URL=${auths[host].platform_url}, Client ID=${auths[host].client_id}`);
+        }
       }
 
       try {
         fs.unlinkSync(tokenPath);
-        console.log('已成功退出登录。');
+        console.log(`已成功退出全部登录（${hosts.join('、')}）。`);
       } catch (err: any) {
         console.error(`退出登录失败: ${err.message}`);
         process.exit(1);

package/client/lib/status.ts

@@ -13,10 +13,11 @@ export function registerStatusCommand(program: Command): void {
   program
     .command('status')
     .description('检查当前登录状态（token 是否存在、是否过期）')
+    .option('--host <string>', '查看指定平台域名的登录态（如 cnb.cool），不传则按当前仓库 git remote、CNB_API_ENDPOINT 的顺序推断，都没有对应登录态时取最近一次保存的')
     .option('--debug', '打印调试信息', false)
     .helpOption('-h, --help', '显示帮助文档')
     .action(async (opts) => {
-      const result = await resolveTokenSource();
+      const result = await resolveTokenSource(opts.host);
 
       if (result.source === 'none') {
         console.log(result.reason ? `⚠️  ${result.reason}` : '⚠️  未登录，请执行 `cnb login` 进行授权。');

package/client/utils/load-token.ts

@@ -1,14 +1,84 @@
 import fs from 'fs';
 import { getTokenPath } from './token-path';
+import { getGitRemoteHosts, normalizeTokenHost } from './resolve-platform-host';
 import type { TokenStore } from './types';
 
-/** 读取已保存的 token */
-export function loadToken(): TokenStore | null {
+/**
+ * ~/.cnb/token 文件格式，参照 docker 的 ~/.docker/config.json，按 host 存放各自的登录态：
+ * { "auths": { "<host>": { "access_token": ..., "expires_at": ..., ... } } }
+ * host 为平台域名（可带端口），每个 host 一条，互不覆盖。
+ */
+export interface TokenFile {
+  auths: Record<string, TokenStore>;
+}
+
+/** 登录态所属的 host：取 platform_url 的 host，缺失或非法时用 login_host，都不可用时返回 null */
+export function hostOfStore(store: Pick<TokenStore, 'platform_url' | 'login_host'>): string | null {
+  for (const url of [store.platform_url, store.login_host]) {
+    if (!url) continue;
+    try {
+      const host = new URL(url).host;
+      if (host) return normalizeTokenHost(host);
+    } catch {
+      // 非法 URL，尝试下一个
+    }
+  }
+  return null;
+}
+
+/** 读取整个 token 文件；兼容旧版单条 TokenStore 格式 */
+export function readTokenFile(): TokenFile {
   const tokenPath = getTokenPath();
-  if (!fs.existsSync(tokenPath)) return null;
+  if (!fs.existsSync(tokenPath)) return { auths: {} };
   try {
-    return JSON.parse(fs.readFileSync(tokenPath, 'utf-8'));
+    const data = JSON.parse(fs.readFileSync(tokenPath, 'utf-8'));
+    if (data?.auths && typeof data.auths === 'object') return data;
+    // 旧版格式：文件本身就是一个 TokenStore。按 platform_url / login_host 归到对应 host；
+    // 仅当两者都没有可用域名时才归到 cnb.cool（与旧版 refresh 时 platform_url 缺省为 cnb.cool 的行为一致）
+    const host = data?.access_token ? hostOfStore(data) ?? 'cnb.cool' : null;
+    return host ? { auths: { [host]: data } } : { auths: {} };
+  } catch {
+    return { auths: {} };
+  }
+}
+
+let remoteHostsCache: string[] | undefined;
+
+/** 当前仓库 git remote 的 host，同一进程只执行一次 git remote -v，避免每次取 token 都 fork 子进程 */
+function remoteHosts(): string[] {
+  remoteHostsCache ??= getGitRemoteHosts();
+  return remoteHostsCache;
+}
+
+/** CNB_API_ENDPOINT（完整 URL）对应的 host key，未设置或非法时返回 null */
+function endpointHost(): string | null {
+  const raw = process.env.CNB_API_ENDPOINT?.trim();
+  if (!raw) return null;
+  try {
+    return normalizeTokenHost(new URL(raw).host);
   } catch {
     return null;
   }
 }
+
+/**
+ * 读取已保存的 token。
+ *
+ * 指定 host 时只返回该 host 的登录态。不指定时按以下优先级选择，与 login 默认取 host 的顺序一致：
+ * 1. 当前仓库 git remote 的 host
+ * 2. CNB_API_ENDPOINT 的 host
+ * 3. 以上都没有对应登录态时，回退到最近一次保存的登录态
+ */
+export function loadToken(host?: string): TokenStore | null {
+  const { auths } = readTokenFile();
+  if (host) {
+    return auths[normalizeTokenHost(host)] ?? null;
+  }
+  const candidates = [...remoteHosts(), endpointHost()];
+  for (const candidate of candidates) {
+    const store = candidate ? auths[normalizeTokenHost(candidate)] : undefined;
+    if (store) return store;
+  }
+  const hosts = Object.keys(auths);
+  return hosts.length ? auths[hosts[hosts.length - 1]] : null;
+}

package/client/utils/refresh-token.ts

@@ -37,7 +37,8 @@ export async function refreshAccessToken(store: TokenStore): Promise<string> {
     access_token: tok.access_token,
     expires_at: nowSec + tok.expires_in,
     refresh_token: tok.refresh_token || store.refresh_token,
-    platform_url: store.platform_url,
+    // 旧版登录态可能缺少 platform_url，回填本次刷新实际使用的平台地址，保证能按 host 写回
+    platform_url: store.platform_url || platformURL,
     client_id: store.client_id,
     login_host: store.login_host,
   };

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

@@ -52,6 +52,17 @@ function parseUrlHost(url: string): string | null {
   }
 }
 
+/**
+ * 平台 host 的规范化规则（唯一来源）：去空白、小写、去掉 api. 前缀，保留端口。
+ * login 写入登录态与 logout / status / git-credential 读取登录态都以此作为 host key，
+ * 修改时两侧会同步生效。
+ * @param raw 主机名，如 cnb.cool、API.cnb.cool、example.com:8080
+ * @returns 如 cnb.cool、example.com:8080
+ */
+export function normalizeTokenHost(raw: string): string {
+  return raw.trim().toLowerCase().replace(/^api\./, '');
+}
+
 /**
  * 将 --host 传入的裸主机名规范化为平台 base URL。
  * --host 仅接受裸域名（如 cnb.cool、example.com:8080），不兼容 https:// 这类协议头；
@@ -71,10 +82,7 @@ function normalizeHost(raw: string): string | null {
   if (!/^[a-z0-9]([a-z0-9-]*[a-z0-9])?(\.[a-z0-9]([a-z0-9-]*[a-z0-9])?)*(:\d{1,5})?$/i.test(trimmed)) {
     return null;
   }
-  // 主机名大小写不敏感，统一转小写；去掉常见的 api. 前缀（api.cnb.cool -> cnb.cool）
-  // 使用 host 形式以保留自定义端口，如 example.com:8080
-  const host = trimmed.toLowerCase().replace(/^api\./, '');
-  return `https://${host}`;
+  return `https://${normalizeTokenHost(trimmed)}`;
 }
 
 /**
@@ -87,10 +95,8 @@ function normalizeEndpoint(raw: string): string | null {
   try {
     const u = new URL(raw.trim());
     if (!u.host) return null;
-    // 去掉常见的 api. 前缀，统一得到平台域名（api.cnb.cool -> cnb.cool）
     // 使用 host（而非 hostname）以保留自定义端口，如 example.com:8080
-    const host = u.host.replace(/^api\./, '');
-    return `https://${host}`;
+    return `https://${normalizeTokenHost(u.host)}`;
   } catch {
     return null;
   }

package/client/utils/resolve-token-source.ts

@@ -17,9 +17,9 @@ export type TokenSourceResult =
  * 1. WorkBuddy 沙箱环境 → 远程接口获取
  * 2. CNB_TOKEN_FOR_CODEBUDDY 环境变量
  * 3. CNB_TOKEN 环境变量
- * 4. 文件 token（~/.cnb/token）
+ * 4. 文件 token（~/.cnb/token），指定 host 时取该 host 的登录态
  */
-export async function resolveTokenSource(): Promise<TokenSourceResult> {
+export async function resolveTokenSource(host?: string): Promise<TokenSourceResult> {
   // WorkBuddy 沙箱环境，从远程接口获取 token
   if (isWorkBuddySandbox()) {
     const tokenURL = process.env[getTokenEnvKey()];
@@ -73,7 +73,7 @@ export async function resolveTokenSource(): Promise<TokenSourceResult> {
   }
 
   // 文件 token
-  const store = loadToken();
+  const store = loadToken(host);
   if (!store || !store.access_token) {
     return { source: 'none' };
   }

package/client/utils/resolve-token.ts

@@ -7,12 +7,14 @@ import { refreshAccessToken } from './refresh-token';
  * 优先使用环境变量 CNB_TOKEN（不检查过期）。
  * 否则从 ~/.cnb/token 读取，若已过期则尝试用 refresh_token 刷新；
  * 刷新失败或无 refresh_token 时终止进程并提示用户重新登录。
+ * 指定 host 时只取该 host 的登录态（git-credential 按请求域取凭据）。
  */
-export async function resolveToken(): Promise<string> {
-  const result = await resolveTokenSource();
+export async function resolveToken(host?: string): Promise<string> {
+  const result = await resolveTokenSource(host);
+  const loginCmd = host ? `cnb login --host ${host}` : 'cnb login';
 
   if (result.source === 'none') {
-    console.error(result.reason || '未找到有效的登录凭证，请先执行 `cnb login` 进行授权。');
+    console.error(result.reason || `未找到有效的登录凭证，请先执行 \`${loginCmd}\` 进行授权。`);
     process.exit(1);
   }
 
@@ -36,11 +38,11 @@ export async function resolveToken(): Promise<string> {
       return await refreshAccessToken(store);
     } catch (err: any) {
       console.error(`Token 已过期，刷新失败: ${err.message}`);
-      console.error('请重新执行 `cnb login` 进行授权。');
+      console.error(`请重新执行 \`${loginCmd}\` 进行授权。`);
       process.exit(1);
     }
   }
 
-  console.error('Token 已过期且无 refresh_token，请重新执行 `cnb login` 进行授权。');
+  console.error(`Token 已过期且无 refresh_token，请重新执行 \`${loginCmd}\` 进行授权。`);
   process.exit(1);
 }

package/client/utils/save-token.ts

@@ -1,14 +1,38 @@
 import fs from 'fs';
 import path from 'path';
 import { getTokenPath } from './token-path';
+import { hostOfStore, readTokenFile, type TokenFile } from './load-token';
+import { normalizeTokenHost } from './resolve-platform-host';
 import type { TokenStore } from './types';
 
-/** 持久化 token 到 ~/.cnb/token */
-export function saveToken(store: TokenStore): void {
+function writeTokenFile(file: TokenFile): void {
   const tokenPath = getTokenPath();
   const dir = path.dirname(tokenPath);
   if (!fs.existsSync(dir)) {
     fs.mkdirSync(dir, { recursive: true });
   }
-  fs.writeFileSync(tokenPath, JSON.stringify(store, null, 2), { mode: 0o600 });
+  fs.writeFileSync(tokenPath, JSON.stringify(file, null, 2), { mode: 0o600 });
+}
+
+/** 持久化 token 到 ~/.cnb/token 中对应 host 的条目，其他 host 的条目保持不变 */
+export function saveToken(store: TokenStore): void {
+  const host = hostOfStore(store);
+  if (!host) {
+    throw new Error('无法从登录态推断所属 host（缺少有效的 platform_url）');
+  }
+  const file = readTokenFile();
+  // 先删再加，保证最近保存的 host 排在最后，作为未指定 host 时的兜底登录态
+  delete file.auths[host];
+  file.auths[host] = store;
+  writeTokenFile(file);
+}
+
+/** 删除某个 host 的登录态，返回是否存在该条目 */
+export function removeToken(host: string): boolean {
+  const file = readTokenFile();
+  const key = normalizeTokenHost(host);
+  if (!file.auths[key]) return false;
+  delete file.auths[key];
+  writeTokenFile(file);
+  return true;
 }

package/package.json

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