@yalieny/pi-better-cost-display-footer 1.1.0 → 2.0.0

package/README.md

@@ -1,12 +1,12 @@
 # @yalieny/pi-better-cost-display-footer
 
-pi 扩展:峰谷动态计价 + footer 增强。
+pi 扩展:脚本化多段计价动态列表 + footer 增强。
 
-- **动态计价**:按配置时区与峰谷窗口,在 `session_start` 和每次发送消息时重注册 provider 的 `cost`(仅影响 pi 本地用量统计)。
-- **档位标签**:价格后紧跟峰值/谷值标签(如 `¥0.047(梁文峰)`),文案与颜色可配置。
+- **多段动态计价**:每个 provider 一个 `tierFn` 脚本函数(JS 表达式字符串)判定当前档位,**档位是任意字符串 id**(`peak` / `flat` / `valley` / `sharp`…),不绑定峰谷语义;在 `session_start` 和每次发送消息时重注册 provider 的 `cost`(仅影响 pi 本地用量统计)。
+- **档位标签**:价格后紧跟当前档位标签(如 `¥0.047(梁文峰)`),按档位 id 配置文案与颜色;未配置标签的档位不显示。
 - **CH 精度**:footer 缓存命中率小数位可配(默认 1 位)。
 - **货币符号**:按 provider 配置计费金额符号(默认 `$`)。
-- **配置升级**:旧版顶层计价配置首次启动自动迁移到 provider 分层格式。
+- **配置升级**:旧版顶层计价配置与 `peakWindows` 首次启动自动迁移为 provider 分层 + `tierFn` 格式(语义不变)。
 
 ## 安装
 
@@ -26,12 +26,10 @@ pi install npm:@yalieny/pi-better-cost-display-footer
   "providers": {
     "deepseek": {
       "timezone": "Asia/Shanghai",
-      "effectiveFrom": "2026-08-17T00:00:00+08:00",
+      "effectiveFrom": "2026-08-23T00:00:00+08:00",
       "currencySymbol": "¥",
-      "peakWindows": [
-        { "start": "09:00", "end": "12:00" },
-        { "start": "14:00", "end": "18:00" }
-      ],
+      "tierFn": "(ctx) => ctx.isWeekend() ? 'offPeak' : (ctx.inWindow('09:00','12:00') || ctx.inWindow('14:00','18:00')) ? 'peak' : 'offPeak'",
+      "defaultTier": "offPeak",
       "labels": {
         "peak": { "text": "（梁文峰）", "color": "error" },
         "offPeak": { "text": "（梁文谷）", "color": "success" }
@@ -39,7 +37,7 @@ pi install npm:@yalieny/pi-better-cost-display-footer
       "models": {
         "deepseek-v4-flash": {
           "offPeak": { "input": 1.5, "output": 4.5, "cacheRead": 0.05, "cacheWrite": 0 },
-          "peak": { "input": 3.0, "output": 9.0, "cacheRead": 0.1, "cacheWrite": 0 }
+          "peak":    { "input": 3.0, "output": 9.0, "cacheRead": 0.1, "cacheWrite": 0 }
         }
       }
     }
@@ -47,7 +45,37 @@ pi install npm:@yalieny/pi-better-cost-display-footer
 }
 ```
 
-价格单位:每百万 tokens(元),与 pi 的 `calculateCost` 约定一致。窗口含起点不含终点。`effectiveFrom`(含)之前不做任何事。
+价格单位:每百万 tokens(元),与 pi 的 `calculateCost` 约定一致。`effectiveFrom`(含)之前不做任何事。
+
+### 档位判定(tierFn)
+
+`tierFn` 是 JS 函数表达式字符串(编译执行),返回**任意档位 id**(字符串)。上下文 `ctx` 均为配置时区本地值:
+
+| 字段/方法 | 说明 |
+| --- | --- |
+| `ctx.now` | 当前时刻 `Date` |
+| `ctx.timezone` | 计费时区 |
+| `ctx.weekday` | 0=周日 … 6=周六 |
+| `ctx.hour` / `ctx.minute` | 时区本地时钟 |
+| `ctx.dateStr` | `YYYY-MM-DD`(如 `'2026-10-01'`,可做日期区间比较) |
+| `ctx.isWeekend()` | 周六/周日 |
+| `ctx.inWindow(start, end)` | `HH:mm` 窗口命中,含起点不含终点 |
+| `ctx.afterISO(iso)` / `ctx.beforeISO(iso)` | ISO 日期比较(节假日窗口、调价日等) |
+
+任一声场(语法错误、执行异常、返回非字符串、未配置 `tierFn`)→ 回退 `defaultTier`(默认 `"offPeak"`),绝不崩溃。旧字段 `peakWindows` 首次加载自动迁移为等价 `tierFn` 并写回配置文件(语义不变)。
+
+示例(档位数不限):
+
+- 工作日峰谷 + 周末全天低谷(DeepSeek 2026-08-23 起官方新规):
+  `"(ctx) => ctx.isWeekend() ? 'offPeak' : (ctx.inWindow('09:00','12:00') || ctx.inWindow('14:00','18:00')) ? 'peak' : 'offPeak'"`
+- 三段计价(尖峰/高峰/平段):
+  `"(ctx) => ctx.inWindow('11:00','12:00') ? 'sharp' : ctx.inWindow('09:00','12:00') ? 'peak' : 'valley'"`
+  对应每个模型三方价格 `"sharp": {...}, "peak": {...}, "valley": {...}`,`labels` 同样按档位 id 添加。
+- 单一平价(官方取消峰谷):`"(ctx) => 'offPeak'"`
+- 国庆节假日全天低谷:
+  `"(ctx) => ctx.dateStr >= '2026-10-01' && ctx.dateStr <= '2026-10-07' ? 'offPeak' : (ctx.inWindow('09:00','12:00') ? 'peak' : 'offPeak')"`
+
+> `tierFn` 为本地执行脚本(`new Function` 编译),仅应来自受信任的配置文件。
 
 ## 一键更新官方定价
 
@@ -55,10 +83,17 @@ pi install npm:@yalieny/pi-better-cost-display-footer
 /deepseek-pricing-update
 ```
 
-该命令把抓取官方价目页、解析价格、更新配置文件的完整任务交给 agent 执行(自然语言指令,无需手动维护价格表)。命令会告知 agent 配置文件位置与 JSON 格式;agent 抓取 [DeepSeek 官方价目页](https://api-docs.deepseek.com/zh-cn/quick_start/pricing) 后更新 `providers.deepseek.models` 下各模型的 offPeak/peak 价格,其余字段不动。
+该命令把抓取官方价目页、解析价格、更新配置文件的完整任务交给 agent 执行(自然语言指令,无需手动维护价格表)。命令会告知 agent 配置文件位置与 JSON 格式;agent 抓取 [DeepSeek 官方价目页](https://api-docs.deepseek.com/zh-cn/quick_start/pricing) 后更新 `providers.deepseek.models` 下各模型的分档价格,其余字段不动。
+
+## 测试
+
+```bash
+npm install && npm test
+```
 
 ## 说明
 
-- 顶层仅配置全局 `cacheHitRatePrecision`;`timezone`/`effectiveFrom`/`currencySymbol`/`peakWindows`/`labels`/`models` 均配置在对应 provider 下。
+- 顶层仅配置全局 `cacheHitRatePrecision`;`timezone`/`effectiveFrom`/`currencySymbol`/`tierFn`/`defaultTier`/`labels`/`models` 均配置在对应 provider 下。
 - provider 必须在 `models.json` 中有定义;未列出的模型保持原价。
+- 模型缺少当前档位价格时保持原价并打日志。
 - cost 仅影响 pi 本地用量统计,不影响 API 实际账单。

package/extensions/pi-better-cost-display-footer.ts

@@ -1,56 +1,44 @@
-// pi-better-cost-display-footer — 峰谷动态计价 + footer 增强扩展（原 pi-dynamic-cost / pi-better-footer）
+// pi-better-cost-display-footer — 脚本化多段计价 + footer 增强扩展（原 pi-dynamic-cost / pi-better-footer）
 // namespace: @YalienY
 //
 // 配置驱动：读取 <configDir>/pi-better-cost-display-footer.json（configDir 默认 ~/.pi/agent，
 // 可用 PI_CODING_AGENT_DIR 覆盖）。配置文件不存在时自动生成默认配置（内置 DeepSeek 官方价目）落盘。
-// 在 session_start 和每次发送消息（input 事件）时，
-// 按各 provider 配置的时区与峰谷窗口计算当前档位并重注册其 cost。
+// 在 session_start 和每次发送消息（input 事件）时，按各 provider 的 tierFn 脚本函数计算当前档位并重注册其 cost。
 // 跨时段边界的请求按发送时刻计费。effectiveFrom（含）之前不做任何事。
 //
+// 档位 = 任意字符串 id（如 "peak" / "flat" / "valley" / "sharp"），不绑定峰值/谷值语义：
+// {
+//   "timezone": "Asia/Shanghai",                  // 计费时区，默认 Asia/Shanghai
+//   "effectiveFrom": "2026-08-17T00:00:00+08:00", // 可选，生效时间（含）
+//   "tierFn": "(ctx) => ctx.isWeekend() ? 'offPeak' : (ctx.inWindow('09:00','12:00') || ctx.inWindow('14:00','18:00')) ? 'peak' : 'offPeak'",
+//   "defaultTier": "offPeak",                     // 可选，tierFn 缺失/语法错误/异常/返回非字符串时的兜底档位
+//   "labels": { "peak": { "text": "（梁文峰）", "color": "error" } },  // 可选，各档位 footer 标签
+//   "currencySymbol": "¥"
+// }
+// tierFn 上下文 ctx（均为配置时区本地值）：
+//   now / timezone / weekday（0=周日…6=周六）/ hour / minute / dateStr（YYYY-MM-DD）
+//   isWeekend()            周六/周日
+//   inWindow(start, end)   HH:mm 窗口命中，含起点不含终点
+//   afterISO(iso) / beforeISO(iso)   ISO 日期比较（节假日、调价日等）
+// 价格按 models.<modelId>.<tierId> { input, output, cacheRead, cacheWrite } 配置，档位数不限。
+// 旧字段 peakWindows 首次读取时自动迁移为等价 tierFn 并写回配置文件（语义不变，写回失败仍走兜底逻辑）；
+// 旧"offPeak/peak 两段"结构本身即是多段特例，无需迁移。
+//
 // 生效范围：
 // - 动态计价与档位标签：所有在 providers 中配置了计价模型的 provider 生效
 //   （不再限定 deepseek 官方，其余 provider 同样适用）。
 // - 自定义 footer 与 CH 精度：所有 provider 适用；未配置模型仅修改 CH，其余保持内置默认。
 //
-// 配置结构（pi-better-cost-display-footer.json）：
-// {
-//   "cacheHitRatePrecision": 3,                       // 全局：footer CH 命中率小数位，默认 1
-//   "providers": {                                     // provider 特定配置
-//     "deepseek": {
-//       "timezone": "Asia/Shanghai",                  // 计费时区，默认 Asia/Shanghai
-//       "effectiveFrom": "2026-08-17T00:00:00+08:00", // 可选，生效时间（含）
-//       "currencySymbol": "¥",                        // 可选，计费金额货币符号，默认 $
-//       "peakWindows": [                               // 高峰窗口，含起点不含终点
-//         { "start": "09:00", "end": "12:00" },
-//         { "start": "14:00", "end": "18:00" }
-//       ],
-//       "labels": {                                    // 可选，档位标签文案与颜色
-//         "peak":    { "text": "（梁文峰）", "color": "error" },
-//         "offPeak": { "text": "（梁文谷）", "color": "success" }
-//       },
-//       "models": {
-//         "<modelId>": {
-//           "offPeak": { "input": 1.5, "output": 4.5, "cacheRead": 0.05, "cacheWrite": 0 },
-//           "peak":    { "input": 3.0, "output": 9.0, "cacheRead": 0.1, "cacheWrite": 0 }
-//         }
-//       }
-//     },
-//     "agent-plan": { "models": { ... } }              // 同样生效
-//   }
-// }
-// 价格单位：每百万 tokens（元），与 pi 的 calculateCost 约定一致（rate/1e6 × tokens），
-// 即官方价格表原值直接填入：如输入谷值 1.5 元/M → "input": 1.5。
-//
 // 计费显示：用 ctx.ui.setFooter 替换内置 footer（升级安全，扩展代码不随 pi 更新）。
 // 复刻内置 footer 渲染（footer.js），并加两处增强：
-//   1. 激活模型有动态计价配置时，价格后紧跟档位标签：
-//      峰值 →（梁文峰）默认红色，谷值 →（梁文谷）默认绿色，如 ¥0.047（梁文峰）。
-//      文案与颜色可由 labels 配置覆盖（color 支持主题色名或 #rrggbb）。
+//   1. 激活模型有动态计价配置时，价格后紧跟档位标签（labels 按档位 id 查，未配置则不显示标签）：
+//      如 ¥0.047（梁文峰）。color 支持主题色名或 #rrggbb。
 //   2. CH 缓存命中率小数位由 cacheHitRatePrecision 控制（内置硬编码 1 位），所有 provider 适用。
 //   3. 计费金额符号由各 provider 的 currencySymbol 控制（内置硬编码 $）。
 //
 // 注意：
-// - 旧版顶层计价字段首次读取时自动下沉到各 provider 并写回配置文件。
+// - 旧版顶层计价字段与 provider 级 peakWindows 首次读取时自动迁移并写回配置文件。
+// - tierFn 用 new Function 编译执行（本地受信任配置，无沙箱；语法错误启动时即报错并回退 defaultTier）。
 // - provider 必须在 models.json 中有定义：扩展读取其完整模型定义，仅替换 cost 后整体重注册
 //   （registerProvider 的 models 是整体替换语义）。未列出的模型保持原价。
 // - cost 仅影响 pi 本地用量统计，不影响 API 实际账单。
@@ -74,12 +62,83 @@ import type {
 const CONFIG_DIR = process.env.PI_CODING_AGENT_DIR || join(homedir(), ".pi", "agent");
 const CONFIG_FILE = join(CONFIG_DIR, "pi-better-cost-display-footer.json");
 
-/** 档位标签默认文案/颜色（labels 未配置时兜底） */
-const DEFAULT_LABELS: Record<Tier, TierLabel> = {
+/** 档位 id：任意字符串。内置默认仅用到 "peak"/"offPeak"，但不绑定该语义 */
+type Tier = string;
+
+interface TierRates {
+  input: number;
+  output: number;
+  cacheRead: number;
+  cacheWrite: number;
+}
+
+interface TierLabel {
+  text: string;
+  /** ThemeColor 主题色名（如 "error"）或 #rrggbb 十六进制 */
+  color: string;
+}
+
+/** 档位标签默认文案/颜色（labels 未配置时兜底；仅对旧 peak/offPeak 生效） */
+const DEFAULT_LABELS: Record<string, TierLabel> = {
   peak: { text: "（梁文峰）", color: "error" },
   offPeak: { text: "（梁文谷）", color: "success" },
 };
 
+/** tierFn 脚本函数运行上下文（全部为配置时区本地值） */
+interface TierCtx {
+  now: Date;
+  timezone: string;
+  /** 0=周日 … 6=周六 */
+  weekday: number;
+  hour: number;
+  minute: number;
+  /** YYYY-MM-DD */
+  dateStr: string;
+  isWeekend(): boolean;
+  /** HH:mm 窗口命中，含起点不含终点 */
+  inWindow(start: string, end: string): boolean;
+  afterISO(iso: string): boolean;
+  beforeISO(iso: string): boolean;
+}
+
+type TierFn = (ctx: TierCtx) => unknown;
+
+interface FooterLabel {
+  text: string;
+  color: string;
+}
+
+interface ProviderPricing {
+  timezone?: string;
+  effectiveFrom?: string;
+  currencySymbol?: string;
+  /** 档位判定脚本函数（JS 函数表达式字符串），返回任意档位 id */
+  tierFn?: string;
+  /** tierFn 缺失/语法错误/执行异常/返回非字符串时的兜底档位，默认 "offPeak" */
+  defaultTier?: Tier;
+  /** 旧格式峰值窗口（含起点不含终点）；首次加载自动迁移为 tierFn 并写回 */
+  peakWindows?: { start: string; end: string }[];
+  /** 各档位 footer 标签（按档位 id 查找），未配置的档位不显示标签 */
+  labels?: Record<string, TierLabel>;
+  /** 每个模型的多段价格：模型 id → 档位 id → 价格 */
+  models: Record<string, Record<Tier, TierRates>>;
+}
+
+interface DynamicConfig {
+  cacheHitRatePrecision?: number;
+  providers?: Record<string, ProviderPricing>;
+}
+
+interface LegacyDynamicConfig extends DynamicConfig {
+  timezone?: string;
+  effectiveFrom?: string;
+  currencySymbol?: string;
+  peakWindows?: { start: string; end: string }[];
+  labels?: { peak?: TierLabel; offPeak?: TierLabel };
+  tierFn?: string;
+  defaultTier?: Tier;
+}
+
 /** 内置默认配置：配置文件缺失/损坏时的兜底，开箱即用（DeepSeek 官方价目，2026-08 抓取） */
 const DEFAULT_CONFIG: DynamicConfig = {
   cacheHitRatePrecision: 1,
@@ -87,10 +146,9 @@ const DEFAULT_CONFIG: DynamicConfig = {
     deepseek: {
       timezone: "Asia/Shanghai",
       currencySymbol: "¥",
-      peakWindows: [
-        { start: "09:00", end: "12:00" },
-        { start: "14:00", end: "18:00" },
-      ],
+      // 周末全天低谷（DeepSeek 2026-08-23 起官方新规）+ 工作日峰谷窗口
+      tierFn: "(ctx) => ctx.isWeekend() ? 'offPeak' : (ctx.inWindow('09:00','12:00') || ctx.inWindow('14:00','18:00')) ? 'peak' : 'offPeak'",
+      defaultTier: "offPeak",
       labels: DEFAULT_LABELS,
       models: {
         "deepseek-v4-flash": {
@@ -106,6 +164,14 @@ const DEFAULT_CONFIG: DynamicConfig = {
   },
 };
 
+let lastAppliedKey: string | null = null;
+/** 最近一次事件的 ctx，footer 闭包读取；随 input/model_select 事件刷新 */
+let activeCtx: ExtensionContext | null = null;
+/** 当前生效配置，apply() 时刷新，footer 每帧直接读，避免频繁读盘 */
+let currentCfg: DynamicConfig | null = null;
+/** 当前档位标签（null = 不显示），apply() 时刷新 */
+let footerLabel: FooterLabel | null = null;
+
 /** 深合并：文件配置覆盖内置默认（全局 + provider + model 三级），部分配置文件也能生效 */
 function mergeConfig(base: DynamicConfig, override: DynamicConfig): DynamicConfig {
   const merged: DynamicConfig = { ...base, ...override };
@@ -123,20 +189,22 @@ function mergeConfig(base: DynamicConfig, override: DynamicConfig): DynamicConfi
 
 /** 旧格式字段；命中任一字段即执行一次磁盘配置升级。 */
 function isLegacyConfig(fileCfg: LegacyDynamicConfig): boolean {
-  return ["timezone", "effectiveFrom", "currencySymbol", "peakWindows", "labels"].some((key) =>
+  return ["timezone", "effectiveFrom", "currencySymbol", "peakWindows", "labels", "tierFn", "defaultTier"].some((key) =>
     Object.prototype.hasOwnProperty.call(fileCfg, key),
   );
 }
 
 /** 把旧格式顶层 provider 配置下沉到每个 provider。 */
 function normalizeConfig(fileCfg: LegacyDynamicConfig): DynamicConfig {
-  const { timezone, effectiveFrom, currencySymbol, peakWindows, labels, providers, cacheHitRatePrecision } = fileCfg;
+  const { timezone, effectiveFrom, currencySymbol, peakWindows, labels, tierFn, defaultTier, providers, cacheHitRatePrecision } = fileCfg;
   const legacyProviderDefaults: Partial<ProviderPricing> = {};
   if (timezone !== undefined) legacyProviderDefaults.timezone = timezone;
   if (effectiveFrom !== undefined) legacyProviderDefaults.effectiveFrom = effectiveFrom;
   if (currencySymbol !== undefined) legacyProviderDefaults.currencySymbol = currencySymbol;
   if (peakWindows !== undefined) legacyProviderDefaults.peakWindows = peakWindows;
-  if (labels !== undefined) legacyProviderDefaults.labels = labels;
+  if (labels !== undefined) legacyProviderDefaults.labels = labels as Record<string, TierLabel>;
+  if (tierFn !== undefined) legacyProviderDefaults.tierFn = tierFn;
+  if (defaultTier !== undefined) legacyProviderDefaults.defaultTier = defaultTier;
   return {
     cacheHitRatePrecision,
     providers: Object.fromEntries(
@@ -148,56 +216,6 @@ function normalizeConfig(fileCfg: LegacyDynamicConfig): DynamicConfig {
   };
 }
 
-interface TierRates {
-  input: number;
-  output: number;
-  cacheRead: number;
-  cacheWrite: number;
-}
-
-interface TierLabel {
-  text: string;
-  /** ThemeColor 主题色名（如 "error"）或 #rrggbb 十六进制 */
-  color: string;
-}
-
-interface ProviderPricing {
-  timezone?: string;
-  effectiveFrom?: string;
-  currencySymbol?: string;
-  peakWindows?: { start: string; end: string }[];
-  labels?: { peak?: TierLabel; offPeak?: TierLabel };
-  models: Record<string, { offPeak: TierRates; peak: TierRates }>;
-}
-
-interface DynamicConfig {
-  cacheHitRatePrecision?: number;
-  providers?: Record<string, ProviderPricing>;
-}
-
-interface LegacyDynamicConfig extends DynamicConfig {
-  timezone?: string;
-  effectiveFrom?: string;
-  currencySymbol?: string;
-  peakWindows?: { start: string; end: string }[];
-  labels?: { peak?: TierLabel; offPeak?: TierLabel };
-}
-
-type Tier = "peak" | "offPeak";
-
-interface FooterLabel {
-  text: string;
-  color: string;
-}
-
-let lastAppliedKey: string | null = null;
-/** 最近一次事件的 ctx，footer 闭包读取；随 input/model_select 事件刷新 */
-let activeCtx: ExtensionContext | null = null;
-/** 当前生效配置，apply() 时刷新，footer 每帧直接读，避免频繁读盘 */
-let currentCfg: DynamicConfig | null = null;
-/** 当前档位标签（null = 不显示），apply() 时刷新 */
-let footerLabel: FooterLabel | null = null;
-
 export function loadDynamicConfig(): DynamicConfig | null {
   try {
     if (!existsSync(CONFIG_FILE)) {
@@ -209,10 +227,11 @@ export function loadDynamicConfig(): DynamicConfig | null {
     }
     const fileCfg = JSON.parse(readFileSync(CONFIG_FILE, "utf8")) as LegacyDynamicConfig;
     const normalizedCfg = normalizeConfig(fileCfg);
-    if (isLegacyConfig(fileCfg)) {
+    const migrated = migratePeakWindows(normalizedCfg);
+    if (isLegacyConfig(fileCfg) || migrated) {
       try {
         writeFileSync(CONFIG_FILE, JSON.stringify(normalizedCfg, null, 2) + "\n");
-        console.log(`[pi-better-cost-display-footer] 旧配置已自动升级为 provider 分层格式 ${CONFIG_FILE}`);
+        console.log(`[pi-better-cost-display-footer] 旧配置已自动升级（provider 分层 + peakWindows → tierFn）${CONFIG_FILE}`);
       } catch (err) {
         console.error(
           `[pi-better-cost-display-footer] 旧配置升级写入失败：${err instanceof Error ? err.message : String(err)}`,
@@ -228,17 +247,35 @@ export function loadDynamicConfig(): DynamicConfig | null {
   }
 }
 
-/** 当前时刻（配置时区）的 { hour, minute }，date 参数仅供测试 */
-export function nowClock(timezone: string, date: Date = new Date()): { hour: number; minute: number } {
+/** 构建 tierFn 运行上下文（全部为配置时区本地值） */
+export function buildTierCtx(timezone: string, date: Date = new Date()): TierCtx {
   const parts = new Intl.DateTimeFormat("en-US", {
     timeZone: timezone,
+    weekday: "short",
+    year: "numeric",
+    month: "2-digit",
+    day: "2-digit",
     hour: "2-digit",
     minute: "2-digit",
     hourCycle: "h23",
   }).formatToParts(date);
-  const get = (t: Intl.DateTimeFormatPartTypes): string =>
-    parts.find((p) => p.type === t)?.value ?? "00";
-  return { hour: Number(get("hour")), minute: Number(get("minute")) };
+  const get = (t: string): string => parts.find((p) => p.type === t)?.value ?? "";
+  const weekdayMap: Record<string, number> = { Sun: 0, Mon: 1, Tue: 2, Wed: 3, Thu: 4, Fri: 5, Sat: 6 };
+  const weekday = weekdayMap[get("weekday")] ?? 0;
+  const hour = Number(get("hour"));
+  const minute = Number(get("minute"));
+  return {
+    now: date,
+    timezone,
+    weekday,
+    hour,
+    minute,
+    dateStr: `${get("year")}-${get("month")}-${get("day")}`,
+    isWeekend: () => weekday === 0 || weekday === 6,
+    inWindow: (start, end) => inWindow({ hour, minute }, { start, end }),
+    afterISO: (iso) => date.getTime() >= Date.parse(iso),
+    beforeISO: (iso) => date.getTime() < Date.parse(iso),
+  };
 }
 
 /** 窗口含起点不含终点，如 { start: "09:00", end: "12:00" } */
@@ -252,13 +289,77 @@ export function inWindow(
   return t >= sh * 60 + sm && t < eh * 60 + em;
 }
 
-/** 命中任一高峰窗口返回 "peak"，否则 "offPeak" */
-export function currentTier(
-  cfg: Pick<ProviderPricing, "peakWindows">,
-  clock: { hour: number; minute: number },
-): Tier {
+/** tierFn 编译缓存：源串 → 函数（null = 编译失败，仅报错一次） */
+const tierFnCache = new Map<string, TierFn | null>();
+
+function compileTierFn(src: string): TierFn | null {
+  try {
+    const fn = new Function(`return (${src})`)();
+    if (typeof fn === "function") return fn as TierFn;
+  } catch (err) {
+    console.error(
+      `[pi-better-cost-display-footer] tierFn 语法错误: ${err instanceof Error ? err.message : String(err)}；tierFn: ${src.slice(0, 80)}`,
+    );
+    return null;
+  }
+  console.error(`[pi-better-cost-display-footer] tierFn 不是函数表达式: ${src.slice(0, 80)}`);
+  return null;
+}
+
+function getTierFn(src: string): TierFn | null {
+  let fn = tierFnCache.get(src);
+  if (fn === undefined) {
+    fn = compileTierFn(src);
+    tierFnCache.set(src, fn);
+  }
+  return fn;
+}
+
+/** 把旧 peakWindows 转成等价 tierFn 源码（语义不变，返回 "peak"/"offPeak"） */
+export function windowsToTierFn(windows: { start: string; end: string }[]): string {
+  if (!windows.length) return "(ctx) => 'offPeak'";
+  const conds = windows.map((w) => `ctx.inWindow('${w.start}','${w.end}')`);
+  return `(ctx) => ${conds.join(" || ")} ? 'peak' : 'offPeak'`;
+}
+
+/** 旧配置（provider 级 peakWindows）→ tierFn；返回是否有变更 */
+export function migratePeakWindows(cfg: DynamicConfig): boolean {
+  let changed = false;
+  for (const provider of Object.values(cfg.providers || {})) {
+    if (provider.tierFn || !provider.peakWindows?.length) continue;
+    provider.tierFn = windowsToTierFn(provider.peakWindows);
+    delete provider.peakWindows;
+    changed = true;
+  }
+  return changed;
+}
+
+/**
+ * 计算当前档位（任意字符串 id）。优先级：tierFn 脚本 → 旧 peakWindows 兜底 → defaultTier。
+ * tierFn 抛出异常或返回非字符串（空串也算）时回退 defaultTier（默认 "offPeak"）。
+ */
+export function currentTier(cfg: ProviderPricing, date: Date = new Date()): Tier {
+  const defaultTier: Tier = typeof cfg.defaultTier === "string" && cfg.defaultTier ? cfg.defaultTier : "offPeak";
+  if (cfg.tierFn) {
+    const fn = getTierFn(cfg.tierFn);
+    if (!fn) return defaultTier;
+    try {
+      const result = fn(buildTierCtx(cfg.timezone || "Asia/Shanghai", date));
+      if (typeof result === "string" && result) return result;
+      if (result !== undefined && result !== null) {
+        console.warn(`[pi-better-cost-display-footer] tierFn 返回未知档位 "${String(result)}"，回退 ${defaultTier}`);
+      }
+      return defaultTier;
+    } catch (err) {
+      console.error(
+        `[pi-better-cost-display-footer] tierFn 执行异常: ${err instanceof Error ? err.message : String(err)}，回退 ${defaultTier}`,
+      );
+      return defaultTier;
+    }
+  }
+  const ctx = buildTierCtx(cfg.timezone || "Asia/Shanghai", date);
   const windows = cfg.peakWindows || [];
-  return windows.some((w) => inWindow(clock, w)) ? "peak" : "offPeak";
+  return windows.some((w) => inWindow({ hour: ctx.hour, minute: ctx.minute }, w)) ? "peak" : "offPeak";
 }
 
 export function isEffective(cfg: Pick<ProviderPricing, "effectiveFrom">, date: Date = new Date()): boolean {
@@ -283,7 +384,11 @@ function patchProviderConfig(
   const rates = cfg.providers?.[providerId]?.models || {};
   const models = provider.models.map((m) => {
     const rate = rates[m.id]?.[tier];
-    return rate ? { ...m, cost: rate } : m;
+    if (!rate) {
+      console.warn(`[pi-better-cost-display-footer] ${providerId}.${m.id} 没有档位 "${tier}" 的价格，保持原价`);
+      return m;
+    }
+    return { ...m, cost: rate };
   });
   return { ...provider, models };
 }
@@ -344,12 +449,12 @@ function getActiveModelPricing(
   return provider;
 }
 
-/** 计算档位标签；当前模型无动态计价配置 / 配置缺失 / 未生效 → null */
+/** 计算档位标签；当前模型无动态计价配置 / 配置缺失 / 未生效 / 档位未配置标签 → null */
 function computeFooterLabel(cfg: DynamicConfig | null, ctx: ExtensionContext | null): FooterLabel | null {
   const provider = getActiveModelPricing(cfg, ctx);
   if (!provider) return null;
-  const tier = currentTier(provider, nowClock(provider.timezone || "Asia/Shanghai"));
-  return provider.labels?.[tier] ?? DEFAULT_LABELS[tier];
+  const tier = currentTier(provider);
+  return provider.labels?.[tier] ?? DEFAULT_LABELS[tier] ?? null;
 }
 
 function installCustomFooter(ctx: ExtensionContext): void {
@@ -489,9 +594,7 @@ function apply(pi: ExtensionAPI, ctx: ExtensionContext): void {
   footerLabel = computeFooterLabel(cfg, ctx);
   if (!cfg) return;
   const providerTiers = Object.entries(cfg.providers || {}).map(([providerId, provider]) => {
-    const tier = isEffective(provider)
-      ? currentTier(provider, nowClock(provider.timezone || "Asia/Shanghai"))
-      : null;
+    const tier = isEffective(provider) ? currentTier(provider) : null;
     return [providerId, tier] as const;
   });
   const key = `${JSON.stringify(cfg)}|${JSON.stringify(providerTiers)}`;
@@ -526,7 +629,8 @@ export default function (pi: ExtensionAPI): void {
 3. 配置 JSON 格式（只动 providers 部分）：
    {
      "timezone": "Asia/Shanghai",
-     "peakWindows": [{ "start": "09:00", "end": "12:00" }, { "start": "14:00", "end": "18:00" }],
+     "tierFn": "(ctx) => ctx.isWeekend() ? 'offPeak' : (ctx.inWindow('09:00','12:00') || ctx.inWindow('14:00','18:00')) ? 'peak' : 'offPeak'",
+     "defaultTier": "offPeak",
      "labels": { "peak": { "text": "（梁文峰）", "color": "error" }, "offPeak": { "text": "（梁文谷）", "color": "success" } },
      "providers": {
        "<providerId>": {
@@ -540,12 +644,15 @@ export default function (pi: ExtensionAPI): void {
      }
    }
    价格单位：元/百万tokens；cacheWrite 恒为 0。页面"空闲时段"价格 → offPeak，"高峰时段"价格 → peak。
-4. 页面未列出的模型保持原样；配置文件中其他所有字段（timezone/peakWindows/labels/cacheHitRatePrecision/currencySymbol 及其他 provider）一律不动。
+   档位是任意字符串 id，可多段：若官方页出现更多价格档（如 尖峰/平段），给对应模型每档都配价格，并在 tierFn 中按 ctx（isWeekend/inWindow/afterISO/beforeISO 等）判定返回哪个档位 id。
+4. 页面未列出的模型保持原样；配置文件中其他所有字段（timezone/tierFn/defaultTier/labels/cacheHitRatePrecision/currencySymbol 及其他 provider）一律不动。
 5. 若配置文件不存在：扩展在启动时会自动生成默认配置文件，一般不会出现该情况；若确实不存在，按默认格式新建一份即可。
 6. 用 edit 精确修改，不要整体重写文件；完成后核对一遍数字与官方页面一致，并简要报告改了什么。
-7. 峰谷价取消处理（两种情况）：
-   a. 官方页取消峰谷价（不再区分空闲/高峰时段，只有单一价格）→ 各模型 offPeak 与 peak 填相同价格，cacheWrite 仍为 0，并把 peakWindows 清空为 []（扩展恒按 offPeak 计价 = 平价格，配置结构保持有效）。
-   b. 用户要求彻底取消动态计价 → 直接删除该 provider 的整个 providers.<providerId> 配置项（扩展会回退 models.json 原始 cost，且不显示档位标签）。`,
+7. 峰谷价调整（两种情况）：
+   a. 官方取消峰谷价（不再区分空闲/高峰时段，只有单一价格）→ 各模型只保留一个档位（如 offPeak 填该价格，其余档位删除），并把 tierFn 改为 "(ctx) => 'offPeak'"（恒按低谷价 = 平价格，配置结构保持有效）。
+   b. 用户要求彻底取消动态计价 → 直接删除该 provider 的整个 providers.<providerId> 配置项（扩展会回退 models.json 原始 cost，且不显示档位标签）。
+   8. 档位判定 tierFn 是 JS 函数表达式字符串（JSON 内用单引号引字符串），上下文 ctx 提供：now / timezone / weekday（0=周日）/ hour / minute / dateStr / isWeekend() / inWindow(start,end) / afterISO(iso) / beforeISO(iso)。DeepSeek 2026-08-23 起周末全天低谷的示例："(ctx) => ctx.isWeekend() ? 'offPeak' : (ctx.inWindow('09:00','12:00') || ctx.inWindow('14:00','18:00')) ? 'peak' : 'offPeak'"。
+   注意：配置中已有 tierFn 时只更新价格，不要改 tierFn 与 defaultTier。`,
         { deliverAs: "followUp" },
       );
     },

package/package.json

@@ -1,14 +1,20 @@
 {
   "name": "@yalieny/pi-better-cost-display-footer",
-  "version": "1.1.0",
-  "description": "pi extension: peak/off-peak dynamic pricing per provider with an enhanced footer (tier label, cache-hit-rate precision, custom currency symbol)",
+  "version": "2.0.0",
+  "description": "pi extension: scriptable peak/off-peak tier rules per provider with an enhanced footer (tier label, cache-hit-rate precision, custom currency symbol)",
   "keywords": ["pi-package", "pi-extension"],
   "license": "MIT",
   "type": "module",
+  "scripts": {
+    "test": "node --test test/tier.test.ts"
+  },
   "peerDependencies": {
     "@earendil-works/pi-coding-agent": "*",
     "@earendil-works/pi-tui": "*"
   },
+  "devDependencies": {
+    "@earendil-works/pi-tui": "^0.80.10"
+  },
   "files": ["extensions", "README.md"],
   "pi": {
     "extensions": ["./extensions"]