@aliou/pi-synthetic 0.8.4 → 0.9.0

package/src/utils/quotas.ts

@@ -1,17 +1,18 @@
 import type { QuotasResponse } from "../types/quotas";
 
-const API_KEY = process.env.SYNTHETIC_API_KEY;
-
-export async function fetchQuotas(): Promise<QuotasResponse | null> {
-  if (!API_KEY) return null;
+export async function fetchQuotas(
+  apiKey: string,
+): Promise<QuotasResponse | null> {
+  if (!apiKey) return null;
 
   try {
     const response = await fetch("https://api.synthetic.new/v2/quotas", {
-      headers: { Authorization: `Bearer ${API_KEY}` },
+      headers: { Authorization: `Bearer ${apiKey}` },
     });
 
     if (!response.ok) return null;
-    return (await response.json()) as QuotasResponse;
+    const data: QuotasResponse = await response.json();
+    return data;
   } catch {
     return null;
   }

package/src/commands/quotas.ts

@@ -1,68 +0,0 @@
-import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
-import type { Component } from "@mariozechner/pi-tui";
-import { QuotasDisplayComponent } from "../components/quotas-display";
-import { QuotasErrorComponent } from "../components/quotas-error";
-import { QuotasLoadingComponent } from "../components/quotas-loading";
-import { fetchQuotas } from "../utils/quotas";
-
-export function registerQuotasCommand(pi: ExtensionAPI): void {
-  pi.registerCommand("synthetic:quotas", {
-    description: "Display Synthetic API usage quotas",
-    handler: async (_args, ctx) => {
-      const result = await ctx.ui.custom<null>((tui, theme, _kb, done) => {
-        let currentComponent: Component = new QuotasLoadingComponent(theme);
-
-        fetchQuotas()
-          .then((quotas) => {
-            if (!quotas) {
-              currentComponent = new QuotasErrorComponent(
-                theme,
-                "Failed to fetch quotas",
-              );
-            } else {
-              currentComponent = new QuotasDisplayComponent(
-                theme,
-                quotas,
-                () => {
-                  done(null);
-                },
-              );
-            }
-            tui.requestRender();
-          })
-          .catch(() => {
-            currentComponent = new QuotasErrorComponent(
-              theme,
-              "Failed to fetch quotas",
-            );
-            tui.requestRender();
-          });
-
-        return {
-          render: (width: number) => currentComponent.render(width),
-          invalidate: () => currentComponent.invalidate(),
-          handleInput: (data: string) => {
-            if (currentComponent.handleInput) {
-              return currentComponent.handleInput(data);
-            }
-            done(null);
-            return true;
-          },
-        };
-      });
-
-      // RPC fallback: return JSON
-      if (result === undefined) {
-        const quotas = await fetchQuotas();
-        if (!quotas) {
-          ctx.ui.notify(
-            JSON.stringify({ error: "Failed to fetch quotas" }),
-            "error",
-          );
-          return;
-        }
-        ctx.ui.notify(JSON.stringify(quotas, null, 2), "info");
-      }
-    },
-  });
-}

package/src/components/quotas-display.ts

@@ -1,240 +0,0 @@
-import type { Theme } from "@mariozechner/pi-coding-agent";
-import type { Component, TUI } from "@mariozechner/pi-tui";
-import { matchesKey } from "@mariozechner/pi-tui";
-import type { QuotasResponse } from "../types/quotas";
-import { TabbedScrollablePanel } from "./tabbed-panel";
-
-export class QuotasDisplayComponent implements Component {
-  private panel: TabbedScrollablePanel;
-  private onClose: () => void;
-
-  constructor(theme: Theme, quotas: QuotasResponse, onClose: () => void) {
-    this.onClose = onClose;
-
-    this.panel = new TabbedScrollablePanel(
-      {
-        title: "Synthetic API Quotas",
-        tabs: [
-          {
-            label: "Completions",
-            buildContent: () =>
-              buildHybridLayout(theme, quotas.subscription, 5), // 5-hour window
-          },
-          {
-            label: "Search",
-            buildContent: () =>
-              buildHybridLayout(theme, quotas.search.hourly, 1), // 1 hour
-          },
-          {
-            label: "Free tool call",
-            buildContent: () =>
-              buildToolCallsLayout(theme, quotas.freeToolCalls, 24), // 24 hours (daily)
-          },
-        ],
-        onClose: onClose,
-      },
-      null as unknown as TUI,
-      theme,
-    );
-  }
-
-  handleInput(data: string): boolean {
-    if (matchesKey(data, "escape") || data === "q") {
-      this.onClose();
-      return true;
-    }
-    return this.panel.handleInput(data);
-  }
-
-  render(width: number): string[] {
-    return this.panel.render(width);
-  }
-
-  invalidate(): void {
-    this.panel.invalidate();
-  }
-}
-
-// Layout: bar + pct/cur-total + pace/reset
-function buildHybridLayout(
-  theme: Theme,
-  quota: { limit: number; requests: number; renewsAt: string },
-  periodHours: number,
-): string[] {
-  const percentUsed = Math.round((quota.requests / quota.limit) * 100);
-  const renewsAt = new Date(quota.renewsAt);
-  const now = new Date();
-
-  const lines: string[] = [];
-  const barWidth = 50;
-
-  const usedStr = quota.requests.toLocaleString();
-  const limitStr = quota.limit.toLocaleString();
-
-  // Color based on usage
-  const usedColor =
-    percentUsed >= 100 ? "error" : percentUsed > 75 ? "warning" : "success";
-
-  // Calculate pace with known period
-  const totalPeriod = periodHours * 60 * 60 * 1000;
-  const timeUntilReset = Math.max(0, renewsAt.getTime() - now.getTime());
-  const timeElapsed = totalPeriod - timeUntilReset;
-  const percentTimeElapsed = (timeElapsed / totalPeriod) * 100;
-  const paceDiff = percentUsed - percentTimeElapsed;
-
-  let paceStr: string;
-  let paceColor: "success" | "dim" | "error";
-  if (paceDiff < -10) {
-    paceStr = `${Math.round(Math.abs(paceDiff))}% behind pace`;
-    paceColor = "success";
-  } else if (paceDiff > 10) {
-    paceStr = `${Math.round(paceDiff)}% ahead of pace`;
-    paceColor = "error";
-  } else {
-    paceStr = "within pace";
-    paceColor = "dim";
-  }
-
-  // Row above bar: pct left, cur/total right
-  const pctStr = `${percentUsed}% used`;
-  const totalDisplay = `${usedStr}/${limitStr}`;
-  const spacing = " ".repeat(
-    Math.max(1, barWidth - pctStr.length - totalDisplay.length),
-  );
-  lines.push(
-    `  ${theme.fg(usedColor, pctStr)}${spacing}${theme.fg("dim", totalDisplay)}`,
-  );
-
-  // Bar
-  const usedWidth = Math.round((percentUsed / 100) * barWidth);
-  let bar: string;
-  if (usedWidth >= barWidth) {
-    bar = theme.fg("error", "█".repeat(barWidth));
-  } else if (percentUsed > 75) {
-    bar =
-      theme.fg("warning", "█".repeat(usedWidth)) +
-      theme.fg("dim", "█".repeat(barWidth - usedWidth));
-  } else {
-    bar =
-      theme.fg("success", "█".repeat(usedWidth)) +
-      theme.fg("dim", "█".repeat(barWidth - usedWidth));
-  }
-  lines.push(`  ${bar}`);
-
-  // Row below bar: pace left, reset right
-  const resetStr = formatShortTime(renewsAt);
-  const paceSpacing = " ".repeat(
-    Math.max(1, barWidth - paceStr.length - resetStr.length),
-  );
-  lines.push(
-    `  ${theme.fg(paceColor, paceStr)}${paceSpacing}${theme.fg("dim", resetStr)}`,
-  );
-
-  return lines;
-}
-
-// Layout for free tool calls - shows remaining, not usage
-function buildToolCallsLayout(
-  theme: Theme,
-  quota: { limit: number; requests: number; renewsAt: string },
-  periodHours: number,
-): string[] {
-  const remaining = quota.limit - quota.requests;
-  const percentRemaining = Math.round((remaining / quota.limit) * 100);
-  const renewsAt = new Date(quota.renewsAt);
-  const now = new Date();
-
-  const lines: string[] = [];
-  const barWidth = 50;
-
-  const remainingStr = remaining.toLocaleString();
-
-  // Color based on remaining (inverse of usage)
-  const remainingColor =
-    remaining <= 0 ? "error" : percentRemaining < 25 ? "warning" : "success";
-
-  // Calculate pace (how fast you're consuming free calls)
-  const totalPeriod = periodHours * 60 * 60 * 1000;
-  const timeUntilReset = Math.max(0, renewsAt.getTime() - now.getTime());
-  const timeElapsed = totalPeriod - timeUntilReset;
-  const percentTimeElapsed = (timeElapsed / totalPeriod) * 100;
-  const expectedRemaining = Math.round(
-    quota.limit * (1 - percentTimeElapsed / 100),
-  );
-  const remainingDiff = remaining - expectedRemaining;
-
-  let paceStr: string;
-  let paceColor: "success" | "dim" | "error";
-  if (remainingDiff > quota.limit * 0.1) {
-    paceStr = `${remainingDiff.toLocaleString()} more than expected`;
-    paceColor = "success";
-  } else if (remainingDiff < -quota.limit * 0.1) {
-    paceStr = `${Math.abs(remainingDiff).toLocaleString()} fewer than expected`;
-    paceColor = "error";
-  } else {
-    paceStr = "on track";
-    paceColor = "dim";
-  }
-
-  // Row above bar: pct remaining left, ratio right (like other tabs)
-  const pctStr = `${percentRemaining}% remaining`;
-  const ratioStr = `${remainingStr}/${quota.limit.toLocaleString()}`;
-  const spacing = " ".repeat(
-    Math.max(1, barWidth - pctStr.length - ratioStr.length),
-  );
-  lines.push(
-    `  ${theme.fg(remainingColor, pctStr)}${spacing}${theme.fg("dim", ratioStr)}`,
-  );
-
-  // Bar (shows remaining, not used - so full bar = all remaining)
-  const remainingWidth = Math.round((percentRemaining / 100) * barWidth);
-  let bar: string;
-  if (remaining <= 0) {
-    bar = theme.fg("dim", "█".repeat(barWidth));
-  } else if (percentRemaining < 25) {
-    bar =
-      theme.fg("warning", "█".repeat(remainingWidth)) +
-      theme.fg("dim", "█".repeat(barWidth - remainingWidth));
-  } else {
-    bar =
-      theme.fg("success", "█".repeat(remainingWidth)) +
-      theme.fg("dim", "█".repeat(barWidth - remainingWidth));
-  }
-  lines.push(`  ${bar}`);
-
-  // Row below bar: pace left, reset right
-  const resetStr = formatShortTime(renewsAt);
-  const paceSpacing = " ".repeat(
-    Math.max(1, barWidth - paceStr.length - resetStr.length),
-  );
-  lines.push(
-    `  ${theme.fg(paceColor, paceStr)}${paceSpacing}${theme.fg("dim", resetStr)}`,
-  );
-
-  // If depleted, show note about paid calls
-  if (remaining <= 0) {
-    lines.push(`  ${theme.fg("dim", "Additional calls will be charged")}`);
-  }
-
-  return lines;
-}
-
-function formatShortTime(date: Date): string {
-  const now = new Date();
-  const diffMs = date.getTime() - now.getTime();
-
-  if (diffMs <= 0) {
-    return "soon";
-  }
-
-  const diffHours = Math.ceil(diffMs / (1000 * 60 * 60));
-  const diffDays = Math.ceil(diffMs / (1000 * 60 * 60 * 24));
-
-  if (diffHours < 24) {
-    return `in ${diffHours}h`;
-  } else if (diffDays < 7) {
-    return `in ${diffDays}d`;
-  } else {
-    return date.toLocaleDateString("en-US", { month: "short", day: "numeric" });
-  }
-}

package/src/components/quotas-error.ts

@@ -1,32 +0,0 @@
-import type { Theme } from "@mariozechner/pi-coding-agent";
-import { DynamicBorder } from "@mariozechner/pi-coding-agent";
-import { type Component, Container, Text } from "@mariozechner/pi-tui";
-
-export class QuotasErrorComponent implements Component {
-  private container: Container;
-
-  constructor(theme: Theme, message: string) {
-    this.container = new Container();
-    const border = new DynamicBorder((s: string) => theme.fg("accent", s));
-
-    this.container.addChild(border);
-    this.container.addChild(
-      new Text(theme.fg("accent", theme.bold(" Synthetic API Quotas ")), 1, 0),
-    );
-    this.container.addChild(new Text("", 0, 0));
-    this.container.addChild(new Text(theme.fg("error", `  ${message}`), 1, 0));
-    this.container.addChild(new Text("", 0, 0));
-    this.container.addChild(
-      new Text(theme.fg("dim", "  Press any key to close"), 1, 0),
-    );
-    this.container.addChild(border);
-  }
-
-  render(width: number): string[] {
-    return this.container.render(width);
-  }
-
-  invalidate(): void {
-    this.container.invalidate();
-  }
-}

package/src/components/quotas-loading.ts

@@ -1,30 +0,0 @@
-import type { Theme } from "@mariozechner/pi-coding-agent";
-import { DynamicBorder } from "@mariozechner/pi-coding-agent";
-import { type Component, Container, Text } from "@mariozechner/pi-tui";
-
-export class QuotasLoadingComponent implements Component {
-  private container: Container;
-
-  constructor(theme: Theme) {
-    this.container = new Container();
-    const border = new DynamicBorder((s: string) => theme.fg("accent", s));
-
-    this.container.addChild(border);
-    this.container.addChild(
-      new Text(theme.fg("accent", theme.bold(" Synthetic API Quotas ")), 1, 0),
-    );
-    this.container.addChild(new Text("", 0, 0));
-    this.container.addChild(
-      new Text(theme.fg("dim", "  Loading quotas..."), 1, 0),
-    );
-    this.container.addChild(border);
-  }
-
-  render(width: number): string[] {
-    return this.container.render(width);
-  }
-
-  invalidate(): void {
-    this.container.invalidate();
-  }
-}

package/src/components/tabbed-panel.ts

@@ -1,161 +0,0 @@
-import type { Theme } from "@mariozechner/pi-coding-agent";
-import { DynamicBorder } from "@mariozechner/pi-coding-agent";
-import type { Component, TUI } from "@mariozechner/pi-tui";
-import { matchesKey, truncateToWidth } from "@mariozechner/pi-tui";
-
-export interface PanelTab {
-  label: string;
-  buildContent: () => string[];
-}
-
-export interface TabbedScrollablePanelOptions {
-  title: string;
-  tabs: PanelTab[];
-  onClose: () => void;
-  maxVisible?: number;
-  keymap?: "vim" | "default";
-}
-
-export class TabbedScrollablePanel implements Component {
-  private activeTab = 0;
-  private scrollOffset = 0;
-  private cachedLines: string[] | null = null;
-  private cachedWidth = 0;
-  private border: DynamicBorder;
-  private options: TabbedScrollablePanelOptions;
-  private theme: Theme;
-
-  constructor(options: TabbedScrollablePanelOptions, _tui: TUI, theme: Theme) {
-    this.options = options;
-    this.theme = theme;
-    this.border = new DynamicBorder((segment) => theme.fg("border", segment));
-  }
-
-  handleInput(data: string): boolean {
-    if (matchesKey(data, "escape") || data === "q") {
-      this.options.onClose();
-      return true;
-    }
-
-    if (matchesKey(data, "tab")) {
-      this.activeTab = (this.activeTab + 1) % this.options.tabs.length;
-      this.scrollOffset = 0;
-      this.invalidate();
-      return true;
-    }
-
-    if (matchesKey(data, "shift+tab")) {
-      this.activeTab =
-        (this.activeTab - 1 + this.options.tabs.length) %
-        this.options.tabs.length;
-      this.scrollOffset = 0;
-      this.invalidate();
-      return true;
-    }
-
-    const maxVisible = this.options.maxVisible ?? 16;
-    const totalLines = this.cachedLines?.length ?? 0;
-    const maxScroll = Math.max(0, totalLines - maxVisible);
-
-    if (data === "j" || matchesKey(data, "down")) {
-      if (this.scrollOffset < maxScroll) {
-        this.scrollOffset++;
-      }
-      return true;
-    }
-
-    if (data === "k" || matchesKey(data, "up")) {
-      if (this.scrollOffset > 0) {
-        this.scrollOffset--;
-      }
-      return true;
-    }
-
-    if (data === " " || matchesKey(data, "pageDown")) {
-      this.scrollOffset = Math.min(this.scrollOffset + maxVisible, maxScroll);
-      return true;
-    }
-
-    if (matchesKey(data, "pageUp")) {
-      this.scrollOffset = Math.max(0, this.scrollOffset - maxVisible);
-      return true;
-    }
-
-    if (matchesKey(data, "home")) {
-      this.scrollOffset = 0;
-      return true;
-    }
-
-    if (matchesKey(data, "end")) {
-      this.scrollOffset = maxScroll;
-      return true;
-    }
-
-    return false;
-  }
-
-  invalidate(): void {
-    this.cachedLines = null;
-    this.cachedWidth = 0;
-  }
-
-  render(width: number): string[] {
-    const tab = this.options.tabs[this.activeTab];
-
-    if (!this.cachedLines || this.cachedWidth !== width) {
-      this.cachedLines = tab ? tab.buildContent() : [];
-      this.cachedWidth = width;
-    }
-
-    const lines: string[] = [];
-
-    lines.push(...this.border.render(width));
-    lines.push(
-      truncateToWidth(
-        ` ${this.theme.fg("accent", this.theme.bold(this.options.title))}`,
-        width,
-      ),
-    );
-    lines.push(this.renderTabBar(width));
-    lines.push("");
-
-    // Content - no forced padding, just render what we have
-    for (const line of this.cachedLines) {
-      lines.push(truncateToWidth(`  ${line}`, width));
-    }
-
-    // Footer directly after content
-    lines.push("");
-    lines.push(
-      truncateToWidth(
-        this.theme.fg("dim", "  Tab/S-Tab switch tabs  q/Esc close"),
-        width,
-      ),
-    );
-    lines.push(...this.border.render(width));
-
-    return lines;
-  }
-
-  private renderTabBar(width: number): string {
-    const parts: string[] = [];
-
-    for (let i = 0; i < this.options.tabs.length; i++) {
-      const tab = this.options.tabs[i];
-      if (!tab) continue;
-      const active = i === this.activeTab;
-
-      if (active) {
-        parts.push(this.theme.fg("accent", this.theme.bold(` ${tab.label} `)));
-      } else {
-        parts.push(this.theme.fg("dim", ` ${tab.label} `));
-      }
-
-      if (i < this.options.tabs.length - 1) {
-        parts.push(this.theme.fg("borderMuted", "│"));
-      }
-    }
-
-    return truncateToWidth(`  ${parts.join("")}`, width);
-  }
-}

package/src/hooks/search-tool-availability.ts

@@ -1,104 +0,0 @@
-import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
-import { SYNTHETIC_WEB_SEARCH_TOOL } from "../tools/search";
-
-async function checkSubscriptionAccess(
-  apiKey: string,
-): Promise<{ ok: true } | { ok: false; reason: string }> {
-  try {
-    const response = await fetch("https://api.synthetic.new/v2/quotas", {
-      method: "GET",
-      headers: {
-        Authorization: `Bearer ${apiKey}`,
-      },
-    });
-
-    if (!response.ok) {
-      return {
-        ok: false,
-        reason: `Quotas check failed (HTTP ${response.status})`,
-      };
-    }
-
-    const data = await response.json();
-    if (data?.subscription?.limit > 0) {
-      return { ok: true };
-    }
-
-    return {
-      ok: false,
-      reason: "No active subscription (search requires a subscription plan)",
-    };
-  } catch (error) {
-    const message =
-      error instanceof Error ? error.message : "Unknown error occurred";
-    return { ok: false, reason: `Quotas check failed: ${message}` };
-  }
-}
-
-export function registerSyntheticWebSearchHooks(pi: ExtensionAPI): void {
-  let accessCheckPromise:
-    | Promise<{ ok: true } | { ok: false; reason: string }>
-    | undefined;
-  let hasAccess = false;
-  let deniedReason: string | undefined;
-  let didNotifyDenied = false;
-
-  // Keep tool inactive at session start. Availability is decided before each agent run.
-  pi.on("session_start", () => {
-    const current = pi.getActiveTools();
-    if (current.includes(SYNTHETIC_WEB_SEARCH_TOOL)) {
-      pi.setActiveTools(
-        current.filter((toolName) => toolName !== SYNTHETIC_WEB_SEARCH_TOOL),
-      );
-    }
-  });
-
-  // Verify subscription only when user starts agent execution.
-  pi.on("before_agent_start", async (_event, ctx) => {
-    const apiKey = process.env.SYNTHETIC_API_KEY;
-    if (!apiKey) {
-      hasAccess = false;
-      deniedReason = "SYNTHETIC_API_KEY is not configured";
-      accessCheckPromise = undefined;
-    } else {
-      if (deniedReason === "SYNTHETIC_API_KEY is not configured") {
-        deniedReason = undefined;
-      }
-
-      if (!hasAccess && !deniedReason) {
-        accessCheckPromise ??= checkSubscriptionAccess(apiKey);
-        const access = await accessCheckPromise;
-
-        if (!access.ok) {
-          deniedReason = access.reason;
-        } else {
-          hasAccess = true;
-          didNotifyDenied = false;
-        }
-      }
-    }
-
-    if (deniedReason) {
-      const current = pi.getActiveTools();
-      if (current.includes(SYNTHETIC_WEB_SEARCH_TOOL)) {
-        pi.setActiveTools(
-          current.filter((toolName) => toolName !== SYNTHETIC_WEB_SEARCH_TOOL),
-        );
-      }
-
-      if (ctx.hasUI && !didNotifyDenied) {
-        ctx.ui.notify(
-          `Synthetic web search disabled: ${deniedReason}`,
-          "warning",
-        );
-        didNotifyDenied = true;
-      }
-      return;
-    }
-
-    const current = pi.getActiveTools();
-    if (!current.includes(SYNTHETIC_WEB_SEARCH_TOOL)) {
-      pi.setActiveTools([...current, SYNTHETIC_WEB_SEARCH_TOOL]);
-    }
-  });
-}

package/src/index.ts

@@ -1,17 +0,0 @@
-import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
-import { registerQuotasCommand } from "./commands/quotas";
-import { registerSyntheticWebSearchHooks } from "./hooks/search-tool-availability";
-import { registerSubIntegration } from "./hooks/sub-integration";
-import { registerSyntheticProvider } from "./providers/index";
-import { registerSyntheticWebSearchTool } from "./tools/search";
-
-export default async function (pi: ExtensionAPI) {
-  registerSyntheticProvider(pi);
-  registerSyntheticWebSearchTool(pi);
-  registerSyntheticWebSearchHooks(pi);
-
-  if (process.env.SYNTHETIC_API_KEY) {
-    registerQuotasCommand(pi);
-    registerSubIntegration(pi);
-  }
-}