@99percentpeople/pi-codex-api 0.1.3 → 0.2.0

package/dist/{index.js → index.ts}

@@ -1202,6 +1202,11 @@ import {
   getAgentDir
 } from "@earendil-works/pi-coding-agent";
 var USAGE_PATH = "../wham/usage";
+var REDEEM_CREDITS_PATH = "../wham/rate-limit-reset-credits";
+var REDEEM_PATH = "../wham/rate-limit-reset-credits/consume";
+var REDEEM_CONFIRM_WINDOW_MS = 1e4;
+var REDEEM_DIALOG_TIMEOUT_MS = 30000;
+var REDEEM_RETRY_WINDOW_MS = 5 * 60000;
 var USAGE_REFRESH_INTERVAL_MS = 60000;
 var AUTH_WATCH_DEBOUNCE_MS = 100;
 var STATUS_KEY = "codex-api-usage";
@@ -1215,6 +1220,18 @@ function payloadNumber(value) {
   const number = typeof value === "number" ? value : typeof value === "string" ? Number(value) : NaN;
   return Number.isFinite(number) ? number : undefined;
 }
+function payloadDate(value) {
+  if (typeof value === "number")
+    return Number.isFinite(value) ? value : undefined;
+  if (typeof value === "string") {
+    const ms = Date.parse(value);
+    return Number.isFinite(ms) ? ms : undefined;
+  }
+  return;
+}
+function payloadString(value) {
+  return typeof value === "string" && value.trim() ? value.trim() : undefined;
+}
 function payloadBool(value) {
   if (typeof value === "boolean")
     return value;
@@ -1253,14 +1270,65 @@ function payloadCredits(value) {
     balance: typeof balance === "string" && balance ? balance : undefined
   };
 }
-function payloadSnapshot(limitId, limitName, rateLimitValue, creditsValue) {
+function payloadSnapshot(limitId, limitName, rateLimitValue, creditsValue, limitReached) {
   const rateLimit = object(rateLimitValue);
   return {
     limitId,
     limitName,
     primary: payloadWindow(rateLimit && property(rateLimit, "primary_window", "primaryWindow")),
     secondary: payloadWindow(rateLimit && property(rateLimit, "secondary_window", "secondaryWindow")),
-    credits: payloadCredits(creditsValue)
+    credits: payloadCredits(creditsValue),
+    limitReached
+  };
+}
+function parseCodexAccountInfo(value) {
+  const input = object(value);
+  if (!input)
+    return;
+  const planType = payloadString(property(input, "plan_type", "planType"));
+  const email = payloadString(property(input, "email", "email"));
+  if (planType === undefined && email === undefined)
+    return;
+  return { planType, email };
+}
+function maskCodexEmail(email) {
+  const [local, domain] = email.split("@");
+  if (!domain)
+    return "***";
+  const head = local.length > 3 ? local.slice(0, 3) : local.slice(0, 1);
+  return `${head}***@${domain}`;
+}
+function parseCodexRedeemCredits(value) {
+  const input = object(value);
+  if (!input)
+    return;
+  const availableCount = payloadNumber(property(input, "available_count", "availableCount"));
+  if (availableCount === undefined)
+    return;
+  const credits = [];
+  const list = property(input, "credits", "credits");
+  if (Array.isArray(list)) {
+    for (const item of list) {
+      const credit = object(item);
+      if (!credit)
+        continue;
+      const id = payloadString(property(credit, "id", "id"));
+      if (!id)
+        continue;
+      credits.push({
+        id,
+        title: payloadString(property(credit, "title", "title")),
+        description: payloadString(property(credit, "description", "description")),
+        status: payloadString(property(credit, "status", "status")),
+        grantedAt: payloadDate(property(credit, "granted_at", "grantedAt")),
+        expiresAt: payloadDate(property(credit, "expires_at", "expiresAt"))
+      });
+    }
+  }
+  return {
+    availableCount,
+    totalEarnedCount: payloadNumber(property(input, "total_earned_count", "totalEarnedCount")),
+    credits
   };
 }
 function parseCodexUsagePayload(value) {
@@ -1268,7 +1336,9 @@ function parseCodexUsagePayload(value) {
   if (!input)
     return [];
   const rateLimit = property(input, "rate_limit", "rateLimit");
-  const snapshots = rateLimit !== undefined || input.credits !== undefined ? [payloadSnapshot("codex", undefined, rateLimit, input.credits)] : [];
+  const rateLimitObject = object(rateLimit);
+  const limitReached = payloadBool(rateLimitObject && property(rateLimitObject, "limit_reached", "limitReached"));
+  const snapshots = rateLimit !== undefined || input.credits !== undefined ? [payloadSnapshot("codex", undefined, rateLimit, input.credits, limitReached)] : [];
   const additional = property(input, "additional_rate_limits", "additionalRateLimits");
   if (Array.isArray(additional)) {
     for (const value2 of additional) {
@@ -1330,14 +1400,16 @@ function parseCodexRateLimits(input) {
       unlimited,
       balance: headers["x-codex-credits-balance"]
     } : undefined;
-    if (!primary && !secondary && !credits)
+    const limitReached = prefix === "x-codex" ? headers["x-codex-rate-limit-reached-type"] !== undefined : undefined;
+    if (!primary && !secondary && !credits && !limitReached)
       return [];
     return [{
       limitId: prefix.slice(2).replace(/-/g, "_"),
       limitName: headers[`${prefix}-limit-name`],
       primary,
       secondary,
-      credits
+      credits,
+      limitReached
     }];
   });
 }
@@ -1350,13 +1422,17 @@ function resetText(epochSeconds, now = Date.now()) {
   const remainingMs = epochSeconds * 1000 - now;
   if (remainingMs <= 0)
     return;
-  const minutes = Math.ceil(remainingMs / 60000);
-  if (minutes < 60)
-    return `${minutes}m`;
-  const hours = Math.ceil(minutes / 60);
-  if (hours < 48)
-    return `${hours}h`;
-  return `${Math.ceil(hours / 24)}d`;
+  const totalMinutes = Math.ceil(remainingMs / 60000);
+  if (totalMinutes < 60)
+    return `${totalMinutes}m`;
+  if (totalMinutes < 24 * 60) {
+    const hours2 = Math.floor(totalMinutes / 60);
+    const minutes = totalMinutes % 60;
+    return minutes > 0 ? `${hours2}h ${minutes}m` : `${hours2}h`;
+  }
+  const days = Math.floor(totalMinutes / (24 * 60));
+  const hours = Math.floor(totalMinutes % (24 * 60) / 60);
+  return hours > 0 ? `${days}d ${hours}h` : `${days}d`;
 }
 var KNOWN_WINDOWS = [
   { minutes: 5 * 60, label: "5h" },
@@ -1393,10 +1469,11 @@ function usageBar(remaining) {
   const filled = Math.round(remaining / 100 * USAGE_BAR_WIDTH);
   return `[${"█".repeat(filled)}${"░".repeat(USAGE_BAR_WIDTH - filled)}]`;
 }
-function windowText(item, labelWidth, now) {
+function windowText(item, labelWidth, now, limitReached) {
   const reset = resetText(item.window.resetsAt, now);
   const remaining = remainingPercent(item.window);
-  return `${item.label.padEnd(labelWidth)} ${usageBar(remaining)} ${percent(remaining)}% left${reset ? ` resets in ${reset}` : ""}`;
+  const state = limitReached ? "limit reached" : `${percent(remaining)}% left`;
+  return `${item.label.padEnd(labelWidth)} ${usageBar(limitReached ? 0 : remaining)} ${state}${reset ? ` resets in ${reset}` : ""}`;
 }
 function creditsText(credits) {
   if (credits.unlimited)
@@ -1406,11 +1483,41 @@ function creditsText(credits) {
   }
   return "no additional credits";
 }
-function formatCodexUsage(snapshots, now = Date.now()) {
+function planLabel(planType) {
+  return planType.charAt(0).toUpperCase() + planType.slice(1);
+}
+function formatDateTime(epochMs) {
+  const date = new Date(epochMs);
+  const pad = (value) => String(value).padStart(2, "0");
+  const offsetMinutes = -new Date(epochMs).getTimezoneOffset();
+  const sign = offsetMinutes >= 0 ? "+" : "-";
+  const abs = Math.abs(offsetMinutes);
+  const offset = abs % 60 === 0 ? `UTC${sign}${abs / 60}` : `UTC${sign}${Math.floor(abs / 60)}:${pad(abs % 60)}`;
+  return `${date.getFullYear()}-${pad(date.getMonth() + 1)}-${pad(date.getDate())} ${pad(date.getHours())}:${pad(date.getMinutes())} ${offset}`;
+}
+function formatCodexRedeemCredits(redeemCredits, now = Date.now()) {
+  if (!redeemCredits || redeemCredits.availableCount <= 0)
+    return [];
+  const lines = [`rate limit redeem${redeemCredits.availableCount === 1 ? "" : ` ×${redeemCredits.availableCount}`}`];
+  const available = redeemCredits.credits.filter((credit) => credit.status === undefined || credit.status === "available").sort((left, right) => (left.expiresAt ?? Number.POSITIVE_INFINITY) - (right.expiresAt ?? Number.POSITIVE_INFINITY));
+  for (const credit of available) {
+    const details = ["available"];
+    if (credit.expiresAt !== undefined) {
+      details.push(credit.expiresAt > now ? `expires ${formatDateTime(credit.expiresAt)}` : "expired");
+    }
+    lines.push(`  ${credit.title ?? "reset credit"} (${details.join(", ")})`);
+  }
+  return lines.length > 1 ? lines : [];
+}
+function formatCodexUsage(snapshots, now = Date.now(), extras = {}) {
   if (snapshots.length === 0) {
     return "No Codex usage data is available. Run /codex-usage with an active Codex subscription model to refresh it.";
   }
   const lines = ["Codex usage"];
+  if (extras.account) {
+    const plan = extras.account.planType ? planLabel(extras.account.planType) : "unknown plan";
+    lines.push("", `account · ${plan}${extras.account.email ? ` (${maskCodexEmail(extras.account.email)})` : ""}`);
+  }
   for (const snapshot of snapshots) {
     const name = snapshot.limitName ?? snapshot.limitId;
     const windows = activeWindows(snapshot, now);
@@ -1419,10 +1526,13 @@ function formatCodexUsage(snapshots, now = Date.now()) {
     if (windows.length === 0)
       lines.push("  no active usage windows");
     else
-      lines.push(...windows.map((window) => `  ${windowText(window, labelWidth, now)}`));
+      lines.push(...windows.map((window) => `  ${windowText(window, labelWidth, now, snapshot.limitReached === true)}`));
     if (snapshot.credits)
       lines.push(`  ${creditsText(snapshot.credits)}`);
   }
+  const redeemLines = formatCodexRedeemCredits(extras.redeemCredits, now);
+  if (redeemLines.length > 0)
+    lines.push("", ...redeemLines);
   return lines.join(`
 `);
 }
@@ -1441,7 +1551,8 @@ function formatCodexStatus(snapshots, fastMode, now = Date.now()) {
     return;
   const remaining = remainingPercent(shortest.window);
   const reset = resetText(shortest.window.resetsAt, now);
-  return `Codex ${shortest.label} ${percent(remaining)}%${reset ? ` ${reset}` : ""}${fastMode ? " Fast" : ""}`;
+  const usage = snapshot.limitReached === true ? "limit reached" : `${percent(remaining)}%`;
+  return `Codex ${shortest.label} ${usage}${reset ? ` ${reset}` : ""}${fastMode ? " Fast" : ""}`;
 }
 function applyFastModePayload(payload, enabled) {
   if (!enabled || !payload || typeof payload !== "object" || Array.isArray(payload))
@@ -1450,6 +1561,7 @@ function applyFastModePayload(payload, enabled) {
 }
 function registerCodexUsageAndFast(pi, controller, options = {}) {
   const usageByAccount = new Map;
+  const pendingRedeemByAccount = new Map;
   let activeAccountId;
   let credentialRevision = 0;
   let latestContext;
@@ -1481,6 +1593,7 @@ function registerCodexUsageAndFast(pi, controller, options = {}) {
     credentialRevision += 1;
     activeAccountId = undefined;
     usageByAccount.clear();
+    pendingRedeemByAccount.clear();
     if (action === "set")
       showSyncingStatus(ctx);
     else
@@ -1492,6 +1605,7 @@ function registerCodexUsageAndFast(pi, controller, options = {}) {
     credentialRevision += 1;
     activeAccountId = accountId;
     usageByAccount.clear();
+    pendingRedeemByAccount.clear();
     showSyncingStatus(ctx);
     return true;
   };
@@ -1542,6 +1656,7 @@ function registerCodexUsageAndFast(pi, controller, options = {}) {
         if (parsed.length === 0)
           throw new Error("Codex usage API returned no usage data");
         state.snapshots = parsed;
+        state.account = parseCodexAccountInfo(payload);
         state.lastFetchAt = Date.now();
       })();
       let nextFetch;
@@ -1646,20 +1761,120 @@ function registerCodexUsageAndFast(pi, controller, options = {}) {
     }
   };
   pi.registerCommand("codex-usage", {
-    description: "Refresh and show Codex subscription usage limits and credits",
+    description: "Refresh and show Codex subscription usage, plan, and rate limit redeems",
     handler: async (_args, ctx) => {
       try {
         await refreshUsage(ctx, true);
       } catch (error) {
-        const message = error instanceof Error ? error.message : String(error);
+        const message2 = error instanceof Error ? error.message : String(error);
         const snapshots = currentState()?.snapshots ?? [];
         if (snapshots.length === 0) {
-          ctx.ui.notify(`Failed to refresh Codex usage: ${message}`, "error");
+          ctx.ui.notify(`Failed to refresh Codex usage: ${message2}`, "error");
+          return;
+        }
+        ctx.ui.notify(`Failed to refresh Codex usage; showing the latest snapshot: ${message2}`, "warning");
+      }
+      try {
+        const resolved = await resolveActiveClient(ctx, controller.getConfig());
+        const state2 = accountState(resolved.accountId);
+        const payload = await resolved.client.get(REDEEM_CREDITS_PATH);
+        state2.redeemCredits = parseCodexRedeemCredits(payload);
+      } catch {}
+      const state = currentState();
+      const message = formatCodexUsage(state?.snapshots ?? [], Date.now(), {
+        account: state?.account,
+        redeemCredits: state?.redeemCredits
+      });
+      ctx.ui.notify(ctx.ui.theme ? ctx.ui.theme.fg("muted", message) : message, "info");
+    }
+  });
+  pi.registerCommand("codex-redeem", {
+    description: "Preview and redeem an earned Codex rate limit reset credit (confirmation required)",
+    handler: async (_args, ctx) => {
+      const config = controller.getConfig();
+      if (ctx.model?.provider !== "openai-codex" && !config.allowOtherProviders) {
+        ctx.ui.notify("An active openai-codex model is required to redeem a rate limit reset. " + "Enable Other providers in /99settings to use the logged-in Codex subscription from another model.", "error");
+        return;
+      }
+      let resolved;
+      try {
+        resolved = await resolveActiveClient(ctx, config);
+      } catch (error) {
+        const message = error instanceof Error ? error.message : String(error);
+        ctx.ui.notify(`Failed to resolve the Codex subscription: ${message}`, "error");
+        return;
+      }
+      const state = accountState(resolved.accountId);
+      try {
+        const payload = await resolved.client.get(REDEEM_CREDITS_PATH);
+        const redeemCredits = parseCodexRedeemCredits(payload);
+        state.redeemCredits = redeemCredits;
+        const now = Date.now();
+        if (!redeemCredits || redeemCredits.availableCount <= 0) {
+          pendingRedeemByAccount.delete(resolved.accountId);
+          ctx.ui.notify("No Codex rate limit reset credits are available to redeem.", "info");
+          return;
+        }
+        const availableCredits = [...redeemCredits.credits].filter((item) => item.status === undefined || item.status === "available").sort((left, right) => (left.expiresAt ?? Number.POSITIVE_INFINITY) - (right.expiresAt ?? Number.POSITIVE_INFINITY));
+        const credit = availableCredits[0] ?? redeemCredits.credits[0];
+        const expiryOf = (item) => item?.expiresAt !== undefined && item.expiresAt > now ? ` (expires ${formatDateTime(item.expiresAt)})` : "";
+        let selected = credit;
+        if (ctx.hasUI) {
+          if (availableCredits.length > 1) {
+            const options2 = availableCredits.map((item) => `${item.title ?? "reset credit"}${expiryOf(item)}`);
+            const choice2 = await ctx.ui.select("Select a reset credit to redeem", options2, {
+              timeout: REDEEM_DIALOG_TIMEOUT_MS
+            });
+            if (choice2 === undefined) {
+              pendingRedeemByAccount.delete(resolved.accountId);
+              ctx.ui.notify("Redeem cancelled — no reset credit was consumed.", "info");
+              return;
+            }
+            const index = options2.indexOf(choice2);
+            selected = availableCredits[index] ?? credit;
+          }
+          const confirmOptions = ["No", "Yes"];
+          const choice = await ctx.ui.select(`Redeem ${selected?.title ?? "Full reset"}${expiryOf(selected)}?`, confirmOptions, { timeout: REDEEM_DIALOG_TIMEOUT_MS });
+          if (choice !== confirmOptions[1]) {
+            pendingRedeemByAccount.delete(resolved.accountId);
+            ctx.ui.notify("Redeem cancelled — no reset credit was consumed.", "info");
+            return;
+          }
+        }
+        const targetId = selected?.id;
+        const existing = pendingRedeemByAccount.get(resolved.accountId);
+        const pending = existing && existing.expiresAt > now && existing.creditId === targetId ? existing : {
+          redeemRequestId: crypto.randomUUID(),
+          creditId: targetId,
+          expiresAt: now + REDEEM_CONFIRM_WINDOW_MS
+        };
+        pendingRedeemByAccount.set(resolved.accountId, pending);
+        if (!ctx.hasUI && (!existing || existing.expiresAt <= now || existing.creditId !== targetId)) {
+          ctx.ui.notify(`${redeemCredits.availableCount} rate limit reset redeem available: ${credit?.title ?? "Full reset"}${expiryOf(credit)}. ` + `Run /codex-redeem again within ${REDEEM_CONFIRM_WINDOW_MS / 1000}s to confirm.`, "warning");
           return;
         }
-        ctx.ui.notify(`Failed to refresh Codex usage; showing the latest snapshot: ${message}`, "warning");
+        try {
+          await resolved.client.post(REDEEM_PATH, {
+            redeem_request_id: pending.redeemRequestId,
+            credit_id: pending.creditId
+          });
+          pendingRedeemByAccount.delete(resolved.accountId);
+          try {
+            await refreshUsage(ctx, true);
+          } catch {}
+          const status = formatCodexStatus(currentState()?.snapshots ?? [], config.fastMode);
+          ctx.ui.notify(`✓ Rate limit reset redeemed — usage reset.${status ? `
+${status}` : ""}`, "info");
+        } catch (error) {
+          pending.expiresAt = Date.now() + REDEEM_RETRY_WINDOW_MS;
+          pendingRedeemByAccount.set(resolved.accountId, pending);
+          const message = error instanceof Error ? error.message : String(error);
+          ctx.ui.notify(`Failed to redeem a rate limit reset: ${message}. ` + "Run /codex-redeem again to retry with the same request ID.", "error");
+        }
+      } catch (error) {
+        const message = error instanceof Error ? error.message : String(error);
+        ctx.ui.notify(`Failed to redeem a rate limit reset: ${message}`, "error");
       }
-      ctx.ui.notify(formatCodexUsage(currentState()?.snapshots ?? []), "info");
     }
   });
   pi.on("before_provider_request", (event, ctx) => {
@@ -1691,6 +1906,7 @@ function registerCodexUsageAndFast(pi, controller, options = {}) {
     credentialRevision += 1;
     activeAccountId = undefined;
     usageByAccount.clear();
+    pendingRedeemByAccount.clear();
     latestContext = undefined;
     accountObserverActive = false;
     if (authWatchDebounce)
@@ -1741,14 +1957,18 @@ export {
   registerCodexImageTool,
   registerCodexApiSettings,
   parseCodexUsagePayload,
+  parseCodexRedeemCredits,
   parseCodexRateLimits,
+  parseCodexAccountInfo,
   normalizeCodexImageSize,
   normalizeCodexApiConfig,
+  maskCodexEmail,
   loadCodexApiConfig,
   getCodexApiConfigPath,
   formatCodexUsage,
   formatCodexStatus,
   formatCodexSearchDisplay,
+  formatCodexRedeemCredits,
   extractCodexAccountId,
   codex_api_default as default,
   createCodexSearchDisplay,
@@ -1765,5 +1985,5 @@ export {
   CODEX_API_SETTINGS_NAMESPACE
 };
 
-//# debugId=C13FC844FE4EE5BA64756E2164756E21
-//# sourceMappingURL=index.js.map
+//# debugId=356A61CA111E3F1D64756E2164756E21
+//# sourceMappingURL=index.ts.map