import { describe, it, expect, vi, afterEach } from 'vitest';
import { openAICompatibleProvider } from '../src/index.js';

const opts = { baseUrl: 'https://api.example.com/v1', apiKey: 'k', model: 'm' };

function mockFetchJson(content: unknown) {
  return vi.fn().mockResolvedValue({
    ok: true,
    json: async () => ({ choices: [{ message: { content: JSON.stringify(content) } }] }),
  });
}

afterEach(() => {
  vi.unstubAllGlobals();
});

describe('openAICompatibleProvider', () => {
  it('isConfigured() reflects whether baseUrl/apiKey/model are present', () => {
    expect(openAICompatibleProvider(opts).isConfigured()).toBe(true);
    expect(openAICompatibleProvider({ ...opts, apiKey: '' }).isConfigured()).toBe(false);
  });

  it('anonymize() parses {texte_anonymise, mapping} and posts temperature 0 + json_object', async () => {
    const fetchMock = mockFetchJson({ texte_anonymise: '[EMAIL_1]', mapping: { '[EMAIL_1]': 'a@b.ch' } });
    vi.stubGlobal('fetch', fetchMock);

    const r = await openAICompatibleProvider(opts).anonymize('a@b.ch');
    expect(r).toEqual({ anon: '[EMAIL_1]', mapping: { '[EMAIL_1]': 'a@b.ch' } });

    const [url, init] = fetchMock.mock.calls[0];
    expect(url).toBe('https://api.example.com/v1/chat/completions');
    const body = JSON.parse(init.body);
    expect(body.temperature).toBe(0);
    expect(body.response_format).toEqual({ type: 'json_object' });
    expect(init.headers.Authorization).toBe('Bearer k');
  });

  it('anonymizeBatch() returns {segments, mapping} and includes used ids in the prompt', async () => {
    const fetchMock = mockFetchJson({ segments: ['[PER_2.NOM:M]'], mapping: { '[PER_2.NOM:M]': 'Bob' } });
    vi.stubGlobal('fetch', fetchMock);

    const r = await openAICompatibleProvider(opts).anonymizeBatch(['Bob'], ['[PER_1.NOM:M]']);
    expect(r.segments).toEqual(['[PER_2.NOM:M]']);

    const system = JSON.parse(fetchMock.mock.calls[0][1].body).messages[0].content;
    expect(system).toContain('[PER_1.NOM:M]');
  });

  it('throws on a non-OK HTTP status', async () => {
    vi.stubGlobal('fetch', vi.fn().mockResolvedValue({ ok: false, status: 500 }));
    await expect(openAICompatibleProvider(opts).anonymize('x')).rejects.toThrow('LLM_HTTP_500');
  });

  it('throws on a malformed response shape', async () => {
    vi.stubGlobal('fetch', mockFetchJson({ wrong: true }));
    await expect(openAICompatibleProvider(opts).anonymize('x')).rejects.toThrow('LLM_BAD_SHAPE');
  });
});