@forcome/ai-cli 0.1.0 → 0.1.1

package/bin/fai.js

@@ -4,14 +4,15 @@
 // 不用 isMain 守卫——本文件只作为 bin 运行（纯逻辑在 ../src/index.js）。
 
 import { spawn } from 'node:child_process';
-import { FORCOME_SERVER, resolveLhPath, resolveLhVersion } from '../src/index.js';
+import { FORCOME_SERVER, resolveLhPath, resolveLhVersion, withLoginServer } from '../src/index.js';
 
 // 默认指向 Forcome 门户（员工无需 --server）；env LOBEHUB_SERVER 可覆盖（staging 等）
 if (!process.env.LOBEHUB_SERVER) {
   process.env.LOBEHUB_SERVER = FORCOME_SERVER;
 }
 
-const args = process.argv.slice(2);
+// lh 的 login 读 --server flag（不读 env，默认 app.lobehub.com）→ 必须显式注入，否则登错服务器
+const args = withLoginServer(process.argv.slice(2), process.env.LOBEHUB_SERVER);
 
 // wrapper 自身状态（不透传给 lh）
 if (args[0] === '--forcome-version') {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@forcome/ai-cli",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "description": "Forcome AI 门户 CLI 薄壳——默认指向 ai.forcome.com，透传官方 @lobehub/cli (lh)",
   "type": "module",
   "license": "MIT",
@@ -45,5 +45,6 @@
   },
   "publishConfig": {
     "access": "public"
-  }
+  },
+  "packageManager": "pnpm@9.7.0+sha1.8f12c476122faede7aed9a2126ef551c0dc65d7e"
 }

package/src/index.js

@@ -25,3 +25,16 @@ export function resolveLhVersion() {
     return 'unknown';
   }
 }
+
+/**
+ * lh 的 `login` 命令读 `--server` 命令行参数（默认 OFFICIAL_SERVER_URL=app.lobehub.com），
+ * **不读 LOBEHUB_SERVER env**。其它命令（agent/gen/file）经 tRPC client 读 env，不受影响。
+ * 故 `fai login` 必须显式注入 `--server`，否则会登到 LobeHub 公有云而非我们的门户。
+ * 用户已带 `--server` 时不重复注入（允许覆盖到 staging 等）。
+ */
+export function withLoginServer(args, server) {
+  if (args[0] === 'login' && !args.includes('--server')) {
+    return ['login', '--server', server, ...args.slice(1)];
+  }
+  return args;
+}

package/src/index.test.js

@@ -1,6 +1,6 @@
 import { test } from 'node:test';
 import assert from 'node:assert/strict';
-import { resolveServerUrl, FORCOME_SERVER } from './index.js';
+import { resolveServerUrl, FORCOME_SERVER, withLoginServer } from './index.js';
 
 test('resolveServerUrl 默认指向 Forcome 门户', () => {
   delete process.env.LOBEHUB_SERVER;
@@ -13,3 +13,21 @@ test('resolveServerUrl 受 LOBEHUB_SERVER env 覆盖（staging 等）', () => {
   assert.equal(resolveServerUrl(), 'https://staging.forcome.com');
   delete process.env.LOBEHUB_SERVER;
 });
+
+const SRV = 'https://ai.forcome.com';
+
+test('withLoginServer: login 无 --server 时注入默认 server（避免登到 app.lobehub.com）', () => {
+  assert.deepEqual(withLoginServer(['login'], SRV), ['login', '--server', SRV]);
+});
+
+test('withLoginServer: login 已带 --server 不重复注入（允许覆盖到 staging）', () => {
+  assert.deepEqual(
+    withLoginServer(['login', '--server', 'https://staging.forcome.com'], SRV),
+    ['login', '--server', 'https://staging.forcome.com'],
+  );
+});
+
+test('withLoginServer: 非 login 命令不动（agent/gen/file 走 env）', () => {
+  assert.deepEqual(withLoginServer(['agent', 'list'], SRV), ['agent', 'list']);
+  assert.deepEqual(withLoginServer(['gen', 'text', '-m', 'newapi/gpt-5.5'], SRV), ['gen', 'text', '-m', 'newapi/gpt-5.5']);
+});