import { describe, it, expect, beforeEach, afterEach, vi } from 'vitest';
import { mkdtempSync, statSync } from 'fs';
import { tmpdir } from 'os';
import { join } from 'path';

describe('auth/anthropic', () => {
  const originalHome = process.env.HOME;
  const originalEnvKey = process.env.ANTHROPIC_API_KEY;
  const originalEnvToken = process.env.ANTHROPIC_AUTH_TOKEN;

  let homeDir: string;

  beforeEach(() => {
    homeDir = mkdtempSync(join(tmpdir(), 'flynn-auth-anthropic-'));
    process.env.HOME = homeDir;
    delete process.env.ANTHROPIC_API_KEY;
    delete process.env.ANTHROPIC_AUTH_TOKEN;
    vi.resetModules();
  });

  afterEach(() => {
    process.env.HOME = originalHome;
    if (originalEnvKey) {
      process.env.ANTHROPIC_API_KEY = originalEnvKey;
    } else {
      delete process.env.ANTHROPIC_API_KEY;
    }

    if (originalEnvToken) {
      process.env.ANTHROPIC_AUTH_TOKEN = originalEnvToken;
    } else {
      delete process.env.ANTHROPIC_AUTH_TOKEN;
    }
  });

  it('stores, loads, and clears Anthropic API key', async () => {
    const mod = await import('./anthropic.js');

    expect(mod.loadStoredAnthropicAuth()).toBeNull();

    mod.storeAnthropicAuth('sk-test');
    expect(mod.loadStoredAnthropicAuth()?.api_key).toBe('sk-test');

    const authFile = join(homeDir, '.config/flynn/auth.json');
    const mode = statSync(authFile).mode & 0o777;
    expect(mode).toBe(0o600);

    mod.clearAnthropicAuth();
    expect(mod.loadStoredAnthropicAuth()).toBeNull();
  });

  it('stores, loads, and clears Anthropic auth token', async () => {
    const mod = await import('./anthropic.js');

    expect(mod.loadStoredAnthropicAuthToken()).toBeNull();

    mod.storeAnthropicAuthToken('tok-test');
    expect(mod.loadStoredAnthropicAuthToken()).toBe('tok-test');

    const authFile = join(homeDir, '.config/flynn/auth.json');
    const mode = statSync(authFile).mode & 0o777;
    expect(mode).toBe(0o600);

    mod.clearAnthropicAuthToken();
    expect(mod.loadStoredAnthropicAuthToken()).toBeNull();
  });

  it('getAnthropicApiKey prefers environment variable', async () => {
    process.env.ANTHROPIC_API_KEY = 'sk-env';
    const mod = await import('./anthropic.js');
    expect(mod.getAnthropicApiKey()).toBe('sk-env');
  });

  it('getAnthropicAuthToken prefers environment variable', async () => {
    process.env.ANTHROPIC_AUTH_TOKEN = 'tok-env';
    const mod = await import('./anthropic.js');
    expect(mod.getAnthropicAuthToken()).toBe('tok-env');
  });
});
