dsh-plugin-subscription-oauth 0.4.0 → 0.5.0

package/lib/client.js

@@ -632,6 +632,8 @@ const RPC_PREFIX = "subscription-oauth.";
 //#region src/client/subscriptions-rpc.ts
 /** The shared channel the node half's routes are mounted on. */
 const CHANNEL = "/api";
+/** The host's own DeepSeek account controller, from which the chip reads a balance. */
+const ACCOUNT_CHANNEL = "/api";
 /** A failed call: either the transport rejected or the endpoint answered `ok: false`. */
 var SubscriptionsRpcError = class extends Error {
 	constructor(message, code) {
@@ -661,6 +663,55 @@ function createSubscriptionsCall(rpc) {
 		return result.value;
 	};
 }
+/**
+* Read the DeepSeek account's balance through the host's own account controller.
+*
+* The host owns the DeepSeek OAuth grant and the platform endpoint, so this asks it
+* rather than re-deriving either. It returns `null` when the account is signed out or
+* the read fails, so a chip can simply not render in that case.
+*
+* @param rpc - the browser connection's RPC caller.
+* @returns the balance, or null when there is nothing to show.
+*/
+function createAccountBalanceCall(rpc) {
+	return async function call() {
+		let result;
+		try {
+			result = await rpc.call(ACCOUNT_CHANNEL, "account/getBalance", {
+				version: "1",
+				locale: Intl.DateTimeFormat().resolvedOptions().locale ?? "en-US",
+				timezoneOffsetSeconds: -(/* @__PURE__ */ new Date()).getTimezoneOffset()
+			});
+		} catch (error) {
+			throw new SubscriptionsRpcError(messageOf$1(error), "RPC_TRANSPORT");
+		}
+		if (!result.ok) throw new SubscriptionsRpcError(result.error.message, result.error.code);
+		return result.value;
+	};
+}
+/**
+* Read a manually added provider's balance through the plugin's cc-switch endpoint.
+*
+* The caller names the credential reference the provider's profile uses (its
+* `apiKeyEnv`), and the host's credential service resolves it to the API key. The result
+* is `undefined` when there is no key or the provider did not answer — never a wrong
+* figure.
+*
+* @param rpc - the browser connection's RPC caller.
+* @returns a narrow caller taking the credential reference.
+*/
+function createCcSwitchBalanceCall(rpc) {
+	return async function call(credentialRef) {
+		let result;
+		try {
+			result = await rpc.call(CHANNEL, `${RPC_PREFIX}ccswitch-balance`, { credentialRef });
+		} catch (error) {
+			throw new SubscriptionsRpcError(messageOf$1(error), "RPC_TRANSPORT");
+		}
+		if (!result.ok) throw new SubscriptionsRpcError(result.error.message, result.error.code);
+		return result.value;
+	};
+}
 
 //#endregion
 //#region src/client/SubscriptionsSection.tsx
@@ -2221,6 +2272,43 @@ function providerOf(projection) {
 	return (projection?.next ?? projection?.lastUsed)?.provider;
 }
 /**
+* Reduce the host's account-balance payload to the wallets the chip shows.
+*
+* Bonus wallets are shown after the topped-up ones, so a reader sees their own credit
+* first and any promotional credit as a secondary figure.
+*
+* @param value - the host's ready balance, or undefined when there is none.
+* @returns the wallets, topped-up first then bonus, or undefined when nothing is usable.
+*/
+function balanceOf(value) {
+	if (value === null || value === void 0 || !("value" in value) || value.value === void 0) return void 0;
+	const wallets = [...value.value.map((wallet) => ({
+		currency: wallet.currency,
+		balance: wallet.balance,
+		bonus: false
+	})), ...(value.bonusWallets ?? []).map((wallet) => ({
+		currency: wallet.currency,
+		balance: wallet.balance,
+		bonus: true
+	}))];
+	return wallets.length === 0 ? void 0 : wallets;
+}
+/**
+* Format one wallet for the chip's compact line.
+*
+* A balance is money, not a percentage, so it renders as `¥6.89` rather than a meter;
+* the currency prefix keeps two currencies from being mistaken for one sum.
+*
+* @param wallet - the wallet to show.
+* @returns the short form, e.g. `¥6.89` or `+¥2` for a bonus wallet.
+*/
+function formatBalance(wallet) {
+	const amount = Number.parseFloat(wallet.balance);
+	const trimmed = Number.isFinite(amount) ? String(Math.round(amount * 100) / 100) : wallet.balance;
+	const symbol = wallet.currency === "CNY" ? "¥" : wallet.currency === "USD" ? "$" : `${wallet.currency} `;
+	return `${wallet.bonus ? "+" : ""}${symbol}${trimmed}`;
+}
+/**
 * The short form of a route's name.
 *
 * @param provider - the route id.
@@ -2309,12 +2397,37 @@ function formatReset(at, now) {
 * @param props - injected RPC caller, copy, and the session's provider reader.
 * @returns the chip, or null when there is nothing true to say.
 */
-function UsageChip({ call, t, isOwnProvider, useProjection }) {
+function UsageChip({ call, t, isOwnProvider, accountBalance, ccSwitchBalance, ccSwitchCredentialRef, useProjection }) {
 	const [reading, setReading] = (0, react.useState)(void 0);
+	const [balance, setBalance] = (0, react.useState)(void 0);
+	const [ccBalance, setCcBalance] = (0, react.useState)(void 0);
 	const [now, setNow] = (0, react.useState)(() => Date.now());
 	const [open, setOpen] = (0, react.useState)(false);
 	const lastOwn = (0, react.useRef)(void 0);
 	const selected = useProjection("modelSelection", (value) => providerOf(value));
+	const isDeepSeek = selected === "deepseek-account";
+	const ccRef = selected === void 0 || isOwnProvider(selected) || isDeepSeek ? void 0 : ccSwitchCredentialRef(selected);
+	(0, react.useEffect)(() => {
+		if (ccRef === void 0) {
+			setCcBalance(void 0);
+			return;
+		}
+		let live = true;
+		const refresh = () => {
+			ccSwitchBalance(ccRef).then((value) => {
+				if (live) setCcBalance(value === void 0 ? void 0 : {
+					currency: value.currency,
+					total: value.total
+				});
+			}).catch(() => {});
+		};
+		refresh();
+		const timer = setInterval(refresh, REFRESH_INTERVAL_MS);
+		return () => {
+			live = false;
+			clearInterval(timer);
+		};
+	}, [ccRef, ccSwitchBalance]);
 	const load = (0, react.useCallback)(async (target) => {
 		try {
 			const result = await call("usage", { provider: target });
@@ -2323,6 +2436,24 @@ function UsageChip({ call, t, isOwnProvider, useProjection }) {
 			return;
 		}
 	}, [call]);
+	(0, react.useEffect)(() => {
+		if (!isDeepSeek) {
+			setBalance(void 0);
+			return;
+		}
+		let live = true;
+		const refresh = () => {
+			accountBalance().then((value) => {
+				if (live) setBalance(balanceOf(value));
+			}).catch(() => {});
+		};
+		refresh();
+		const timer = setInterval(refresh, REFRESH_INTERVAL_MS);
+		return () => {
+			live = false;
+			clearInterval(timer);
+		};
+	}, [isDeepSeek, accountBalance]);
 	(0, react.useEffect)(() => {
 		const target = selected !== void 0 && isOwnProvider(selected) ? selected : lastOwn.current;
 		if (target === void 0) {
@@ -2359,6 +2490,100 @@ function UsageChip({ call, t, isOwnProvider, useProjection }) {
 			document.removeEventListener("click", close);
 		};
 	}, [open]);
+	if (ccRef !== void 0) {
+		if (ccBalance === void 0) return null;
+		const short$1 = shortLabel(selected ?? "", selected ?? "");
+		return /* @__PURE__ */ (0, react_jsx_runtime.jsxs)("span", {
+			className: "dsh-subscription-usage-chip",
+			children: [
+				/* @__PURE__ */ (0, react_jsx_runtime.jsx)("style", { children: usageChipCss }),
+				/* @__PURE__ */ (0, react_jsx_runtime.jsxs)("button", {
+					type: "button",
+					className: "dsh-subscription-usage-chip-trigger",
+					"aria-expanded": open,
+					title: t("usageChipTitle", { provider: selected ?? "" }),
+					onClick: (event) => {
+						event.stopPropagation();
+						setOpen((was) => !was);
+					},
+					children: [/* @__PURE__ */ (0, react_jsx_runtime.jsx)("span", {
+						className: "dsh-subscription-usage-chip-label",
+						children: short$1
+					}), /* @__PURE__ */ (0, react_jsx_runtime.jsx)("span", {
+						className: "dsh-subscription-usage-chip-cell",
+						children: /* @__PURE__ */ (0, react_jsx_runtime.jsx)("span", {
+							className: "dsh-subscription-usage-chip-value",
+							children: `${ccBalance.currency === "CNY" ? "¥" : ccBalance.currency === "USD" ? "$" : `${ccBalance.currency} `}${ccBalance.total}`
+						})
+					})]
+				}),
+				open && /* @__PURE__ */ (0, react_jsx_runtime.jsxs)("span", {
+					className: "dsh-subscription-usage-chip-panel",
+					role: "status",
+					children: [/* @__PURE__ */ (0, react_jsx_runtime.jsx)("span", {
+						className: "dsh-subscription-usage-chip-full",
+						children: t("usageChipFull", { provider: selected ?? "" })
+					}), /* @__PURE__ */ (0, react_jsx_runtime.jsx)("span", {
+						className: "dsh-subscription-usage-chip-meter",
+						children: /* @__PURE__ */ (0, react_jsx_runtime.jsxs)("span", {
+							className: "dsh-subscription-usage-chip-meterHead",
+							children: [/* @__PURE__ */ (0, react_jsx_runtime.jsx)("span", { children: ccBalance.currency }), /* @__PURE__ */ (0, react_jsx_runtime.jsx)("span", {
+								className: "dsh-subscription-usage-chip-detail",
+								children: ccBalance.total
+							})]
+						})
+					})]
+				})
+			]
+		});
+	}
+	if (isDeepSeek) {
+		if (balance === void 0) return null;
+		const short$1 = shortLabel("deepseek-account", "DeepSeek Account");
+		return /* @__PURE__ */ (0, react_jsx_runtime.jsxs)("span", {
+			className: "dsh-subscription-usage-chip",
+			children: [
+				/* @__PURE__ */ (0, react_jsx_runtime.jsx)("style", { children: usageChipCss }),
+				/* @__PURE__ */ (0, react_jsx_runtime.jsxs)("button", {
+					type: "button",
+					className: "dsh-subscription-usage-chip-trigger",
+					"aria-expanded": open,
+					title: t("usageChipTitle", { provider: "DeepSeek Account" }),
+					onClick: (event) => {
+						event.stopPropagation();
+						setOpen((was) => !was);
+					},
+					children: [/* @__PURE__ */ (0, react_jsx_runtime.jsx)("span", {
+						className: "dsh-subscription-usage-chip-label",
+						children: short$1
+					}), balance.map((wallet) => /* @__PURE__ */ (0, react_jsx_runtime.jsx)("span", {
+						className: "dsh-subscription-usage-chip-cell",
+						children: /* @__PURE__ */ (0, react_jsx_runtime.jsx)("span", {
+							className: "dsh-subscription-usage-chip-value",
+							children: formatBalance(wallet)
+						})
+					}, `${wallet.currency}${wallet.bonus ? ":bonus" : ""}`))]
+				}),
+				open && /* @__PURE__ */ (0, react_jsx_runtime.jsxs)("span", {
+					className: "dsh-subscription-usage-chip-panel",
+					role: "status",
+					children: [/* @__PURE__ */ (0, react_jsx_runtime.jsx)("span", {
+						className: "dsh-subscription-usage-chip-full",
+						children: t("usageChipFull", { provider: "DeepSeek Account" })
+					}), balance.map((wallet) => /* @__PURE__ */ (0, react_jsx_runtime.jsx)("span", {
+						className: "dsh-subscription-usage-chip-meter",
+						children: /* @__PURE__ */ (0, react_jsx_runtime.jsxs)("span", {
+							className: "dsh-subscription-usage-chip-meterHead",
+							children: [/* @__PURE__ */ (0, react_jsx_runtime.jsx)("span", { children: `${wallet.bonus ? "Bonus · " : ""}${wallet.currency}` }), /* @__PURE__ */ (0, react_jsx_runtime.jsx)("span", {
+								className: "dsh-subscription-usage-chip-detail",
+								children: formatBalance(wallet)
+							})]
+						})
+					}, `${wallet.currency}${wallet.bonus ? ":bonus" : ""}`))]
+				})
+			]
+		});
+	}
 	if (reading === void 0) return null;
 	if (reading.window.usedPercent === void 0) return null;
 	formatReset(reading.window.resetsAt, now);
@@ -2455,6 +2680,17 @@ const SECTION_ORDER = 92;
 */
 const CHIP_ORDER = 10;
 /**
+* The credential reference a manually added provider's balance query uses, by provider
+* id.
+*
+* A manually added provider is not one this plugin serves, so the only way to read its
+* balance is cc-switch's query, which needs the provider's API key. The provider's
+* profile names that key by reference (`apiKeyEnv`), and the host's credential service
+* resolves the name; this map is the one place a provider id is tied to that name. A
+* provider with no entry gets no balance.
+*/
+const CC_SWITCH_CREDENTIAL_REFS = { huanling: "HUANLING_API_KEY" };
+/**
 * Required services: `slots` carries the registration seat, `connection` the
 * RPC caller, and `locale` the copy dictionaries.
 */
@@ -2496,7 +2732,10 @@ function apply(ctx) {
 		inject: () => ({
 			call,
 			t,
-			isOwnProvider: (provider) => OWN_PROVIDER_IDS.has(provider)
+			isOwnProvider: (provider) => OWN_PROVIDER_IDS.has(provider),
+			accountBalance: createAccountBalanceCall(connection.rpc),
+			ccSwitchBalance: createCcSwitchBalanceCall(connection.rpc),
+			ccSwitchCredentialRef: (provider) => CC_SWITCH_CREDENTIAL_REFS[provider]
 		})
 	}, UsageChip));
 }