dsh-plugin-token-heatmap 0.1.0

package/lib/index.js

@@ -0,0 +1,335 @@
+var __create = Object.create;
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __knownSymbol = (name2, symbol) => (symbol = Symbol[name2]) ? symbol : Symbol.for("Symbol." + name2);
+var __typeError = (msg) => {
+  throw TypeError(msg);
+};
+var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+var __decoratorStart = (base) => [, , , __create(base?.[__knownSymbol("metadata")] ?? null)];
+var __decoratorStrings = ["class", "method", "getter", "setter", "accessor", "field", "value", "get", "set"];
+var __expectFn = (fn) => fn !== void 0 && typeof fn !== "function" ? __typeError("Function expected") : fn;
+var __decoratorContext = (kind, name2, done, metadata, fns) => ({ kind: __decoratorStrings[kind], name: name2, metadata, addInitializer: (fn) => done._ ? __typeError("Already initialized") : fns.push(__expectFn(fn || null)) });
+var __decoratorMetadata = (array, target) => __defNormalProp(target, __knownSymbol("metadata"), array[3]);
+var __runInitializers = (array, flags, self, value) => {
+  for (var i = 0, fns = array[flags >> 1], n = fns && fns.length; i < n; i++) flags & 1 ? fns[i].call(self) : value = fns[i].call(self, value);
+  return value;
+};
+var __decorateElement = (array, flags, name2, decorators, target, extra) => {
+  var fn, it, done, ctx, access, k = flags & 7, s = !!(flags & 8), p = !!(flags & 16);
+  var j = k > 3 ? array.length + 1 : k ? s ? 1 : 2 : 0, key = __decoratorStrings[k + 5];
+  var initializers = k > 3 && (array[j - 1] = []), extraInitializers = array[j] || (array[j] = []);
+  var desc = k && (!p && !s && (target = target.prototype), k < 5 && (k > 3 || !p) && __getOwnPropDesc(k < 4 ? target : { get [name2]() {
+    return __privateGet(this, extra);
+  }, set [name2](x) {
+    return __privateSet(this, extra, x);
+  } }, name2));
+  k ? p && k < 4 && __name(extra, (k > 2 ? "set " : k > 1 ? "get " : "") + name2) : __name(target, name2);
+  for (var i = decorators.length - 1; i >= 0; i--) {
+    ctx = __decoratorContext(k, name2, done = {}, array[3], extraInitializers);
+    if (k) {
+      ctx.static = s, ctx.private = p, access = ctx.access = { has: p ? (x) => __privateIn(target, x) : (x) => name2 in x };
+      if (k ^ 3) access.get = p ? (x) => (k ^ 1 ? __privateGet : __privateMethod)(x, target, k ^ 4 ? extra : desc.get) : (x) => x[name2];
+      if (k > 2) access.set = p ? (x, y) => __privateSet(x, target, y, k ^ 4 ? extra : desc.set) : (x, y) => x[name2] = y;
+    }
+    it = (0, decorators[i])(k ? k < 4 ? p ? extra : desc[key] : k > 4 ? void 0 : { get: desc.get, set: desc.set } : target, ctx), done._ = 1;
+    if (k ^ 4 || it === void 0) __expectFn(it) && (k > 4 ? initializers.unshift(it) : k ? p ? extra = it : desc[key] = it : target = it);
+    else if (typeof it !== "object" || it === null) __typeError("Object expected");
+    else __expectFn(fn = it.get) && (desc.get = fn), __expectFn(fn = it.set) && (desc.set = fn), __expectFn(fn = it.init) && initializers.unshift(fn);
+  }
+  return k || __decoratorMetadata(array, target), desc && __defProp(target, name2, desc), p ? k ^ 4 ? extra : desc : target;
+};
+var __publicField = (obj, key, value) => __defNormalProp(obj, typeof key !== "symbol" ? key + "" : key, value);
+var __accessCheck = (obj, member, msg) => member.has(obj) || __typeError("Cannot " + msg);
+var __privateIn = (member, obj) => Object(obj) !== obj ? __typeError('Cannot use the "in" operator on this value') : member.has(obj);
+var __privateGet = (obj, member, getter) => (__accessCheck(obj, member, "read from private field"), getter ? getter.call(obj) : member.get(obj));
+var __privateSet = (obj, member, value, setter) => (__accessCheck(obj, member, "write to private field"), setter ? setter.call(obj, value) : member.set(obj, value), value);
+var __privateMethod = (obj, member, method) => (__accessCheck(obj, member, "access private method"), method);
+
+// src/index.ts
+import { dshHomePath } from "@deepseek-ai/dsh-home-paths";
+
+// src/day.ts
+function dayKeyOf(time) {
+  const d = new Date(time);
+  const m = String(d.getMonth() + 1).padStart(2, "0");
+  const day = String(d.getDate()).padStart(2, "0");
+  return d.getFullYear() + "-" + m + "-" + day;
+}
+function cutoffKey(now) {
+  const d = new Date(now);
+  return dayKeyOf(new Date(d.getFullYear(), d.getMonth() - 12, 1).getTime());
+}
+
+// src/backfill.ts
+import { mkdirSync, readFileSync, writeFileSync } from "node:fs";
+import { join } from "node:path";
+
+// src/usage.ts
+function emptyTotals() {
+  return { total: 0, input: 0, output: 0, cacheRead: 0, cacheWrite: 0 };
+}
+function addUsage(map, day, input, output, cacheRead = 0, cacheWrite = 0) {
+  const entry = map[day] ?? (map[day] = emptyTotals());
+  entry.input += input;
+  entry.output += output;
+  entry.cacheRead += cacheRead;
+  entry.cacheWrite += cacheWrite;
+  entry.total = entry.input + entry.output + entry.cacheRead + entry.cacheWrite;
+}
+function mergeMaps(target, source) {
+  for (const [day, t] of Object.entries(source)) {
+    addUsage(target, day, t.input, t.output, t.cacheRead, t.cacheWrite);
+  }
+}
+function prune(map, cutoff) {
+  for (const key of Object.keys(map)) {
+    if (key < cutoff) delete map[key];
+  }
+}
+
+// src/backfill.ts
+function extractSamples(events) {
+  const samples = [];
+  for (const event of events) {
+    if (event.type !== "assistant/message") continue;
+    const usage = event.data.usage;
+    if (usage === void 0) continue;
+    samples.push({
+      day: dayKeyOf(event.time),
+      input: usage.inputTokens,
+      output: usage.outputTokens,
+      cacheRead: usage.cacheReadTokens ?? 0,
+      cacheWrite: usage.cacheWriteTokens ?? 0
+    });
+  }
+  return samples;
+}
+function aggregateSamples(global, bySession, sessionId, samples) {
+  const map = bySession[sessionId] ?? (bySession[sessionId] = {});
+  for (const sample of samples) {
+    addUsage(map, sample.day, sample.input, sample.output, sample.cacheRead, sample.cacheWrite);
+    addUsage(global, sample.day, sample.input, sample.output, sample.cacheRead, sample.cacheWrite);
+  }
+}
+var BackfillRunner = class {
+  dir;
+  persistence;
+  constructor(dir, persistence) {
+    this.dir = dir;
+    this.persistence = persistence;
+  }
+  meta() {
+    try {
+      return JSON.parse(readFileSync(join(this.dir, "meta.json"), "utf8"));
+    } catch {
+      return { done: false, scanned: 0, skipped: 0, startedAt: null, finishedAt: null };
+    }
+  }
+  async run() {
+    const prior = this.meta();
+    if (prior.done) {
+      return { global: {}, bySession: {}, scannedIds: [], scanned: prior.scanned, skipped: prior.skipped, enumerated: false };
+    }
+    if (this.persistence === void 0) {
+      this.writeMeta({ done: true, scanned: 0, skipped: 0, startedAt: Date.now(), finishedAt: Date.now() });
+      return { global: {}, bySession: {}, scannedIds: [], scanned: 0, skipped: 0, enumerated: false };
+    }
+    this.writeMeta({ done: false, scanned: 0, skipped: 0, startedAt: Date.now(), finishedAt: null });
+    const headers = await this.persistence.list();
+    const global = {};
+    const bySession = {};
+    const scannedIds = [];
+    let skipped = 0;
+    for (const header of headers) {
+      scannedIds.push(header.id);
+      try {
+        const { events } = await this.persistence.inspect(header.id);
+        aggregateSamples(global, bySession, header.id, extractSamples(events));
+      } catch (error) {
+        skipped += 1;
+        console.error(`[token-heatmap] backfill skipped session ${header.id}:`, error);
+      }
+    }
+    const cutoff = cutoffKey(Date.now());
+    prune(global, cutoff);
+    for (const map of Object.values(bySession)) prune(map, cutoff);
+    this.writeMeta({ done: true, scanned: headers.length, skipped, startedAt: prior.startedAt ?? Date.now(), finishedAt: Date.now() });
+    return { global, bySession, scannedIds, scanned: headers.length, skipped, enumerated: true };
+  }
+  writeMeta(meta) {
+    try {
+      mkdirSync(this.dir, { recursive: true });
+      writeFileSync(join(this.dir, "meta.json"), JSON.stringify(meta), "utf8");
+    } catch (error) {
+      console.error("[token-heatmap] meta write failed:", error);
+    }
+  }
+};
+
+// src/service.ts
+import { Remote, TypertRemoteService } from "@deepseek-ai/dsh-typert-protocol";
+var _getBackfillStatus_dec, _getSessionUsage_dec, _getGlobalUsage_dec, _a, _init;
+var TokenHeatmapService = class extends (_a = TypertRemoteService, _getGlobalUsage_dec = [Remote], _getSessionUsage_dec = [Remote], _getBackfillStatus_dec = [Remote], _a) {
+  constructor(ctx, store, backfillMeta) {
+    super(ctx, "tokenHeatmap");
+    __runInitializers(_init, 5, this);
+    __publicField(this, "store");
+    __publicField(this, "backfillMeta");
+    this.store = store;
+    this.backfillMeta = backfillMeta;
+  }
+  async getGlobalUsage() {
+    return { version: this.store.getVersion(), days: this.store.snapshotGlobal() };
+  }
+  async getSessionUsage(sessionId) {
+    return { version: this.store.getVersion(), days: this.store.snapshotSession(sessionId) };
+  }
+  async getBackfillStatus() {
+    return this.backfillMeta();
+  }
+};
+_init = __decoratorStart(_a);
+__decorateElement(_init, 1, "getGlobalUsage", _getGlobalUsage_dec, TokenHeatmapService);
+__decorateElement(_init, 1, "getSessionUsage", _getSessionUsage_dec, TokenHeatmapService);
+__decorateElement(_init, 1, "getBackfillStatus", _getBackfillStatus_dec, TokenHeatmapService);
+__decoratorMetadata(_init, TokenHeatmapService);
+
+// src/usage-store.ts
+import { mkdirSync as mkdirSync2, readFileSync as readFileSync2, renameSync, writeFileSync as writeFileSync2 } from "node:fs";
+import { join as join2 } from "node:path";
+var WRITE_DELAY_MS = 5e3;
+var UsageStore = class {
+  dir;
+  global = {};
+  bySession = {};
+  dirty = false;
+  timer = null;
+  version = 0;
+  constructor(dir) {
+    this.dir = dir;
+  }
+  getVersion() {
+    return this.version;
+  }
+  load() {
+    const read = (name2) => {
+      try {
+        return JSON.parse(readFileSync2(join2(this.dir, name2), "utf8"));
+      } catch (error) {
+        console.error(`[token-heatmap] failed to load ${name2}, starting empty:`, error);
+        return void 0;
+      }
+    };
+    this.global = read("global.json")?.days ?? {};
+    this.bySession = read("sessions.json")?.bySession ?? {};
+    this.pruneAll();
+  }
+  addUsage(sessionId, day, input, output, cacheRead = 0, cacheWrite = 0) {
+    addUsage(this.global, day, input, output, cacheRead, cacheWrite);
+    if (sessionId !== null) {
+      const map = this.bySession[sessionId] ?? (this.bySession[sessionId] = {});
+      addUsage(map, day, input, output, cacheRead, cacheWrite);
+    }
+    this.version += 1;
+    this.scheduleWrite();
+  }
+  mergeBackfill(global, bySession) {
+    mergeMaps(this.global, global);
+    for (const [id, map] of Object.entries(bySession)) {
+      mergeMaps(this.bySession[id] ?? (this.bySession[id] = {}), map);
+    }
+    this.version += 1;
+    this.scheduleWrite();
+  }
+  pruneSessions(liveIds) {
+    for (const id of Object.keys(this.bySession)) {
+      if (!liveIds.has(id)) delete this.bySession[id];
+    }
+  }
+  /** 启动时应用回填结果：只有真正枚举过会话库的结果才有资格 prune，否则会清空全部按会话数据。 */
+  applyBackfill(result) {
+    if (result.enumerated) {
+      this.pruneSessions(new Set(result.scannedIds));
+    }
+    this.mergeBackfill(result.global, result.bySession);
+    this.flush();
+  }
+  flush() {
+    if (!this.dirty) return;
+    this.dirty = false;
+    this.pruneAll();
+    try {
+      mkdirSync2(this.dir, { recursive: true });
+      this.atomicWrite("global.json", JSON.stringify({ version: 1, days: this.global }));
+      this.atomicWrite("sessions.json", JSON.stringify({ version: 1, bySession: this.bySession }));
+    } catch (error) {
+      console.error("[token-heatmap] store write failed (degraded to memory):", error);
+    }
+  }
+  snapshotGlobal() {
+    return this.deepCopy(this.global);
+  }
+  snapshotSession(id) {
+    return this.deepCopy(this.bySession[id] ?? {});
+  }
+  pruneAll() {
+    const cutoff = cutoffKey(Date.now());
+    prune(this.global, cutoff);
+    for (const map of Object.values(this.bySession)) prune(map, cutoff);
+  }
+  scheduleWrite() {
+    this.dirty = true;
+    if (this.timer !== null) return;
+    this.timer = setTimeout(() => {
+      this.timer = null;
+      this.flush();
+    }, WRITE_DELAY_MS);
+  }
+  atomicWrite(name2, content) {
+    const target = join2(this.dir, name2);
+    const tmp = `${target}.tmp`;
+    writeFileSync2(tmp, content, "utf8");
+    renameSync(tmp, target);
+  }
+  deepCopy(map) {
+    const out = {};
+    for (const [key, value] of Object.entries(map)) out[key] = { ...value };
+    return out;
+  }
+};
+
+// src/index.ts
+var name = "token-heatmap";
+var inject = ["sessionPersistence"];
+function apply(ctx) {
+  const dir = dshHomePath("token-heatmap");
+  const store = new UsageStore(dir);
+  store.load();
+  const persistence = ctx.get("sessionPersistence");
+  const runner = new BackfillRunner(dir, persistence);
+  new TokenHeatmapService(ctx, store, () => runner.meta());
+  ctx.on("session/event", (session, event) => {
+    if (event.type !== "assistant/message") return;
+    const usage = event.data.usage;
+    if (usage === void 0) return;
+    store.addUsage(
+      session.id,
+      dayKeyOf(event.time),
+      usage.inputTokens,
+      usage.outputTokens,
+      usage.cacheReadTokens ?? 0,
+      usage.cacheWriteTokens ?? 0
+    );
+  });
+  void runner.run().then((result) => {
+    store.applyBackfill(result);
+  });
+  ctx.effect(() => () => store.flush());
+}
+export {
+  apply,
+  inject,
+  name
+};

package/package.json

@@ -0,0 +1,85 @@
+{
+  "name": "dsh-plugin-token-heatmap",
+  "version": "0.1.0",
+  "description": "GitHub-style daily token usage heatmap for DeepSeek Harness Web",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/P-CAmilE/dsh-plugin-token-heatmap.git"
+  },
+  "keywords": [
+    "deepseek",
+    "harness",
+    "dsh",
+    "dsh-plugin",
+    "token-usage",
+    "heatmap"
+  ],
+  "type": "module",
+  "main": "lib/index.js",
+  "types": "lib/types/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./lib/types/index.d.ts",
+      "default": "./lib/index.js"
+    },
+    "./client": {
+      "types": "./lib/types/client/index.d.ts",
+      "default": "./lib/client.js"
+    },
+    "./package.json": "./package.json"
+  },
+  "files": [
+    "lib",
+    "cordis.patch.yml"
+  ],
+  "dsh": {
+    "bundle": {
+      "patch": "./cordis.patch.yml"
+    },
+    "client": {
+      "inject": [
+        "@deepseek-ai/dsh-client-runtime",
+        "@deepseek-ai/dsh-api-gateway"
+      ],
+      "platform": "web"
+    }
+  },
+  "scripts": {
+    "build": "node build.mjs",
+    "prepare": "pnpm build",
+    "prepack": "pnpm build",
+    "typecheck": "tsc --noEmit",
+    "test": "node --import ./test/helpers/ts-decorators-preload.mjs --test \"test/**/*.test.ts\"",
+    "test:ui": "vitest run"
+  },
+  "peerDependencies": {
+    "@deepseek-ai/cordis": "^4.0.1",
+    "@deepseek-ai/dsh-api-gateway": "^0.1.1-rc.2",
+    "@deepseek-ai/dsh-client-runtime": "^0.1.1-rc.2",
+    "@deepseek-ai/dsh-home-paths": "^0.1.1-rc.2",
+    "@deepseek-ai/dsh-session": "^0.1.1-rc.2",
+    "@deepseek-ai/dsh-session-persistence": "^0.1.1-rc.2",
+    "@deepseek-ai/dsh-typert-protocol": "^0.1.1-rc.2",
+    "react": "^18.2.0"
+  },
+  "devDependencies": {
+    "@deepseek-ai/cordis": "4.0.1",
+    "@deepseek-ai/dsh-api-gateway": "0.1.1-rc.2",
+    "@deepseek-ai/dsh-client-runtime": "0.1.1-rc.2",
+    "@deepseek-ai/dsh-home-paths": "0.1.1-rc.2",
+    "@deepseek-ai/dsh-session": "0.1.1-rc.2",
+    "@deepseek-ai/dsh-session-persistence": "0.1.1-rc.2",
+    "@deepseek-ai/dsh-typert-protocol": "0.1.1-rc.2",
+    "@testing-library/react": "^16.3.2",
+    "@types/node": "^26.2.0",
+    "@types/react": "~18.3.1",
+    "@types/react-dom": "~18.3.1",
+    "esbuild": "^0.25.0",
+    "jsdom": "^29.1.1",
+    "react": "18.3.1",
+    "react-dom": "18.3.1",
+    "typescript": "^5.9.0",
+    "vitest": "^4.0.0"
+  }
+}