ck-pi-zen-session 0.1.3 → 0.1.5

package/README.md

@@ -6,7 +6,7 @@ OpenCode Zen 免费模型同步、请求头伪装与 Session 会话自动维护
 
 ## 🌟 核心特性
 
-- ⚡ **/zen 指令一键激活**：输入一次 API Key (`/zen oc_sk_...`)，全自动验证、探测 9 款可用免费模型并注入 Pi 与 CC-Switch。
+- ⚡ **/zen 指令一键激活与动态免费模型全量探测**：输入一次 API Key (`/zen oc_sk_...`)，全自动在线探测所有带有 `free` / `zero` / `pickle` 标识的模型，全量标记 0 额度消耗 (`cost: { input: 0, output: 0 }`) 并动态热挂载进 Pi 与 CC-Switch。
 - 🔄 **官方算法深度逆向**：完整对齐 OpenCode 客户端 `Identifier.descending("ses")` 与 `Identifier.ascending("msg")` 算法，生成 30 位降序会话 ID 与升序请求 ID，支持毫秒级时间戳反解。
 - 🛡️ **全套官方级伪装请求头**：自动注入全部 7 个官方客户端对齐请求头：
   - `User-Agent`: `opencode/1.18.32 ai-sdk/provider-utils/4.0.23 runtime/bun/1.3.14`
@@ -19,12 +19,13 @@ OpenCode Zen 免费模型同步、请求头伪装与 Session 会话自动维护
 - ⏱️ **30 分钟无感前置平滑轮换**：官方后端限制免费会话时长约为 1 小时（超时直接 403 `FreeTierError`）。本插件在底层请求拦截器（`before_provider_headers`）和 Agent 循环守卫（`before_agent_start`）中设置 30 分钟主动换新阈值，兼顾会话上下文平滑与长期高频调用稳定性，确保永不触碰 1 小时硬限制。
 - 🚑 **网关异常自愈守卫 (`after_provider_response`)**：新增底层响应拦截，一旦检测到远端网关返回 401 或 403 `FreeTierError`，立即当场触发 Session 换新与后台持久化，并提示用户，彻底终结会话失效死锁。
 - 🔑 **多源凭据自适应感知**：支持优先读取 `OPENCODE_API_KEY`、`OPENCODE_ZEN_API_KEY` 环境变量，亦可直接使用 `/zen <key>` 配置，无缝兼容各类容器与终端环境。
+- 🌐 **原生代理支持解除限速与额度**：由于 OpenCode 端点受 Cloudflare WAF 保护，在 HTTP 头部伪造假 IP 无法欺骗底层 TCP 握手并会招致 WAF 封禁；如需轮换出口解除 IP 限制，直接配置标准网络代理（如 `export HTTPS_PROXY=http://127.0.0.1:7890`）即可让物理连接从新节点发起，安全解除限速与额度。
 - 🧰 **官方全套 6 大核心工具链守卫与字母序规约**：OpenCode Zen 免费端点对工具集有硬性要求（无工具直接拒绝）。插件在 `before_provider_request` 钩子中实时监控请求体：
   - 若用户处于纯文本或无工具模式，自动补齐官方完整的 6 大核心工具（`bash`, `edit`, `glob`, `grep`, `read`, `write`）与 `tool_choice: "auto"`，彻底根除 403 拦截；
   - 若请求已有工具，严格按照官方客户端规约（`localeCompare`）按函数名升序重排；
   - 自动对齐流式用量元数据 `stream_options: { include_usage: true }`。
-- 🌐 **独立供应商隔离命名空间**：注册为独立供应商 `opencode-zen-free`，彻底避开并兼容用户自带的官方登录 `opencode` / `opencode-zen` 套餐，互不干扰、平稳共存。
-- 🧠 **精准上下文与思维链适配**：严格对齐 9 款免费模型的 `contextWindow`、`maxTokens` 与 `thinkingLevelMap`（涵盖小米 MiMo、英伟达 Nemotron、Meta Muse Spark、Ling、Jev 等）。
+- 🌐 **独立供应商隔离与零干扰保障 (`isZenModelTarget`)**：严格将生命周期守卫、请求头与请求体拦截限制在 `opencode-zen-free` 命名空间内，对 `relayhub`、`deepseek`、`anthropic`、`openai`、`onerouter`、`apmix` 等其他任何模型及 Claude Code（`pi-cc-extensions`）插件 100% 保持完全静默放行，零副作用、零冲突。
+- 🧠 **精准上下文与思维链适配**：严格对齐免费模型的 `contextWindow`、`maxTokens` 与 `thinkingLevelMap`（涵盖小米 MiMo、英伟达 Nemotron、Meta Muse Spark、Ling、Jev 等）。
 - 🔌 **全自动多端持久化同步**：自动双写 Pi 本地配置（`~/.pi/agent/models.json` 和 `~/.pi/agent/auth.json`），若检测到 CC-Switch 亦无缝同步其 SQLite 数据库。
 
 ---

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ck-pi-zen-session",
-  "version": "0.1.3",
+  "version": "0.1.5",
   "description": "OpenCode Zen session auto-refresh and free-tier model catalog synchronization for the Pi coding agent.",
   "keywords": [
     "pi-package",

package/src/index.ts

@@ -19,6 +19,7 @@ import {
 } from "./models-registry.js";
 import {
 	getStoredZenApiKey,
+	getStoredZenModels,
 	getStoredZenSessionId,
 	getZenStatusInfo,
 	syncZenConfiguration,
@@ -31,11 +32,13 @@ import {
 	isZenSessionExpired,
 	isValidZenSessionId,
 } from "./session.js";
+import type { ZenModelDefinition } from "./types.js";
 
 export function registerZenProviderToPi(
 	pi: ExtensionAPI,
 	apiKey: string,
 	sessionId: string = getStoredZenSessionId() || generateZenSessionId(),
+	models: ZenModelDefinition[] = getStoredZenModels(),
 ): void {
 	try {
 		pi.registerProvider(ZEN_PROVIDER_ID, {
@@ -58,13 +61,27 @@ export function registerZenProviderToPi(
 				supportsStore: false,
 				supportsUsageInStreaming: true,
 			},
-			models: Object.values(KNOWN_ZEN_FREE_MODELS),
+			models: models && models.length > 0 ? models : Object.values(KNOWN_ZEN_FREE_MODELS),
 		});
 	} catch {
 		// 忽略重复注册
 	}
 }
 
+export function isZenModelTarget(model?: { provider?: string; id?: string }): boolean {
+	if (!model) return false;
+	// 严格隔离：仅处理 opencode-zen-free 或明确属于 opencode 官方的免费模型
+	// 绝不干涉 relayhub、deepseek、anthropic、openai 等其他任何供应商或 CC 插件
+	if (model.provider === ZEN_PROVIDER_ID) return true;
+	if (
+		model.provider === "opencode" &&
+		(model.id?.includes("-free") || model.id === "big-pickle")
+	) {
+		return true;
+	}
+	return false;
+}
+
 export default function piZenSession(pi: ExtensionAPI): void {
 	// 1. 若本地已配置 Key，启动时即刻注册独立供应商 opencode-zen-free
 	// 独立命名空间，彻底避开并兼容用户自带的官方登录 opencode / opencode-zen 套餐
@@ -75,13 +92,7 @@ export default function piZenSession(pi: ExtensionAPI): void {
 
 	// 2. 核心请求头拦截钩子：在每一次请求发往 Provider 前注入完整官方客户端对齐请求头
 	pi.on("before_provider_headers", (event, ctx) => {
-		const model = ctx.model;
-		const isZenModel =
-			model?.provider === ZEN_PROVIDER_ID ||
-			model?.id?.includes("-free") ||
-			model?.id === "big-pickle";
-
-		if (isZenModel) {
+		if (isZenModelTarget(ctx.model)) {
 			// 检查 x-opencode-session 是否过期 (>30分钟) 或缺失
 			let currentSession = event.headers["x-opencode-session"];
 			if (
@@ -108,13 +119,7 @@ export default function piZenSession(pi: ExtensionAPI): void {
 
 	// 3. 核心请求体守卫钩子：深度对齐 OpenCode 官方请求体结构与工具集
 	pi.on("before_provider_request", (event, ctx) => {
-		const model = ctx.model;
-		const isZenModel =
-			model?.provider === ZEN_PROVIDER_ID ||
-			model?.id?.includes("-free") ||
-			model?.id === "big-pickle";
-
-		if (isZenModel && event.payload && typeof event.payload === "object") {
+		if (isZenModelTarget(ctx.model) && event.payload && typeof event.payload === "object") {
 			const payload = event.payload as Record<string, unknown>;
 			const tools = payload.tools;
 			let modified = false;
@@ -148,13 +153,7 @@ export default function piZenSession(pi: ExtensionAPI): void {
 
 	// 4. 网关响应守卫：遇 401/403 会话受限或过期时，立即自愈换新 Session ID
 	pi.on("after_provider_response", (event, ctx) => {
-		const model = ctx.model;
-		const isZenModel =
-			model?.provider === ZEN_PROVIDER_ID ||
-			model?.id?.includes("-free") ||
-			model?.id === "big-pickle";
-
-		if (isZenModel && (event.status === 401 || event.status === 403)) {
+		if (isZenModelTarget(ctx.model) && (event.status === 401 || event.status === 403)) {
 			// 自动异步刷新 Session
 			syncZenConfiguration({ forceSession: true }).catch(() => {});
 			if (ctx.hasUI) {
@@ -167,8 +166,11 @@ export default function piZenSession(pi: ExtensionAPI): void {
 	});
 
 	// 5. 常驻生命周期守卫：在每次 Agent 循环执行前，自动检测 Session 有效性并无感续期
-	pi.on("before_agent_start", async () => {
+	pi.on("before_agent_start", async (_event, ctx) => {
 		try {
+			// 严格隔离：仅在当前激活/选择的是 Zen 模型时才触发轮换检测，绝不干扰其他模型或 CC 插件
+			if (!isZenModelTarget(ctx.model)) return;
+
 			const currentSession = getStoredZenSessionId();
 			// 若当前会话不存在或已使用超过 30 分钟，后台自动续期
 			if (!currentSession || isZenSessionExpired(currentSession, DEFAULT_SESSION_MAX_AGE_MS)) {
@@ -258,12 +260,13 @@ export async function handleZenCommand(
 			notify(ctx, "正在生成全新 OpenCode Zen 会话 ID 并同步...", "info");
 			const result = await syncZenConfiguration({ forceSession: true });
 			if (pi) {
-				registerZenProviderToPi(pi, result.apiKey, result.sessionId);
+				registerZenProviderToPi(pi, result.apiKey, result.sessionId, result.resolvedModels);
 			}
 			const successMsg = [
 				"✓ OpenCode Zen Session 已成功刷新！",
 				`• 全新 Session: ${result.sessionId}`,
-				`• 请求头伪装: User-Agent, x-opencode-session 已注入`,
+				`• 免费模型: 已对齐 ${result.modelsCount} 款 0 额度消耗模型`,
+				`• 请求头伪装: 7 维官方客户端签名已注入`,
 				`• 同步状态: models.json ✓ | CC-Switch ${result.ccSwitchUpdated ? "✓" : "-(未安装或无此条目)"}`,
 			].join("\n");
 			notify(ctx, successMsg, "info");
@@ -276,16 +279,17 @@ export async function handleZenCommand(
 	// 4. 输入了具体的 API Key: /zen oc_sk_... 或其它 key
 	if (rawArg.length > 0 && !["status", "refresh", "list", "models"].includes(command)) {
 		try {
-			notify(ctx, "正在验证 OpenCode Zen API Key 并探测免费模型...", "info");
+			notify(ctx, "正在验证 OpenCode Zen API Key 并探测免费与零额度模型...", "info");
 			const result = await syncZenConfiguration({ apiKey: rawArg, forceSession: true });
 			if (pi) {
-				registerZenProviderToPi(pi, result.apiKey, result.sessionId);
+				registerZenProviderToPi(pi, result.apiKey, result.sessionId, result.resolvedModels);
 			}
 			const msg = [
-				"🎉 OpenCode Zen 配置成功！免费套餐与请求头已全部就绪：",
+				"🎉 OpenCode Zen 配置成功！免费套餐与零额度模型已自动识别并全量对接：",
 				`• API Key: ${maskKey(result.apiKey)} (已验证并保存)`,
 				`• 会话 Session: ${result.sessionId} (官方降序时间戳逆向算法)`,
-				`• 发现免费模型: ${result.modelsCount} 个已自动配置 (含思维链 thinkingLevelMap 适配)`,
+				`• 自动对接免费模型 (${result.modelsCount} 款，全部标记 0 额度消耗)：`,
+				...result.models.map((id) => `  • ${id}`),
 				`• 同步路径: ${result.modelsPath}`,
 				result.ccSwitchUpdated ? "• CC-Switch: 本地数据库已同步更新" : "",
 			].filter(Boolean).join("\n");
@@ -304,13 +308,13 @@ export async function handleZenCommand(
 			notify(ctx, "正在刷新 OpenCode Zen 会话与请求头...", "info");
 			const result = await syncZenConfiguration({ forceSession: true });
 			if (pi) {
-				registerZenProviderToPi(pi, result.apiKey, result.sessionId);
+				registerZenProviderToPi(pi, result.apiKey, result.sessionId, result.resolvedModels);
 			}
 			const msg = [
 				"✨ [OpenCode Zen 已自动刷新并就绪]",
 				`• API Key: ${maskKey(result.apiKey)}`,
 				`• 全新 Session: ${result.sessionId} (有效且已持久化)`,
-				`• 免费模型: 已同步 ${result.modelsCount} 个可用免费模型`,
+				`• 自动对接免费模型: 已同步 ${result.modelsCount} 个 0 额度消耗模型`,
 				`• 请求头保护: 30分钟自动轮换 + 7维官方签名 + 6大核心工具全注入`,
 				"",
 				"常用操作：",

package/src/sync.ts

@@ -29,7 +29,7 @@ import {
 	isZenSessionExpired,
 	isValidZenSessionId,
 } from "./session.js";
-import type { ZenStatusInfo, ZenSyncOptions, ZenSyncResult } from "./types.js";
+import type { ZenModelDefinition, ZenStatusInfo, ZenSyncOptions, ZenSyncResult } from "./types.js";
 
 export function defaultModelsPath(): string {
 	return join(homedir(), ".pi", "agent", "models.json");
@@ -44,7 +44,7 @@ export function defaultCcSwitchDbPath(): string {
 }
 
 /**
- * 校验 API Key 并拉取 OpenCode Zen 在线模型清单
+ * 校验 API Key 并拉取 OpenCode Zen 在线模型清单，自动识别所有免费与零额度模型
  */
 export async function fetchZenFreeModels(apiKey: string): Promise<string[]> {
 	const res = await fetch(`${ZEN_BASE_URL}/models`, {
@@ -60,18 +60,47 @@ export async function fetchZenFreeModels(apiKey: string): Promise<string[]> {
 		throw new Error(`OpenCode Zen API 验证失败 (HTTP ${res.status}): ${text || res.statusText}`);
 	}
 
-	const data = (await res.json()) as { data?: Array<{ id: string }> };
+	const data = (await res.json()) as { data?: Array<{ id: string; [key: string]: unknown }> };
 	if (!data || !Array.isArray(data.data)) {
 		return Object.keys(KNOWN_ZEN_FREE_MODELS);
 	}
 
+	// 自动识别所有带 free / zero / pickle 标识的零额度消耗免费模型
 	const freeIds = data.data
-		.map((item) => item.id)
-		.filter((id) => id.includes("free") || id === "big-pickle" || id.includes("pickle"));
+		.map((item) => String(item.id))
+		.filter((id) => {
+			const lower = id.toLowerCase();
+			return (
+				lower.includes("free") ||
+				lower.includes("pickle") ||
+				lower.includes("zero") ||
+				lower.endsWith("-free")
+			);
+		});
 
 	return freeIds.length > 0 ? freeIds : Object.keys(KNOWN_ZEN_FREE_MODELS);
 }
 
+/**
+ * 读取当前系统已持久化配置的 OpenCode Zen 模型清单
+ */
+export function getStoredZenModels(modelsPath = defaultModelsPath()): ZenModelDefinition[] {
+	try {
+		if (existsSync(modelsPath)) {
+			const modelsDoc = JSON.parse(readFileSync(modelsPath, "utf8")) as {
+				providers?: Record<string, { models?: ZenModelDefinition[] }>;
+			};
+			const list = modelsDoc.providers?.[ZEN_PROVIDER_ID]?.models;
+			if (Array.isArray(list) && list.length > 0) {
+				return list;
+			}
+		}
+	} catch {
+		// 忽略读取错误
+	}
+	return Object.values(KNOWN_ZEN_FREE_MODELS);
+}
+
 /**
  * 读取当前系统中已配置的 OpenCode Zen API Key
  */
@@ -282,6 +311,7 @@ export async function syncZenConfiguration(options: ZenSyncOptions = {}): Promis
 		isNewSession,
 		modelsCount: resolvedModels.length,
 		models: resolvedModels.map((m) => m.id),
+		resolvedModels,
 		modelsPath,
 		ccSwitchUpdated,
 	};

package/src/types.ts

@@ -52,6 +52,7 @@ export interface ZenSyncResult {
 	isNewSession: boolean;
 	modelsCount: number;
 	models: string[];
+	resolvedModels: ZenModelDefinition[];
 	modelsPath: string;
 	ccSwitchUpdated: boolean;
 }