@anthonyhaussman/opencode-agy-auth 1.0.11-alpha.0 → 1.0.11-alpha.2

package/dist/index.js

@@ -212,6 +212,23 @@ async function retrieveUserQuota(accessToken, projectId, userAgentModel) {
     return null;
   }
 }
+async function retrieveUserQuotaSummary(accessToken, projectId, userAgentModel) {
+  const url2 = `${AGY_CODE_ASSIST_ENDPOINT}/v1internal:retrieveUserQuotaSummary`;
+  const headers = buildCodeAssistHeaders(accessToken, userAgentModel);
+  try {
+    const response = await agyFetch(url2, {
+      method: "POST",
+      headers,
+      body: JSON.stringify({ project: projectId })
+    });
+    if (!response.ok) {
+      return null;
+    }
+    return await response.json();
+  } catch {
+    return null;
+  }
+}
 function buildCodeAssistHeaders(accessToken, userAgentModel) {
   const userAgent = buildAgyCliUserAgent(userAgentModel);
   return {
@@ -14142,6 +14159,65 @@ async function fetchTokenRefresh(refreshToken) {
   return agyFetch(tokenUrl, init);
 }
 
+// src/plugin/quota-utils.ts
+function clamp(value, min, max) {
+  if (Number.isNaN(value)) {
+    return min;
+  }
+  if (value < min) {
+    return min;
+  }
+  if (value > max) {
+    return max;
+  }
+  return value;
+}
+function pad(value, width) {
+  if (value.length >= width) {
+    return value;
+  }
+  return value.padEnd(width, " ");
+}
+function buildProgressBar(fraction, width = 20) {
+  const clamped = clamp(fraction, 0, 1);
+  const filled = clamped >= 1 ? width : Math.max(0, Math.min(width, Math.max(clamped > 0 ? 1 : 0, Math.floor(clamped * width))));
+  const empty = width - filled;
+  return `${"\u2593".repeat(filled)}${"\u2591".repeat(empty)}`;
+}
+function formatRemainingAmount(value) {
+  if (!value) {
+    return void 0;
+  }
+  const parsed = Number(value);
+  if (!Number.isFinite(parsed)) {
+    return value;
+  }
+  return parsed.toLocaleString("en-US");
+}
+function formatRelativeResetTime(resetTime) {
+  if (!resetTime) {
+    return void 0;
+  }
+  const resetAt = new Date(resetTime).getTime();
+  if (Number.isNaN(resetAt)) {
+    return void 0;
+  }
+  const diffMs = resetAt - Date.now();
+  if (diffMs <= 0) {
+    return "reset pending";
+  }
+  const totalMinutes = Math.ceil(diffMs / (1e3 * 60));
+  const hours = Math.floor(totalMinutes / 60);
+  const minutes = totalMinutes % 60;
+  if (hours > 0 && minutes > 0) {
+    return `resets in ${hours}h ${minutes}m`;
+  }
+  if (hours > 0) {
+    return `resets in ${hours}h`;
+  }
+  return `resets in ${minutes}m`;
+}
+
 // src/plugin/quota.ts
 var AGY_QUOTA_TOOL_NAME = "agy_quota";
 function createAgyQuotaTool({
@@ -14269,60 +14345,6 @@ function formatUsageRemaining(bucket) {
   }
   return "unknown";
 }
-function formatRemainingAmount(value) {
-  if (!value) {
-    return void 0;
-  }
-  const parsed = Number.parseInt(value, 10);
-  if (!Number.isFinite(parsed)) {
-    return value;
-  }
-  return parsed.toLocaleString("en-US");
-}
-function formatRelativeResetTime(resetTime) {
-  if (!resetTime) {
-    return void 0;
-  }
-  const resetAt = new Date(resetTime).getTime();
-  if (Number.isNaN(resetAt)) {
-    return void 0;
-  }
-  const diffMs = resetAt - Date.now();
-  if (diffMs <= 0) {
-    return "reset pending";
-  }
-  const totalMinutes = Math.ceil(diffMs / (1e3 * 60));
-  const hours = Math.floor(totalMinutes / 60);
-  const minutes = totalMinutes % 60;
-  if (hours > 0 && minutes > 0) {
-    return `resets in ${hours}h ${minutes}m`;
-  }
-  if (hours > 0) {
-    return `resets in ${hours}h`;
-  }
-  return `resets in ${minutes}m`;
-}
-function buildProgressBar(fraction, width = 20) {
-  const clamped = clamp(fraction, 0, 1);
-  const filled = clamped >= 1 ? width : Math.max(0, Math.min(width, Math.max(clamped > 0 ? 1 : 0, Math.floor(clamped * width))));
-  const empty = width - filled;
-  return `${"\u2593".repeat(filled)}${"\u2591".repeat(empty)}`;
-}
-function pad(value, width) {
-  if (value.length >= width) {
-    return value;
-  }
-  return value.padEnd(width, " ");
-}
-function clamp(value, min, max) {
-  if (value < min) {
-    return min;
-  }
-  if (value > max) {
-    return max;
-  }
-  return value;
-}
 function normalizeTokenType(bucket) {
   const value = bucket.tokenType?.trim();
   return value ? value.toUpperCase() : "REQUESTS";
@@ -14435,6 +14457,208 @@ function splitModelVariant(modelId) {
   };
 }
 
+// src/plugin/quota-summary.ts
+var AGY_QUOTA_SUMMARY_TOOL_NAME = "agy_quota_summary";
+function createAgyQuotaSummaryTool({
+  client,
+  getAuthResolver,
+  getConfiguredProjectId,
+  getUserAgentModel
+}) {
+  return tool({
+    description: "Retrieve Agy Code Assist quota summary with weekly and 5-hour limits grouped by model family.",
+    args: {},
+    async execute() {
+      const getAuth = getAuthResolver();
+      if (!getAuth) {
+        return "Agy quota summary is unavailable before Google auth is initialized. Authenticate with the Google provider and retry.";
+      }
+      const auth = await getAuth();
+      if (!isOAuthAuth(auth)) {
+        return "Agy quota summary requires OAuth with Google. Run `opencode auth login` and choose `Google OAuth (Antigravity CLI)` or `Google OAuth (Gemini CLI)`.";
+      }
+      let authRecord = resolveCachedAuth(auth);
+      if (accessTokenExpired(authRecord)) {
+        const refreshed = await refreshAccessToken(authRecord, client);
+        if (!refreshed?.access) {
+          return "Agy quota summary lookup failed because the access token could not be refreshed. Re-authenticate and retry.";
+        }
+        authRecord = refreshed;
+      }
+      if (!authRecord.access) {
+        return "Agy quota summary lookup failed because no access token is available. Re-authenticate and retry.";
+      }
+      try {
+        const projectContext = await ensureProjectContext(
+          authRecord,
+          client,
+          getConfiguredProjectId(),
+          getUserAgentModel()
+        );
+        if (!projectContext.effectiveProjectId) {
+          return "Agy quota summary lookup failed because no Google Cloud project could be resolved.";
+        }
+        const summary = await retrieveUserQuotaSummary(
+          authRecord.access,
+          projectContext.effectiveProjectId,
+          getUserAgentModel()
+        );
+        if (!summary) {
+          return `No Agy quota summary available for project \`${projectContext.effectiveProjectId}\`.`;
+        }
+        return formatAgyQuotaSummaryOutput(
+          projectContext.effectiveProjectId,
+          summary
+        );
+      } catch (error45) {
+        const message = error45 instanceof Error ? error45.message : "unknown error";
+        return `Agy quota summary lookup failed: ${message}`;
+      }
+    }
+  });
+}
+var BAR_WIDTH = 50;
+function windowLabel(window) {
+  switch (window?.toUpperCase()) {
+    case "WEEKLY":
+      return "Weekly Limit";
+    case "FIVE_HOUR":
+      return "Five Hour Limit";
+    default:
+      return "Other Limit";
+  }
+}
+function formatSummaryBucket(bucket, indent) {
+  if (bucket.disabled) {
+    const defaultDesc = `${windowLabel(bucket.window).toLowerCase()} exhausted`;
+    const desc = bucket.description?.trim() || defaultDesc;
+    return [`${indent}Disabled: ${desc}`];
+  }
+  const lines = [];
+  const fraction = bucket.remainingFraction;
+  const hasFraction = typeof fraction === "number" && Number.isFinite(fraction);
+  if (hasFraction) {
+    const clamped = clamp(fraction, 0, 1);
+    const percent = (clamped * 100).toFixed(2);
+    const bar = buildProgressBar(clamped, BAR_WIDTH);
+    lines.push(`${indent}[${bar}] ${percent}%`);
+    const remaining = formatRemainingAmount(bucket.remainingAmount);
+    if (remaining) {
+      const pctWhole = (clamped * 100).toFixed(0);
+      lines.push(`${indent}${pctWhole}% remaining \xB7 ${remaining} left`);
+    } else {
+      const pctWhole = (clamped * 100).toFixed(0);
+      lines.push(`${indent}${pctWhole}% remaining`);
+    }
+  } else {
+    const remaining = formatRemainingAmount(bucket.remainingAmount);
+    lines.push(remaining ? `${indent}${remaining} remaining` : `${indent}unknown remaining`);
+  }
+  const resetLabel = formatRelativeResetTime(bucket.resetTime);
+  if (resetLabel) {
+    const formattedReset = resetLabel.startsWith("resets in ") ? resetLabel.replace("resets in ", "Refreshes in ") : resetLabel.charAt(0).toUpperCase() + resetLabel.slice(1);
+    lines.push(`${indent}${formattedReset}`);
+  }
+  return lines;
+}
+function groupBucketsByWindow(buckets) {
+  const groups = /* @__PURE__ */ new Map();
+  const order = ["WEEKLY", "FIVE_HOUR"];
+  for (const bucket of buckets) {
+    const key = bucket.window?.toUpperCase() || "UNKNOWN";
+    const existing = groups.get(key);
+    if (existing) {
+      existing.push(bucket);
+    } else {
+      groups.set(key, [bucket]);
+    }
+  }
+  const sorted = /* @__PURE__ */ new Map();
+  for (const key of order) {
+    const group = groups.get(key);
+    if (group) {
+      sorted.set(key, group);
+    }
+  }
+  for (const [key, group] of groups) {
+    if (!sorted.has(key)) {
+      sorted.set(key, group);
+    }
+  }
+  return sorted;
+}
+function formatSummaryGroup(group) {
+  const lines = [];
+  const name = group.displayName?.trim();
+  if (name) {
+    lines.push(name);
+  }
+  const desc = group.description?.trim();
+  if (desc) {
+    lines.push(`  Models within this group: ${desc}`);
+  }
+  const buckets = group.buckets;
+  if (buckets?.length) {
+    const windowGroups = groupBucketsByWindow(buckets);
+    let firstWindow = true;
+    for (const [, windowBuckets] of windowGroups) {
+      if (!firstWindow) {
+        lines.push("");
+      }
+      for (const bucket of windowBuckets) {
+        const baseLabel = windowLabel(bucket.window);
+        const label = bucket.displayName ? `${baseLabel} (${bucket.displayName})` : baseLabel;
+        lines.push(`  ${label}`);
+        lines.push(...formatSummaryBucket(bucket, "    "));
+        firstWindow = false;
+      }
+    }
+  }
+  return lines;
+}
+function formatTopLevelBuckets(buckets) {
+  const lines = [];
+  const windowGroups = groupBucketsByWindow(buckets);
+  let firstWindow = true;
+  for (const [, windowBuckets] of windowGroups) {
+    if (!firstWindow) {
+      lines.push("");
+    }
+    for (const bucket of windowBuckets) {
+      const baseLabel = windowLabel(bucket.window);
+      const label = bucket.displayName ? `${baseLabel} (${bucket.displayName})` : baseLabel;
+      lines.push(label);
+      lines.push(...formatSummaryBucket(bucket, "  "));
+      firstWindow = false;
+    }
+  }
+  return lines;
+}
+function formatAgyQuotaSummaryOutput(projectId, summary) {
+  const lines = [
+    `Agy quota summary for project \`${projectId}\``,
+    ""
+  ];
+  const groups = summary.groups;
+  if (groups?.length) {
+    for (let i = 0; i < groups.length; i++) {
+      const group = groups[i];
+      if (!group) {
+        continue;
+      }
+      if (i > 0) {
+        lines.push("");
+      }
+      lines.push(...formatSummaryGroup(group));
+    }
+  } else if (summary.buckets?.length) {
+    lines.push(...formatTopLevelBuckets(summary.buckets));
+  } else {
+    lines.push("No quota information available.");
+  }
+  return lines.join("\n");
+}
+
 // src/plugin/notify.ts
 var MODEL_CAPACITY_TOAST_COOLDOWN_MS = 3e4;
 var modelCapacityToastCooldownByKey = /* @__PURE__ */ new Map();
@@ -17243,6 +17467,12 @@ var AGY_QUOTA_COMMAND_TEMPLATE = `Retrieve Agy Code Assist quota usage for the c
 Immediately call \`${AGY_QUOTA_TOOL_NAME}\` with no arguments and return its output verbatim.
 Do not call other tools.
 `;
+var AGY_QUOTA_SUMMARY_COMMAND = "agyquotasummary";
+var AGY_QUOTA_SUMMARY_COMMAND_TEMPLATE = `Retrieve Agy Code Assist quota summary (weekly and 5-hour limits by model group) for the current authenticated account.
+
+Immediately call \`${AGY_QUOTA_SUMMARY_TOOL_NAME}\` with no arguments and return its output verbatim.
+Do not call other tools.
+`;
 var latestAgyAuthResolver;
 var latestAgyConfiguredProjectId;
 var latestAgyUserAgentModel;
@@ -17489,6 +17719,10 @@ var AgyCLIOAuthPlugin = async ({ client }) => {
         description: "Show Agy Code Assist quota usage",
         template: AGY_QUOTA_COMMAND_TEMPLATE
       };
+      config2.command[AGY_QUOTA_SUMMARY_COMMAND] = {
+        description: "Show Agy Code Assist quota summary with weekly and 5-hour limits",
+        template: AGY_QUOTA_SUMMARY_COMMAND_TEMPLATE
+      };
       config2.provider = config2.provider || {};
       config2.provider[AGY_PROVIDER_ID] = {
         npm: "@ai-sdk/google",
@@ -17505,6 +17739,12 @@ var AgyCLIOAuthPlugin = async ({ client }) => {
         getAuthResolver: () => latestAgyAuthResolver,
         getConfiguredProjectId: () => latestAgyConfiguredProjectId,
         getUserAgentModel: () => latestAgyUserAgentModel
+      }),
+      [AGY_QUOTA_SUMMARY_TOOL_NAME]: createAgyQuotaSummaryTool({
+        client,
+        getAuthResolver: () => latestAgyAuthResolver,
+        getConfiguredProjectId: () => latestAgyConfiguredProjectId,
+        getUserAgentModel: () => latestAgyUserAgentModel
       })
     },
     auth: {