dsh-balance-widget 0.1.0 → 0.2.0

package/README.en.md

@@ -3,6 +3,7 @@
 [中文](README.md) | **English**
 
 <p align="center">
+  <img src="https://img.shields.io/npm/v/dsh-balance-widget?style=flat-square&label=npm" alt="npm">
   <img src="https://img.shields.io/github/stars/LL-cmyk-so/dsh-balance-widget?style=flat-square&label=Stars" alt="Stars">
   <img src="https://img.shields.io/github/license/LL-cmyk-so/dsh-balance-widget?style=flat-square" alt="License">
   <img src="https://img.shields.io/github/last-commit/LL-cmyk-so/dsh-balance-widget?style=flat-square" alt="Last commit">
@@ -29,8 +30,10 @@ A balance & cost widget for the [DeepSeek Harness](https://github.com/deepseek-a
 
 - **Account balance** — On click, the host proxies DeepSeek's official `GET /user/balance` and shows the `¥` balance. The API key is resolved through the host credentials service and never leaves the host process; the browser only talks to same-origin routes.
 - **Session cost (estimate)** — Computed from the session's `tokenUsage` projection × DeepSeek's official peak/off-peak price table. Follows the configured model (default `deepseek-v4-flash`, switchable to `deepseek-v4-pro`) and the Beijing-time peak/off-peak windows automatically.
+- **Last prompt cost (estimate)** — Parses the current session file and prices the last turn's token usage, answering "how much did that last prompt cost".
+- **Today total cost (estimate)** — Walks every session under `~/.dsh/sessions/` and sums today's (calendar day) token usage × price.
 - **Token usage** — Also shows the session's input (incl. cache hits) / output tokens.
-- **On-demand refresh** — No polling, no background requests; the balance endpoint is only hit when you click the icon. Costs zero tokens to use.
+- **On-demand refresh** — No polling, no background requests; endpoints are only hit when you click the icon. Costs zero tokens to use.
 
 ## Why this plugin
 
@@ -42,13 +45,14 @@ A balance & cost widget for the [DeepSeek Harness](https://github.com/deepseek-a
 ```
 host half (lib/index.js)
   ctx.webServer.register:
-    GET /api/dsh-balance/balance → official /user/balance (loopback-only guard)
-    GET /api/dsh-balance/cost    → reserved route (cost priced client-side)
+    GET /api/dsh-balance/balance    → official /user/balance (loopback-only guard)
+    GET /api/dsh-balance/last-cost  → last prompt cost (session file + zstd decode)
+    GET /api/dsh-balance/today-cost → today's total across all sessions
 
 client half (lib/client.js)
   ctx.slots.inject("conversation.session.header.utilities")
     → 💰 icon (session header, top-right corner)
-    → click fetches same-origin /api/dsh-balance/balance → popover with balance + cost + tokens
+    → click fetches same-origin APIs → popover with balance + three costs + tokens
 ```
 
 ## Installation
@@ -100,6 +104,7 @@ Built-in DeepSeek official peak/off-peak pricing (CNY per 1M tokens), effective
 
 - Config tree: `dsh --profile web --dump-config` should show a `balance-widget` entry.
 - Balance route: after restarting dsh web, `curl -s http://127.0.0.1:3080/api/dsh-balance/balance` should return `{ ok, balance_infos, modelId }`.
+- Cost routes: `curl -s "http://127.0.0.1:3080/api/dsh-balance/last-cost?session=<sessionId>"` and `curl -s http://127.0.0.1:3080/api/dsh-balance/today-cost` should return `{ cost, inputTokens, outputTokens, modelId }`.
 
 ## License
 

package/README.md

@@ -3,6 +3,7 @@
 **中文** | [English](README.en.md)
 
 <p align="center">
+  <img src="https://img.shields.io/npm/v/dsh-balance-widget?style=flat-square&label=npm" alt="npm">
   <img src="https://img.shields.io/github/stars/LL-cmyk-so/dsh-balance-widget?style=flat-square&label=Stars" alt="Stars">
   <img src="https://img.shields.io/github/license/LL-cmyk-so/dsh-balance-widget?style=flat-square" alt="License">
   <img src="https://img.shields.io/github/last-commit/LL-cmyk-so/dsh-balance-widget?style=flat-square" alt="Last commit">
@@ -29,22 +30,25 @@ DeepSeek Harness (DSH) Web GUI 的余额与成本小部件：在会话头部右
 
 - **账户余额** — 点击图标时经宿主代理查询 DeepSeek 官方 `GET /user/balance`，展示 `¥` 余额；API key 只在宿主进程内读取（凭据服务），浏览器不接触密钥。
 - **本会话成本（估算）** — 由会话的 `tokenUsage` 投影 × DeepSeek 官方峰谷定价表计算，随当前会话模型（默认 `deepseek-v4-flash`，可在配置中改为 `deepseek-v4-pro`）与北京时间高峰/空闲时段自动切换。
+- **最近一次提问成本（估算）** — 从当前会话文件解析最后一个 turn 的 token 用量 × 单价，回答"刚才那条提问花了多少"。
+- **今天总成本（估算）** — 遍历 `~/.dsh/sessions/` 下所有会话，累加今天（自然日）的 token 用量 × 单价。
 - **Token 用量** — 同时展示本会话输入（含缓存命中）/ 输出 token 数。
-- **按需刷新** — 无轮询、无后台请求；只有点击图标时才发起余额查询，不消耗任何 token。
+- **按需刷新** — 无轮询、无后台请求；只有点击图标时才发起查询，不消耗任何 token。
 
 ## 架构
 
 ```
 host 半区 (lib/index.js)
   ctx.webServer.register:
-    GET /api/dsh-balance/balance → 官方 /user/balance（loopback-only 守卫）
-    GET /api/dsh-balance/cost    → 保留路由（成本目前客户端计价）
+    GET /api/dsh-balance/balance    → 官方 /user/balance（loopback-only 守卫）
+    GET /api/dsh-balance/last-cost  → 当前会话最近一次提问成本（读会话文件 + zstd 解压）
+    GET /api/dsh-balance/today-cost → 今天所有会话总成本
   依赖：零外部 @deepseek-ai/* import，任何 profile 布局均可解析
 
 client 半区 (lib/client.js)
   ctx.slots.inject("conversation.session.header.utilities")
     → 💰 图标（会话头部右上角）
-    → 点击 fetch 同源 /api/dsh-balance/balance → 弹层展示余额 + 成本 + token
+    → 点击 fetch 同源 API → 弹层展示余额 + 三种成本 + token
 ```
 
 ## 安装
@@ -96,6 +100,7 @@ dsh plugin --profile web add "link:$(pwd)"
 
 - 配置树：`dsh --profile web --dump-config` 应出现 `balance-widget` 条目
 - 余额路由：重启 dsh web 后 `curl -s http://127.0.0.1:3080/api/dsh-balance/balance` 应返回 `{ ok, balance_infos, modelId }`
+- 成本路由：`curl -s "http://127.0.0.1:3080/api/dsh-balance/last-cost?session=<会话id>"` 与 `curl -s http://127.0.0.1:3080/api/dsh-balance/today-cost` 应返回 `{ cost, inputTokens, outputTokens, modelId }`
 
 ## License
 

package/lib/client.js

@@ -104,9 +104,12 @@ window.__ModuleLoader__.load({
 		//#region lib/types/client/balance-api.js
 		/** Same-origin balance route (host half registers it over ctx.webServer). */
 		const BALANCE_ROUTE = "/api/dsh-balance/balance";
+		/** Same-origin cost routes. */
+		const LAST_COST_ROUTE = "/api/dsh-balance/last-cost";
+		const TODAY_COST_ROUTE = "/api/dsh-balance/today-cost";
 		/**
 		* Fetch the account balance from the host proxy.
-		* @returns { ok, balanceInfos?, error? }
+		* @returns { ok, balanceInfos?, modelId?, error? }
 		*/
 		async function fetchBalance() {
 			try {
@@ -127,6 +130,25 @@ window.__ModuleLoader__.load({
 				return { ok: false, error: error instanceof Error ? error.message : String(error) };
 			}
 		}
+		/**
+		* Fetch a cost figure from the host (last turn or today's total).
+		* @param {string} route - LAST_COST_ROUTE or TODAY_COST_ROUTE.
+		* @param {string} [sessionId] - required for last-cost.
+		* @returns { ok, cost?, inputTokens?, outputTokens?, error? }
+		*/
+		async function fetchCost(route, sessionId) {
+			try {
+				const url = sessionId === void 0 || sessionId === ""
+					? route
+					: `${route}?session=${encodeURIComponent(sessionId)}`;
+				const response = await fetch(url, { headers: { accept: "application/json" } });
+				const payload = await response.json().catch(() => ({}));
+				if (!response.ok) return { ok: false, error: payload.error ?? `cost endpoint responded ${response.status}` };
+				return { ok: true, cost: payload.cost ?? 0, inputTokens: payload.inputTokens ?? 0, outputTokens: payload.outputTokens ?? 0 };
+			} catch (error) {
+				return { ok: false, error: error instanceof Error ? error.message : String(error) };
+			}
+		}
 		/** Pick the display balance from balance_infos (prefer CNY, else first). */
 		function displayBalance(balanceInfos) {
 			if (!Array.isArray(balanceInfos) || balanceInfos.length === 0) return null;
@@ -149,6 +171,8 @@ window.__ModuleLoader__.load({
 			const [open, setOpen] = (0, react.useState)(false);
 			const [balance, setBalance] = (0, react.useState)(null);
 			const [modelId, setModelId] = (0, react.useState)("deepseek-v4-flash");
+			const [lastCost, setLastCost] = (0, react.useState)(null);
+			const [todayCost, setTodayCost] = (0, react.useState)(null);
 			const [error, setError] = (0, react.useState)(null);
 			const [loading, setLoading] = (0, react.useState)(false);
 			const usage = useProjection("tokenUsage");
@@ -159,14 +183,22 @@ window.__ModuleLoader__.load({
 				if (next) {
 					setLoading(true);
 					setError(null);
-					const result = await fetchBalance();
-					if (result.ok) {
-						setBalance(displayBalance(result.balanceInfos));
-						if (typeof result.modelId === "string" && result.modelId !== "") setModelId(result.modelId);
+					const [balanceResult, lastResult, todayResult] = await Promise.all([
+						fetchBalance(),
+						fetchCost(LAST_COST_ROUTE, sessionId),
+						fetchCost(TODAY_COST_ROUTE)
+					]);
+					if (balanceResult.ok) {
+						setBalance(displayBalance(balanceResult.balanceInfos));
+						if (typeof balanceResult.modelId === "string" && balanceResult.modelId !== "") setModelId(balanceResult.modelId);
 					} else {
 						setBalance(null);
-						setError(result.error);
+						setError(balanceResult.error);
 					}
+					if (lastResult.ok) setLastCost(lastResult.cost);
+					else if (!lastResult.ok && lastResult.error) setError((error ?? "") + (error ? " " : "") + `last: ${lastResult.error}`);
+					if (todayResult.ok) setTodayCost(todayResult.cost);
+					else if (!todayResult.ok && todayResult.error) setError((error ?? "") + (error ? " " : "") + `today: ${todayResult.error}`);
 					setLoading(false);
 				}
 			};
@@ -206,6 +238,26 @@ window.__ModuleLoader__.load({
 										children: priced !== null ? formatCny(priced.total) : "—"
 									})]
 								})
+							}), (0, react_jsx_runtime.jsx)("div", {
+								className: styles.row,
+								children: (0, react_jsx_runtime.jsxs)(react_jsx_runtime.Fragment, {
+									children: [(0, react_jsx_runtime.jsx)("span", {
+										children: t("lastPromptCost")
+									}), (0, react_jsx_runtime.jsx)("span", {
+										className: styles.val,
+										children: loading ? "…" : lastCost !== null ? formatCny(lastCost) : "—"
+									})]
+								})
+							}), (0, react_jsx_runtime.jsx)("div", {
+								className: styles.row,
+								children: (0, react_jsx_runtime.jsxs)(react_jsx_runtime.Fragment, {
+									children: [(0, react_jsx_runtime.jsx)("span", {
+										children: t("todayCost")
+									}), (0, react_jsx_runtime.jsx)("span", {
+										className: styles.val,
+										children: loading ? "…" : todayCost !== null ? formatCny(todayCost) : "—"
+									})]
+								})
 							}), priced !== null && (0, react_jsx_runtime.jsx)("div", {
 								className: styles.row,
 								children: (0, react_jsx_runtime.jsxs)(react_jsx_runtime.Fragment, {
@@ -243,20 +295,24 @@ window.__ModuleLoader__.load({
 		const NS = "balance-widget";
 		/** Simplified Chinese dictionary (the key-set source of truth). */
 		const zh = {
-			"aria": "查看账户余额与本轮成本",
+			"aria": "查看账户余额与成本",
 			"title": "余额与成本",
 			"balance": "账户余额",
 			"sessionCost": "本会话成本（估算）",
+			"lastPromptCost": "最近一次提问（估算）",
+			"todayCost": "今天总成本（估算）",
 			"tokens": "Token 用量",
 			"hint": "价格按 DeepSeek 官方峰谷定价估算，实际以账单为准。",
 			"close": "关闭"
 		};
 		/** English dictionary, checked complete against the zh key set. */
 		const en = {
-			"aria": "View account balance and session cost",
+			"aria": "View account balance and costs",
 			"title": "Balance & cost",
 			"balance": "Account balance",
 			"sessionCost": "This session cost (est.)",
+			"lastPromptCost": "Last prompt (est.)",
+			"todayCost": "Today total (est.)",
 			"tokens": "Token usage",
 			"hint": "Prices estimated from DeepSeek official peak/off-peak rates; billing is authoritative.",
 			"close": "Close"

package/lib/index.js

@@ -22,7 +22,9 @@ export const inject = ["webServer"];
 /** Route paths shared with the browser half (spelled here, not imported). */
 export const API = {
 	balance: "/api/dsh-balance/balance",
-	cost: "/api/dsh-balance/cost"
+	cost: "/api/dsh-balance/cost",
+	lastCost: "/api/dsh-balance/last-cost",
+	todayCost: "/api/dsh-balance/today-cost"
 };
 
 /** Default DeepSeek API base (mirrors dsh-llm-deepseek's PUBLIC_BASE_URL). */
@@ -206,6 +208,202 @@ async function fetchBalance(ctx, config) {
 	}
 }
 
+/**
+ * Pricing table (CNY per 1M tokens, official 2026-08-17 peak/off-peak).
+ * Mirrors the client half; kept here so host-computed costs use the same
+ * numbers.
+ */
+const PRICING = {
+	"deepseek-v4-flash": { peak: { inputMiss: 3.0, inputHit: 0.10, output: 9.0 }, offPeak: { inputMiss: 1.5, inputHit: 0.05, output: 4.5 } },
+	"deepseek-v4-pro": { peak: { inputMiss: 9.0, inputHit: 0.30, output: 27.0 }, offPeak: { inputMiss: 4.5, inputHit: 0.15, output: 13.5 } },
+	"deepseek-chat": { peak: { inputMiss: 3.0, inputHit: 0.10, output: 9.0 }, offPeak: { inputMiss: 1.5, inputHit: 0.05, output: 4.5 } },
+	"deepseek-reasoner": { peak: { inputMiss: 9.0, inputHit: 0.30, output: 27.0 }, offPeak: { inputMiss: 4.5, inputHit: 0.15, output: 13.5 } }
+};
+const PRICING_ALIASES = {
+	"deepseek-vision": "deepseek-v4-flash",
+	"deepseek-official": "deepseek-v4-flash"
+};
+
+/** Whether `date` falls inside a Beijing-time peak window. */
+function isPeak(date) {
+	const hour = date.getHours();
+	return (hour >= 9 && hour < 12) || (hour >= 14 && hour < 18);
+}
+
+/** Map a provider/model id to a pricing key. */
+function pricingKey(modelId) {
+	if (modelId === void 0 || modelId === null) return "deepseek-v4-flash";
+	if (PRICING[modelId] !== void 0) return modelId;
+	const alias = PRICING_ALIASES[modelId];
+	if (alias !== void 0) return alias;
+	// heuristic: pro/reasoner-ish names price as pro
+	if (/pro|reasoner|r1/i.test(modelId)) return "deepseek-v4-pro";
+	return "deepseek-v4-flash";
+}
+
+/** Price one usage sample in CNY. */
+function priceUsage(usage, modelId, timeMs) {
+	if (usage === void 0 || usage === null) return 0;
+	const input = usage.inputTokens ?? usage.uncachedInputTokens ?? 0;
+	const hit = usage.cacheReadTokens ?? 0;
+	const output = usage.outputTokens ?? 0;
+	if (input + hit + output <= 0) return 0;
+	const table = isPeak(new Date(timeMs ?? Date.now())) ? PRICING[pricingKey(modelId)].peak : PRICING[pricingKey(modelId)].offPeak;
+	return (input * table.inputMiss + hit * table.inputHit + output * table.output) / 1e6;
+}
+
+/**
+ * Resolve the DSH sessions root directory (defaults to ~/.dsh/sessions).
+ * Avoids importing dsh-home-paths to keep zero external dependencies.
+ */
+function sessionsRoot() {
+	return `${process.env.HOME ?? ""}/.dsh/sessions`;
+}
+
+/** Strip a leading "session-" prefix if present (callers may pass either form). */
+function normalizeSessionId(sessionId) {
+	return sessionId.startsWith("session-") ? sessionId.slice("session-".length) : sessionId;
+}
+
+/** Resolve the current session's JSONL path by walking the sessions tree. */
+async function resolveSessionFile(sessionId) {
+	if (sessionId === void 0 || sessionId === "") return void 0;
+	const bareId = normalizeSessionId(sessionId);
+	const root = sessionsRoot();
+	try {
+		const fs = await awaitImportFs();
+		const dirs = fs.readdirSync(root, { withFileTypes: true });
+		for (const dir of dirs) {
+			if (!dir.isDirectory()) continue;
+			const candidate = `${root}/${dir.name}/session-${bareId}/session.jsonl.zstd`;
+			if (fs.existsSync(candidate)) return candidate;
+		}
+	} catch {
+		/* ignore filesystem errors */
+	}
+	return void 0;
+}
+
+let _fs;
+function awaitImportFs() {
+	if (_fs === void 0) _fs = import("node:fs");
+	return _fs;
+}
+
+/**
+ * Decompress a .zstd session file to text via the zstd CLI (present on
+ * macOS/Linux with zstd installed). Returns undefined on failure.
+ */
+async function decompressSession(path) {
+	const { execFile } = await import("node:child_process");
+	const { promisify } = await import("node:util");
+	const execFileP = promisify(execFile);
+	try {
+		const { stdout } = await execFileP("zstd", ["-d", "-c", path], { maxBuffer: 512 * 1024 * 1024 });
+		return stdout;
+	} catch {
+		return void 0;
+	}
+}
+
+/**
+ * Parse a decompressed session JSONL and return every assistant usage sample
+ * with its turn, time, and the model id from the nearest request/header.
+ */
+function parseSessionUsages(text) {
+	const samples = [];
+	let currentModel = void 0;
+	for (const line of text.split("\n")) {
+		if (line === "") continue;
+		let event;
+		try {
+			event = JSON.parse(line);
+		} catch {
+			continue;
+		}
+		const type = event.type;
+		if (type === "request/header") {
+			currentModel = event.data?.header?.config?.model ?? currentModel;
+		} else if (type === "assistant/message") {
+			const usage = event.data?.usage;
+			if (usage !== void 0) {
+				samples.push({
+					turn: event.data?.turn ?? 0,
+					time: event.time ?? Date.now(),
+					model: currentModel,
+					usage
+				});
+			}
+		}
+	}
+	return samples;
+}
+
+/** Cost of the last turn (the most recent user prompt) in one session file. */
+function lastTurnCost(samples) {
+	if (samples.length === 0) return { cost: 0, inputTokens: 0, outputTokens: 0 };
+	let lastTurn = -1;
+	for (const s of samples) if (s.turn > lastTurn) lastTurn = s.turn;
+	let cost = 0;
+	let input = 0;
+	let output = 0;
+	for (const s of samples) {
+		if (s.turn !== lastTurn) continue;
+		cost += priceUsage(s.usage, s.model, s.time);
+		input += (s.usage.inputTokens ?? s.usage.uncachedInputTokens ?? 0) + (s.usage.cacheReadTokens ?? 0);
+		output += s.usage.outputTokens ?? 0;
+	}
+	return { cost, inputTokens: input, outputTokens: output };
+}
+
+/**
+ * Sum today's usage across every session file. "Today" = events whose time
+ * falls on the same calendar day (local time) as now.
+ */
+function todayCost(samplesBySession) {
+	const now = new Date();
+	const dayStart = new Date(now.getFullYear(), now.getMonth(), now.getDate()).getTime();
+	const dayEnd = dayStart + 24 * 60 * 60 * 1000;
+	let cost = 0;
+	let input = 0;
+	let output = 0;
+	for (const samples of samplesBySession) {
+		for (const s of samples) {
+			if (s.time < dayStart || s.time >= dayEnd) continue;
+			cost += priceUsage(s.usage, s.model, s.time);
+			input += (s.usage.inputTokens ?? s.usage.uncachedInputTokens ?? 0) + (s.usage.cacheReadTokens ?? 0);
+			output += s.usage.outputTokens ?? 0;
+		}
+	}
+	return { cost, inputTokens: input, outputTokens: output };
+}
+
+/** Walk every session file under the sessions root, decompress, and parse. */
+async function collectAllSessionSamples() {
+	const root = sessionsRoot();
+	const fs = (await awaitImportFs());
+	const files = [];
+	try {
+		const walk = (dir) => {
+			for (const entry of fs.readdirSync(dir, { withFileTypes: true })) {
+				const full = `${dir}/${entry.name}`;
+				if (entry.isDirectory()) walk(full);
+				else if (entry.name === "session.jsonl.zstd") files.push(full);
+			}
+		};
+		walk(root);
+	} catch {
+		return [];
+	}
+	const perSession = [];
+	for (const file of files) {
+		const text = await decompressSession(file);
+		if (text === void 0) continue;
+		perSession.push(parseSessionUsages(text));
+	}
+	return perSession;
+}
+
 /** Build the route table for the balance widget. */
 function makeRoutes(ctx, config) {
 	return [
@@ -227,9 +425,41 @@ function makeRoutes(ctx, config) {
 			path: API.cost,
 			handler: async (req, res) => {
 				if (!guard(req, res, "GET")) return;
-				// Cost is priced client-side from the tokenUsage projection; this
-				// route exists as the canonical ask point for future server pricing.
-				writeJson(res, 200, { ok: true, note: "cost is priced client-side from tokenUsage projection" });
+				// Legacy route: cost is priced per-session; keep responding for
+				// compatibility with older client builds.
+				writeJson(res, 200, { ok: true, note: "use /last-cost and /today-cost" });
+			}
+		},
+		{
+			kind: "exact",
+			path: API.lastCost,
+			handler: async (req, res) => {
+				if (!guard(req, res, "GET")) return;
+				const url = new URL(req.url ?? "/", "http://localhost");
+				const sessionId = url.searchParams.get("session") ?? "";
+				const file = await resolveSessionFile(sessionId);
+				if (file === void 0) {
+					writeJson(res, 404, { error: `session ${sessionId} not found`, modelId: config.modelId });
+					return;
+				}
+				const text = await decompressSession(file);
+				if (text === void 0) {
+					writeJson(res, 502, { error: "failed to read session file", modelId: config.modelId });
+					return;
+				}
+				const samples = parseSessionUsages(text);
+				const result = lastTurnCost(samples);
+				writeJson(res, 200, { ...result, modelId: config.modelId });
+			}
+		},
+		{
+			kind: "exact",
+			path: API.todayCost,
+			handler: async (req, res) => {
+				if (!guard(req, res, "GET")) return;
+				const perSession = await collectAllSessionSamples();
+				const result = todayCost(perSession);
+				writeJson(res, 200, { ...result, modelId: config.modelId });
 			}
 		}
 	];

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "dsh-balance-widget",
   "description": "Balance & cost widget for the dsh web GUI: a corner icon in the conversation session header that shows the DeepSeek account balance and the current session's estimated spend on click. Host half proxies the official /user/balance API over ctx.webServer (loopback-only); client half renders the icon and popover. Zero external dependencies — works on Node 24.",
-  "version": "0.1.0",
+  "version": "0.2.0",
   "type": "module",
   "main": "lib/index.js",
   "files": [