import fs from "node:fs";
import os from "node:os";
import path from "node:path";
import { afterAll, beforeAll, describe, expect, it, vi } from "vitest";

vi.mock("./cli-credentials.js", () => ({
  readClaudeCliCredentialsCached: () => null,
  readCodexCliCredentialsCached: () => null,
  readMiniMaxCliCredentialsCached: () => null,
}));

vi.mock("../plugins/provider-runtime.js", () => ({
  resolveExternalAuthProfilesWithPlugins: () => [],
}));

import {
  clearRuntimeAuthProfileStoreSnapshots,
  ensureAuthProfileStore,
} from "./auth-profiles/store.js";
import { calculateAuthProfileCooldownMs, markAuthProfileFailure } from "./auth-profiles/usage.js";

type AuthProfileStore = ReturnType<typeof ensureAuthProfileStore>;

let tempRoot = "";
let tempCaseIndex = 0;

beforeAll(() => {
  tempRoot = fs.mkdtempSync(path.join(os.tmpdir(), "openclaw-auth-"));
});

afterAll(() => {
  clearRuntimeAuthProfileStoreSnapshots();
  fs.rmSync(tempRoot, { recursive: true, force: true });
});

function makeAgentDir(label = "case") {
  tempCaseIndex += 1;
  const agentDir = path.join(tempRoot, `${tempCaseIndex}-${label}`);
  fs.mkdirSync(agentDir, { recursive: true });
  return agentDir;
}

async function withAuthProfileStore(
  fn: (ctx: { agentDir: string; store: AuthProfileStore }) => Promise<void>,
): Promise<void> {
  const agentDir = makeAgentDir("store");
  const authPath = path.join(agentDir, "auth-profiles.json");
  fs.writeFileSync(
    authPath,
    JSON.stringify({
      version: 1,
      profiles: {
        "anthropic:default": {
          type: "api_key",
          provider: "anthropic",
          key: "sk-default",
        },
        "openrouter:default": {
          type: "api_key",
          provider: "openrouter",
          key: "sk-or-default",
        },
      },
    }),
  );

  const store = ensureAuthProfileStore(agentDir);
  await fn({ agentDir, store });
}

function expectCooldownInRange(remainingMs: number, minMs: number, maxMs: number): void {
  expect(remainingMs).toBeGreaterThan(minMs);
  expect(remainingMs).toBeLessThan(maxMs);
}

describe("markAuthProfileFailure", () => {
  it("does not overwrite fresher on-disk credentials with a stale runtime snapshot", async () => {
    const agentDir = makeAgentDir("stale-snapshot");
    const authPath = path.join(agentDir, "auth-profiles.json");
    fs.writeFileSync(
      authPath,
      JSON.stringify({
        version: 1,
        profiles: {
          "openai:default": {
            type: "api_key",
            provider: "openai",
            key: "sk-expired-old",
          },
        },
      }),
    );

    const staleRuntimeStore: AuthProfileStore = {
      version: 1,
      profiles: {
        "openai:default": {
          type: "api_key",
          provider: "openai",
          key: "sk-expired-old",
        },
      },
    };

    fs.writeFileSync(
      authPath,
      JSON.stringify({
        version: 1,
        profiles: {
          "openai:default": {
            type: "api_key",
            provider: "openai",
            key: "sk-fresh-new",
          },
        },
      }),
    );

    const staleCredential = staleRuntimeStore.profiles["openai:default"];
    expect(staleCredential?.type).toBe("api_key");
    expect(staleCredential && "key" in staleCredential ? staleCredential.key : undefined).toBe(
      "sk-expired-old",
    );

    await markAuthProfileFailure({
      store: staleRuntimeStore,
      profileId: "openai:default",
      reason: "rate_limit",
      agentDir,
    });

    clearRuntimeAuthProfileStoreSnapshots();
    const reloaded = ensureAuthProfileStore(agentDir);
    const reloadedCredential = reloaded.profiles["openai:default"];
    expect(reloadedCredential?.type).toBe("api_key");
    expect(
      reloadedCredential && "key" in reloadedCredential ? reloadedCredential.key : undefined,
    ).toBe("sk-fresh-new");
    expect(typeof reloaded.usageStats?.["openai:default"]?.cooldownUntil).toBe("number");
  });

  it("disables billing failures for ~5 hours by default", async () => {
    await withAuthProfileStore(async ({ agentDir, store }) => {
      const startedAt = Date.now();
      await markAuthProfileFailure({
        store,
        profileId: "anthropic:default",
        reason: "billing",
        agentDir,
      });

      const disabledUntil = store.usageStats?.["anthropic:default"]?.disabledUntil;
      expect(typeof disabledUntil).toBe("number");
      const remainingMs = (disabledUntil as number) - startedAt;
      expectCooldownInRange(remainingMs, 4.5 * 60 * 60 * 1000, 5.5 * 60 * 60 * 1000);
    });
  });
  it("honors per-provider billing backoff overrides", async () => {
    await withAuthProfileStore(async ({ agentDir, store }) => {
      const startedAt = Date.now();
      await markAuthProfileFailure({
        store,
        profileId: "anthropic:default",
        reason: "billing",
        agentDir,
        cfg: {
          auth: {
            cooldowns: {
              billingBackoffHoursByProvider: { Anthropic: 1 },
              billingMaxHours: 2,
            },
          },
        } as never,
      });

      const disabledUntil = store.usageStats?.["anthropic:default"]?.disabledUntil;
      expect(typeof disabledUntil).toBe("number");
      const remainingMs = (disabledUntil as number) - startedAt;
      expectCooldownInRange(remainingMs, 0.8 * 60 * 60 * 1000, 1.2 * 60 * 60 * 1000);
    });
  });
  it("keeps persisted cooldownUntil unchanged across mid-window retries", async () => {
    await withAuthProfileStore(async ({ agentDir, store }) => {
      await markAuthProfileFailure({
        store,
        profileId: "anthropic:default",
        reason: "rate_limit",
        agentDir,
      });

      const firstCooldownUntil = store.usageStats?.["anthropic:default"]?.cooldownUntil;
      expect(typeof firstCooldownUntil).toBe("number");

      await markAuthProfileFailure({
        store,
        profileId: "anthropic:default",
        reason: "rate_limit",
        agentDir,
      });

      const secondCooldownUntil = store.usageStats?.["anthropic:default"]?.cooldownUntil;
      expect(secondCooldownUntil).toBe(firstCooldownUntil);

      const reloaded = ensureAuthProfileStore(agentDir);
      expect(reloaded.usageStats?.["anthropic:default"]?.cooldownUntil).toBe(firstCooldownUntil);
    });
  });
  it("records overloaded failures in the cooldown bucket", async () => {
    await withAuthProfileStore(async ({ agentDir, store }) => {
      await markAuthProfileFailure({
        store,
        profileId: "anthropic:default",
        reason: "overloaded",
        agentDir,
      });

      const stats = store.usageStats?.["anthropic:default"];
      expect(typeof stats?.cooldownUntil).toBe("number");
      expect(stats?.disabledUntil).toBeUndefined();
      expect(stats?.disabledReason).toBeUndefined();
      expect(stats?.failureCounts?.overloaded).toBe(1);
    });
  });
  it("disables auth_permanent failures for ~10 minutes by default", async () => {
    await withAuthProfileStore(async ({ agentDir, store }) => {
      const startedAt = Date.now();
      await markAuthProfileFailure({
        store,
        profileId: "anthropic:default",
        reason: "auth_permanent",
        agentDir,
      });

      const stats = store.usageStats?.["anthropic:default"];
      expect(typeof stats?.disabledUntil).toBe("number");
      expect(stats?.disabledReason).toBe("auth_permanent");
      // Should NOT set cooldownUntil (that's for transient errors)
      expect(stats?.cooldownUntil).toBeUndefined();
      const remainingMs = (stats?.disabledUntil as number) - startedAt;
      expectCooldownInRange(remainingMs, 9 * 60 * 1000, 11 * 60 * 1000);
    });
  });

  it("honors auth_permanent backoff overrides", async () => {
    await withAuthProfileStore(async ({ agentDir, store }) => {
      const startedAt = Date.now();
      await markAuthProfileFailure({
        store,
        profileId: "anthropic:default",
        reason: "auth_permanent",
        agentDir,
        cfg: {
          auth: {
            cooldowns: {
              authPermanentBackoffMinutes: 15,
              authPermanentMaxMinutes: 45,
            },
          },
        } as never,
      });

      const disabledUntil = store.usageStats?.["anthropic:default"]?.disabledUntil;
      expect(typeof disabledUntil).toBe("number");
      const remainingMs = (disabledUntil as number) - startedAt;
      expectCooldownInRange(remainingMs, 14 * 60 * 1000, 16 * 60 * 1000);
    });
  });
  it("resets backoff counters outside the failure window", async () => {
    const agentDir = makeAgentDir("reset-window");
    const authPath = path.join(agentDir, "auth-profiles.json");
    const now = Date.now();
    fs.writeFileSync(
      authPath,
      JSON.stringify({
        version: 1,
        profiles: {
          "anthropic:default": {
            type: "api_key",
            provider: "anthropic",
            key: "sk-default",
          },
        },
        usageStats: {
          "anthropic:default": {
            errorCount: 9,
            failureCounts: { billing: 3 },
            lastFailureAt: now - 48 * 60 * 60 * 1000,
          },
        },
      }),
    );

    const store = ensureAuthProfileStore(agentDir);
    await markAuthProfileFailure({
      store,
      profileId: "anthropic:default",
      reason: "billing",
      agentDir,
      cfg: {
        auth: { cooldowns: { failureWindowHours: 24 } },
      } as never,
    });

    expect(store.usageStats?.["anthropic:default"]?.errorCount).toBe(1);
    expect(store.usageStats?.["anthropic:default"]?.failureCounts?.billing).toBe(1);
  });

  it("resets error count when previous cooldown has expired to prevent escalation", async () => {
    const agentDir = makeAgentDir("expired-cooldown");
    const authPath = path.join(agentDir, "auth-profiles.json");
    const now = Date.now();
    // Simulate state left on disk after 3 rapid failures within a 1-min cooldown
    // window. The cooldown has since expired, but clearExpiredCooldowns() only
    // ran in-memory and never persisted - so disk still carries errorCount: 3.
    fs.writeFileSync(
      authPath,
      JSON.stringify({
        version: 1,
        profiles: {
          "anthropic:default": {
            type: "api_key",
            provider: "anthropic",
            key: "sk-default",
          },
        },
        usageStats: {
          "anthropic:default": {
            errorCount: 3,
            failureCounts: { rate_limit: 3 },
            lastFailureAt: now - 120_000, // 2 minutes ago
            cooldownUntil: now - 60_000, // expired 1 minute ago
          },
        },
      }),
    );

    const store = ensureAuthProfileStore(agentDir);
    await markAuthProfileFailure({
      store,
      profileId: "anthropic:default",
      reason: "rate_limit",
      agentDir,
    });

    const stats = store.usageStats?.["anthropic:default"];
    // Error count should reset to 1 (not escalate to 4) because the
    // previous cooldown expired. Cooldown should be ~30s, not ~5 min.
    expect(stats?.errorCount).toBe(1);
    expect(stats?.failureCounts?.rate_limit).toBe(1);
    const cooldownMs = (stats?.cooldownUntil ?? 0) - now;
    // calculateAuthProfileCooldownMs(1) = 30_000 (stepped: 30s -> 1m -> 5m)
    expectCooldownInRange(cooldownMs, 25_000, 35_000);
  });

  it("does not persist cooldown windows for OpenRouter profiles", async () => {
    await withAuthProfileStore(async ({ agentDir, store }) => {
      await markAuthProfileFailure({
        store,
        profileId: "openrouter:default",
        reason: "rate_limit",
        agentDir,
      });

      await markAuthProfileFailure({
        store,
        profileId: "openrouter:default",
        reason: "billing",
        agentDir,
      });

      expect(store.usageStats?.["openrouter:default"]).toBeUndefined();

      const reloaded = ensureAuthProfileStore(agentDir);
      expect(reloaded.usageStats?.["openrouter:default"]).toBeUndefined();
    });
  });
});

describe("calculateAuthProfileCooldownMs", () => {
  it("applies stepped backoff with a 5-min cap", () => {
    expect(calculateAuthProfileCooldownMs(1)).toBe(30_000); // 30 seconds
    expect(calculateAuthProfileCooldownMs(2)).toBe(60_000); // 1 minute
    expect(calculateAuthProfileCooldownMs(3)).toBe(5 * 60_000); // 5 minutes
    expect(calculateAuthProfileCooldownMs(4)).toBe(5 * 60_000); // 5 minutes (cap)
    expect(calculateAuthProfileCooldownMs(5)).toBe(5 * 60_000); // 5 minutes (cap)
  });
});
