import Keystore from "@keystore/sdk";
import OpenAI from "openai";
import Anthropic from "@anthropic-ai/sdk";

const ks = new Keystore({ agentToken: process.env.KS_TOKEN! });
ks.interceptAll();

const openai = new OpenAI();
const claude = new Anthropic();

async function callWithRetry<T>(
  fn: () => Promise<T>,
  maxRetries = 3
): Promise<T> {
  for (let attempt = 0; attempt <= maxRetries; attempt++) {
    try {
      return await fn();
    } catch (err: any) {
      // Budget exceeded — stop immediately, no point retrying
      if (err.status === 402) {
        console.error("Budget exceeded. Stopping agent.");
        throw new Error("BUDGET_EXCEEDED");
      }

      // Rate limited — wait and retry
      if (err.status === 429) {
        const retryAfter = parseInt(err.headers?.["retry-after"] || "5", 10);
        console.warn(`Rate limited. Retrying in ${retryAfter}s...`);

        if (attempt < maxRetries) {
          await new Promise((r) => setTimeout(r, retryAfter * 1000));
          continue;
        }
      }

      // Other errors — retry with exponential backoff
      if (attempt < maxRetries) {
        const delay = Math.pow(2, attempt) * 1000;
        console.warn(`Error (attempt ${attempt + 1}). Retrying in ${delay}ms...`);
        await new Promise((r) => setTimeout(r, delay));
        continue;
      }

      throw err;
    }
  }
  throw new Error("Max retries exceeded");
}

async function runAgent() {
  try {
    const result = await callWithRetry(() =>
      openai.chat.completions.create({
        model: "gpt-4o",
        messages: [{ role: "user", content: "Analyze today's metrics." }],
      })
    );

    console.log(result.choices[0].message.content);
  } catch (err: any) {
    if (err.message === "BUDGET_EXCEEDED") {
      // Notify ops team, pause agent, etc.
      console.error("Agent paused: budget exceeded");
      return;
    }
    throw err;
  }
}

import { createHmac } from "crypto";

// Express / Hono / any HTTP framework
app.post("/api/keystore-webhooks", async (req, res) => {
  // Verify the webhook signature
  const signature = req.headers["x-keystore-signature"] as string;
  const payload = JSON.stringify(req.body);
  const expected = createHmac("sha256", process.env.WEBHOOK_SECRET!)
    .update(payload)
    .digest("hex");

  if (signature !== expected) {
    return res.status(401).json({ error: "Invalid signature" });
  }

  const event = req.body;

  switch (event.type) {
    case "budget.warning":
      console.warn(
        `Agent ${event.agentId} at ${event.percentUsed}% of budget`
      );
      // Send Slack notification, etc.
      break;

    case "budget.exceeded":
      console.error(`Agent ${event.agentId} exceeded budget`);
      // Trigger incident, pause agent, notify team
      break;

    case "agent.rate_limited":
      console.warn(`Agent ${event.agentId} rate limited`);
      break;
  }

  res.json({ received: true });
});