@slkiser/opencode-quota 4.0.1 → 4.2.0

package/dist/lib/mimo-config.js

@@ -0,0 +1,147 @@
+import { readFile } from "fs/promises";
+import { join } from "path";
+import { getOpencodeRuntimeDirCandidates } from "./opencode-runtime-paths.js";
+const MIMO_COOKIE_ENV = "MIMO_USAGE_COOKIE";
+const RETAINED_COOKIE_NAMES = [
+    "api-platform_serviceToken",
+    "userId",
+    "api-platform_ph",
+    "api-platform_slh",
+];
+const RETAINED_COOKIE_NAME_SET = new Set(RETAINED_COOKIE_NAMES);
+const REQUIRED_COOKIE_NAMES = ["api-platform_serviceToken", "userId"];
+function getConfigCandidatePaths() {
+    const { configDirs } = getOpencodeRuntimeDirCandidates();
+    return configDirs.map((dir) => join(dir, "opencode-quota", "mimo.json"));
+}
+export function normalizeMimoCookieHeader(raw) {
+    if (raw.includes("\r") || raw.includes("\n"))
+        return null;
+    const withoutPrefix = raw.trim().replace(/^cookie\s*:\s*/iu, "");
+    if (!withoutPrefix)
+        return null;
+    const retained = new Map();
+    for (const rawPair of withoutPrefix.split(";")) {
+        const pair = rawPair.trim();
+        if (!pair)
+            continue;
+        const separator = pair.indexOf("=");
+        if (separator <= 0)
+            return null;
+        const name = pair.slice(0, separator).trim();
+        const value = pair.slice(separator + 1).trim();
+        if (!name)
+            return null;
+        if (!RETAINED_COOKIE_NAME_SET.has(name))
+            continue;
+        if (!value || retained.has(name))
+            return null;
+        retained.set(name, value);
+    }
+    if (REQUIRED_COOKIE_NAMES.some((name) => !retained.has(name)))
+        return null;
+    return RETAINED_COOKIE_NAMES.flatMap((name) => {
+        const value = retained.get(name);
+        return value === undefined ? [] : [`${name}=${value}`];
+    }).join("; ");
+}
+async function readConfigFile(path) {
+    try {
+        const data = await readFile(path, "utf8");
+        let parsed;
+        try {
+            parsed = JSON.parse(data);
+        }
+        catch {
+            return { state: "invalid", error: "Failed to parse JSON" };
+        }
+        if (!parsed || typeof parsed !== "object" || Array.isArray(parsed)) {
+            return { state: "invalid", error: "Config file must contain a JSON object" };
+        }
+        const record = parsed;
+        return { state: "loaded", cookie: record.cookie, keys: Object.keys(record) };
+    }
+    catch (error) {
+        if (error?.code === "ENOENT") {
+            return { state: "missing" };
+        }
+        return { state: "invalid", error: "Failed to read config file" };
+    }
+}
+export function resolveMimoConfigFromEnv(env = process.env) {
+    if (env[MIMO_COOKIE_ENV] === undefined)
+        return null;
+    const cookie = normalizeMimoCookieHeader(env[MIMO_COOKIE_ENV] ?? "");
+    if (!cookie) {
+        return {
+            state: "invalid",
+            source: `env:${MIMO_COOKIE_ENV}`,
+            error: "Invalid cookie header",
+        };
+    }
+    return {
+        state: "configured",
+        config: { cookie },
+        source: `env:${MIMO_COOKIE_ENV}`,
+    };
+}
+export async function resolveMimoConfig() {
+    const envResult = resolveMimoConfigFromEnv();
+    if (envResult)
+        return envResult;
+    for (const path of getConfigCandidatePaths()) {
+        const fileResult = await readConfigFile(path);
+        if (fileResult.state === "missing")
+            continue;
+        if (fileResult.state === "invalid") {
+            return { state: "invalid", source: path, error: fileResult.error };
+        }
+        if (fileResult.keys.length !== 1 || fileResult.keys[0] !== "cookie") {
+            return {
+                state: "invalid",
+                source: path,
+                error: "Config file must contain only the cookie field",
+            };
+        }
+        if (typeof fileResult.cookie !== "string") {
+            return {
+                state: "invalid",
+                source: path,
+                error: "Config cookie field must be a string",
+            };
+        }
+        const cookie = normalizeMimoCookieHeader(fileResult.cookie);
+        if (!cookie) {
+            return { state: "invalid", source: path, error: "Invalid cookie header" };
+        }
+        return {
+            state: "configured",
+            config: { cookie },
+            source: path,
+        };
+    }
+    return { state: "none" };
+}
+let cachedConfig = null;
+let cachedAt = 0;
+export const DEFAULT_MIMO_CONFIG_CACHE_MAX_AGE_MS = 30_000;
+export async function resolveMimoConfigCached(params) {
+    const maxAgeMs = Math.max(0, params?.maxAgeMs ?? DEFAULT_MIMO_CONFIG_CACHE_MAX_AGE_MS);
+    const now = Date.now();
+    if (cachedConfig && now - cachedAt < maxAgeMs) {
+        return cachedConfig;
+    }
+    cachedConfig = await resolveMimoConfig();
+    cachedAt = now;
+    return cachedConfig;
+}
+export async function getMimoConfigDiagnostics() {
+    const resolved = await resolveMimoConfig();
+    return {
+        state: resolved.state,
+        source: "source" in resolved ? resolved.source : null,
+        error: "error" in resolved ? resolved.error : null,
+        checkedPaths: getConfigCandidatePaths(),
+    };
+}
+//# sourceMappingURL=mimo-config.js.map

package/dist/lib/mimo-config.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mimo-config.js","sourceRoot":"","sources":["../../src/lib/mimo-config.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AACvC,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAE5B,OAAO,EAAE,+BAA+B,EAAE,MAAM,6BAA6B,CAAC;AAuB9E,MAAM,eAAe,GAAG,mBAAmB,CAAC;AAC5C,MAAM,qBAAqB,GAAG;IAC5B,2BAA2B;IAC3B,QAAQ;IACR,iBAAiB;IACjB,kBAAkB;CACV,CAAC;AACX,MAAM,wBAAwB,GAAG,IAAI,GAAG,CAAS,qBAAqB,CAAC,CAAC;AACxE,MAAM,qBAAqB,GAAG,CAAC,2BAA2B,EAAE,QAAQ,CAAU,CAAC;AAE/E,SAAS,uBAAuB;IAC9B,MAAM,EAAE,UAAU,EAAE,GAAG,+BAA+B,EAAE,CAAC;IACzD,OAAO,UAAU,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE,gBAAgB,EAAE,WAAW,CAAC,CAAC,CAAC;AAC3E,CAAC;AAED,MAAM,UAAU,yBAAyB,CAAC,GAAW;IACnD,IAAI,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC;QAAE,OAAO,IAAI,CAAC;IAE1D,MAAM,aAAa,GAAG,GAAG,CAAC,IAAI,EAAE,CAAC,OAAO,CAAC,kBAAkB,EAAE,EAAE,CAAC,CAAC;IACjE,IAAI,CAAC,aAAa;QAAE,OAAO,IAAI,CAAC;IAEhC,MAAM,QAAQ,GAAG,IAAI,GAAG,EAAkB,CAAC;IAC3C,KAAK,MAAM,OAAO,IAAI,aAAa,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;QAC/C,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC;QAC5B,IAAI,CAAC,IAAI;YAAE,SAAS;QAEpB,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACpC,IAAI,SAAS,IAAI,CAAC;YAAE,OAAO,IAAI,CAAC;QAEhC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC,IAAI,EAAE,CAAC;QAC7C,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAC/C,IAAI,CAAC,IAAI;YAAE,OAAO,IAAI,CAAC;QACvB,IAAI,CAAC,wBAAwB,CAAC,GAAG,CAAC,IAAI,CAAC;YAAE,SAAS;QAClD,IAAI,CAAC,KAAK,IAAI,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC;YAAE,OAAO,IAAI,CAAC;QAE9C,QAAQ,CAAC,GAAG,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IAC5B,CAAC;IAED,IAAI,qBAAqB,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAAE,OAAO,IAAI,CAAC;IAE3E,OAAO,qBAAqB,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;QAC5C,MAAM,KAAK,GAAG,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACjC,OAAO,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,GAAG,IAAI,IAAI,KAAK,EAAE,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAChB,CAAC;AAED,KAAK,UAAU,cAAc,CAAC,IAAY;IACxC,IAAI,CAAC;QACH,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAC1C,IAAI,MAAe,CAAC;QACpB,IAAI,CAAC;YACH,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC5B,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,sBAAsB,EAAE,CAAC;QAC7D,CAAC;QAED,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;YACnE,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,wCAAwC,EAAE,CAAC;QAC/E,CAAC;QAED,MAAM,MAAM,GAAG,MAAiC,CAAC;QACjD,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;IAC/E,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAK,KAA2C,EAAE,IAAI,KAAK,QAAQ,EAAE,CAAC;YACpE,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;QAC9B,CAAC;QACD,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,4BAA4B,EAAE,CAAC;IACnE,CAAC;AACH,CAAC;AAED,MAAM,UAAU,wBAAwB,CACtC,MAAyB,OAAO,CAAC,GAAG;IAEpC,IAAI,GAAG,CAAC,eAAe,CAAC,KAAK,SAAS;QAAE,OAAO,IAAI,CAAC;IAEpD,MAAM,MAAM,GAAG,yBAAyB,CAAC,GAAG,CAAC,eAAe,CAAC,IAAI,EAAE,CAAC,CAAC;IACrE,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,OAAO;YACL,KAAK,EAAE,SAAS;YAChB,MAAM,EAAE,OAAO,eAAe,EAAE;YAChC,KAAK,EAAE,uBAAuB;SAC/B,CAAC;IACJ,CAAC;IAED,OAAO;QACL,KAAK,EAAE,YAAY;QACnB,MAAM,EAAE,EAAE,MAAM,EAAE;QAClB,MAAM,EAAE,OAAO,eAAe,EAAE;KACjC,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,iBAAiB;IACrC,MAAM,SAAS,GAAG,wBAAwB,EAAE,CAAC;IAC7C,IAAI,SAAS;QAAE,OAAO,SAAS,CAAC;IAEhC,KAAK,MAAM,IAAI,IAAI,uBAAuB,EAAE,EAAE,CAAC;QAC7C,MAAM,UAAU,GAAG,MAAM,cAAc,CAAC,IAAI,CAAC,CAAC;QAC9C,IAAI,UAAU,CAAC,KAAK,KAAK,SAAS;YAAE,SAAS;QAC7C,IAAI,UAAU,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;YACnC,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,UAAU,CAAC,KAAK,EAAE,CAAC;QACrE,CAAC;QAED,IAAI,UAAU,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,IAAI,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,QAAQ,EAAE,CAAC;YACpE,OAAO;gBACL,KAAK,EAAE,SAAS;gBAChB,MAAM,EAAE,IAAI;gBACZ,KAAK,EAAE,gDAAgD;aACxD,CAAC;QACJ,CAAC;QACD,IAAI,OAAO,UAAU,CAAC,MAAM,KAAK,QAAQ,EAAE,CAAC;YAC1C,OAAO;gBACL,KAAK,EAAE,SAAS;gBAChB,MAAM,EAAE,IAAI;gBACZ,KAAK,EAAE,sCAAsC;aAC9C,CAAC;QACJ,CAAC;QAED,MAAM,MAAM,GAAG,yBAAyB,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QAC5D,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,uBAAuB,EAAE,CAAC;QAC5E,CAAC;QAED,OAAO;YACL,KAAK,EAAE,YAAY;YACnB,MAAM,EAAE,EAAE,MAAM,EAAE;YAClB,MAAM,EAAE,IAAI;SACb,CAAC;IACJ,CAAC;IAED,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;AAC3B,CAAC;AAED,IAAI,YAAY,GAA8B,IAAI,CAAC;AACnD,IAAI,QAAQ,GAAG,CAAC,CAAC;AAEjB,MAAM,CAAC,MAAM,oCAAoC,GAAG,MAAM,CAAC;AAE3D,MAAM,CAAC,KAAK,UAAU,uBAAuB,CAAC,MAE7C;IACC,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,MAAM,EAAE,QAAQ,IAAI,oCAAoC,CAAC,CAAC;IACvF,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACvB,IAAI,YAAY,IAAI,GAAG,GAAG,QAAQ,GAAG,QAAQ,EAAE,CAAC;QAC9C,OAAO,YAAY,CAAC;IACtB,CAAC;IAED,YAAY,GAAG,MAAM,iBAAiB,EAAE,CAAC;IACzC,QAAQ,GAAG,GAAG,CAAC;IACf,OAAO,YAAY,CAAC;AACtB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,wBAAwB;IAC5C,MAAM,QAAQ,GAAG,MAAM,iBAAiB,EAAE,CAAC;IAC3C,OAAO;QACL,KAAK,EAAE,QAAQ,CAAC,KAAK;QACrB,MAAM,EAAE,QAAQ,IAAI,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI;QACrD,KAAK,EAAE,OAAO,IAAI,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI;QAClD,YAAY,EAAE,uBAAuB,EAAE;KACxC,CAAC;AACJ,CAAC"}

package/dist/lib/mimo.d.ts

@@ -0,0 +1,35 @@
+export interface MimoMonthlyQuota {
+    used: number;
+    limit: number;
+}
+export interface MimoPlanDetail {
+    planName: string | null;
+    planCode: string | null;
+    expired: boolean;
+}
+export interface MimoBalance {
+    total: number | null;
+    cash: number | null;
+    gift: number | null;
+    currency: string | null;
+}
+export type MimoEndpointResult<T> = {
+    state: "success";
+    data: T;
+} | {
+    state: "error";
+    error: string;
+};
+export interface MimoDashboardResult {
+    usage: MimoEndpointResult<MimoMonthlyQuota>;
+    detail: MimoEndpointResult<MimoPlanDetail>;
+    balance: MimoEndpointResult<MimoBalance>;
+}
+export declare function parseMimoUsageResponse(json: unknown): MimoMonthlyQuota;
+export declare function parseMimoDetailResponse(json: unknown): MimoPlanDetail;
+export declare function parseMimoBalanceResponse(json: unknown): MimoBalance;
+export declare function formatMimoDashboardTimeZone(timezoneOffsetMinutes?: number): string;
+export declare function queryMimoDashboard(cookie: string, options?: {
+    requestTimeoutMs?: number;
+}): Promise<MimoDashboardResult>;
+//# sourceMappingURL=mimo.d.ts.map

package/dist/lib/mimo.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mimo.d.ts","sourceRoot":"","sources":["../../src/lib/mimo.ts"],"names":[],"mappings":"AAWA,MAAM,WAAW,gBAAgB;IAC/B,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;CACf;AAED,MAAM,WAAW,cAAc;IAC7B,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,OAAO,EAAE,OAAO,CAAC;CAClB;AAED,MAAM,WAAW,WAAW;IAC1B,KAAK,EAAE,MAAM,GAAG,IAAI,CAAC;IACrB,IAAI,EAAE,MAAM,GAAG,IAAI,CAAC;IACpB,IAAI,EAAE,MAAM,GAAG,IAAI,CAAC;IACpB,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;CACzB;AAED,MAAM,MAAM,kBAAkB,CAAC,CAAC,IAC5B;IAAE,KAAK,EAAE,SAAS,CAAC;IAAC,IAAI,EAAE,CAAC,CAAA;CAAE,GAC7B;IAAE,KAAK,EAAE,OAAO,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,CAAC;AAEtC,MAAM,WAAW,mBAAmB;IAClC,KAAK,EAAE,kBAAkB,CAAC,gBAAgB,CAAC,CAAC;IAC5C,MAAM,EAAE,kBAAkB,CAAC,cAAc,CAAC,CAAC;IAC3C,OAAO,EAAE,kBAAkB,CAAC,WAAW,CAAC,CAAC;CAC1C;AA0BD,wBAAgB,sBAAsB,CAAC,IAAI,EAAE,OAAO,GAAG,gBAAgB,CAkBtE;AAED,wBAAgB,uBAAuB,CAAC,IAAI,EAAE,OAAO,GAAG,cAAc,CASrE;AAED,wBAAgB,wBAAwB,CAAC,IAAI,EAAE,OAAO,GAAG,WAAW,CA8BnE;AA6CD,wBAAgB,2BAA2B,CAAC,qBAAqB,CAAC,EAAE,MAAM,GAAG,MAAM,CAQlF;AAiFD,wBAAsB,kBAAkB,CACtC,MAAM,EAAE,MAAM,EACd,OAAO,GAAE;IAAE,gBAAgB,CAAC,EAAE,MAAM,CAAA;CAAO,GAC1C,OAAO,CAAC,mBAAmB,CAAC,CA+B9B"}

package/dist/lib/mimo.js

@@ -0,0 +1,236 @@
+import { sanitizeSingleLineDisplaySnippet } from "./display-sanitize.js";
+import { fetchWithTimeout } from "./http.js";
+const MIMO_USAGE_URL = "https://platform.xiaomimimo.com/api/v1/tokenPlan/usage";
+const MIMO_DETAIL_URL = "https://platform.xiaomimimo.com/api/v1/tokenPlan/detail";
+const MIMO_BALANCE_URL = "https://platform.xiaomimimo.com/api/v1/balance";
+const MIMO_REQUEST_TIMEOUT_MS = 10_000;
+const MIMO_RESPONSE_MAX_BYTES = 256 * 1024;
+const MIMO_USER_AGENT = "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/143.0.0.0 Safari/537.36";
+function isRecord(value) {
+    return Boolean(value) && typeof value === "object" && !Array.isArray(value);
+}
+function parseEnvelope(json) {
+    if (!isRecord(json) || json.code !== 0 || !isRecord(json.data)) {
+        throw new Error("invalid envelope");
+    }
+    return json.data;
+}
+function parseNonNegativeNumber(value) {
+    return typeof value === "number" && Number.isFinite(value) && value >= 0 ? value : null;
+}
+function parseBalanceNumber(value) {
+    if (typeof value === "number") {
+        return Number.isFinite(value) && value >= 0 ? value : null;
+    }
+    if (typeof value !== "string" || !/^\d+(?:\.\d+)?$/u.test(value.trim()))
+        return null;
+    const parsed = Number(value);
+    return Number.isFinite(parsed) ? parsed : null;
+}
+export function parseMimoUsageResponse(json) {
+    const data = parseEnvelope(json);
+    if (!isRecord(data.monthUsage) || !Array.isArray(data.monthUsage.items)) {
+        throw new Error("missing monthly usage");
+    }
+    const item = data.monthUsage.items.find((candidate) => isRecord(candidate) && candidate.name === "month_total_token");
+    if (!isRecord(item))
+        throw new Error("missing monthly quota");
+    const used = parseNonNegativeNumber(item.used);
+    const limit = parseNonNegativeNumber(item.limit);
+    if (used === null || limit === null || limit <= 0) {
+        throw new Error("invalid monthly quota");
+    }
+    return { used, limit };
+}
+export function parseMimoDetailResponse(json) {
+    const data = parseEnvelope(json);
+    if (typeof data.expired !== "boolean")
+        throw new Error("invalid plan state");
+    return {
+        planName: typeof data.planName === "string" && data.planName.trim() ? data.planName : null,
+        planCode: typeof data.planCode === "string" && data.planCode.trim() ? data.planCode : null,
+        expired: data.expired,
+    };
+}
+export function parseMimoBalanceResponse(json) {
+    const data = parseEnvelope(json);
+    const fields = [
+        ["balance", data.balance],
+        ["cashBalance", data.cashBalance],
+        ["giftBalance", data.giftBalance],
+    ];
+    const parsed = new Map();
+    for (const [name, value] of fields) {
+        if (value === undefined || value === null || value === "") {
+            parsed.set(name, null);
+            continue;
+        }
+        const amount = parseBalanceNumber(value);
+        if (amount === null)
+            throw new Error("invalid balance");
+        parsed.set(name, amount);
+    }
+    const currency = typeof data.currency === "string" && /^[A-Za-z]{3}$/u.test(data.currency.trim())
+        ? data.currency.trim().toUpperCase()
+        : null;
+    return {
+        total: parsed.get("balance") ?? null,
+        cash: parsed.get("cashBalance") ?? null,
+        gift: parsed.get("giftBalance") ?? null,
+        currency,
+    };
+}
+async function readBoundedJson(response) {
+    if (!response.body)
+        throw new Error("empty response");
+    const reader = response.body.getReader();
+    const chunks = [];
+    let byteLength = 0;
+    try {
+        const contentLength = Number(response.headers.get("content-length"));
+        if (Number.isFinite(contentLength) && contentLength > MIMO_RESPONSE_MAX_BYTES) {
+            throw new Error("response too large");
+        }
+        while (true) {
+            const { done, value } = await reader.read();
+            if (done)
+                break;
+            byteLength += value.byteLength;
+            if (byteLength > MIMO_RESPONSE_MAX_BYTES) {
+                throw new Error("response too large");
+            }
+            chunks.push(value);
+        }
+    }
+    catch (error) {
+        try {
+            await reader.cancel();
+        }
+        catch {
+            // Preserve the original read or size-limit error.
+        }
+        throw error;
+    }
+    finally {
+        reader.releaseLock();
+    }
+    const body = new Uint8Array(byteLength);
+    let offset = 0;
+    for (const chunk of chunks) {
+        body.set(chunk, offset);
+        offset += chunk.byteLength;
+    }
+    return JSON.parse(new TextDecoder().decode(body));
+}
+export function formatMimoDashboardTimeZone(timezoneOffsetMinutes) {
+    const javascriptOffset = timezoneOffsetMinutes ?? new Date().getTimezoneOffset();
+    const utcOffset = -javascriptOffset;
+    const sign = utcOffset >= 0 ? "+" : "-";
+    const absoluteMinutes = Math.abs(utcOffset);
+    const hours = Math.floor(absoluteMinutes / 60);
+    const minutes = absoluteMinutes % 60;
+    return `UTC${sign}${String(hours).padStart(2, "0")}:${String(minutes).padStart(2, "0")}`;
+}
+function getCookieValues(cookie) {
+    return cookie
+        .split(";")
+        .map((part) => part.slice(part.indexOf("=") + 1).trim())
+        .filter(Boolean);
+}
+function sanitizeRequestError(error, cookie) {
+    let message = error instanceof Error ? error.message : String(error);
+    for (const secret of [cookie, ...getCookieValues(cookie)]) {
+        message = message.split(secret).join("[redacted]");
+    }
+    return sanitizeSingleLineDisplaySnippet(message, 120) || "request failed";
+}
+async function fetchMimoEndpoint(params) {
+    try {
+        return await fetchWithTimeout(params.url, {
+            request: {
+                method: "GET",
+                redirect: "manual",
+                headers: {
+                    Accept: "application/json, text/plain, */*",
+                    "Accept-Language": "en-US,en;q=0.9",
+                    Cookie: params.cookie,
+                    Origin: "https://platform.xiaomimimo.com",
+                    Referer: "https://platform.xiaomimimo.com/#/console/balance",
+                    "User-Agent": MIMO_USER_AGENT,
+                    "x-timeZone": params.timeZone,
+                },
+            },
+            timeoutMs: params.requestTimeoutMs,
+            consume: async (response, timeoutSignal) => {
+                if (response.status >= 300 && response.status < 400) {
+                    return { state: "error", error: `Xiaomi MiMo ${params.name} request requires login` };
+                }
+                if (!response.ok) {
+                    return {
+                        state: "error",
+                        error: `Xiaomi MiMo ${params.name} request failed (HTTP ${response.status})`,
+                    };
+                }
+                let json;
+                try {
+                    json = await readBoundedJson(response);
+                }
+                catch (error) {
+                    if (timeoutSignal.aborted)
+                        throw error;
+                    return {
+                        state: "error",
+                        error: `Xiaomi MiMo ${params.name} response could not be parsed`,
+                    };
+                }
+                try {
+                    return { state: "success", data: params.parse(json) };
+                }
+                catch {
+                    return {
+                        state: "error",
+                        error: `Xiaomi MiMo ${params.name} response did not match the expected schema`,
+                    };
+                }
+            },
+        });
+    }
+    catch (error) {
+        return {
+            state: "error",
+            error: `Xiaomi MiMo ${params.name} request failed: ${sanitizeRequestError(error, params.cookie)}`,
+        };
+    }
+}
+export async function queryMimoDashboard(cookie, options = {}) {
+    const requestTimeoutMs = options.requestTimeoutMs ?? MIMO_REQUEST_TIMEOUT_MS;
+    const timeZone = formatMimoDashboardTimeZone();
+    const [usage, detail, balance] = await Promise.all([
+        fetchMimoEndpoint({
+            name: "usage",
+            url: MIMO_USAGE_URL,
+            cookie,
+            requestTimeoutMs,
+            timeZone,
+            parse: parseMimoUsageResponse,
+        }),
+        fetchMimoEndpoint({
+            name: "detail",
+            url: MIMO_DETAIL_URL,
+            cookie,
+            requestTimeoutMs,
+            timeZone,
+            parse: parseMimoDetailResponse,
+        }),
+        fetchMimoEndpoint({
+            name: "balance",
+            url: MIMO_BALANCE_URL,
+            cookie,
+            requestTimeoutMs,
+            timeZone,
+            parse: parseMimoBalanceResponse,
+        }),
+    ]);
+    return { usage, detail, balance };
+}
+//# sourceMappingURL=mimo.js.map

package/dist/lib/mimo.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mimo.js","sourceRoot":"","sources":["../../src/lib/mimo.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gCAAgC,EAAE,MAAM,uBAAuB,CAAC;AACzE,OAAO,EAAE,gBAAgB,EAAE,MAAM,WAAW,CAAC;AAE7C,MAAM,cAAc,GAAG,wDAAwD,CAAC;AAChF,MAAM,eAAe,GAAG,yDAAyD,CAAC;AAClF,MAAM,gBAAgB,GAAG,gDAAgD,CAAC;AAC1E,MAAM,uBAAuB,GAAG,MAAM,CAAC;AACvC,MAAM,uBAAuB,GAAG,GAAG,GAAG,IAAI,CAAC;AAC3C,MAAM,eAAe,GACnB,uHAAuH,CAAC;AA8B1H,SAAS,QAAQ,CAAC,KAAc;IAC9B,OAAO,OAAO,CAAC,KAAK,CAAC,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AAC9E,CAAC;AAED,SAAS,aAAa,CAAC,IAAa;IAClC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QAC/D,MAAM,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC;IACtC,CAAC;IACD,OAAO,IAAI,CAAC,IAAI,CAAC;AACnB,CAAC;AAED,SAAS,sBAAsB,CAAC,KAAc;IAC5C,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC;AAC1F,CAAC;AAED,SAAS,kBAAkB,CAAC,KAAc;IACxC,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;QAC9B,OAAO,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC;IAC7D,CAAC;IACD,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,kBAAkB,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;QAAE,OAAO,IAAI,CAAC;IACrF,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IAC7B,OAAO,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC;AACjD,CAAC;AAED,MAAM,UAAU,sBAAsB,CAAC,IAAa;IAClD,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;IACjC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;QACxE,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;IAC3C,CAAC;IAED,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,CACrC,CAAC,SAAS,EAAE,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,SAAS,CAAC,IAAI,KAAK,mBAAmB,CAC7E,CAAC;IACF,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;QAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;IAE9D,MAAM,IAAI,GAAG,sBAAsB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC/C,MAAM,KAAK,GAAG,sBAAsB,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACjD,IAAI,IAAI,KAAK,IAAI,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;QAClD,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;IAC3C,CAAC;IAED,OAAO,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC;AACzB,CAAC;AAED,MAAM,UAAU,uBAAuB,CAAC,IAAa;IACnD,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;IACjC,IAAI,OAAO,IAAI,CAAC,OAAO,KAAK,SAAS;QAAE,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;IAE7E,OAAO;QACL,QAAQ,EAAE,OAAO,IAAI,CAAC,QAAQ,KAAK,QAAQ,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI;QAC1F,QAAQ,EAAE,OAAO,IAAI,CAAC,QAAQ,KAAK,QAAQ,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI;QAC1F,OAAO,EAAE,IAAI,CAAC,OAAO;KACtB,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,wBAAwB,CAAC,IAAa;IACpD,MAAM,IAAI,GAAG,aAAa,CAAC,IAAI,CAAC,CAAC;IACjC,MAAM,MAAM,GAAG;QACb,CAAC,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC;QACzB,CAAC,aAAa,EAAE,IAAI,CAAC,WAAW,CAAC;QACjC,CAAC,aAAa,EAAE,IAAI,CAAC,WAAW,CAAC;KACzB,CAAC;IAEX,MAAM,MAAM,GAAG,IAAI,GAAG,EAAyB,CAAC;IAChD,KAAK,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,MAAM,EAAE,CAAC;QACnC,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,KAAK,EAAE,EAAE,CAAC;YAC1D,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;YACvB,SAAS;QACX,CAAC;QACD,MAAM,MAAM,GAAG,kBAAkB,CAAC,KAAK,CAAC,CAAC;QACzC,IAAI,MAAM,KAAK,IAAI;YAAE,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;QACxD,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IAC3B,CAAC;IAED,MAAM,QAAQ,GACZ,OAAO,IAAI,CAAC,QAAQ,KAAK,QAAQ,IAAI,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QAC9E,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE;QACpC,CAAC,CAAC,IAAI,CAAC;IAEX,OAAO;QACL,KAAK,EAAE,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,IAAI;QACpC,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,aAAa,CAAC,IAAI,IAAI;QACvC,IAAI,EAAE,MAAM,CAAC,GAAG,CAAC,aAAa,CAAC,IAAI,IAAI;QACvC,QAAQ;KACT,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,eAAe,CAAC,QAAkB;IAC/C,IAAI,CAAC,QAAQ,CAAC,IAAI;QAAE,MAAM,IAAI,KAAK,CAAC,gBAAgB,CAAC,CAAC;IAEtD,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;IACzC,MAAM,MAAM,GAAiB,EAAE,CAAC;IAChC,IAAI,UAAU,GAAG,CAAC,CAAC;IAEnB,IAAI,CAAC;QACH,MAAM,aAAa,GAAG,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC,CAAC;QACrE,IAAI,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,IAAI,aAAa,GAAG,uBAAuB,EAAE,CAAC;YAC9E,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;QACxC,CAAC;QAED,OAAO,IAAI,EAAE,CAAC;YACZ,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;YAC5C,IAAI,IAAI;gBAAE,MAAM;YAChB,UAAU,IAAI,KAAK,CAAC,UAAU,CAAC;YAC/B,IAAI,UAAU,GAAG,uBAAuB,EAAE,CAAC;gBACzC,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAC;YACxC,CAAC;YACD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACrB,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,CAAC;YACH,MAAM,MAAM,CAAC,MAAM,EAAE,CAAC;QACxB,CAAC;QAAC,MAAM,CAAC;YACP,kDAAkD;QACpD,CAAC;QACD,MAAM,KAAK,CAAC;IACd,CAAC;YAAS,CAAC;QACT,MAAM,CAAC,WAAW,EAAE,CAAC;IACvB,CAAC;IAED,MAAM,IAAI,GAAG,IAAI,UAAU,CAAC,UAAU,CAAC,CAAC;IACxC,IAAI,MAAM,GAAG,CAAC,CAAC;IACf,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACxB,MAAM,IAAI,KAAK,CAAC,UAAU,CAAC;IAC7B,CAAC;IAED,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,WAAW,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;AACpD,CAAC;AAED,MAAM,UAAU,2BAA2B,CAAC,qBAA8B;IACxE,MAAM,gBAAgB,GAAG,qBAAqB,IAAI,IAAI,IAAI,EAAE,CAAC,iBAAiB,EAAE,CAAC;IACjF,MAAM,SAAS,GAAG,CAAC,gBAAgB,CAAC;IACpC,MAAM,IAAI,GAAG,SAAS,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC;IACxC,MAAM,eAAe,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IAC5C,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,eAAe,GAAG,EAAE,CAAC,CAAC;IAC/C,MAAM,OAAO,GAAG,eAAe,GAAG,EAAE,CAAC;IACrC,OAAO,MAAM,IAAI,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC;AAC3F,CAAC;AAED,SAAS,eAAe,CAAC,MAAc;IACrC,OAAO,MAAM;SACV,KAAK,CAAC,GAAG,CAAC;SACV,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;SACvD,MAAM,CAAC,OAAO,CAAC,CAAC;AACrB,CAAC;AAED,SAAS,oBAAoB,CAAC,KAAc,EAAE,MAAc;IAC1D,IAAI,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACrE,KAAK,MAAM,MAAM,IAAI,CAAC,MAAM,EAAE,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC;QAC1D,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IACrD,CAAC;IACD,OAAO,gCAAgC,CAAC,OAAO,EAAE,GAAG,CAAC,IAAI,gBAAgB,CAAC;AAC5E,CAAC;AAED,KAAK,UAAU,iBAAiB,CAAI,MAOnC;IACC,IAAI,CAAC;QACH,OAAO,MAAM,gBAAgB,CAAC,MAAM,CAAC,GAAG,EAAE;YACxC,OAAO,EAAE;gBACP,MAAM,EAAE,KAAK;gBACb,QAAQ,EAAE,QAAQ;gBAClB,OAAO,EAAE;oBACP,MAAM,EAAE,mCAAmC;oBAC3C,iBAAiB,EAAE,gBAAgB;oBACnC,MAAM,EAAE,MAAM,CAAC,MAAM;oBACrB,MAAM,EAAE,iCAAiC;oBACzC,OAAO,EAAE,mDAAmD;oBAC5D,YAAY,EAAE,eAAe;oBAC7B,YAAY,EAAE,MAAM,CAAC,QAAQ;iBAC9B;aACF;YACD,SAAS,EAAE,MAAM,CAAC,gBAAgB;YAClC,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,aAAa,EAAkC,EAAE;gBACzE,IAAI,QAAQ,CAAC,MAAM,IAAI,GAAG,IAAI,QAAQ,CAAC,MAAM,GAAG,GAAG,EAAE,CAAC;oBACpD,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,KAAK,EAAE,eAAe,MAAM,CAAC,IAAI,yBAAyB,EAAE,CAAC;gBACxF,CAAC;gBACD,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;oBACjB,OAAO;wBACL,KAAK,EAAE,OAAO;wBACd,KAAK,EAAE,eAAe,MAAM,CAAC,IAAI,yBAAyB,QAAQ,CAAC,MAAM,GAAG;qBAC7E,CAAC;gBACJ,CAAC;gBAED,IAAI,IAAa,CAAC;gBAClB,IAAI,CAAC;oBACH,IAAI,GAAG,MAAM,eAAe,CAAC,QAAQ,CAAC,CAAC;gBACzC,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,IAAI,aAAa,CAAC,OAAO;wBAAE,MAAM,KAAK,CAAC;oBACvC,OAAO;wBACL,KAAK,EAAE,OAAO;wBACd,KAAK,EAAE,eAAe,MAAM,CAAC,IAAI,+BAA+B;qBACjE,CAAC;gBACJ,CAAC;gBAED,IAAI,CAAC;oBACH,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;gBACxD,CAAC;gBAAC,MAAM,CAAC;oBACP,OAAO;wBACL,KAAK,EAAE,OAAO;wBACd,KAAK,EAAE,eAAe,MAAM,CAAC,IAAI,6CAA6C;qBAC/E,CAAC;gBACJ,CAAC;YACH,CAAC;SACF,CAAC,CAAC;IACL,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO;YACL,KAAK,EAAE,OAAO;YACd,KAAK,EAAE,eAAe,MAAM,CAAC,IAAI,oBAAoB,oBAAoB,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,EAAE;SAClG,CAAC;IACJ,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACtC,MAAc,EACd,UAAyC,EAAE;IAE3C,MAAM,gBAAgB,GAAG,OAAO,CAAC,gBAAgB,IAAI,uBAAuB,CAAC;IAC7E,MAAM,QAAQ,GAAG,2BAA2B,EAAE,CAAC;IAC/C,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,OAAO,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;QACjD,iBAAiB,CAAC;YAChB,IAAI,EAAE,OAAO;YACb,GAAG,EAAE,cAAc;YACnB,MAAM;YACN,gBAAgB;YAChB,QAAQ;YACR,KAAK,EAAE,sBAAsB;SAC9B,CAAC;QACF,iBAAiB,CAAC;YAChB,IAAI,EAAE,QAAQ;YACd,GAAG,EAAE,eAAe;YACpB,MAAM;YACN,gBAAgB;YAChB,QAAQ;YACR,KAAK,EAAE,uBAAuB;SAC/B,CAAC;QACF,iBAAiB,CAAC;YAChB,IAAI,EAAE,SAAS;YACf,GAAG,EAAE,gBAAgB;YACrB,MAAM;YACN,gBAAgB;YAChB,QAAQ;YACR,KAAK,EAAE,wBAAwB;SAChC,CAAC;KACH,CAAC,CAAC;IAEH,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;AACpC,CAAC"}

package/dist/lib/modelsdev-pricing.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"modelsdev-pricing.d.ts","sourceRoot":"","sources":["../../src/lib/modelsdev-pricing.ts"],"names":[],"mappings":"AAMA,OAAO,EAA0B,KAAK,mBAAmB,EAAE,MAAM,6BAA6B,CAAC;AAC/F,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,YAAY,CAAC;AAWxD,MAAM,MAAM,WAAW,GAAG;IACxB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB,CAAC;AAEF,MAAM,MAAM,eAAe,GAAG;IAC5B,KAAK,EAAE;QACL,MAAM,EAAE,MAAM,CAAC;QACf,WAAW,EAAE,MAAM,CAAC;QACpB,SAAS,EAAE,MAAM,EAAE,CAAC;QACpB,KAAK,EAAE,MAAM,CAAC;KACf,CAAC;IACF,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC,CAAC;CACxD,CAAC;AAEF,eAAO,MAAM,mCAAmC,QAA0B,CAAC;AAE3E,MAAM,MAAM,qBAAqB,GAAG;IAClC,WAAW,EAAE,MAAM,CAAC;IACpB,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,OAAO,CAAC;CAChB,CAAC;AAEF,MAAM,WAAW,qBAAqB;IACpC,OAAO,EAAE,CAAC,CAAC;IACX,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,UAAU,CAAC,EAAE,SAAS,GAAG,cAAc,GAAG,eAAe,GAAG,mBAAmB,GAAG,QAAQ,CAAC;IAC3F,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,oBAAoB;IACnC,OAAO,EAAE,OAAO,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,oBAAoB,EAAE,MAAM,CAAC;IAC7B,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,qBAAqB;IACpC,MAAM,CAAC,EAAE,MAAM,GAAG,QAAQ,GAAG,QAAQ,GAAG,QAAQ,CAAC;IACjD,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,mBAAmB,CAAC;IAClC,OAAO,CAAC,EAAE,OAAO,KAAK,CAAC;IACvB,yBAAyB,CAAC,EAAE,eAAe,CAAC;IAC5C,iBAAiB,CAAC,EAAE,MAAM,EAAE,CAAC;IAC7B,iBAAiB,CAAC,EAAE,qBAAqB,CAAC;IAC1C,gCAAgC,CAAC,EAAE,OAAO,CAAC;CAC5C;AAED,MAAM,WAAW,oBAAoB;IACnC,SAAS,EAAE,OAAO,CAAC;IACnB,OAAO,EAAE,OAAO,CAAC;IACjB,KAAK,EAAE,qBAAqB,CAAC;IAC7B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAuCD,wBAAgB,uBAAuB,IAAI,oBAAoB,CAO9D;AAED,wBAAgB,6BAA6B,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI,CAEhE;AA6ED,wBAAgB,6BAA6B,CAAC,WAAW,CAAC,EAAE,mBAAmB,GAAG,MAAM,CAGvF;AAED,wBAAgB,iCAAiC,CAAC,WAAW,CAAC,EAAE,mBAAmB,GAAG,MAAM,CAG3F;AAyID,wBAAsB,uBAAuB,CAC3C,WAAW,CAAC,EAAE,mBAAmB,GAChC,OAAO,CAAC,qBAAqB,GAAG,IAAI,CAAC,CAEvC;AAsGD,wBAAsB,2BAA2B,CAC/C,IAAI,GAAE,qBAA0B,GAC/B,OAAO,CAAC,oBAAoB,CAAC,CAkN/B;AAED,wBAAgB,2BAA2B,CAAC,SAAS,EAAE,qBAAqB,GAAG,IAAI,CAKlF;AAED,wBAAgB,2BAA2B,IAAI,qBAAqB,CAEnE;AAED,wBAAgB,sBAAsB,IAAI,eAAe,CAAC,OAAO,CAAC,CAEjE;AAED,wBAAgB,wBAAwB,IAAI,SAAS,GAAG,SAAS,GAAG,OAAO,CAG1E;AAED,wBAAgB,wBAAwB,CAAC,IAAI,CAAC,EAAE;IAC9C,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB,GAAG,qBAAqB,CAWxB;AAED,wBAAgB,WAAW,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAEvD;AAED,wBAAgB,qBAAqB,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAEjE;AAED,wBAAgB,QAAQ,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAIrE;AAED;;;GAGG;AACH,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAItE;AAED,wBAAgB,qBAAqB,CAAC,UAAU,EAAE,MAAM,GAAG,MAAM,CAEhE;AAED,wBAAgB,aAAa,IAAI,MAAM,EAAE,CAExC;AAED,wBAAgB,qBAAqB,CAAC,UAAU,EAAE,MAAM,GAAG,MAAM,EAAE,CAElE;AAED,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,EAAE,CAGjE;AAED,wBAAgB,UAAU,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,WAAW,GAAG,IAAI,CAMlF;AAED,wBAAgB,OAAO,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAEpE;AAED,wBAAgB,8BAA8B,IAAI,IAAI,CAQrD"}
+{"version":3,"file":"modelsdev-pricing.d.ts","sourceRoot":"","sources":["../../src/lib/modelsdev-pricing.ts"],"names":[],"mappings":"AAMA,OAAO,EAA0B,KAAK,mBAAmB,EAAE,MAAM,6BAA6B,CAAC;AAC/F,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,YAAY,CAAC;AAWxD,MAAM,MAAM,WAAW,GAAG;IACxB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB,CAAC;AAEF,MAAM,MAAM,eAAe,GAAG;IAC5B,KAAK,EAAE;QACL,MAAM,EAAE,MAAM,CAAC;QACf,WAAW,EAAE,MAAM,CAAC;QACpB,SAAS,EAAE,MAAM,EAAE,CAAC;QACpB,KAAK,EAAE,MAAM,CAAC;KACf,CAAC;IACF,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC,CAAC;CACxD,CAAC;AAEF,eAAO,MAAM,mCAAmC,QAA0B,CAAC;AAE3E,MAAM,MAAM,qBAAqB,GAAG;IAClC,WAAW,EAAE,MAAM,CAAC;IACpB,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,OAAO,CAAC;CAChB,CAAC;AAEF,MAAM,WAAW,qBAAqB;IACpC,OAAO,EAAE,CAAC,CAAC;IACX,SAAS,EAAE,MAAM,CAAC;IAClB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,UAAU,CAAC,EAAE,SAAS,GAAG,cAAc,GAAG,eAAe,GAAG,mBAAmB,GAAG,QAAQ,CAAC;IAC3F,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,oBAAoB;IACnC,OAAO,EAAE,OAAO,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,oBAAoB,EAAE,MAAM,CAAC;IAC7B,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,qBAAqB;IACpC,MAAM,CAAC,EAAE,MAAM,GAAG,QAAQ,GAAG,QAAQ,GAAG,QAAQ,CAAC;IACjD,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,oBAAoB,CAAC,EAAE,MAAM,CAAC;IAC9B,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,mBAAmB,CAAC;IAClC,OAAO,CAAC,EAAE,OAAO,KAAK,CAAC;IACvB,yBAAyB,CAAC,EAAE,eAAe,CAAC;IAC5C,iBAAiB,CAAC,EAAE,MAAM,EAAE,CAAC;IAC7B,iBAAiB,CAAC,EAAE,qBAAqB,CAAC;IAC1C,gCAAgC,CAAC,EAAE,OAAO,CAAC;CAC5C;AAED,MAAM,WAAW,oBAAoB;IACnC,SAAS,EAAE,OAAO,CAAC;IACnB,OAAO,EAAE,OAAO,CAAC;IACjB,KAAK,EAAE,qBAAqB,CAAC;IAC7B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAuCD,wBAAgB,uBAAuB,IAAI,oBAAoB,CAO9D;AAED,wBAAgB,6BAA6B,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI,CAEhE;AA+ED,wBAAgB,6BAA6B,CAAC,WAAW,CAAC,EAAE,mBAAmB,GAAG,MAAM,CAGvF;AAED,wBAAgB,iCAAiC,CAAC,WAAW,CAAC,EAAE,mBAAmB,GAAG,MAAM,CAG3F;AAkJD,wBAAsB,uBAAuB,CAC3C,WAAW,CAAC,EAAE,mBAAmB,GAChC,OAAO,CAAC,qBAAqB,GAAG,IAAI,CAAC,CAEvC;AA6HD,wBAAsB,2BAA2B,CAC/C,IAAI,GAAE,qBAA0B,GAC/B,OAAO,CAAC,oBAAoB,CAAC,CAiN/B;AAED,wBAAgB,2BAA2B,CAAC,SAAS,EAAE,qBAAqB,GAAG,IAAI,CAKlF;AAED,wBAAgB,2BAA2B,IAAI,qBAAqB,CAEnE;AAED,wBAAgB,sBAAsB,IAAI,eAAe,CAAC,OAAO,CAAC,CAEjE;AAED,wBAAgB,wBAAwB,IAAI,SAAS,GAAG,SAAS,GAAG,OAAO,CAG1E;AAED,wBAAgB,wBAAwB,CAAC,IAAI,CAAC,EAAE;IAC9C,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB,GAAG,qBAAqB,CAWxB;AAED,wBAAgB,WAAW,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAEvD;AAED,wBAAgB,qBAAqB,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAEjE;AAED,wBAAgB,QAAQ,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAIrE;AAED;;;GAGG;AACH,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,GAAG,IAAI,CAItE;AAED,wBAAgB,qBAAqB,CAAC,UAAU,EAAE,MAAM,GAAG,MAAM,CAEhE;AAED,wBAAgB,aAAa,IAAI,MAAM,EAAE,CAExC;AAED,wBAAgB,qBAAqB,CAAC,UAAU,EAAE,MAAM,GAAG,MAAM,EAAE,CAElE;AAED,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,EAAE,CAGjE;AAED,wBAAgB,UAAU,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,WAAW,GAAG,IAAI,CAMlF;AAED,wBAAgB,OAAO,CAAC,UAAU,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,CAEpE;AAED,wBAAgB,8BAA8B,IAAI,IAAI,CAQrD"}

package/dist/lib/modelsdev-pricing.js

@@ -230,7 +230,13 @@ function normalizeRefreshState(raw) {
     if (Number.isFinite(lastFailureAt) && lastFailureAt > 0)
         out.lastFailureAt = Math.trunc(lastFailureAt);
     if (typeof obj.lastResult === "string") {
-        const allowed = new Set(["success", "not_modified", "skipped_fresh", "skipped_throttled", "failed"]);
+        const allowed = new Set([
+            "success",
+            "not_modified",
+            "skipped_fresh",
+            "skipped_throttled",
+            "failed",
+        ]);
         if (allowed.has(obj.lastResult)) {
             out.lastResult = obj.lastResult;
         }
@@ -321,20 +327,28 @@ async function fetchModelsDevSnapshot(params) {
         headers.set("If-None-Match", params.state.etag);
     if (params.state.lastModified)
         headers.set("If-Modified-Since", params.state.lastModified);
-    if (!params.fetchFn) {
-        return await fetchWithTimeout(SOURCE_URL, { headers }, params.timeoutMs);
-    }
-    const controller = new AbortController();
-    const timeoutId = setTimeout(() => controller.abort(), params.timeoutMs);
-    try {
-        return await params.fetchFn(SOURCE_URL, {
-            headers,
-            signal: controller.signal,
-        });
-    }
-    finally {
-        clearTimeout(timeoutId);
-    }
+    return await fetchWithTimeout(SOURCE_URL, {
+        request: { headers },
+        timeoutMs: params.timeoutMs,
+        fetchFn: params.fetchFn,
+        consume: async (response) => {
+            const responseMetadata = {
+                etag: response.headers.get("etag") ?? undefined,
+                lastModified: response.headers.get("last-modified") ?? undefined,
+            };
+            if (response.status === 304) {
+                return { kind: "not_modified", ...responseMetadata };
+            }
+            if (!response.ok) {
+                throw new Error(`Failed to fetch ${SOURCE_URL}: ${response.status} ${response.statusText}`);
+            }
+            return {
+                kind: "success",
+                api: await response.json(),
+                ...responseMetadata,
+            };
+        },
+    });
 }
 export async function maybeRefreshPricingSnapshot(opts = {}) {
     if (REFRESH_IN_FLIGHT)
@@ -403,7 +417,9 @@ export async function maybeRefreshPricingSnapshot(opts = {}) {
                 },
             };
         }
-        if (!force && previousState.lastAttemptAt && nowMs - previousState.lastAttemptAt < minAttemptIntervalMs) {
+        if (!force &&
+            previousState.lastAttemptAt &&
+            nowMs - previousState.lastAttemptAt < minAttemptIntervalMs) {
             return {
                 attempted: false,
                 updated: false,
@@ -422,12 +438,12 @@ export async function maybeRefreshPricingSnapshot(opts = {}) {
             lastAttemptAt: nowMs,
         };
         try {
-            const response = await fetchModelsDevSnapshot({
+            const fetchResult = await fetchModelsDevSnapshot({
                 timeoutMs,
                 state: attemptingState,
                 fetchFn: opts.fetchFn,
             });
-            if (response.status === 304) {
+            if (fetchResult.kind === "not_modified") {
                 const baseSnapshot = runtimeSnapshotBeforeRefresh ?? ensureLoaded();
                 const refreshedSnapshot = {
                     _meta: {
@@ -448,8 +464,8 @@ export async function maybeRefreshPricingSnapshot(opts = {}) {
                     lastSuccessAt: nowMs,
                     lastResult: "not_modified",
                     lastError: undefined,
-                    etag: response.headers.get("etag") ?? attemptingState.etag,
-                    lastModified: response.headers.get("last-modified") ?? attemptingState.lastModified,
+                    etag: fetchResult.etag ?? attemptingState.etag,
+                    lastModified: fetchResult.lastModified ?? attemptingState.lastModified,
                 };
                 try {
                     await writeJsonAtomic(statePath, nextState, { trailingNewline: true });
@@ -463,11 +479,7 @@ export async function maybeRefreshPricingSnapshot(opts = {}) {
                     state: nextState,
                 };
             }
-            if (!response.ok) {
-                throw new Error(`Failed to fetch ${SOURCE_URL}: ${response.status} ${response.statusText}`);
-            }
-            const api = await response.json();
-            const snapshot = buildSnapshotFromApi(api, opts.providerAllowlist ?? DEFAULT_MODELSDEV_PROVIDERS, nowMs);
+            const snapshot = buildSnapshotFromApi(fetchResult.api, opts.providerAllowlist ?? DEFAULT_MODELSDEV_PROVIDERS, nowMs);
             if (countPricedModels(snapshot) === 0) {
                 throw new Error("Refusing to persist empty pricing snapshot from models.dev");
             }
@@ -483,8 +495,8 @@ export async function maybeRefreshPricingSnapshot(opts = {}) {
                 lastSuccessAt: nowMs,
                 lastResult: "success",
                 lastError: undefined,
-                etag: response.headers.get("etag") ?? attemptingState.etag,
-                lastModified: response.headers.get("last-modified") ?? attemptingState.lastModified,
+                etag: fetchResult.etag ?? attemptingState.etag,
+                lastModified: fetchResult.lastModified ?? attemptingState.lastModified,
             };
             try {
                 await writeJsonAtomic(statePath, nextState, { trailingNewline: true });