paseo-plugin-zcode-provider 0.3.0 → 0.4.0

package/server/vendor/zcode/packages/shared/src/zcodeEndpoint.ts

@@ -1,40 +1,72 @@
 import type { ZCodeEnv } from "./env.js";
 
 export const DEFAULT_ZCODE_ENDPOINT_ORIGIN = "https://zcode.z.ai";
+export const TEST_ZCODE_ENDPOINT_ORIGIN = "https://zcode.z.ai";
+export const CODING_PLAN_LOCAL_WEBVIEW_ORIGIN = "http://localhost:3000";
 export const DEFAULT_BIGMODEL_API_ORIGIN = "https://bigmodel.cn";
+export const TEST_BIGMODEL_API_ORIGIN = "https://bigmodel.cn";
 export const DEFAULT_ZAI_OAUTH_ORIGIN = "https://chat.z.ai";
+export const TEST_ZAI_OAUTH_ORIGIN = "https://chat.z.ai";
 export const DEFAULT_ZAI_BUSINESS_BASE_URL = "https://api.z.ai";
+export const TEST_ZAI_BUSINESS_BASE_URL = "https://api.z.ai";
 export const DEFAULT_ZAI_OAUTH_CLIENT_ID = "client_P8X5CMWmlaRO9gyO-KSqtg";
+export const TEST_ZAI_OAUTH_CLIENT_ID = "client_P8X5CMWmlaRO9gyO-KSqtg";
+export const DEFAULT_WEB_REMOTE_CONTROL_RELAY_WS_URL = "wss://zcode.z.ai/api/v1/remote/ws2";
+export const WEB_REMOTE_CONTROL_V4_MIN_APP_VERSION = "3.4.0";
 
-// 构建仅注入公开链接；Node 调用方仍可显式传 env，避免读取另一进程的配置。
-declare const __ZCODE_ENDPOINT_ENV__: Record<string, string | undefined> | undefined;
-export function pickProductEndpointEnv(
-  env: Record<string, string | undefined>,
-): Record<string, string> {
-  const keys = [
-    "ZCODE_BASE_URL",
-    "ZCODE_ENDPOINT_ORIGIN",
-    "BIGMODEL_API_BASE_URL",
-    "ZAI_OAUTH_ORIGIN",
-    "ZAI_BUSINESS_BASE_URL",
-    "ZAI_OAUTH_CLIENT_ID",
-    "ZAI_OAUTH_APP_ID",
-  ];
-  return Object.fromEntries(
-    keys.flatMap((key) => (env[key]?.trim() ? [[key, env[key]!.trim()]] : [])),
-  );
+export interface BuildZCodeEndpointUrlsOptions {
+  appVersion?: string;
 }
-export function readProductEndpointEnv(): Record<string, string | undefined> {
-  return {
-    ...(typeof __ZCODE_ENDPOINT_ENV__ === "undefined" ? {} : __ZCODE_ENDPOINT_ENV__),
-    ...pickProductEndpointEnv(typeof process === "undefined" ? {} : process.env),
-  };
+
+function isWebRemoteControlV4AppVersion(appVersion: string | undefined): boolean {
+  if (!appVersion) {
+    return false;
+  }
+
+  // zcodeEndpoint 会被 Vite 配置在 Node 加载期直接导入，不能依赖源码态 `.js` runtime import。
+  // 这里只比较固定的稳定版门槛；高于 3.4.0 的预发布版本仍满足门槛，3.4.0 自身的预发布版本不满足。
+  const normalizedVersion = appVersion.trim().replace(/^v/i, "");
+  const match = normalizedVersion.match(
+    /^(0|[1-9]\d*)\.(0|[1-9]\d*)\.(0|[1-9]\d*)(?:-([^+]+))?(?:\+(.+))?$/,
+  );
+  if (!match) {
+    return false;
+  }
+
+  const prereleaseIdentifiers = match[4]?.split(".");
+  const hasValidPrerelease =
+    prereleaseIdentifiers === undefined ||
+    prereleaseIdentifiers.every(
+      (identifier) =>
+        /^[0-9A-Za-z-]+$/.test(identifier) &&
+        (!/^\d+$/.test(identifier) || /^(0|[1-9]\d*)$/.test(identifier)),
+    );
+  const buildIdentifiers = match[5]?.split(".");
+  const hasValidBuild =
+    buildIdentifiers === undefined ||
+    buildIdentifiers.every((identifier) => /^[0-9A-Za-z-]+$/.test(identifier));
+  if (!hasValidPrerelease || !hasValidBuild) {
+    return false;
+  }
+
+  const versionParts = [Number(match[1]), Number(match[2]), Number(match[3])];
+  const floorParts = WEB_REMOTE_CONTROL_V4_MIN_APP_VERSION.split(".").map(Number);
+  for (let index = 0; index < floorParts.length; index += 1) {
+    if (versionParts[index]! !== floorParts[index]!) {
+      return versionParts[index]! > floorParts[index]!;
+    }
+  }
+
+  return prereleaseIdentifiers === undefined;
 }
 
 export interface ZCodeEndpointUrls {
   origin: string;
   apiBaseUrl: string;
+  remoteUrl: string;
+  webRemoteCallbackUrl: string;
   webShareCallbackUrl: string;
+  relayWsUrl: string;
   zcodePlanOpenAiBaseUrl: string;
   zcodePlanAnthropicBaseUrl: string;
   zcodePlanBillingCurrentUrl: string;
@@ -46,20 +78,30 @@ export interface RuntimeZCodeEndpointEnv {
   ZCODE_ENV?: string;
   ZCODE_BASE_URL?: string;
   ZCODE_ENDPOINT_ORIGIN?: string;
+  ZCODE_TEST_BASE_URL?: string;
+  ZCODE_PRODUCTION_BASE_URL?: string;
 }
 
 export interface RuntimeBigModelApiEnv {
   [key: string]: string | undefined;
   ZCODE_ENV?: string;
   BIGMODEL_API_BASE_URL?: string;
+  BIGMODEL_TEST_API_BASE_URL?: string;
+  BIGMODEL_PRODUCTION_API_BASE_URL?: string;
 }
 
 export interface RuntimeZaiEndpointEnv {
   [key: string]: string | undefined;
   ZCODE_ENV?: string;
   ZAI_OAUTH_ORIGIN?: string;
+  ZAI_TEST_OAUTH_ORIGIN?: string;
+  ZAI_PRODUCTION_OAUTH_ORIGIN?: string;
   ZAI_BUSINESS_BASE_URL?: string;
+  ZAI_TEST_BUSINESS_BASE_URL?: string;
+  ZAI_PRODUCTION_BUSINESS_BASE_URL?: string;
   ZAI_OAUTH_CLIENT_ID?: string;
+  ZAI_TEST_OAUTH_CLIENT_ID?: string;
+  ZAI_PRODUCTION_OAUTH_CLIENT_ID?: string;
   ZAI_OAUTH_APP_ID?: string;
 }
 
@@ -84,6 +126,15 @@ function readRuntimeEnvValue(
   return value ? value : undefined;
 }
 
+function readRuntimeZCodeEnvScopedValue(
+  env: Record<string, string | undefined>,
+  zcodeEnv: ZCodeEnv,
+  testKey: string,
+  productionKey: string,
+): string | undefined {
+  return readRuntimeEnvValue(env, zcodeEnv === "production" ? productionKey : testKey);
+}
+
 export function normalizeZCodeEndpointOrigin(value: string): string {
   const trimmed = value.trim();
   if (!trimmed) {
@@ -112,7 +163,8 @@ export function isTrustedCodingPlanWebviewOrigin(
     const origin = normalizeZCodeEndpointOrigin(value);
     if (
       origin === DEFAULT_ZCODE_ENDPOINT_ORIGIN ||
-      origin === resolveRuntimeZCodeEndpointOrigin()
+      origin === TEST_ZCODE_ENDPOINT_ORIGIN ||
+      origin === CODING_PLAN_LOCAL_WEBVIEW_ORIGIN
     ) {
       return true;
     }
@@ -128,95 +180,152 @@ export function resolveZCodeEndpointOrigin(options?: {
   envBaseOrigin?: string | null;
   overrideOrigin?: string | null;
 }): string {
-  const origin = options?.overrideOrigin?.trim() || options?.envBaseOrigin?.trim();
-  return origin ? normalizeZCodeEndpointOrigin(origin) : DEFAULT_ZCODE_ENDPOINT_ORIGIN;
+  const env = options?.env ?? "test";
+  const envBaseOrigin = options?.envBaseOrigin?.trim();
+  if (env === "production") {
+    return envBaseOrigin
+      ? normalizeZCodeEndpointOrigin(envBaseOrigin)
+      : DEFAULT_ZCODE_ENDPOINT_ORIGIN;
+  }
+
+  const overrideOrigin = options?.overrideOrigin?.trim();
+  if (overrideOrigin) {
+    return normalizeZCodeEndpointOrigin(overrideOrigin);
+  }
+
+  return envBaseOrigin ? normalizeZCodeEndpointOrigin(envBaseOrigin) : TEST_ZCODE_ENDPOINT_ORIGIN;
 }
 
-export function resolveRuntimeZCodeEnv(
-  env: RuntimeZCodeEndpointEnv = readProductEndpointEnv(),
-): ZCodeEnv {
-  // 产品身份仅用于既有展示与安装标识，不参与地址解析。
+export function resolveRuntimeZCodeEnv(env: RuntimeZCodeEndpointEnv = {}): ZCodeEnv {
+  // 独立 CLI / 服务测试未注入 ZCODE_ENV 时默认生产域；桌面测试包会显式下发 ZCODE_ENV=test。
   return env.ZCODE_ENV?.trim().toLowerCase() === "test" ? "test" : "production";
 }
 
 export function resolveRuntimeZCodeEndpointOrigin(
-  env: RuntimeZCodeEndpointEnv = readProductEndpointEnv(),
+  env: RuntimeZCodeEndpointEnv = {},
   options?: { overrideOrigin?: string | null },
 ): string {
+  const zcodeEnv = resolveRuntimeZCodeEnv(env);
+  const scopedBaseOrigin =
+    zcodeEnv === "production"
+      ? readRuntimeEnvValue(env, "ZCODE_PRODUCTION_BASE_URL")
+      : readRuntimeEnvValue(env, "ZCODE_TEST_BASE_URL");
+
   return resolveZCodeEndpointOrigin({
+    env: zcodeEnv,
     envBaseOrigin:
       readRuntimeEnvValue(env, "ZCODE_BASE_URL") ??
-      readRuntimeEnvValue(env, "ZCODE_ENDPOINT_ORIGIN"),
+      readRuntimeEnvValue(env, "ZCODE_ENDPOINT_ORIGIN") ??
+      scopedBaseOrigin,
     overrideOrigin: options?.overrideOrigin,
   });
 }
 
 export function buildRuntimeZCodeEndpointUrls(
-  env: RuntimeZCodeEndpointEnv = readProductEndpointEnv(),
+  env: RuntimeZCodeEndpointEnv = {},
 ): ZCodeEndpointUrls {
   return buildZCodeEndpointUrls(resolveRuntimeZCodeEndpointOrigin(env));
 }
 
-export function buildRuntimeZCodeApiUrl(
-  env: RuntimeZCodeEndpointEnv = readProductEndpointEnv(),
-  path: string,
-): string {
+export function buildRuntimeZCodeApiUrl(env: RuntimeZCodeEndpointEnv = {}, path: string): string {
   const normalizedPath = path.startsWith("/") ? path : `/${path}`;
   return `${resolveRuntimeZCodeEndpointOrigin(env)}${normalizedPath}`;
 }
 
-export function resolveBigModelApiOrigin(
-  env: RuntimeBigModelApiEnv = readProductEndpointEnv(),
-): string {
+export function resolveWebRemoteControlRelayWsUrl(options?: {
+  endpointOrigin?: string | null | undefined;
+  overrideUrl?: string | null | undefined;
+}): string {
+  const overrideUrl = options?.overrideUrl?.trim();
+  if (overrideUrl) {
+    return overrideUrl;
+  }
+
+  const origin = new URL(
+    normalizeZCodeEndpointOrigin(options?.endpointOrigin?.trim() || DEFAULT_ZCODE_ENDPOINT_ORIGIN),
+  );
+  origin.protocol = origin.protocol === "https:" ? "wss:" : "ws:";
+  origin.pathname = "/api/v1/remote/ws2";
+  return origin.toString();
+}
+
+export function resolveBigModelApiOrigin(env: RuntimeBigModelApiEnv = {}): string {
+  const zcodeEnv = resolveRuntimeZCodeEnv(env);
+  const scopedBaseOrigin =
+    zcodeEnv === "production"
+      ? readRuntimeEnvValue(env, "BIGMODEL_PRODUCTION_API_BASE_URL")
+      : readRuntimeEnvValue(env, "BIGMODEL_TEST_API_BASE_URL");
+  const fallbackOrigin =
+    zcodeEnv === "production" ? DEFAULT_BIGMODEL_API_ORIGIN : TEST_BIGMODEL_API_ORIGIN;
+
   return normalizeZCodeEndpointOrigin(
-    readRuntimeEnvValue(env, "BIGMODEL_API_BASE_URL") ?? DEFAULT_BIGMODEL_API_ORIGIN,
+    readRuntimeEnvValue(env, "BIGMODEL_API_BASE_URL") ?? scopedBaseOrigin ?? fallbackOrigin,
   );
 }
 
-export function buildBigModelApiUrl(
-  env: RuntimeBigModelApiEnv = readProductEndpointEnv(),
-  path: string,
-): string {
+export function buildBigModelApiUrl(env: RuntimeBigModelApiEnv = {}, path: string): string {
   const normalizedPath = path.startsWith("/") ? path : `/${path}`;
   return `${resolveBigModelApiOrigin(env)}${normalizedPath}`;
 }
 
-export function buildBigModelCodingPlanPersonalManageUrl(
-  env: RuntimeBigModelApiEnv = readProductEndpointEnv(),
-): string {
-  // 管理页与业务 API 共用显式 origin，避免把已登录账号带到另一个部署。
+export function buildBigModelCodingPlanPersonalManageUrl(env: RuntimeBigModelApiEnv = {}): string {
+  // 修复原因：BigModel Coding Plan 管理页要跟随 ZCODE_ENV；
+  // 测试环境如果继续写死 bigmodel.cn，会把测试账号带到生产套餐页。
   return buildBigModelApiUrl(env, "/coding-plan/personal/overview");
 }
 
-export function buildBigModelCodingPlanTeamManageUrl(
-  env: RuntimeBigModelApiEnv = readProductEndpointEnv(),
-): string {
+export function buildBigModelCodingPlanTeamManageUrl(env: RuntimeBigModelApiEnv = {}): string {
   return buildBigModelApiUrl(env, "/coding-plan/team/plans");
 }
 
-export function resolveZaiOAuthOrigin(
-  env: RuntimeZaiEndpointEnv = readProductEndpointEnv(),
-): string {
+export function resolveZaiOAuthOrigin(env: RuntimeZaiEndpointEnv = {}): string {
+  const zcodeEnv = resolveRuntimeZCodeEnv(env);
+  const scopedOrigin = readRuntimeZCodeEnvScopedValue(
+    env,
+    zcodeEnv,
+    "ZAI_TEST_OAUTH_ORIGIN",
+    "ZAI_PRODUCTION_OAUTH_ORIGIN",
+  );
+  const fallbackOrigin =
+    zcodeEnv === "production" ? DEFAULT_ZAI_OAUTH_ORIGIN : TEST_ZAI_OAUTH_ORIGIN;
+
   return normalizeZCodeEndpointOrigin(
-    readRuntimeEnvValue(env, "ZAI_OAUTH_ORIGIN") ?? DEFAULT_ZAI_OAUTH_ORIGIN,
+    readRuntimeEnvValue(env, "ZAI_OAUTH_ORIGIN") ?? scopedOrigin ?? fallbackOrigin,
   );
 }
 
-export function resolveZaiBusinessBaseUrl(
-  env: RuntimeZaiEndpointEnv = readProductEndpointEnv(),
-): string {
+export function resolveZaiBusinessBaseUrl(env: RuntimeZaiEndpointEnv = {}): string {
+  const zcodeEnv = resolveRuntimeZCodeEnv(env);
+  const scopedOrigin = readRuntimeZCodeEnvScopedValue(
+    env,
+    zcodeEnv,
+    "ZAI_TEST_BUSINESS_BASE_URL",
+    "ZAI_PRODUCTION_BUSINESS_BASE_URL",
+  );
+  const fallbackOrigin =
+    zcodeEnv === "production" ? DEFAULT_ZAI_BUSINESS_BASE_URL : TEST_ZAI_BUSINESS_BASE_URL;
+
   return normalizeZCodeEndpointOrigin(
-    readRuntimeEnvValue(env, "ZAI_BUSINESS_BASE_URL") ?? DEFAULT_ZAI_BUSINESS_BASE_URL,
+    readRuntimeEnvValue(env, "ZAI_BUSINESS_BASE_URL") ?? scopedOrigin ?? fallbackOrigin,
   );
 }
 
-export function resolveZaiOAuthClientId(
-  env: RuntimeZaiEndpointEnv = readProductEndpointEnv(),
-): string {
+export function resolveZaiOAuthClientId(env: RuntimeZaiEndpointEnv = {}): string {
+  const zcodeEnv = resolveRuntimeZCodeEnv(env);
+  const scopedClientId = readRuntimeZCodeEnvScopedValue(
+    env,
+    zcodeEnv,
+    "ZAI_TEST_OAUTH_CLIENT_ID",
+    "ZAI_PRODUCTION_OAUTH_CLIENT_ID",
+  );
+  const fallbackClientId =
+    zcodeEnv === "production" ? DEFAULT_ZAI_OAUTH_CLIENT_ID : TEST_ZAI_OAUTH_CLIENT_ID;
+
   return (
     readRuntimeEnvValue(env, "ZAI_OAUTH_CLIENT_ID") ??
+    scopedClientId ??
     readRuntimeEnvValue(env, "ZAI_OAUTH_APP_ID") ??
-    DEFAULT_ZAI_OAUTH_CLIENT_ID
+    fallbackClientId
   );
 }
 
@@ -225,23 +334,17 @@ export function buildZaiOAuthUrl(origin: string, path: string): string {
   return `${normalizeZCodeEndpointOrigin(origin)}${normalizedPath}`;
 }
 
-export function buildRuntimeZaiOAuthUrl(
-  env: RuntimeZaiEndpointEnv = readProductEndpointEnv(),
-  path: string,
-): string {
+export function buildRuntimeZaiOAuthUrl(env: RuntimeZaiEndpointEnv = {}, path: string): string {
   return buildZaiOAuthUrl(resolveZaiOAuthOrigin(env), path);
 }
 
-export function buildRuntimeZaiBusinessUrl(
-  env: RuntimeZaiEndpointEnv = readProductEndpointEnv(),
-  path: string,
-): string {
+export function buildRuntimeZaiBusinessUrl(env: RuntimeZaiEndpointEnv = {}, path: string): string {
   const normalizedPath = path.startsWith("/") ? path : `/${path}`;
   return `${resolveZaiBusinessBaseUrl(env)}${normalizedPath}`;
 }
 
 export function resolveRuntimeProductEndpointConfig(
-  env: RuntimeProductEndpointEnv = readProductEndpointEnv(),
+  env: RuntimeProductEndpointEnv = {},
 ): RuntimeProductEndpointConfig {
   const zcodeEnv = resolveRuntimeZCodeEnv(env);
   const zcodeEndpointOrigin = resolveRuntimeZCodeEndpointOrigin(env);
@@ -257,12 +360,24 @@ export function resolveRuntimeProductEndpointConfig(
   };
 }
 
-export function buildZCodeEndpointUrls(origin: string): ZCodeEndpointUrls {
+export function buildZCodeEndpointUrls(
+  origin: string,
+  options: BuildZCodeEndpointUrlsOptions = {},
+): ZCodeEndpointUrls {
   const normalizedOrigin = normalizeZCodeEndpointOrigin(origin);
+  const parsed = new URL(normalizedOrigin);
+  const wsProtocol = parsed.protocol === "https:" ? "wss:" : "ws:";
+  const wsOrigin = `${wsProtocol}//${parsed.host}`;
+  // 无法解析或低于分界版本时保守使用 v3，避免开发版本和旧客户端进入未兼容的 v4 页面。
+  const remoteVersion = isWebRemoteControlV4AppVersion(options.appVersion) ? "v4" : "v3";
+
   return {
     origin: normalizedOrigin,
     apiBaseUrl: `${normalizedOrigin}/api/v1`,
+    remoteUrl: `${normalizedOrigin}/remote/${remoteVersion}`,
+    webRemoteCallbackUrl: `${normalizedOrigin}/remote/v4`,
     webShareCallbackUrl: `${normalizedOrigin}/cn/share/callback`,
+    relayWsUrl: `${wsOrigin}/api/v1/remote/ws2`,
     zcodePlanOpenAiBaseUrl: `${normalizedOrigin}/api/v1/zcode-plan`,
     zcodePlanAnthropicBaseUrl: `${normalizedOrigin}/api/v1/zcode-plan/anthropic`,
     zcodePlanBillingCurrentUrl: `${normalizedOrigin}/api/v1/zcode-plan/billing/current`,