@slkiser/opencode-quota 2.17.0 → 3.0.1

package/dist/plugin.js

@@ -13,12 +13,11 @@ import { formatQuotaCommand } from "./lib/quota-command-format.js";
 import { getProviders } from "./providers/registry.js";
 import { tool } from "@opencode-ai/plugin";
 import { aggregateUsage, resolveSessionTree, SessionNotFoundError, } from "./lib/quota-stats.js";
-import { fetchSessionTokensForDisplay } from "./lib/session-tokens.js";
 import { formatQuotaStatsReport } from "./lib/quota-stats-format.js";
 import { buildQuotaStatusReport } from "./lib/quota-status.js";
+import { inspectTuiConfig } from "./lib/tui-config-diagnostics.js";
 import { getPricingSnapshotMeta, getPricingSnapshotSource, getRuntimePricingRefreshStatePath, getRuntimePricingSnapshotPath, maybeRefreshPricingSnapshot, setPricingSnapshotAutoRefresh, setPricingSnapshotSelection, } from "./lib/modelsdev-pricing.js";
 import { refreshGoogleTokensForAllAccounts } from "./lib/google.js";
-import { getQuotaProviderDisplayLabel } from "./lib/provider-metadata.js";
 import { DEFAULT_ALIBABA_AUTH_CACHE_MAX_AGE_MS, isAlibabaModelId, resolveAlibabaCodingPlanAuthCached, } from "./lib/alibaba-auth.js";
 import { isQwenCodeModelId, resolveQwenLocalPlanCached } from "./lib/qwen-auth.js";
 import { recordAlibabaCodingPlanCompletion, recordQwenCompletion } from "./lib/qwen-local-quota.js";
@@ -27,7 +26,7 @@ import { parseOptionalJsonArgs, parseQuotaBetweenArgs, startOfLocalDayMs, startO
 import { handled } from "./lib/command-handled.js";
 import { renderCommandHeading } from "./lib/format-utils.js";
 import { sanitizeDisplayText } from "./lib/display-sanitize.js";
-import { getAnthropicNoDataMessage } from "./providers/anthropic.js";
+import { collectQuotaRenderData, matchesQuotaProviderCurrentSelection, resolveQuotaRenderSelection, } from "./lib/quota-render-data.js";
 /** All token report command specifications */
 const TOKEN_REPORT_COMMANDS = [
     {
@@ -243,101 +242,6 @@ export const QuotaToastPlugin = async ({ client }) => {
             return false;
         return true;
     }
-    function makeProviderFetchCacheKey(providerId, ctx) {
-        const style = ctx.config.toastStyle ?? "classic";
-        const googleModels = ctx.config.googleModels.join(",");
-        const alibabaCodingPlanTier = ctx.config.alibabaCodingPlanTier;
-        const cursorPlan = ctx.config.cursorPlan;
-        const cursorIncludedApiUsd = ctx.config.cursorIncludedApiUsd ?? "";
-        const cursorBillingCycleStartDay = ctx.config.cursorBillingCycleStartDay ?? "";
-        const onlyCurrentModel = ctx.config.onlyCurrentModel ? "yes" : "no";
-        const currentModel = ctx.config.currentModel ?? "";
-        const currentProviderID = ctx.config.currentProviderID ?? "";
-        const anthropicBinaryPath = ctx.config.anthropicBinaryPath ?? "";
-        return `${providerId}|style=${style}|anthropicBinaryPath=${anthropicBinaryPath}|googleModels=${googleModels}|alibabaTier=${alibabaCodingPlanTier}|cursorPlan=${cursorPlan}|cursorIncludedApiUsd=${cursorIncludedApiUsd}|cursorBillingCycleStartDay=${cursorBillingCycleStartDay}|onlyCurrentModel=${onlyCurrentModel}|currentModel=${currentModel}|currentProviderID=${currentProviderID}`;
-    }
-    async function fetchProviderWithCache(params) {
-        const { provider, ctx, ttlMs } = params;
-        // Live local-usage providers should update per completion for accurate local reports.
-        if (LIVE_LOCAL_USAGE_PROVIDER_IDS.has(provider.id)) {
-            return await provider.fetch(ctx);
-        }
-        const cacheKey = makeProviderFetchCacheKey(provider.id, ctx);
-        const now = Date.now();
-        const existing = providerFetchCache.get(cacheKey);
-        if (existing?.result &&
-            existing.timestamp > 0 &&
-            ttlMs > 0 &&
-            now - existing.timestamp < ttlMs) {
-            return existing.result;
-        }
-        if (existing?.inFlight) {
-            return existing.inFlight;
-        }
-        const promise = (async () => {
-            try {
-                const result = await provider.fetch(ctx);
-                if (result.attempted) {
-                    providerFetchCache.set(cacheKey, { timestamp: Date.now(), result });
-                }
-                else {
-                    providerFetchCache.delete(cacheKey);
-                }
-                return result;
-            }
-            catch (err) {
-                providerFetchCache.delete(cacheKey);
-                throw err;
-            }
-        })();
-        providerFetchCache.set(cacheKey, {
-            timestamp: existing?.timestamp ?? 0,
-            result: existing?.result,
-            inFlight: promise,
-        });
-        return promise;
-    }
-    function makeProviderFetchFailure(provider) {
-        return {
-            attempted: true,
-            entries: [],
-            errors: [
-                {
-                    label: getQuotaProviderDisplayLabel(provider.id),
-                    message: "Failed to read quota data",
-                },
-            ],
-        };
-    }
-    async function fetchProviderResults(params) {
-        const settled = await Promise.allSettled(params.providers.map((provider) => fetchProviderWithCache({
-            provider,
-            ctx: params.ctx,
-            ttlMs: params.ttlMs,
-        })));
-        return settled.map((result, index) => result.status === "fulfilled"
-            ? result.value
-            : makeProviderFetchFailure(params.providers[index]));
-    }
-    function getExplicitNoDataMessage(provider) {
-        if (provider.id === "cursor") {
-            return "No local usage yet";
-        }
-        if (provider.id === "anthropic") {
-            return getAnthropicNoDataMessage();
-        }
-        return "Not configured";
-    }
-    function shouldSurfaceNoDataMessage(params) {
-        const { provider, result, isAutoMode, activeProviderCount } = params;
-        if (result.attempted || result.entries.length > 0 || result.errors.length > 0) {
-            return false;
-        }
-        if (!isAutoMode) {
-            return true;
-        }
-        return activeProviderCount === 1 && (provider.id === "anthropic" || provider.id === "cursor");
-    }
     function isProviderEnabled(providerId) {
         return config.enabledProviders === "auto" || config.enabledProviders.includes(providerId);
     }
@@ -514,16 +418,6 @@ export const QuotaToastPlugin = async ({ client }) => {
             return {};
         }
     }
-    function matchesProviderCurrentSelection(params) {
-        if (params.provider.id === "cursor" && isCursorProviderId(params.currentProviderID)) {
-            return true;
-        }
-        if (!params.currentModel)
-            return false;
-        return params.provider.matchesCurrentModel
-            ? params.provider.matchesCurrentModel(params.currentModel)
-            : true;
-    }
     function formatDebugInfo(params) {
         const availability = params.availability
             ? params.availability.map((x) => `${x.id}=${x.ok ? "ok" : "no"}`).join(" ")
@@ -543,55 +437,6 @@ export const QuotaToastPlugin = async ({ client }) => {
             `available=${availability}`,
         ].join("\n");
     }
-    async function resolveQuotaCommandSelection(params = {}) {
-        if (!configLoaded)
-            await refreshConfig();
-        if (!config.enabled)
-            return null;
-        const allProviders = getProviders();
-        const isAutoMode = config.enabledProviders === "auto";
-        const providers = isAutoMode
-            ? allProviders
-            : allProviders.filter((p) => config.enabledProviders.includes(p.id));
-        if (!isAutoMode && providers.length === 0)
-            return null;
-        let currentModel;
-        let currentProviderID;
-        if (config.onlyCurrentModel && params.sessionID) {
-            const currentSession = params.sessionMeta ?? (await getSessionModelMeta(params.sessionID));
-            currentModel = currentSession.modelID;
-            currentProviderID = currentSession.providerID;
-        }
-        const ctx = {
-            client: typedClient,
-            config: {
-                googleModels: config.googleModels,
-                anthropicBinaryPath: config.anthropicBinaryPath,
-                alibabaCodingPlanTier: config.alibabaCodingPlanTier,
-                cursorPlan: config.cursorPlan,
-                cursorIncludedApiUsd: config.cursorIncludedApiUsd,
-                cursorBillingCycleStartDay: config.cursorBillingCycleStartDay,
-                // Always format /quota in grouped mode for a more dashboard-like look.
-                toastStyle: "grouped",
-                onlyCurrentModel: config.onlyCurrentModel,
-                currentModel,
-                currentProviderID,
-            },
-        };
-        const filteringByCurrentSelection = config.onlyCurrentModel && Boolean(currentModel || isCursorProviderId(currentProviderID));
-        const filtered = filteringByCurrentSelection
-            ? providers.filter((p) => matchesProviderCurrentSelection({ provider: p, currentModel, currentProviderID }))
-            : providers;
-        return {
-            isAutoMode,
-            providers,
-            filtered,
-            ctx,
-            currentModel,
-            currentProviderID,
-            filteringByCurrentSelection,
-        };
-    }
     function describeQuotaCommandCurrentSelection(params) {
         if (isCursorProviderId(params.currentProviderID)) {
             return `current provider: ${params.currentProviderID}`;
@@ -602,7 +447,12 @@ export const QuotaToastPlugin = async ({ client }) => {
         return "current session";
     }
     async function buildQuotaCommandUnavailableMessage(params = {}) {
-        const selection = await resolveQuotaCommandSelection(params);
+        const selection = await resolveQuotaRenderSelection({
+            client: typedClient,
+            config,
+            request: params,
+            formatStyle: "grouped",
+        });
         if (!selection) {
             return "Quota unavailable\n\nNo enabled quota providers are configured.\n\nRun /quota_status for diagnostics.";
         }
@@ -648,140 +498,56 @@ export const QuotaToastPlugin = async ({ client }) => {
                 ? formatDebugInfo({ trigger, reason: "disabled", enabledProviders: [] })
                 : null;
         }
-        const allProviders = getProviders();
-        const isAutoMode = config.enabledProviders === "auto";
-        const enabledProviderIds = isAutoMode ? [] : config.enabledProviders;
-        // When enabledProviders is "auto", we'll filter by availability below.
-        // When explicit, filter to just the listed providers.
-        const providers = isAutoMode
-            ? allProviders
-            : allProviders.filter((p) => enabledProviderIds.includes(p.id));
-        // Only bail on empty if user explicitly configured an empty list.
-        if (!isAutoMode && providers.length === 0) {
+        if (config.enabledProviders !== "auto" && config.enabledProviders.length === 0) {
             return config.debug
                 ? formatDebugInfo({ trigger, reason: "enabledProviders empty", enabledProviders: [] })
                 : null;
         }
-        let currentModel;
-        let currentProviderID;
-        if (config.onlyCurrentModel) {
-            const currentSession = await getSessionModelMeta(sessionID);
-            currentModel = currentSession.modelID;
-            currentProviderID = currentSession.providerID;
-        }
-        const ctx = {
-            client: typedClient,
-            config: {
-                googleModels: config.googleModels,
-                anthropicBinaryPath: config.anthropicBinaryPath,
-                alibabaCodingPlanTier: config.alibabaCodingPlanTier,
-                cursorPlan: config.cursorPlan,
-                cursorIncludedApiUsd: config.cursorIncludedApiUsd,
-                cursorBillingCycleStartDay: config.cursorBillingCycleStartDay,
-                toastStyle: config.toastStyle,
-                onlyCurrentModel: config.onlyCurrentModel,
-                currentModel,
-                currentProviderID,
-            },
+        const quotaRequestContext = {
+            sessionID,
+            sessionMeta: config.onlyCurrentModel && sessionID ? await getSessionModelMeta(sessionID) : undefined,
         };
-        const filtered = config.onlyCurrentModel && (currentModel || isCursorProviderId(currentProviderID))
-            ? providers.filter((p) => matchesProviderCurrentSelection({ provider: p, currentModel, currentProviderID }))
-            : providers;
-        // availability checks are cheap, do them in parallel
-        const avail = await Promise.all(filtered.map(async (p) => ({ p, ok: await p.isAvailable(ctx) })));
-        const active = avail.filter((x) => x.ok).map((x) => x.p);
-        if (active.length === 0) {
+        const quotaResult = await collectQuotaRenderData({
+            client: typedClient,
+            config,
+            request: quotaRequestContext,
+            providerFetchCache,
+            surfaceExplicitProviderIssues: true,
+            formatStyle: config.formatStyle,
+        });
+        const { selection, availability, active, attemptedAny, hasExplicitProviderIssues, data } = quotaResult;
+        if (config.showSessionTokens && sessionID) {
+            lastSessionTokenError = quotaResult.sessionTokenError;
+        }
+        const currentModel = selection?.currentModel;
+        const errors = data?.errors ?? [];
+        if (active.length === 0 && !(hasExplicitProviderIssues && errors.length > 0)) {
             return config.debug
                 ? formatDebugInfo({
                     trigger,
                     reason: "no enabled providers available",
                     currentModel,
                     enabledProviders: config.enabledProviders,
-                    availability: avail.map((x) => ({ id: x.p.id, ok: x.ok })),
+                    availability: availability.map((item) => ({
+                        id: item.provider.id,
+                        ok: item.ok,
+                    })),
                 })
                 : null;
         }
-        const results = await fetchProviderResults({
-            providers: active,
-            ctx,
-            ttlMs: config.minIntervalMs,
-        });
-        const entries = results.flatMap((r) => r.entries);
-        const errors = results.flatMap((r) => r.errors);
-        const attemptedAny = results.some((r) => r.attempted);
-        let hasExplicitProviderIssues = false;
-        for (let i = 0; i < active.length; i++) {
-            const provider = active[i];
-            const result = results[i];
-            if (shouldSurfaceNoDataMessage({
-                provider,
-                result,
-                isAutoMode,
-                activeProviderCount: active.length,
-            })) {
-                errors.push({
-                    label: getQuotaProviderDisplayLabel(provider.id),
-                    message: getExplicitNoDataMessage(provider),
-                });
-                if (!isAutoMode) {
-                    hasExplicitProviderIssues = true;
-                }
-            }
-        }
-        // When enabledProviders is an explicit list, surface unavailable/skipped
-        // providers instead of silently omitting them.
-        if (!isAutoMode) {
-            // If a user explicitly enabled providers that are unavailable (or skipped due
-            // to model filtering), surface that instead of silently omitting them.
-            const filteredIds = new Set(filtered.map((p) => p.id));
-            const activeIds = new Set(active.map((p) => p.id));
-            const availById = new Map(avail.map((x) => [x.p.id, x.ok]));
-            for (const p of providers) {
-                if (activeIds.has(p.id))
-                    continue;
-                if (!filteredIds.has(p.id)) {
-                    const detail = config.onlyCurrentModel && currentModel ? `current model: ${currentModel}` : "filtered";
-                    errors.push({
-                        label: getQuotaProviderDisplayLabel(p.id),
-                        message: `Skipped (${detail})`,
-                    });
-                    hasExplicitProviderIssues = true;
-                    continue;
-                }
-                const ok = availById.get(p.id);
-                if (ok === false) {
-                    errors.push({
-                        label: getQuotaProviderDisplayLabel(p.id),
-                        message: "Unavailable (not detected)",
-                    });
-                    hasExplicitProviderIssues = true;
-                }
-            }
-        }
-        // Fetch session tokens if enabled and sessionID is available
-        let sessionTokens;
-        if (config.showSessionTokens && sessionID) {
-            const stResult = await fetchSessionTokensForDisplay({
-                enabled: config.showSessionTokens,
-                sessionID,
-            });
-            sessionTokens = stResult.sessionTokens;
-            // Update diagnostics state: clear on success (no error returned), set on failure
-            lastSessionTokenError = stResult.error;
-        }
-        if (entries.length > 0) {
+        if (data?.entries.length) {
             const formatted = formatQuotaRows({
                 version: "1.0.0",
                 layout: config.layout,
-                entries,
-                errors,
-                style: config.toastStyle,
-                sessionTokens,
+                entries: data.entries,
+                errors: data.errors,
+                style: config.formatStyle,
+                sessionTokens: data.sessionTokens,
             });
             if (!config.debug)
                 return formatted;
-            const debugFooter = `\n\n[debug] src=${configMeta.source} providers=${config.enabledProviders === "auto" ? "(auto)" : config.enabledProviders.join(",") || "(none)"} avail=${avail
-                .map((x) => `${x.p.id}:${x.ok ? "ok" : "no"}`)
+            const debugFooter = `\n\n[debug] src=${configMeta.source} providers=${config.enabledProviders === "auto" ? "(auto)" : config.enabledProviders.join(",") || "(none)"} avail=${availability
+                .map((item) => `${item.provider.id}:${item.ok ? "ok" : "no"}`)
                 .join(" ")}`;
             return formatted + debugFooter;
         }
@@ -789,8 +555,7 @@ export const QuotaToastPlugin = async ({ client }) => {
         // 1. showOnBothFail is enabled and at least one provider attempted (existing behavior)
         // 2. OR we're in explicit mode and have "Not configured"/"Unavailable" errors (new behavior)
         if ((config.showOnBothFail && attemptedAny && errors.length > 0) || hasExplicitProviderIssues) {
-            // Format errors as individual lines
-            const errorLines = errors.map((e) => `${e.label}: ${e.message}`).join("\n");
+            const errorLines = errors.map((error) => `${error.label}: ${error.message}`).join("\n");
             if (!config.debug)
                 return errorLines || "Quota unavailable";
             return ((errorLines || "Quota unavailable") +
@@ -802,7 +567,10 @@ export const QuotaToastPlugin = async ({ client }) => {
                         : "all providers failed",
                     currentModel,
                     enabledProviders: config.enabledProviders,
-                    availability: avail.map((x) => ({ id: x.p.id, ok: x.ok })),
+                    availability: availability.map((item) => ({
+                        id: item.provider.id,
+                        ok: item.ok,
+                    })),
                 }));
         }
         return config.debug
@@ -811,7 +579,10 @@ export const QuotaToastPlugin = async ({ client }) => {
                 reason: "no entries",
                 currentModel,
                 enabledProviders: config.enabledProviders,
-                availability: avail.map((x) => ({ id: x.p.id, ok: x.ok })),
+                availability: availability.map((item) => ({
+                    id: item.provider.id,
+                    ok: item.ok,
+                })),
             })
             : null;
     }
@@ -871,50 +642,18 @@ export const QuotaToastPlugin = async ({ client }) => {
         }
     }
     async function fetchQuotaCommandData(trigger, params = {}) {
-        const selection = await resolveQuotaCommandSelection(params);
-        if (!selection)
-            return null;
-        const { isAutoMode, ctx } = selection;
-        const avail = await Promise.all(selection.filtered.map(async (p) => ({ p, ok: await p.isAvailable(ctx) })));
-        const active = avail.filter((x) => x.ok).map((x) => x.p);
-        if (active.length === 0)
-            return null;
-        const results = await fetchProviderResults({
-            providers: active,
-            ctx,
-            ttlMs: config.minIntervalMs,
+        const quotaResult = await collectQuotaRenderData({
+            client: typedClient,
+            config,
+            request: params,
+            providerFetchCache,
+            surfaceExplicitProviderIssues: false,
+            formatStyle: "grouped",
         });
-        const entries = results.flatMap((r) => r.entries);
-        const errors = results.flatMap((r) => r.errors);
-        for (let i = 0; i < active.length; i++) {
-            const provider = active[i];
-            const result = results[i];
-            if (shouldSurfaceNoDataMessage({
-                provider,
-                result,
-                isAutoMode,
-                activeProviderCount: active.length,
-            })) {
-                errors.push({
-                    label: getQuotaProviderDisplayLabel(provider.id),
-                    message: getExplicitNoDataMessage(provider),
-                });
-            }
-        }
-        // Fetch session tokens if enabled and sessionID is available
-        let sessionTokens;
         if (config.showSessionTokens && params.sessionID) {
-            const stResult = await fetchSessionTokensForDisplay({
-                enabled: config.showSessionTokens,
-                sessionID: params.sessionID,
-            });
-            sessionTokens = stResult.sessionTokens;
-            // Update diagnostics state: clear on success (no error returned), set on failure
-            lastSessionTokenError = stResult.error;
+            lastSessionTokenError = quotaResult.sessionTokenError;
         }
-        if (entries.length === 0 && errors.length === 0 && !sessionTokens)
-            return null;
-        return { entries, errors, sessionTokens };
+        return quotaResult.data;
     }
     async function buildQuotaReport(params) {
         const result = await aggregateUsage({
@@ -976,16 +715,23 @@ export const QuotaToastPlugin = async ({ client }) => {
                 enabled: isAutoMode ? ok : config.enabledProviders.includes(p.id),
                 available: ok,
                 matchesCurrentModel: currentModel || isCursorProviderId(currentProviderID)
-                    ? matchesProviderCurrentSelection({ provider: p, currentModel, currentProviderID })
+                    ? matchesQuotaProviderCurrentSelection({
+                        provider: p,
+                        currentModel,
+                        currentProviderID,
+                    })
                     : undefined,
             };
         }));
         const refresh = params.refreshGoogleTokens
             ? await refreshGoogleTokensForAllAccounts({ skewMs: params.skewMs, force: params.force })
             : null;
+        const tuiDiagnostics = await inspectTuiConfig();
         return await buildQuotaStatusReport({
+            tuiDiagnostics,
             configSource: configMeta.source,
             configPaths: configMeta.paths,
+            networkSettingSources: configMeta.networkSettingSources,
             enabledProviders: config.enabledProviders,
             anthropicBinaryPath: config.anthropicBinaryPath,
             alibabaCodingPlanTier: config.alibabaCodingPlanTier,
@@ -1249,7 +995,7 @@ export const QuotaToastPlugin = async ({ client }) => {
             };
             cfg.command["quota_status"] = {
                 template: "/quota_status",
-                description: "Diagnostics for toast + pricing + local storage (includes unknown pricing report).",
+                description: "Diagnostics for toast + TUI + pricing + local storage (includes unknown pricing report).",
             };
             cfg.command["pricing_refresh"] = {
                 template: "/pricing_refresh",
@@ -1312,7 +1058,7 @@ export const QuotaToastPlugin = async ({ client }) => {
         },
         tool: {
             quota_status: tool({
-                description: "Diagnostics for toast + pricing + local storage (includes unknown pricing report).",
+                description: "Diagnostics for toast + TUI + pricing + local storage (includes unknown pricing report).",
                 args: {
                     refreshGoogleTokens: tool.schema
                         .boolean()