opencode-go-usage-tui 1.1.0 → 1.3.0

package/src/pricing.ts

@@ -0,0 +1,348 @@
+import { readFileSync, writeFileSync, mkdirSync, existsSync } from "node:fs"
+import { join, dirname } from "node:path"
+
+declare const process: { env: Record<string, string | undefined> } | undefined
+
+const DOCS_URL = "https://opencode.ai/docs/go"
+const PRICING_DIR = process?.env?.OPENCODE_CONFIG_DIR
+  || (process?.env?.USERPROFILE ? `${process.env.USERPROFILE}\\.config\\opencode` : "")
+  || process?.env?.HOME + "/.config/opencode"
+const PRICING_FILE = join(PRICING_DIR, "go-pricing-data.json")
+
+export interface ModelPrice {
+  id: string
+  name: string
+  variants: { label: string; pricing: { input: number | null; output: number | null; cachedRead: number | null; cachedWrite: number | null } }[]
+  pricing: {
+    input: number | null
+    output: number | null
+    cachedRead: number | null
+    cachedWrite: number | null
+  }
+  limits: {
+    fiveHour: number
+    weekly: number
+    monthly: number
+  }
+  usageLimit: number | null
+  endpoint: string
+  sdk: string
+  training: string
+  retention: string
+}
+
+export interface PricingData {
+  fetchTime: string
+  models: ModelPrice[]
+}
+
+export interface PricingChange {
+  type: "added" | "removed" | "pricing" | "limits" | "usageLimit"
+  modelId: string
+  modelName: string
+  changes?: { field: string; oldValue: string | number | null; newValue: string | number | null; changeType: string }[]
+  newValue?: ModelPrice
+  oldValue?: ModelPrice
+}
+
+export interface PricingCompareResult {
+  hasChanges: boolean
+  changes: PricingChange[]
+  summary: { total: number; added: number; removed: number; priceChanges: number; limitChanges: number }
+}
+
+export interface PricingStore {
+  lastFetch: string
+  current: PricingData
+  previous: PricingData | null
+  history: { fetchTime: string; changes: PricingChange[]; summary: PricingCompareResult["summary"] }[]
+}
+
+function cleanText(html: string): string {
+  return html
+    .replace(/<[^>]+>/g, "")
+    .replace(/&amp;/g, "&")
+    .replace(/&lt;/g, "<")
+    .replace(/&gt;/g, ">")
+    .replace(/&nbsp;/g, " ")
+    .replace(/&#x27;|&#39;/g, "'")
+    .trim()
+}
+
+function parsePrice(str: string): number | null {
+  const s = (str || "").replace(/[$\s]/g, "")
+  if (!s || s === "-" || s === "null" || s === "—" || s === "–") return null
+  const num = parseFloat(s.replace(/,/g, ""))
+  return isNaN(num) ? null : num
+}
+
+function parseCount(str: string): number {
+  const s = (str || "").replace(/[,\s]/g, "")
+  if (!s || s === "-" || s === "—" || s === "–") return 0
+  const m = s.match(/^([\d.]+)k$/i)
+  if (m) return Math.round(parseFloat(m[1]) * 1000)
+  const num = parseInt(s, 10)
+  return isNaN(num) ? 0 : num
+}
+
+function normalizeName(name: string): string {
+  return name.toLowerCase().replace(/[^a-z0-9]/g, "")
+}
+
+function generateId(name: string): string {
+  return name
+    .toLowerCase()
+    .replace(/[^a-z0-9]+/g, "-")
+    .replace(/^-+|-+$/g, "")
+}
+
+function extractTables(html: string): string[] {
+  const tables: string[] = []
+  const re = /<table[^>]*>([\s\S]*?)<\/table>/gi
+  let m: RegExpExecArray | null
+  while ((m = re.exec(html)) !== null) tables.push(m[1])
+  return tables
+}
+
+function extractRows(tableHtml: string): string[][] {
+  const rows: string[][] = []
+  const rowRe = /<tr[^>]*>([\s\S]*?)<\/tr>/gi
+  let rm: RegExpExecArray | null
+  while ((rm = rowRe.exec(tableHtml)) !== null) {
+    const cells: string[] = []
+    const cellRe = /<t[dh][^>]*>([\s\S]*?)<\/t[dh]>/gi
+    let cm: RegExpExecArray | null
+    while ((cm = cellRe.exec(rm[1])) !== null) cells.push(cleanText(cm[1]))
+    if (cells.length > 0) rows.push(cells)
+  }
+  return rows
+}
+
+function extractLinks(html: string): string[] {
+  const links: string[] = []
+  const re = /<a[^>]*href="([^"]*)"[^>]*>([\s\S]*?)<\/a>/gi
+  let m: RegExpExecArray | null
+  while ((m = re.exec(html)) !== null) {
+    const href = m[1]
+    if (href.startsWith("http") || href.startsWith("/")) links.push(href)
+  }
+  return links
+}
+
+export async function fetchPricing(): Promise<PricingData> {
+  const res = await fetch(DOCS_URL, {
+    headers: {
+      "user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36",
+      "accept": "text/html",
+    },
+    signal: AbortSignal.timeout(15000),
+  })
+  if (!res.ok) throw new Error(`HTTP ${res.status}`)
+  const html = await res.text()
+  const tables = extractTables(html)
+  if (tables.length < 4) throw new Error("Required tables not found")
+
+  const priceTable = tables[1]
+  const limitTable = tables[0]
+  const endpointTable = tables[2]
+  const retentionTable = tables[3]
+
+  const priceRows = extractRows(priceTable)
+  const limitRows = extractRows(limitTable)
+  const endpointRows = extractRows(endpointTable)
+  const retentionRows = extractRows(retentionTable)
+
+  const limitMap = new Map<string, string[]>()
+  for (const row of limitRows.slice(1)) {
+    if (row.length >= 4) limitMap.set(normalizeName(row[0]), row)
+  }
+  const endpointMap = new Map<string, string[]>()
+  for (const row of endpointRows.slice(1)) {
+    if (row.length >= 4) endpointMap.set(normalizeName(row[0]), row)
+  }
+  const retentionMap = new Map<string, string[]>()
+  for (const row of retentionRows.slice(1)) {
+    if (row.length >= 3) retentionMap.set(normalizeName(row[0]), row)
+  }
+
+  // 解析价格行：按基名分组保留变体（价格信息）
+  interface RawRow { baseName: string; variant: string; row: string[] }
+  const priceGroupMap = new Map<string, RawRow[]>()
+  for (const row of priceRows.slice(1)) {
+    if (row.length < 5) continue
+    const rawName = row[0]
+    const m = rawName.match(/^(.*?)\s*[（(]\s*([^（()]*?)\s*[)）]\s*$/)
+    const baseName = m ? m[1].trim() : rawName.trim()
+    const variant = m ? m[2].trim() : ""
+    const key = normalizeName(baseName)
+    if (!priceGroupMap.has(key)) priceGroupMap.set(key, [])
+    priceGroupMap.get(key)!.push({ baseName, variant, row })
+  }
+
+  // 以限额表为准构建模型列表（限额表是权威的当前模型来源）
+  const models: ModelPrice[] = []
+  for (const row of limitRows.slice(1)) {
+    if (row.length < 4) continue
+    const name = row[0]
+    const key = normalizeName(name)
+    const id = generateId(name)
+    const group = priceGroupMap.get(key)
+    const endpointRow = endpointMap.get(key)
+    const retentionRow = retentionMap.get(key)
+
+    const variants = (group ?? []).map((g) => ({
+      label: g.variant || "default",
+      pricing: {
+        input: parsePrice(g.row[1]),
+        output: parsePrice(g.row[2]),
+        cachedRead: parsePrice(g.row[3]),
+        cachedWrite: g.row.length > 4 ? parsePrice(g.row[4]) : null,
+      },
+    }))
+
+    const allInputs = variants.map((v) => v.pricing.input).filter((v): v is number => v !== null)
+    const allOutputs = variants.map((v) => v.pricing.output).filter((v): v is number => v !== null)
+    const allReads = variants.map((v) => v.pricing.cachedRead).filter((v): v is number => v !== null)
+    const allWrites = variants.map((v) => v.pricing.cachedWrite).filter((v): v is number => v !== null)
+
+    models.push({
+      id,
+      name,
+      variants,
+      pricing: {
+        input: variants.length === 1 ? variants[0].pricing.input : (allInputs.length ? Math.min(...allInputs) : null),
+        output: variants.length === 1 ? variants[0].pricing.output : (allOutputs.length ? Math.min(...allOutputs) : null),
+        cachedRead: variants.length === 1 ? variants[0].pricing.cachedRead : (allReads.length ? Math.min(...allReads) : null),
+        cachedWrite: variants.length === 1 ? variants[0].pricing.cachedWrite : (allWrites.length ? Math.min(...allWrites) : null),
+      },
+      limits: {
+        fiveHour: parseCount(row[1]),
+        weekly: parseCount(row[2]),
+        monthly: parseCount(row[3]),
+      },
+      usageLimit: group && group[0].row.length > 5 ? parsePrice(group[0].row[5]) : null,
+      endpoint: endpointRow ? (extractLinks(endpointRow.join(" "))[0] || "") : "",
+      sdk: endpointRow ? (endpointRow[3] || "") : "",
+      training: retentionRow ? (retentionRow[1] || "") : "",
+      retention: retentionRow ? (retentionRow[2] || "") : "",
+    })
+  }
+
+  return { fetchTime: new Date().toISOString(), models }
+}
+
+export function comparePricing(oldData: PricingData | null, newData: PricingData): PricingCompareResult {
+  const changes: PricingChange[] = []
+  if (!oldData || oldData.models.length === 0) {
+    return {
+      hasChanges: true,
+      changes: newData.models.map((m) => ({ type: "added", modelId: m.id, modelName: m.name, newValue: m })),
+      summary: { total: newData.models.length, added: newData.models.length, removed: 0, priceChanges: 0, limitChanges: 0 },
+    }
+  }
+
+  const oldIndex = new Map(oldData.models.map((m) => [m.id, m]))
+  for (const newModel of newData.models) {
+    const oldModel = oldIndex.get(newModel.id)
+    if (!oldModel) {
+      changes.push({ type: "added", modelId: newModel.id, modelName: newModel.name, newValue: newModel })
+      continue
+    }
+    const priceChanges: PricingChange["changes"] = []
+    const limitChanges: PricingChange["changes"] = []
+    for (const field of ["input", "output", "cachedRead", "cachedWrite"] as const) {
+      if (oldModel.pricing[field] !== newModel.pricing[field]) {
+        priceChanges.push({
+          field: `pricing.${field}`,
+          oldValue: oldModel.pricing[field],
+          newValue: newModel.pricing[field],
+          changeType: changeType(oldModel.pricing[field], newModel.pricing[field]),
+        })
+      }
+    }
+    if (priceChanges.length > 0) {
+      changes.push({ type: "pricing", modelId: newModel.id, modelName: newModel.name, changes: priceChanges })
+    }
+    for (const field of ["fiveHour", "weekly", "monthly"] as const) {
+      if (oldModel.limits[field] !== newModel.limits[field]) {
+        limitChanges.push({
+          field: `limits.${field}`,
+          oldValue: oldModel.limits[field],
+          newValue: newModel.limits[field],
+          changeType: changeType(oldModel.limits[field], newModel.limits[field]),
+        })
+      }
+    }
+    if (limitChanges.length > 0) {
+      changes.push({ type: "limits", modelId: newModel.id, modelName: newModel.name, changes: limitChanges })
+    }
+    if (oldModel.usageLimit !== newModel.usageLimit) {
+      changes.push({
+        type: "usageLimit",
+        modelId: newModel.id,
+        modelName: newModel.name,
+        changes: [{
+          field: "usageLimit",
+          oldValue: oldModel.usageLimit,
+          newValue: newModel.usageLimit,
+          changeType: changeType(oldModel.usageLimit, newModel.usageLimit),
+        }],
+      })
+    }
+    oldIndex.delete(newModel.id)
+  }
+  for (const [id, oldModel] of oldIndex) {
+    changes.push({ type: "removed", modelId: id, modelName: oldModel.name, oldValue: oldModel })
+  }
+
+  const summary = { total: changes.length, added: 0, removed: 0, priceChanges: 0, limitChanges: 0 }
+  for (const c of changes) {
+    if (c.type === "added") summary.added++
+    else if (c.type === "removed") summary.removed++
+    else if (c.type === "pricing") summary.priceChanges++
+    else if (c.type === "limits" || c.type === "usageLimit") summary.limitChanges++
+  }
+  return { hasChanges: changes.length > 0, changes, summary }
+}
+
+function changeType(oldVal: number | null, newVal: number | null): string {
+  if (oldVal === null && newVal !== null) return "added"
+  if (oldVal !== null && newVal === null) return "removed"
+  if (newVal !== null && oldVal !== null && newVal > oldVal) return "increase"
+  if (newVal !== null && oldVal !== null && newVal < oldVal) return "decrease"
+  return "unchanged"
+}
+
+export function loadPricingStore(): PricingStore {
+  try {
+    const data = JSON.parse(readFileSync(PRICING_FILE, "utf8")) as PricingStore
+    if (data && Array.isArray(data.current?.models)) return data
+  } catch { /* 无存储或损坏 */ }
+  return { lastFetch: "", current: { fetchTime: "", models: [] }, previous: null, history: [] }
+}
+
+export function savePricingStore(store: PricingStore): void {
+  try {
+    mkdirSync(dirname(PRICING_FILE), { recursive: true })
+    writeFileSync(PRICING_FILE, JSON.stringify(store, null, 2), "utf8")
+  } catch { /* 写入失败忽略 */ }
+}
+
+export function updatePricingStore(newData: PricingData): PricingStore {
+  const store = loadPricingStore()
+  const compare = comparePricing(store.current.models.length > 0 ? store.current : null, newData)
+  const prev = store.current
+  const newStore: PricingStore = {
+    lastFetch: newData.fetchTime,
+    current: newData,
+    previous: prev.models.length > 0 ? prev : null,
+    history: [
+      { fetchTime: newData.fetchTime, changes: compare.changes, summary: compare.summary },
+      ...store.history,
+    ].slice(0, 10),
+  }
+  savePricingStore(newStore)
+  return newStore
+}
+
+export { PRICING_FILE }