ck-pi-zen-session 0.1.2 → 0.1.4

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`
@@ -17,12 +17,15 @@ OpenCode Zen 免费模型同步、请求头伪装与 Session 会话自动维护
   - `x-session-affinity`: 与 session 严格同步
   - `X-Session-Id`: 与 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 等）。
+- 🧠 **精准上下文与思维链适配**：严格对齐免费模型的 `contextWindow`、`maxTokens` 与 `thinkingLevelMap`（涵盖小米 MiMo、英伟达 Nemotron、Meta Muse Spark、Ling、Jev 等）。
 - 🔌 **全自动多端持久化同步**：自动双写 Pi 本地配置（`~/.pi/agent/models.json` 和 `~/.pi/agent/auth.json`），若检测到 CC-Switch 亦无缝同步其 SQLite 数据库。
 
 ---
@@ -50,7 +53,7 @@ pi install git:github.com/1837620622/ck-pi-extension
 | 指令 | 说明 |
 | --- | --- |
 | `/zen <oc_sk_xxx>` | 设置/更新 API Key，自动在线验证并全量同步模型与全套请求头 |
-| `/zen` | 智能检查当前会话状态；若会话临近过期自动换新；若未配置 Key 则弹出交互输入框 |
+| `/zen` | 立即一键强制换新 Session 并全面刷新所有请求头；若未配置 Key 则弹出交互输入框 |
 | `/zen refresh` | 强制生成全新的合法降序 Session ID 并更新所有请求头与落盘配置 |
 | `/zen status` | 查看当前 API Key 掩码、活跃会话精确存活时间及模型就绪状态 |
 | `/zen list`（或 `/zen models`） | 查看当前已激活的所有免费模型列表 |

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ck-pi-zen-session",
-  "version": "0.1.2",
+  "version": "0.1.4",
   "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,7 +61,7 @@ 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 {
 		// 忽略重复注册
@@ -146,7 +149,27 @@ export default function piZenSession(pi: ExtensionAPI): void {
 		}
 	});
 
-	// 4. 常驻生命周期守卫：在每次 Agent 循环执行前，自动检测 Session 有效性并无感续期
+	// 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)) {
+			// 自动异步刷新 Session
+			syncZenConfiguration({ forceSession: true }).catch(() => {});
+			if (ctx.hasUI) {
+				ctx.ui.notify(
+					`检测到 OpenCode Zen 会话受限 (HTTP ${event.status})，已自动轮换新 Session ID。`,
+					"warning",
+				);
+			}
+		}
+	});
+
+	// 5. 常驻生命周期守卫：在每次 Agent 循环执行前，自动检测 Session 有效性并无感续期
 	pi.on("before_agent_start", async () => {
 		try {
 			const currentSession = getStoredZenSessionId();
@@ -162,7 +185,7 @@ export default function piZenSession(pi: ExtensionAPI): void {
 		}
 	});
 
-	// 4. 注册 /zen 指令
+	// 6. 注册 /zen 指令
 	pi.registerCommand("zen", {
 		description: "OpenCode Zen 免费模型与 Session 会话自动维护 (/zen [key|refresh|status|list])",
 		getArgumentCompletions: (prefix: string) => {
@@ -238,12 +261,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");
@@ -256,16 +280,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");
@@ -279,23 +304,24 @@ export async function handleZenCommand(
 	// 5. 无参调用: /zen
 	const status = getZenStatusInfo();
 	if (status.hasKey) {
-		// 已有 Key 时，自动检测并刷新 Session
+		// 已有 Key 时，自动强制刷新 Session 并重新注册热载
 		try {
-			notify(ctx, "正在检测并刷新 OpenCode Zen 会话...", "info");
-			const result = await syncZenConfiguration({ forceSession: status.sessionExpired });
+			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 已连接就绪]",
+				"✨ [OpenCode Zen 已自动刷新并就绪]",
 				`• API Key: ${maskKey(result.apiKey)}`,
-				`• 活跃 Session: ${result.sessionId} (${result.isNewSession ? "已换新" : "仍有效"})`,
-				`• 免费模型: ${result.modelsCount} 个可用`,
+				`• 全新 Session: ${result.sessionId} (有效且已持久化)`,
+				`• 自动对接免费模型: 已同步 ${result.modelsCount} 个 0 额度消耗模型`,
+				`• 请求头保护: 30分钟自动轮换 + 7维官方签名 + 6大核心工具全注入`,
 				"",
 				"常用操作：",
 				"  /zen <key>   - 更换 API Key",
-				"  /zen refresh - 换新会话 ID",
 				"  /zen status  - 详情状态",
+				"  /zen list    - 查看所有可用免费模型",
 			].join("\n");
 			notify(ctx, msg, "info");
 		} catch (error) {
@@ -312,7 +338,7 @@ export async function handleZenCommand(
 				"oc_sk_",
 			);
 			if (enteredKey && enteredKey.trim()) {
-				await handleZenCommand(enteredKey.trim(), ctx);
+				await handleZenCommand(enteredKey.trim(), ctx, pi);
 				return;
 			}
 			notify(ctx, "已取消输入。如需配置请使用 /zen <oc_sk_xxx>", "info");

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,22 +60,60 @@ 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
  */
 export function getStoredZenApiKey(authPath = defaultAuthPath(), modelsPath = defaultModelsPath()): string | null {
+	// 优先读取环境变量，支持免配置直接启动
+	const envKey =
+		process.env.OPENCODE_API_KEY ||
+		process.env.OPENCODE_ZEN_API_KEY ||
+		process.env.ZEN_API_KEY;
+	if (envKey && envKey.trim()) {
+		return envKey.trim();
+	}
+
 	try {
 		if (existsSync(authPath)) {
 			const auth = JSON.parse(readFileSync(authPath, "utf8")) as Record<string, { key?: string }>;
@@ -273,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;
 }