dsh-plugin-deepseek-usage 0.1.0

package/cordis.patch.yml

@@ -0,0 +1,11 @@
+# deepseek-usage — bundle patch layer.
+#
+# This file turns the package into a DSH *profile bundle* (see
+# `package.json` → `dsh.bundle.patch`): the desktop app appends the package to
+# the `web` profile's `dsh.profile.bundles` list on first launch, and this
+# layer inserts the plugin row into the composed tree — the host half mounts
+# the `/deepseek-usage` route, and the client-modules node half scans the entry
+# and serves the sidebar balance button bundle.
+- insert:
+    - id: deepseek-usage
+      name: 'dsh-plugin-deepseek-usage'

package/lib/client.js

@@ -0,0 +1,146 @@
+window.__ModuleLoader__.load({
+  id: "dsh-plugin-deepseek-usage",
+  factory: (require) => {
+    var module = { exports: {} };
+    var exports = module.exports;
+    Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
+
+    var React = require("react");
+    var createElement = React.createElement;
+
+    var css = [
+      ".dshu-trigger{box-sizing:border-box;cursor:pointer;width:100%;height:34px;color:var(--dsw-alias-label-primary,#111);background:transparent;border:none;border-radius:12px;flex:none;display:flex;align-items:center;gap:8px;padding:6px 10px;font-family:inherit;font-size:14px}",
+      ".dshu-trigger:hover{background:var(--dsw-alias-interactive-bg-hover,#f0f0f0)}",
+      ".dshu-trigger.dshu-rail{justify-content:center;gap:0;width:36px;height:36px;padding:0}",
+      ".dshu-trigger.dshu-rail .dshu-status-dot{position:absolute;right:5px;bottom:5px}",
+      ".dshu-trigger-label{white-space:nowrap;overflow:hidden}",
+      ".dshu-amount{margin-left:auto;flex:none;background:rgba(46,160,90,.16);color:#2ea05a;border-radius:999px;padding:1px 8px;font-size:12px;line-height:18px;font-variant-numeric:tabular-nums;white-space:nowrap}",
+      ".dshu-amount-bad{background:rgba(211,60,60,.16);color:var(--dsw-alias-state-error-primary,#d33)}",
+      ".dshu-status-dot{width:8px;height:8px;border-radius:50%;flex:none;display:inline-block}",
+      ".dshu-status-dot.dshu-dot-ok{background:#2ea05a}",
+      ".dshu-status-dot.dshu-dot-bad{background:var(--dsw-alias-state-error-primary,#d33)}",
+      ".dshu-status-dot.dshu-dot-wait{background:var(--dsw-alias-label-tertiary,#999)}"
+    ].join("\n");
+
+    var tagId = "dsh-plugin-deepseek-usage/usage.css";
+    if (typeof document !== "undefined" && document.querySelector('style[data-plugin-css="' + tagId + '"]') === null) {
+      var tag = document.createElement("style");
+      tag.dataset.plugin = "dsh-plugin-deepseek-usage";
+      tag.dataset.pluginCss = tagId;
+      tag.textContent = css;
+      document.head.appendChild(tag);
+    }
+
+    function fmt(value) {
+      if (value === undefined || value === null || value === "") return "—";
+      return String(value);
+    }
+
+    /**
+     * 返回当前「用量」所属人：9:00–12:00、14:00–18:00 为梁文锋，其余时间为梁文谷。
+     * 边界按 [开始, 结束) 处理，即 12:00 / 18:00 起切换为梁文谷。
+     */
+    function getActiveUser(now) {
+      var hour = new Date(now).getHours();
+      var wenfeng = (hour >= 9 && hour < 12) || (hour >= 14 && hour < 18);
+      return wenfeng ? "梁文锋" : "梁文谷";
+    }
+
+    /** Sidebar footer button: shows the DeepSeek account balance directly. */
+    function BalanceButton(props) {
+      var wide = props.wide;
+      var stateRef = React.useState({ phase: "loading", total: null, currency: null, available: false, error: null });
+      var state = stateRef[0];
+      var setState = stateRef[1];
+
+      var nowRef = React.useState(Date.now());
+      var now = nowRef[0];
+      var setNow = nowRef[1];
+
+      var load = React.useCallback(function () {
+        fetch("/deepseek-usage", { method: "GET", cache: "no-store" })
+          .then(function (r) { return r.json(); })
+          .then(function (j) {
+            if (j && j.ok) {
+              var data = j.data || {};
+              var infos = Array.isArray(data.balance_infos) ? data.balance_infos : [];
+              var first = infos.length > 0 ? infos[0] : null;
+              setState({
+                phase: "ready",
+                total: first && first.total_balance !== undefined ? String(first.total_balance) : null,
+                currency: first && first.currency ? String(first.currency) : null,
+                available: data.is_available !== false,
+                error: null
+              });
+            } else {
+              setState({ phase: "error", total: null, currency: null, available: false, error: (j && (j.message || j.error)) || "未知错误" });
+            }
+          })
+          .catch(function (e) {
+            setState({ phase: "error", total: null, currency: null, available: false, error: (e && e.message) || String(e) });
+          });
+      }, []);
+
+      React.useEffect(function () {
+        load();
+        var timer = window.setInterval(load, 60000);
+        var nameTimer = window.setInterval(function () { setNow(Date.now()); }, 30000);
+        return function () {
+          window.clearInterval(timer);
+          window.clearInterval(nameTimer);
+        };
+      }, [load]);
+
+      var phase = state.phase;
+      var amount = state.total;
+      var userName = getActiveUser(now);
+      var titleText = "DeepSeek 余额（" + userName + "）";
+      if (phase === "error") titleText = "余额查询失败（" + userName + "）：" + state.error;
+      else if (phase === "ready" && amount !== null) titleText = "DeepSeek 余额（" + userName + "）：" + amount + (state.currency ? " " + state.currency : "") + (state.available ? "（可用）" : "（不可用）");
+      else if (phase === "ready") titleText = "DeepSeek 余额（" + userName + "）：暂无数据";
+      else titleText = "DeepSeek 余额（" + userName + "）：加载中…";
+
+      var dotClass = phase === "error" || (phase === "ready" && !state.available) ? "dshu-dot-bad"
+        : phase === "ready" && amount !== null ? "dshu-dot-ok"
+        : "dshu-dot-wait";
+
+      return createElement(
+        "button",
+        {
+          type: "button",
+          className: "dshu-trigger" + (wide ? "" : " dshu-rail"),
+          "aria-label": titleText,
+          title: titleText,
+          onClick: load
+        },
+        createElement(
+          "svg",
+          { viewBox: "0 0 16 16", width: wide ? 14 : 18, height: wide ? 14 : 18, "aria-hidden": true },
+          createElement("path", { d: "M2 13h12M3 9h4l2-4 2 2h4", stroke: "currentColor", strokeWidth: "1.5", fill: "none", strokeLinecap: "round", strokeLinejoin: "round" })
+        ),
+        wide ? createElement("span", { className: "dshu-trigger-label" }, "用量 · " + userName) : null,
+        !wide ? createElement("span", { className: "dshu-status-dot " + dotClass }) : null,
+        wide && phase === "ready" && amount !== null
+          ? createElement("span", { className: "dshu-amount" + (!state.available ? " dshu-amount-bad" : "") }, amount + (state.currency ? " " + state.currency : ""))
+          : null
+      );
+    }
+
+    /** Services required by this client plugin. */
+    var inject = ["slots"];
+
+    function apply(ctx) {
+      ctx.slots.inject("sidebar.footer.action", function () {
+        return ctx.slots.register({
+          name: "sidebar.footer.action",
+          id: "deepseek-usage",
+          order: 10
+        }, BalanceButton);
+      });
+    }
+
+    exports.apply = apply;
+    exports.inject = inject;
+    return module.exports;
+  }
+});

package/lib/index.js

@@ -0,0 +1,105 @@
+import z from "@deepseek-ai/schemastery";
+import { credentialRef } from "@deepseek-ai/dsh-credentials";
+
+/**
+ * dsh-plugin-deepseek-usage — host half.
+ *
+ * Registers one exact HTTP route (`/deepseek-usage`) on the Web server that:
+ *   1. resolves the DeepSeek API key through `ctx.credentials` (per request,
+ *      so a rotated key reaches the very next query without a restart),
+ *   2. calls the official balance endpoint `GET <baseURL>/user/balance`,
+ *   3. returns a small JSON envelope the browser half renders.
+ *
+ * The key never leaves the host and never crosses the wire: the route answers
+ * with balance figures only.
+ */
+
+/** Stable Cordis plugin name. */
+const name = "deepseek-usage";
+
+/** Services required before the route can be registered. */
+const inject = ["webServer", "credentials"];
+
+/** Plugin config (schema defaults applied by the loader). */
+const Config = z.object({
+  /** Credential reference (an env-var-shaped name) holding the DeepSeek key. */
+  apiKeyEnv: z.string().default("DEEPSEEK_API_KEY"),
+  /** DeepSeek API base; the balance endpoint is `<baseURL>/user/balance`. */
+  baseURL: z.string().default("https://api.deepseek.com")
+});
+
+function writeJson(res, status, payload) {
+  const body = JSON.stringify(payload);
+  res.writeHead(status, {
+    "content-type": "application/json; charset=utf-8",
+    "cache-control": "no-store"
+  });
+  res.end(body);
+}
+
+function apply(ctx, config) {
+  const apiKeyEnv = config.apiKeyEnv;
+  const baseURL = config.baseURL.replace(/\/+$/, "");
+
+  ctx.effect(() => ctx.webServer.register({
+    kind: "exact",
+    path: "/deepseek-usage",
+    handler: async (req, res) => {
+      if (req.method !== "GET") {
+        writeJson(res, 405, { ok: false, error: "METHOD_NOT_ALLOWED" });
+        return;
+      }
+      try {
+        const hit = await ctx.credentials.resolve(credentialRef(apiKeyEnv));
+        if (hit === undefined || hit.value === "") {
+          writeJson(res, 200, {
+            ok: false,
+            error: "MISSING_CREDENTIAL",
+            message: `API key not configured (${apiKeyEnv})`
+          });
+          return;
+        }
+
+        const upstream = await fetch(`${baseURL}/user/balance`, {
+          method: "GET",
+          headers: {
+            Authorization: `Bearer ${hit.value}`,
+            Accept: "application/json"
+          }
+        });
+
+        const text = await upstream.text();
+        let parsed = null;
+        try {
+          parsed = text ? JSON.parse(text) : null;
+        } catch {
+          parsed = null;
+        }
+
+        if (!upstream.ok) {
+          const upstreamMessage =
+            (parsed && (parsed.error?.message || parsed.message)) ||
+            text ||
+            `HTTP ${upstream.status}`;
+          writeJson(res, 200, {
+            ok: false,
+            error: "UPSTREAM_ERROR",
+            status: upstream.status,
+            message: upstreamMessage
+          });
+          return;
+        }
+
+        writeJson(res, 200, { ok: true, status: upstream.status, data: parsed });
+      } catch (err) {
+        writeJson(res, 200, {
+          ok: false,
+          error: "TRANSPORT",
+          message: err instanceof Error ? err.message : String(err)
+        });
+      }
+    }
+  }), "deepseek-usage: http route");
+}
+
+export { Config, apply, inject, name };

package/package.json

@@ -0,0 +1,31 @@
+{
+  "name": "dsh-plugin-deepseek-usage",
+  "description": "DeepSeek Harness plugin: show the current DeepSeek account balance right on the sidebar button, queried through the resolved API key",
+  "version": "0.1.0",
+  "type": "module",
+  "main": "lib/index.js",
+  "exports": {
+    ".": "./lib/index.js",
+    "./client": "./lib/client.js",
+    "./package.json": "./package.json"
+  },
+  "dsh": {
+    "client": {
+      "inject": [
+        "@deepseek-ai/dsh-client-runtime",
+        "@deepseek-ai/dsh-client-ui-sidebar"
+      ],
+      "platform": "web"
+    },
+    "bundle": {
+      "patch": "./cordis.patch.yml"
+    }
+  },
+  "license": "MIT",
+  "peerDependencies": {
+    "@deepseek-ai/cordis": "^4.0.1",
+    "@deepseek-ai/dsh-credentials": "^0.1.0-rc.6",
+    "@deepseek-ai/dsh-host-webserver": "^0.1.0-rc.6",
+    "@deepseek-ai/schemastery": "^3.18.1"
+  }
+}