dsh-plugin-subscription-usage 0.0.0-stage → 0.1.0

package/lib/usage-providers/kimi.js

@@ -0,0 +1,52 @@
+/** Kimi Code usage from `/coding/v1/usages`. */
+import { monthBefore } from '../providers/copilot.js';
+import { bearerJson, clampPercent, finite, isRecord, KeyRejectedError, timestamp } from './common.js';
+/** Which host serves a key is not documented, so the second is tried when the first rejects it. */
+const URLS = ['https://api.kimi.com/coding/v1/usages', 'https://api.kimi.ai/coding/v1/usages'];
+const HOUR_MS = 3_600_000;
+const RATIOS = [
+    { key: 'limit_5h', kind: 'session', durationMs: 5 * HOUR_MS },
+    { key: 'limit_7d', kind: 'weekly', durationMs: 7 * 24 * HOUR_MS },
+    { key: 'limit_month_total', kind: 'other', durationMs: undefined },
+];
+export function parseKimiUsage(payload) {
+    const usages = isRecord(payload) && isRecord(payload.usages) ? payload.usages : {};
+    const windows = [];
+    for (const { key, kind, durationMs } of RATIOS) {
+        const entry = usages[key];
+        if (!isRecord(entry))
+            continue;
+        const ratio = finite(entry.used_ratio);
+        if (ratio === undefined)
+            continue;
+        const resetsAt = timestamp(entry.reset_time);
+        const length = durationMs ?? (resetsAt === undefined ? undefined : monthBefore(resetsAt));
+        windows.push({
+            kind, usedPercent: clampPercent(ratio * 100),
+            ...length === undefined ? {} : { durationMs: length },
+            ...resetsAt === undefined ? {} : { resetsAt },
+        });
+    }
+    return windows;
+}
+export const kimi = {
+    id: 'kimi',
+    name: 'Kimi Code',
+    async fetchUsage(key, fetchFn, signal) {
+        let rejected;
+        for (const url of URLS) {
+            try {
+                const windows = parseKimiUsage(await bearerJson('Kimi', url, key, fetchFn, signal));
+                if (windows.length === 0)
+                    throw new Error('Kimi usage data could not be parsed');
+                return { supported: true, windows };
+            }
+            catch (error) {
+                if (!(error instanceof KeyRejectedError))
+                    throw error;
+                rejected = error;
+            }
+        }
+        throw new Error(`${rejected?.message ?? 'Kimi rejected the key'}. It may not be a Kimi Code key.`);
+    },
+};

package/lib/usage-providers/ollama-cloud.d.ts

@@ -0,0 +1,4 @@
+import type { UsageWindow } from '../providers/common.js';
+import type { KeyProvider } from './common.js';
+export declare function parseOllamaUsage(payload: unknown): UsageWindow[];
+export declare const ollamaCloud: KeyProvider;

package/lib/usage-providers/ollama-cloud.js

@@ -0,0 +1,42 @@
+/** Ollama Cloud usage from `ollama.com/api/balance`, with the plan from `/api/me`. */
+import { monthBefore } from '../providers/copilot.js';
+import { bearerJson, clampPercent, finite, isRecord, timestamp } from './common.js';
+const BALANCE_URL = 'https://ollama.com/api/balance';
+const ME_URL = 'https://ollama.com/api/me';
+export function parseOllamaUsage(payload) {
+    const included = isRecord(payload) && isRecord(payload.included) ? payload.included : {};
+    const windows = [];
+    const allowance = finite(included.allowance_usd);
+    const balance = finite(included.balance_usd);
+    if (allowance !== undefined && allowance > 0 && balance !== undefined) {
+        const resetsAt = timestamp(isRecord(included.period) ? included.period.until : undefined);
+        windows.push({
+            kind: 'other', usedPercent: clampPercent((allowance - balance) / allowance * 100),
+            ...resetsAt === undefined ? {} : { resetsAt, durationMs: monthBefore(resetsAt) },
+        });
+    }
+    for (const [name, kind] of [['session', 'session'], ['weekly', 'weekly']]) {
+        const limit = included[name];
+        if (!isRecord(limit))
+            continue;
+        const remaining = finite(limit.remaining_percent);
+        if (remaining === undefined)
+            continue;
+        const resetsAt = timestamp(limit.resets_at);
+        windows.push({ kind, usedPercent: clampPercent(100 - remaining), ...resetsAt === undefined ? {} : { resetsAt } });
+    }
+    return windows;
+}
+export const ollamaCloud = {
+    id: 'ollama-cloud',
+    name: 'Ollama Cloud',
+    async fetchUsage(key, fetchFn, signal) {
+        const windows = parseOllamaUsage(await bearerJson('Ollama Cloud', BALANCE_URL, key, fetchFn, signal));
+        if (windows.length === 0)
+            throw new Error('Ollama Cloud usage data could not be parsed');
+        // The plan name is optional: a failure here drops the label, not the reading.
+        const plan = await bearerJson('Ollama Cloud', ME_URL, key, fetchFn, signal, 'POST')
+            .then(body => isRecord(body) && typeof body.Plan === 'string' && body.Plan !== '' ? body.Plan : undefined, () => undefined);
+        return { supported: true, windows, ...plan === undefined ? {} : { plan } };
+    },
+};

package/lib/usage-providers/zai.d.ts

@@ -0,0 +1,5 @@
+/** Z.ai coding plan usage from `api.z.ai/api/monitor/usage/quota/limit`. */
+import type { ProviderUsage } from '../providers/common.js';
+import type { KeyProvider } from './common.js';
+export declare function parseZaiUsage(payload: unknown): ProviderUsage;
+export declare const zai: KeyProvider;

package/lib/usage-providers/zai.js

@@ -0,0 +1,53 @@
+import { bearerJson, clampPercent, finite, isRecord, timestamp } from './common.js';
+const URL = 'https://api.z.ai/api/monitor/usage/quota/limit';
+const HOUR_MS = 3_600_000;
+const DAY_MS = 24 * HOUR_MS;
+/** Z.ai's `unit` codes for a token window, as the length of one unit. */
+const UNIT_MS = { 3: HOUR_MS, 6: 7 * DAY_MS };
+export function parseZaiUsage(payload) {
+    if (!isRecord(payload))
+        throw new Error('Z.ai returned no object');
+    // A bad key or missing plan arrives as HTTP 200 with `success: false`.
+    const code = payload.code;
+    if (payload.success === false || (code !== undefined && code !== null && code !== 200)) {
+        throw new Error(typeof payload.msg === 'string' && payload.msg !== '' ? payload.msg : `Z.ai error ${String(code)}`);
+    }
+    const data = isRecord(payload.data) ? payload.data : {};
+    const limits = Array.isArray(data.limits) ? data.limits : [];
+    const windows = [];
+    for (const limit of limits) {
+        if (!isRecord(limit))
+            continue;
+        const percent = finite(limit.percentage);
+        if (percent === undefined)
+            continue;
+        const resetsAt = timestamp(limit.nextResetTime);
+        if (limit.type === 'TIME_LIMIT') {
+            windows.push({ kind: 'other', scope: 'MCP', usedPercent: clampPercent(percent), durationMs: 30 * DAY_MS, ...resetsAt === undefined ? {} : { resetsAt } });
+        }
+        else if (limit.type === 'TOKENS_LIMIT' || limit.type === 'CREDIT_LIMIT') {
+            const unit = finite(limit.unit);
+            const number = finite(limit.number);
+            const unitMs = unit === undefined ? undefined : UNIT_MS[unit];
+            const durationMs = unitMs === undefined || number === undefined || number <= 0 ? undefined : unitMs * number;
+            const used = finite(limit.currentValue);
+            const total = finite(limit.usage);
+            windows.push({
+                kind: durationMs === 5 * HOUR_MS ? 'session' : durationMs === 7 * DAY_MS ? 'weekly' : 'other',
+                usedPercent: clampPercent(percent),
+                ...durationMs === undefined ? {} : { durationMs },
+                ...resetsAt === undefined ? {} : { resetsAt },
+                ...used === undefined || total === undefined ? {} : { used, limit: total },
+            });
+        }
+    }
+    const plan = typeof data.level === 'string' && data.level !== '' ? data.level : undefined;
+    return { supported: true, windows, ...plan === undefined ? {} : { plan } };
+}
+export const zai = {
+    id: 'zai',
+    name: 'Z.ai',
+    async fetchUsage(key, fetchFn, signal) {
+        return parseZaiUsage(await bearerJson('Z.ai', URL, key, fetchFn, signal));
+    },
+};

package/locale/en.json

@@ -0,0 +1,6 @@
+{
+  "meta": {
+    "title": "Subscription models",
+    "description": "Use ChatGPT (Codex), Claude, Grok, GitHub Copilot and Google Antigravity subscriptions from the Models page, with pace-aware usage meters under the composer."
+  }
+}

package/locale/zh.json

@@ -0,0 +1,6 @@
+{
+  "meta": {
+    "title": "订阅模型",
+    "description": "在模型页面使用 ChatGPT (Codex)、Claude、Grok、GitHub Copilot 和 Google Antigravity 订阅，并在输入框下方显示按速度预测的用量。"
+  }
+}

package/package.json

@@ -1,6 +1,117 @@
 {
   "name": "dsh-plugin-subscription-usage",
-  "version": "0.0.0-stage",
-  "stub": true,
-  "description": "Temporary package placeholder for staged publishing"
-}
+  "version": "0.1.0",
+  "description": "ChatGPT (Codex), Claude, Grok, GitHub Copilot and Google Antigravity subscriptions as rows on the DSH Models page, with pace-aware usage meters",
+  "license": "MIT",
+  "author": "Parth Patel <pp542004@gmail.com>",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/P4rthPat3l/dsh-plugin-subscription-usage.git"
+  },
+  "homepage": "https://github.com/P4rthPat3l/dsh-plugin-subscription-usage#readme",
+  "bugs": {
+    "url": "https://github.com/P4rthPat3l/dsh-plugin-subscription-usage/issues"
+  },
+  "keywords": [
+    "deepseek-harness",
+    "dsh-plugin",
+    "llm",
+    "oauth",
+    "subscriptions",
+    "usage"
+  ],
+  "type": "module",
+  "packageManager": "pnpm@10.26.2",
+  "main": "lib/index.js",
+  "types": "lib/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./lib/index.d.ts",
+      "default": "./lib/index.js"
+    },
+    "./client": {
+      "types": "./lib/client/index.d.ts",
+      "default": "./lib/client.js"
+    },
+    "./locale/*.json": "./locale/*.json",
+    "./icon": "./icon.svg",
+    "./package.json": "./package.json"
+  },
+  "files": [
+    "lib",
+    "cordis.patch.yml",
+    "locale",
+    "icon.svg"
+  ],
+  "dsh": {
+    "bundle": {
+      "patch": "./cordis.patch.yml"
+    },
+    "client": {
+      "platform": "web",
+      "inject": [
+        "@deepseek-ai/dsh-client-ui-renderer",
+        "@deepseek-ai/dsh-client-ui-settings",
+        "@deepseek-ai/dsh-client-ui-settings-models",
+        "@deepseek-ai/dsh-client-ui-conversation",
+        "@deepseek-ai/dsh-client-locale"
+      ]
+    }
+  },
+  "scripts": {
+    "build": "tsc && tsdown",
+    "test": "tsc -p tsconfig.test.json && node --test \"lib-test/test/**/*.spec.js\"",
+    "prepare": "tsdown -c tsdown.prepare.config.ts"
+  },
+  "peerDependencies": {
+    "@deepseek-ai/cordis": "^4.0.1 || 4.0.5-alpha.1 || 4.0.5-alpha.2",
+    "@deepseek-ai/dsh-attachment": ">=0.2.2-rc.1",
+    "@deepseek-ai/dsh-home-paths": ">=0.2.2-rc.1",
+    "@deepseek-ai/dsh-llm": ">=0.2.2-rc.1",
+    "@deepseek-ai/dsh-tools": ">=0.2.2-rc.1",
+    "@deepseek-ai/dsh-web": ">=0.2.2-rc.1",
+    "@deepseek-ai/schemastery": "^3.18.1 || 3.18.5-alpha.1 || 3.18.5-alpha.2"
+  },
+  "devDependencies": {
+    "@deepseek-ai/cordis": "4.0.5",
+    "@deepseek-ai/cordis-plugin-loader": "1.1.0",
+    "@deepseek-ai/dsh-agent": "0.2.2-rc.1",
+    "@deepseek-ai/dsh-api-remotes": "0.2.2-rc.1",
+    "@deepseek-ai/dsh-attachment": "0.2.2-rc.1",
+    "@deepseek-ai/dsh-client-connection": "0.2.2-rc.1",
+    "@deepseek-ai/dsh-client-locale": "0.2.2-rc.1",
+    "@deepseek-ai/dsh-client-ui-commands": "0.2.2-rc.1",
+    "@deepseek-ai/dsh-client-ui-conversation": "0.2.2-rc.1",
+    "@deepseek-ai/dsh-client-ui-layout": "0.2.2-rc.1",
+    "@deepseek-ai/dsh-client-ui-primitives": "0.2.2-rc.1",
+    "@deepseek-ai/dsh-client-ui-renderer": "0.2.2-rc.1",
+    "@deepseek-ai/dsh-client-ui-settings": "0.2.2-rc.1",
+    "@deepseek-ai/dsh-client-ui-settings-models": "0.2.2-rc.1",
+    "@deepseek-ai/dsh-client-ui-slots": "0.2.2-rc.1",
+    "@deepseek-ai/dsh-home-paths": "0.2.2-rc.1",
+    "@deepseek-ai/dsh-invariants": "0.2.0-rc.2",
+    "@deepseek-ai/dsh-llm": "0.2.2-rc.1",
+    "@deepseek-ai/dsh-scope": "0.2.2-rc.1",
+    "@deepseek-ai/dsh-session": "0.2.2-rc.1",
+    "@deepseek-ai/dsh-session-projection": "0.2.2-rc.1",
+    "@deepseek-ai/dsh-settings": "0.2.2-rc.1",
+    "@deepseek-ai/dsh-system-prompt": "0.2.2-rc.1",
+    "@deepseek-ai/dsh-tools": "0.2.2-rc.1",
+    "@deepseek-ai/dsh-user-approval": "0.2.2-rc.1",
+    "@deepseek-ai/dsh-util-values": "0.2.2-rc.1",
+    "@deepseek-ai/dsh-web": "0.2.2-rc.1",
+    "@deepseek-ai/schemastery": "3.18.5",
+    "@types/node": "^24.0.0",
+    "@types/react": "~18.3.1",
+    "@types/react-dom": "~18.3.7",
+    "@types/semver": "^7.8.0",
+    "react": "^18.2.0",
+    "react-dom": "^18.2.0",
+    "semver": "^7.8.5",
+    "tsdown": "^0.15.0",
+    "typescript": "^5.8.0"
+  },
+  "dependencies": {
+    "undici": "^7.0.0"
+  }
+}