dsh-plugin-subscription-oauth 0.10.1 → 0.10.2

package/lib/auth/mimo-flow.d.ts

@@ -114,4 +114,16 @@ export declare function refreshMimoSession(creds: MimoCredentials, fetchFn: type
     cookies: Record<string, string>;
     expiresAt: number;
 }>;
+/** One metered window the Settings card reads. */
+export interface MimoUsageWindow {
+    label: string;
+    usedPercent?: number;
+    resetsAt?: number;
+}
+/** The account's plan and weekly allowance, for the Settings card and chip. */
+export declare function readMimoUsage(session: MimoSessionCookies, creds: MimoCredentials, fetchFn: typeof fetch): Promise<{
+    plan: string;
+    renewsAt?: number;
+    windows: MimoUsageWindow[];
+}>;
 export {};

package/lib/auth/mimo-flow.js

@@ -353,3 +353,109 @@ export async function refreshMimoSession(creds, fetchFn) {
     const session = await mimoSignOn(creds, creds.base, fetchFn);
     return { cookies: session.cookies, expiresAt: Date.now() + MIMO_RENEW_AFTER_MS };
 }
+/** The MiMo server's names for its plans' tiers. */
+const TIERS = { 1: 'Starter', 2: 'Plus', 3: 'Pro', 4: 'Ultra' };
+/** Read one of the MiMo server's pages with the session's cookies; signs on again once when refused. */
+async function mimoPage(session, creds, path, fetchFn) {
+    for (let attempt = 0;; attempt++) {
+        const headers = { Cookie: cookieHeaderOf(session.cookies), ...appHeaders() };
+        const response = await fetchFn(creds.base.replace(/\/+$/, '') + path, {
+            headers,
+            signal: AbortSignal.timeout(20_000),
+        });
+        const text = await response.text();
+        let env = {};
+        try {
+            env = JSON.parse(text);
+        }
+        catch {
+            env = {};
+        }
+        const bad = !isRecord(env) || Array.isArray(env) || (env.code != null && !Number.isInteger(env.code)) ||
+            (env.msg != null && typeof env.msg !== 'string');
+        if (response.status === 401 || (response.status === 200 && bad)) {
+            if (attempt === 0) {
+                // The session went stale: re-sign on once and retry.
+                const renewed = await refreshMimoSession(creds, fetchFn);
+                session = { cookies: renewed.cookies, userId: session.userId };
+                continue;
+            }
+            throw new MimoSignInLapsed('the Xiaomi MiMo sign-in has expired — sign in again');
+        }
+        if (response.status !== 200 || bad) {
+            throw new Error(`Xiaomi MiMo ${path}: HTTP ${response.status}`);
+        }
+        if ((env.code ?? 0) !== 0) {
+            throw new Error(`Xiaomi MiMo ${path}: code ${String(env.code)} ${String(env.msg ?? '')}`.trim());
+        }
+        return isRecord(env.data) ? env.data : null;
+    }
+}
+/** The session's cookies as one Cookie header, in a steady order. */
+function cookieHeaderOf(cookies) {
+    const first = ['serviceToken', 'userId', 'cUserId'];
+    const parts = first.filter((key) => key in cookies).map((key) => `${key}=${cookies[key]}`);
+    for (const [key, value] of Object.entries(cookies)) {
+        if (!first.includes(key))
+            parts.push(`${key}=${value}`);
+    }
+    return parts.join('; ');
+}
+/** Read one of the server's times; they carry no zone and are Beijing/Singapore time. */
+function mimoServerTime(value) {
+    const text = String(value ?? '').trim();
+    let iso = '';
+    if (/^\d{4}-\d\d-\d\dT\d\d:\d\d:\d\d(\.\d+)?(Z|[+-]\d\d:\d\d)$/i.test(text))
+        iso = text;
+    else if (/^\d{4}-\d\d-\d\d[T ]\d\d:\d\d:\d\d(\.\d+)?$/.test(text))
+        iso = `${text.replace(' ', 'T')}+08:00`;
+    else if (/^\d{4}-\d\d-\d\d$/.test(text))
+        iso = `${text}T00:00:00+08:00`;
+    const at = iso === '' ? NaN : Date.parse(iso);
+    return Number.isNaN(at) ? undefined : at;
+}
+/** The account's plan and weekly allowance, for the Settings card and chip. */
+export async function readMimoUsage(session, creds, fetchFn) {
+    // The plan, from the app's subscription page.
+    let self = null;
+    try {
+        self = await mimoPage(session, creds, '/user/xiaomi/subscription/self', fetchFn);
+    }
+    catch {
+        self = null;
+    }
+    const current = self?.current !== undefined && isRecord(self.current) ? self.current : undefined;
+    let plan = 'Free';
+    let renewsAt;
+    if (current !== undefined) {
+        const title = typeof current.title === 'string' ? current.title.trim() : '';
+        const tier = typeof current.planTier === 'number' ? TIERS[current.planTier] : undefined;
+        const code = typeof current.planCode === 'string' ? current.planCode.trim() : '';
+        plan = title || tier || code || 'MiMo';
+        const end = mimoServerTime(current.endTime);
+        if (end !== undefined)
+            renewsAt = end;
+    }
+    // The week's allowance, from the app's usage page.
+    const windows = [];
+    try {
+        const use = await mimoPage(session, creds, '/user/usage', fetchFn);
+        const percent = typeof use?.percent === 'number' ? use.percent : undefined;
+        const reset = typeof use?.resetDate === 'string' ? use.resetDate : undefined;
+        // No plan: the free offer shows no allowance.
+        if (percent !== undefined && reset !== undefined) {
+            const window = {
+                label: '7 days',
+                usedPercent: Math.max(0, Math.min(100, 100 - percent)),
+            };
+            const at = mimoServerTime(reset);
+            if (at !== undefined)
+                window.resetsAt = at;
+            windows.push(window);
+        }
+    }
+    catch {
+        // The plan alone, when the week's allowance can't be read.
+    }
+    return { plan, ...renewsAt === undefined ? {} : { renewsAt }, windows };
+}

package/lib/providers/mimo-provider.js

@@ -8,7 +8,7 @@ import { REAUTH_REQUIRED, ProviderError } from '../errors.js';
 import { TokenStore } from '../auth/store.js';
 import { TokenManager } from '../auth/tokens.js';
 import { LOGIN_CANCELLED_MESSAGE } from '../auth/device-flow.js';
-import { finishMimoSignIn, mimoCredentialsFrom, refreshMimoSession, startMimoSignIn, waitMimoPoll, } from '../auth/mimo-flow.js';
+import { finishMimoSignIn, mimoCredentialsFrom, readMimoUsage, refreshMimoSession, startMimoSignIn, waitMimoPoll, } from '../auth/mimo-flow.js';
 import { MimoAdapter, MIMO_CODE_URL, MIMO_CONTEXT_WINDOW, MIMO_DEFAULT_MODEL, MIMO_MODELS, MIMO_PROVIDER_ID, MIMO_PROVIDER_NAME, parseMimoSession, } from './mimo.js';
 import { ProviderController } from '../auth/provider-controller.js';
 import { proxiedFetch } from '../http.js';
@@ -132,7 +132,22 @@ export function createMimoProvider(options) {
         catalogOwn: () => adapter.ownCatalogEntries(),
         fallbackContextWindow: MIMO_CONTEXT_WINDOW,
         discover: () => Promise.resolve(undefined),
-        usage: () => Promise.resolve({ windows: [] }),
+        usage: async (_accessToken, account) => {
+            const held = await store.get(account);
+            if (held === undefined)
+                return undefined;
+            let creds;
+            let cookies;
+            try {
+                creds = JSON.parse(held.mimoCreds);
+                cookies = JSON.parse(held.mimoCookies);
+            }
+            catch {
+                return undefined;
+            }
+            const read = await readMimoUsage({ cookies, userId: creds.userId }, creds, fetchFn);
+            return { plan: read.plan, ...read.renewsAt === undefined ? {} : { renewsAt: read.renewsAt }, windows: read.windows };
+        },
         details: () => [
             { key: 'provider', value: MIMO_PROVIDER_NAME },
             { key: 'credentialFile', value: options.credentialFile },

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "dsh-plugin-subscription-oauth",
-  "version": "0.10.1",
+  "version": "0.10.2",
   "description": "Use subscription accounts as DeepSeek Harness LLM providers — Kimi Code (OAuth device flow), Grok / X Premium (OIDC + PKCE), Google Antigravity (Google OAuth + Cloud Code Assist) and GLM / BigModel (ZCode's browser sign-in) — with per-model visibility, drag-to-reorder Auto fallback and plan quotas from the web Settings page",
   "license": "MIT",
   "author": "EricFL1998",