dsh-multi-router 0.5.9 → 0.6.0

package/lib/{chunk-I4EW5TI7.js → chunk-OCMOMSB4.js}

@@ -6,21 +6,27 @@ import {
   refreshWorkBuddyCheckinCredential,
   supportsCheckin,
   workbuddyToken
-} from "./chunk-U2MU2M5C.js";
+} from "./chunk-NSKCFVHP.js";
 import {
   discoverLocalAccounts,
   importAccount,
   localAccountKey,
   normalizeLocalAccountKey,
   readPortableCredential
-} from "./chunk-S6ZQ45SS.js";
+} from "./chunk-JUOESKAK.js";
+import {
+  modelWithCurrentPromotion
+} from "./chunk-TEZ2FMEO.js";
+import {
+  ApiService
+} from "./chunk-SNEZMYNU.js";
 import {
   hostIsLoopback
 } from "./chunk-DNQPZMPF.js";
 import {
   Telemetry,
   matchModel
-} from "./chunk-J3XO7RTF.js";
+} from "./chunk-ROZINFTV.js";
 import {
   PROVIDERS,
   PROVIDER_NAMES,
@@ -31,16 +37,13 @@ import {
   routeReasoning,
   validateConfig,
   writePrivateJson
-} from "./chunk-MCHV2R25.js";
-import {
-  modelWithCurrentPromotion
-} from "./chunk-TEZ2FMEO.js";
+} from "./chunk-WVNY4LSL.js";
 
 // src/manager.mjs
 import { createHash, randomBytes as randomBytes3, timingSafeEqual } from "node:crypto";
 
 // package.json
-var version = "0.5.9";
+var version = "0.6.0";
 
 // src/manager.mjs
 import { readFile as readFile4, mkdir as mkdir2, mkdtemp as mkdtemp2, rm as rm3 } from "node:fs/promises";
@@ -102,7 +105,7 @@ var derive = promisify(scrypt);
 var FORMAT = "dsh-multi-router";
 var MAX_BYTES = 512 * 1024;
 var pick = (value, keys) => Object.fromEntries(keys.filter((k) => Object.hasOwn(value ?? {}, k)).map((k) => [k, value[k]]));
-var options = ["cooldownMs", "requestTimeoutMs", "catalogTtlMs", "retryServerErrors", "modelVisibility", "usageRetentionDays", "checkinSchedule"];
+var options = ["cooldownMs", "requestTimeoutMs", "catalogTtlMs", "retryServerErrors", "modelVisibility", "usageRetentionDays", "checkinSchedule", "sidebarUsage"];
 function settings(config) {
   return {
     ...pick(config, options),
@@ -1003,6 +1006,7 @@ var WorkBuddyLogin = class {
 
 // src/manager.mjs
 var MANAGEMENT_PATH = "/.well-known/dsh-multi-router";
+var SUMMARY_PATH = `${MANAGEMENT_PATH}/summary`;
 var revisionOf = (raw) => createHash("sha256").update(JSON.stringify(raw)).digest("hex");
 var ManagementError = class extends Error {
   constructor(status, message) {
@@ -1060,6 +1064,7 @@ var Manager = class {
     this.checkins = new Checkins(join5(dirname3(file), "checkins.json"), this);
     this.workbuddyLogin = new WorkBuddyLogin(join5(dirname3(file), "workbuddy-login.json"), { fetchImpl: pluginAccess.fetchImpl });
     this.setupFile = join5(dirname3(file), "initial-setup.json");
+    this.api = new ApiService(join5(dirname3(file), "api-service.json"), this);
     router.telemetry = this.telemetry;
     router.monitor = (a) => this.monitorAccount(a, { maxAgeMs: 6e4 });
     router.prepareFree = () => this.checkFreeModels(true);
@@ -1087,6 +1092,7 @@ var Manager = class {
     }
     this.telemetry.configure(this.config.usageRetentionDays);
     await this.telemetry.load();
+    await this.api.load();
     await Promise.all(this.config.accounts.map((a) => this.readLogin(a)));
     this.publish();
   }
@@ -1400,6 +1406,19 @@ var Manager = class {
     this.refreshes.set(account.id, promise);
     return promise;
   }
+  summaryState() {
+    const today = new Date(this.telemetry.now()).toLocaleDateString("en-CA", { timeZone: "Asia/Shanghai" });
+    const daily = this.telemetry.data.days[today] ?? {};
+    return {
+      pluginVersion: version,
+      enabled: this.config.sidebarUsage,
+      today,
+      tokens: daily.tokens ?? 0,
+      requests: daily.requests ?? 0,
+      unmetered: daily.unmetered ?? 0,
+      totalTokens: Object.values(this.telemetry.data.accounts).reduce((sum, row) => sum + (row.tokens ?? 0), 0)
+    };
+  }
   state() {
     const status = this.router.status();
     return {
@@ -1412,6 +1431,7 @@ var Manager = class {
       pluginUpdates: this.plugins.state(),
       checkins: this.checkins.state(),
       telemetry: this.telemetry.state(),
+      apiService: this.api.state(),
       accounts: this.config.accounts.map((a) => ({
         id: a.id,
         provider: a.provider,
@@ -1439,12 +1459,22 @@ var Manager = class {
         cooldownMs: this.config.cooldownMs,
         requestTimeoutMs: this.config.requestTimeoutMs,
         retryServerErrors: this.config.retryServerErrors,
-        usageRetentionDays: this.config.usageRetentionDays
+        usageRetentionDays: this.config.usageRetentionDays,
+        sidebarUsage: this.config.sidebarUsage
       }
     };
   }
   async action(input) {
     if (!input || typeof input !== "object" || Array.isArray(input)) bad("\u8BF7\u6C42\u683C\u5F0F\u65E0\u6548");
+    if (input.action?.startsWith("api.")) {
+      let result;
+      try {
+        result = await this.api.action(input);
+      } catch (e) {
+        bad(e.message);
+      }
+      return { ...this.state(), ...result };
+    }
     if (input.action === "plugin.reload") {
       if (!this.runtime) bad("\u5F53\u524D\u540E\u53F0\u4E0D\u652F\u6301\u91CD\u65B0\u52A0\u8F7D");
       this.runtime.retryAt = 0;
@@ -1787,7 +1817,7 @@ var Manager = class {
           if (!input.schedule) bad("\u8BF7\u586B\u5199\u7B7E\u5230\u8BA1\u5212");
           raw.checkinSchedule = input.schedule;
         } else if (input.action === "options.save") {
-          for (const key of ["cooldownMs", "requestTimeoutMs", "retryServerErrors", "usageRetentionDays"]) if (input[key] !== void 0) raw[key] = input[key];
+          for (const key of ["cooldownMs", "requestTimeoutMs", "retryServerErrors", "usageRetentionDays", "sidebarUsage"]) if (input[key] !== void 0) raw[key] = input[key];
         } else bad("\u672A\u77E5\u64CD\u4F5C");
         let config;
         try {
@@ -1854,7 +1884,7 @@ var Manager = class {
     };
   }
 };
-function managementHandler(manager) {
+function managementHandler(manager, { summaryOnly = false } = {}) {
   const key = randomBytes3(32).toString("base64url");
   const json = (res, status, value) => {
     res.writeHead(status, { "Content-Type": "application/json", "Cache-Control": "no-store", "X-Content-Type-Options": "nosniff" });
@@ -1868,9 +1898,10 @@ function managementHandler(manager) {
       if (req.headers.origin && req.headers.origin !== `http://${req.headers.host}` && req.headers.origin !== `https://${req.headers.host}`) throw new ManagementError(403, "\u6765\u6E90\u65E0\u6548");
       if (req.headers["sec-fetch-site"] === "cross-site") throw new ManagementError(403, "\u6765\u6E90\u65E0\u6548");
       if (req.method === "GET") {
-        json(res, 200, { ...manager.state(), key });
+        json(res, 200, summaryOnly ? manager.summaryState() : { ...manager.state(), key });
         return;
       }
+      if (summaryOnly) throw new ManagementError(405, "\u65B9\u6CD5\u65E0\u6548");
       if (req.method !== "POST") throw new ManagementError(405, "\u65B9\u6CD5\u65E0\u6548");
       const actual = Buffer.from(req.headers["x-multi-router-key"] ?? ""), expected = Buffer.from(key);
       if (actual.length !== expected.length || !timingSafeEqual(actual, expected)) throw new ManagementError(401, "\u8BF7\u5237\u65B0\u914D\u7F6E\u9875\u9762");
@@ -1904,9 +1935,10 @@ export {
   version,
   PluginReload,
   MANAGEMENT_PATH,
+  SUMMARY_PATH,
   ManagementError,
   modelInfo,
   Manager,
   managementHandler
 };
-//# sourceMappingURL=chunk-I4EW5TI7.js.map
+//# sourceMappingURL=chunk-OCMOMSB4.js.map