dsh-claude-style 0.17.0 → 0.18.0-dev.35

package/lib/host/session-cost.js

@@ -0,0 +1,306 @@
+// packages/contracts/src/package.ts
+var PACKAGE_NAME = "dsh-claude-style";
+
+// packages/contracts/src/prices.ts
+var CURRENCY_CNY = "CNY";
+var CURRENCY_USD = "USD";
+var CURRENCY_SYMBOL = {
+  [CURRENCY_CNY]: "\xA5",
+  [CURRENCY_USD]: "$"
+};
+var WEEKDAYS = ["Sun", "Mon", "Tue", "Wed", "Thu", "Fri", "Sat"];
+var formatters = /* @__PURE__ */ new Map();
+function formatter(timeZone) {
+  let held = formatters.get(timeZone);
+  if (held === void 0) {
+    held = new Intl.DateTimeFormat("en-CA", {
+      timeZone,
+      weekday: "short",
+      year: "numeric",
+      month: "2-digit",
+      day: "2-digit",
+      hour: "2-digit",
+      minute: "2-digit",
+      hourCycle: "h23"
+    });
+    formatters.set(timeZone, held);
+  }
+  return held;
+}
+function clockAt(atMs, timeZone) {
+  if (!Number.isFinite(atMs)) return null;
+  let parts;
+  try {
+    parts = formatter(timeZone).formatToParts(new Date(atMs));
+  } catch {
+    return null;
+  }
+  const field = (type) => parts.find((part) => part.type === type)?.value ?? "";
+  const weekday = WEEKDAYS.indexOf(field("weekday"));
+  const year = Number(field("year"));
+  const month = Number(field("month"));
+  const day = Number(field("day"));
+  const hour = Number(field("hour"));
+  const minute = Number(field("minute"));
+  if (weekday < 0 || !Number.isFinite(year) || !Number.isFinite(month) || !Number.isFinite(day) || !Number.isFinite(hour) || !Number.isFinite(minute)) return null;
+  const pad = (value) => String(value).padStart(2, "0");
+  return { weekday, date: `${year}-${pad(month)}-${pad(day)}`, minutes: hour * 60 + minute };
+}
+function clockMinutes(value) {
+  const match = /^(\d{1,2}):(\d{2})$/.exec(value);
+  if (match === null) return Number.NaN;
+  return Number(match[1]) * 60 + Number(match[2]);
+}
+function isPeak(atMs, schedule) {
+  const clock = clockAt(atMs, schedule.timeZone);
+  if (clock === null || !schedule.peakDays.includes(clock.weekday)) return false;
+  const holiday = clockAt(atMs, schedule.holidayZone);
+  if (holiday !== null && schedule.holidays.includes(holiday.date)) return false;
+  for (let i = 0; i < schedule.peakWindows.length; i++) {
+    const window = schedule.peakWindows[i];
+    const start = clockMinutes(window.start);
+    const end = clockMinutes(window.end);
+    if (clock.minutes >= start && clock.minutes < end) return true;
+  }
+  return false;
+}
+function bareModelId(modelId) {
+  const raw = modelId.trim().toLowerCase();
+  if (raw === "") return null;
+  return raw.includes("/") ? raw.slice(raw.lastIndexOf("/") + 1) : raw;
+}
+var CATALOG_SUFFIXES = ["-high", "-low", "-fast", "-sg", "-256k", "-128k", "-1m", "-32k", "-200k", "-flex"];
+function catalogIdFor(document, bare) {
+  const catalog = document.catalog;
+  if (catalog === void 0) return null;
+  const candidates = [document.aliases[bare], bare];
+  for (const suffix of CATALOG_SUFFIXES) {
+    if (bare.endsWith(suffix) && bare.length > suffix.length) candidates.push(bare.slice(0, -suffix.length));
+  }
+  for (const candidate of candidates) {
+    if (candidate !== void 0 && Object.hasOwn(catalog, candidate)) return candidate;
+  }
+  return null;
+}
+function priceModelId(document, modelId) {
+  const bare = bareModelId(modelId);
+  if (bare === null) return null;
+  if (Object.hasOwn(document.models, bare)) return bare;
+  const alias = document.aliases[bare];
+  return alias !== void 0 && Object.hasOwn(document.models, alias) ? alias : null;
+}
+function priceFor(document, modelId, currency) {
+  const id = priceModelId(document, modelId);
+  if (id !== null) return document.models[id]?.[currency];
+  const bare = bareModelId(modelId);
+  if (bare === null) return void 0;
+  const catalogId = catalogIdFor(document, bare);
+  if (catalogId === null) return void 0;
+  const published = currency === "CNY" ? document.catalogCny?.[catalogId] : void 0;
+  const parts = (published ?? document.catalog?.[catalogId] ?? "").split(" ").map(Number);
+  if (parts.length !== 3 || !parts.every((value) => Number.isFinite(value) && value >= 0)) return void 0;
+  if (parts[0] === 0 && parts[2] === 0) return void 0;
+  const rate = published !== void 0 ? 1 : currency === "CNY" ? document.usdToCny ?? 7.2 : 1;
+  const tier = { cacheHit: parts[1] * rate, cacheMiss: parts[0] * rate, output: parts[2] * rate };
+  return { offPeak: tier, peak: tier };
+}
+function priceTierAt(price, atMs, schedule) {
+  return isPeak(atMs, schedule) ? price.peak : price.offPeak;
+}
+function costMicro(usage, tier) {
+  const count2 = (value) => Number.isFinite(value) && value > 0 ? value : 0;
+  const cost = count2(usage.input) * tier.cacheMiss + count2(usage.cacheRead) * tier.cacheHit + count2(usage.cacheWrite) * tier.cacheHit + count2(usage.output) * tier.output;
+  return Number.isFinite(cost) && cost > 0 ? cost : 0;
+}
+
+// packages/engine/node/session-events.ts
+function eventsOf(events) {
+  return Array.isArray(events) ? events : null;
+}
+function unreadableLog(error) {
+  const failure = error;
+  return failure?.code === "SESSION_QUERY_CORRUPT_SESSION" || failure?.code === "SESSION_QUERY_SESSION_NOT_FOUND" ? failure : null;
+}
+function reportUnreadable(ctx, packageName, error, sessionId, label) {
+  ctx.logger?.warn?.(`${packageName}: session ${sessionId} left out of ${label}: ${unreadableLog(error)?.message}`);
+}
+function releaseObservation(observation) {
+  const named = observation.dispose;
+  if (typeof named === "function") named.call(observation);
+  else if (Symbol.dispose in observation) observation[Symbol.dispose]();
+}
+async function readEvents(ctx, packageName, sessionId, label) {
+  const query = ctx.get("sessionQuery");
+  if (query === null || query === void 0) return null;
+  try {
+    if (typeof query.observeSession === "function") {
+      const observation = await query.observeSession(sessionId, { projectionMode: "none" });
+      try {
+        return eventsOf(observation.events);
+      } finally {
+        releaseObservation(observation);
+      }
+    }
+    if (typeof query.readSession !== "function") return null;
+    const snapshot = await query.readSession(sessionId);
+    return eventsOf(snapshot?.events);
+  } catch (error) {
+    if (unreadableLog(error) === null) throw error;
+    reportUnreadable(ctx, packageName, error, sessionId, label);
+    return null;
+  }
+}
+
+// packages/host/src/session-cost.ts
+import { modelOf, usageOf } from "./usage-fold.js";
+var KEPT_SESSIONS = 128;
+function count(value) {
+  const number = Number(value);
+  return Number.isFinite(number) && number > 0 ? number : 0;
+}
+function usageOfSample(event) {
+  const sample = usageOf(event);
+  if (sample === void 0) return null;
+  return {
+    input: count(sample.inputTokens),
+    output: count(sample.outputTokens),
+    cacheRead: count(sample.cacheReadTokens),
+    cacheWrite: count(sample.cacheWriteTokens)
+  };
+}
+function sameUsage(settlement, usage) {
+  return settlement.input === usage.input && settlement.output === usage.output && settlement.cacheRead === usage.cacheRead && settlement.cacheWrite === usage.cacheWrite;
+}
+function applyCostEvent(state, event, document) {
+  const type = event?.type;
+  if (type === "llm/retry-started") {
+    const turn2 = event.data?.turn;
+    const step2 = event.data?.step;
+    if (state.last !== null && state.last.turn === turn2 && state.last.step === step2) {
+      return { ...state, last: null };
+    }
+    return state;
+  }
+  if (type !== "assistant/message" && type !== "assistant/attempt") return state;
+  const usage = usageOfSample(event);
+  if (usage === null) {
+    const named = modelOf(event);
+    return named === null || named === state.model ? state : { ...state, model: named };
+  }
+  const time = event.time;
+  const model = modelOf(event) ?? state.model;
+  if (typeof time !== "number" || !Number.isFinite(time)) return state;
+  let cny = 0;
+  let usd = 0;
+  let partial = state.partial;
+  const priced = (currency) => {
+    const price = model === null ? void 0 : priceFor(document, model, currency);
+    if (price === void 0) return null;
+    return costMicro(usage, priceTierAt(price, time, document.schedule));
+  };
+  const cnyCost = priced("CNY");
+  const usdCost = priced("USD");
+  if (cnyCost === null || usdCost === null) partial = true;
+  cny = cnyCost ?? 0;
+  usd = usdCost ?? 0;
+  const turn = event.data?.turn;
+  const step = event.data?.step;
+  const previous = state.last;
+  const replacing = previous !== null && previous.turn === turn && previous.step === step;
+  if (replacing && previous !== null && sameUsage(previous, usage)) return state;
+  const takeBack = replacing && previous !== null ? previous : null;
+  return {
+    cny: Math.max(0, state.cny - (takeBack?.cny ?? 0) + cny),
+    usd: Math.max(0, state.usd - (takeBack?.usd ?? 0) + usd),
+    partial,
+    last: {
+      ...turn === void 0 ? {} : { turn },
+      ...step === void 0 ? {} : { step },
+      ...usage,
+      cny,
+      usd
+    },
+    model: modelOf(event) ?? state.model
+  };
+}
+function foldSessionCost(events, document) {
+  let state = { cny: 0, usd: 0, partial: false, last: null, model: null };
+  for (let i = 0; i < events.length; i++) state = applyCostEvent(state, events[i], document);
+  return state;
+}
+var STALE_MS = 1e4;
+function createSessionCost(ctx, prices) {
+  const states = /* @__PURE__ */ new Map();
+  const touchedAt = /* @__PURE__ */ new Map();
+  const pending = /* @__PURE__ */ new Map();
+  const offPriceChange = prices.onChange(() => {
+    states.clear();
+    touchedAt.clear();
+  });
+  function trim() {
+    while (states.size > KEPT_SESSIONS) {
+      const oldest = states.keys().next().value;
+      if (oldest === void 0 || oldest === "") break;
+      states.delete(oldest);
+      touchedAt.delete(oldest);
+    }
+  }
+  function answer(state) {
+    return { cny: state.cny, usd: state.usd, partial: state.partial, model: state.model };
+  }
+  function observe(sessionId, event) {
+    const state = states.get(sessionId);
+    if (state === void 0) return;
+    const document = prices.document();
+    if (document === null) return;
+    const next = applyCostEvent(state, event, document);
+    if (next !== state) states.set(sessionId, next);
+    touchedAt.set(sessionId, Date.now());
+  }
+  function ensure() {
+    if (typeof ctx.on !== "function") return;
+    ctx.on("session/event", (session, event) => {
+      const id = typeof session?.id === "string" ? session.id : "";
+      if (id !== "") observe(id, event);
+    });
+  }
+  async function get(sessionId) {
+    prices.ensure();
+    const held = states.get(sessionId);
+    const touched = touchedAt.get(sessionId);
+    if (held !== void 0 && touched !== void 0 && Date.now() - touched < STALE_MS) return answer(held);
+    const running = pending.get(sessionId);
+    if (running !== void 0) return running;
+    const document = prices.document();
+    if (document === null) return null;
+    const work = readEvents(ctx, PACKAGE_NAME, sessionId, "the session cost").then((events) => {
+      pending.delete(sessionId);
+      if (events === null) return null;
+      const state = foldSessionCost(events, document);
+      states.set(sessionId, state);
+      touchedAt.set(sessionId, Date.now());
+      trim();
+      return answer(state);
+    }, (error) => {
+      pending.delete(sessionId);
+      throw error;
+    });
+    pending.set(sessionId, work);
+    return work;
+  }
+  return {
+    ensure,
+    get,
+    dispose() {
+      offPriceChange();
+      states.clear();
+      touchedAt.clear();
+      pending.clear();
+    }
+  };
+}
+export {
+  applyCostEvent,
+  createSessionCost,
+  foldSessionCost
+};

package/lib/host/settings.js

@@ -4,7 +4,18 @@ var PACKAGE_NAME = "dsh-claude-style";
 // packages/contracts/src/services.ts
 var OFFICIAL_MODEL_GROUP = "deepseek-official";
 
+// packages/contracts/src/prices.ts
+var CURRENCY_CNY = "CNY";
+var CURRENCY_USD = "USD";
+var CURRENCIES = [CURRENCY_CNY, CURRENCY_USD];
+var CURRENCY_SYMBOL = {
+  [CURRENCY_CNY]: "\xA5",
+  [CURRENCY_USD]: "$"
+};
+
 // packages/contracts/src/prefs.ts
+var BILLING_OFF = "off";
+var BILLING_CHOICES = [BILLING_OFF, ...CURRENCIES];
 var BRAND_CLAUDE = "claude";
 var BRAND_DEEPSEEK = "deepseek";
 var BRAND_DEEPSEEK_LEGACY = "off";
@@ -122,6 +133,8 @@ var PREFS = {
   mascotScope: { type: "choice", values: MASCOT_SCOPES, default: MASCOT_SCOPE_ALL },
   permissionsControl: { type: "boolean", default: true },
   statsPosition: { type: "choice", values: STATS_POSITIONS, default: STATS_POSITION_CONTEXT },
+  /** Which official price table the session's cost follows, or off for no figure at all. */
+  billingCurrency: { type: "choice", values: BILLING_CHOICES, default: CURRENCY_CNY },
   workspaceView: { type: "boolean", default: true },
   dockCards: { type: "boolean", default: true },
   sidebarSearch: { type: "boolean", default: true },

package/lib/host/usage-fold.js

@@ -135,5 +135,7 @@ function listSessionLogs(root) {
 export {
   dayKey,
   foldSession,
-  listSessionLogs
+  listSessionLogs,
+  modelOf,
+  usageOf
 };

package/lib/model-descriptions.json

@@ -2,7 +2,7 @@
   "version": 1,
   "fallback": "en",
   "brands": {
-    "note": "Which brand mark a row shows. Every id names a vendored lockup in packages/assets/src/icons/combine/; scripts/build.mjs fails the build on an id that is not there. This is presentation data, kept out of the bundle like the rest of this document, and it is separate from the copy rules above: a vendor's mark and a vendor's description line change for different reasons. `providers` is deliberately empty: a model no rule claims draws **no** lockup rather than its provider's, because the reseller's own lockup on a model it did not make reads as a wrong answer rather than a missing one. `models` is an ordered, anchored rule list resolved against the model id, exactly like `families` and with the same discipline: first match wins, so a broad rule must sit below the narrow ones and every rule must name its vendor (`step` cannot match `step-3.7-flash` by accident only if no earlier rule already claimed it). A brand with no lockup renders the label without a mark.",
+    "note": "Which brand mark a row shows. Every id names a vendored lockup in packages/assets/src/icons/combine/; scripts/build.mjs fails the build on an id that is not there. This is presentation data, kept out of the bundle like the rest of this document, and it is separate from the copy rules above: a vendor's mark and a vendor's description line change for different reasons. `providers` is deliberately empty: a model no rule claims draws **no** lockup rather than its provider's, because the reseller's own lockup on a model it did not make reads as a wrong answer rather than a missing one. `models` is an ordered, anchored rule list resolved against the model's id and then its display name (D79), exactly like `families` and with the same discipline: first match wins, so a broad rule must sit below the narrow ones and every rule must name its vendor (`step` cannot match `step-3.7-flash` by accident only if no earlier rule already claimed it). A brand with no lockup renders the label without a mark.",
     "providers": {},
     "models": [
       {