@cnbcool/cnb-api-generate 2.14.2 → 2.15.0

package/client/lib/login.ts

@@ -16,13 +16,14 @@ export function registerLoginCommand(program: Command): void {
     .command('login')
     .description('通过 OAuth2 设备授权流登录 CNB，获取并保存 access_token')
     .option('--client-id <string>', 'OAuth2 client_id', process.env.OAUTH2_CLIENT_ID || 'cnb_cli')
+    .option('--host <string>', '平台域名（如 cnb.cool，可带端口如 cnb.cool:8080），优先级高于 git remote 与 CNB_API_ENDPOINT')
     .option('--debug', '打印调试信息', false)
     .helpOption('-h, --help', '显示帮助文档')
     .action(async (opts) => {
       const cfg: LoginConfig = {
         clientID: opts.clientId,
         // 从 git remote -v 动态识别平台域名，兼容不同部署（公开 / 内网 / 自定义域名）
-        platformURL: resolvePlatformUrl(),
+        platformURL: resolvePlatformUrl(opts.host),
         debug: opts.debug,
       };
 
@@ -79,7 +80,8 @@ export function registerLoginCommand(program: Command): void {
         // 持久化 token
         const nowSec = Math.floor(Date.now() / 1000);
         // 从 platformURL 推导 API 域名，例如 https://cnb.cool -> https://api.cnb.cool
-        const platformHost = new URL(cfg.platformURL).hostname;
+        // 使用 host（而非 hostname）以保留自定义端口，如 https://example.com:8080 -> https://api.example.com:8080
+        const platformHost = new URL(cfg.platformURL).host;
         const loginHost = `https://api.${platformHost}`;
         const store: TokenStore = {
           access_token: tok.access_token,

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

@@ -52,13 +52,77 @@ function parseUrlHost(url: string): string | null {
   }
 }
 
+/**
+ * 将 --host 传入的裸主机名规范化为平台 base URL。
+ * --host 仅接受裸域名（如 cnb.cool、example.com:8080），不兼容 https:// 这类协议头；
+ * 含协议头的输入视为非法，返回 null（交由后续优先级处理）。
+ * 内部统一补 https://，并去掉常见的 api. 前缀。
+ * @param raw 裸主机名，如 cnb.cool、example.com:8080、api.cnb.cool
+ * @returns 形如 https://cnb.cool，失败返回 null
+ */
+function normalizeHost(raw: string): string | null {
+  const trimmed = raw.trim();
+  if (!trimmed) return null;
+  // 含协议头（如 https://）视为非法 --host 输入
+  if (/^[a-z][a-z0-9+.-]*:\/\//i.test(trimmed)) return null;
+  // 校验为合法主机名（可带端口），拒绝路径、空格等非法字符；
+  // 否则后续 new URL(`https://${host}`) 可能抛 Invalid URL，或产生带路径的异常平台地址，
+  // 导致 login_host 与 platformURL 脱节。
+  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}`;
+}
+
+/**
+ * 将 CNB_API_ENDPOINT（完整 API URL，如 https://api.cnb.cool）规范化为平台 base URL。
+ * 与 --host 不同，这里是完整 URL，需要按 URL 解析，并去掉常见的 api. 前缀、保留自定义端口。
+ * @param raw 完整 URL，如 https://api.cnb.cool、https://api.example.com:8443
+ * @returns 形如 https://cnb.cool，失败返回 null
+ */
+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}`;
+  } catch {
+    return null;
+  }
+}
+
 /**
  * 解析当前平台的 base URL（形如 https://cnb.cool）。
- * 优先取 git remote 中的 host 拼成 https://{host}；
- * 无 remote 或解析失败时兜底返回 https://cnb.cool。
+ *
+ * 优先级（从高到低）：
+ * 1. 显式传入的参数（login 的 --host，裸域名）
+ * 2. 当前仓库 git remote -v 解析出的 host
+ * 3. 环境变量 CNB_API_ENDPOINT（完整 URL）
+ * 4. 兜底 https://cnb.cool
+ *
+ * @param paramUrl 可选的显式 --host 裸主机名
  * @returns 平台 base URL
  */
-export function resolvePlatformUrl(): string {
+export function resolvePlatformUrl(paramUrl?: string): string {
+  // 1. 显式参数（--host 裸主机名）优先
+  const fromParam = paramUrl?.trim() ? normalizeHost(paramUrl.trim()) : null;
+  if (fromParam) return fromParam;
+
+  // 2. 无参数时从 git remote 取
   const host = getGitRemoteHosts()[0];
-  return host ? `https://${host}` : 'https://cnb.cool';
+  if (host) return `https://${host}`;
+
+  // 3. 都没有则尝试从 CNB_API_ENDPOINT（完整 URL）取
+  const fromEnv = process.env.CNB_API_ENDPOINT?.trim()
+    ? normalizeEndpoint(process.env.CNB_API_ENDPOINT.trim())
+    : null;
+  if (fromEnv) return fromEnv;
+
+  // 4. 最后降级到 cnb.cool
+  return 'https://cnb.cool';
 }

package/package.json

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