package api

import (
    "bytes"
    "mime/multipart"
    "net/http"
    "net/http/httptest"
    "os"
    "path/filepath"
    "strings"
    "testing"

    "go-whisper-api/config"
)

func TestRegistryResolve(t *testing.T) {
    dir := t.TempDir()
    if err := os.WriteFile(filepath.Join(dir, "ggml-small.bin"), []byte("x"), 0o644); err != nil {
        t.Fatal(err)
    }
    if err := os.WriteFile(filepath.Join(dir, "ggml-large-v3-turbo.bin"), []byte("y"), 0o644); err != nil {
        t.Fatal(err)
    }
    reg := NewRegistry(dir)

    id, err := reg.Resolve("whisper-1", "ggml-large-v3-turbo")
    if err != nil || id != "ggml-large-v3-turbo" {
        t.Fatalf("whisper-1: id=%q err=%v", id, err)
    }
    id, err = reg.Resolve("whisper-large-v3-turbo", "")
    if err != nil || id != "large-v3-turbo" {
        t.Fatalf("whisper-large-v3-turbo: id=%q err=%v", id, err)
    }
    id, err = reg.Resolve("ggml-small", "")
    if err != nil || id != "ggml-small" {
        t.Fatalf("ggml-small: id=%q err=%v", id, err)
    }
    id, err = reg.Resolve("", "ggml-small")
    if err != nil || id != "ggml-small" {
        t.Fatalf("empty with default: id=%q err=%v", id, err)
    }
}

func TestHandleOpenAITranscriptionsMissingFile(t *testing.T) {
    dir := t.TempDir()
    if err := os.WriteFile(filepath.Join(dir, "ggml-small.bin"), []byte("x"), 0o644); err != nil {
        t.Fatal(err)
    }
    srv := &Server{
        cfg:    config.API{ModelsDir: dir, Language: "ru"},
        models: NewRegistry(dir),
    }
    body := &bytes.Buffer{}
    w := multipart.NewWriter(body)
    _ = w.WriteField("model", "ggml-small")
    w.Close()

    req := httptest.NewRequest(http.MethodPost, "/v1/audio/transcriptions", body)
    req.Header.Set("Content-Type", w.FormDataContentType())
    rec := httptest.NewRecorder()
    srv.handleOpenAITranscriptions(rec, req)

    if rec.Code != http.StatusBadRequest {
        t.Fatalf("status=%d body=%s", rec.Code, rec.Body.String())
    }
    if !strings.Contains(rec.Body.String(), "file") {
        t.Fatalf("expected file error, got %s", rec.Body.String())
    }
}

func TestHandleOpenAIModels(t *testing.T) {
    dir := t.TempDir()
    if err := os.WriteFile(filepath.Join(dir, "ggml-small.bin"), []byte("x"), 0o644); err != nil {
        t.Fatal(err)
    }
    srv := &Server{
        cfg:    config.API{ModelsDir: dir, Language: "ru"},
        models: NewRegistry(dir),
    }
    req := httptest.NewRequest(http.MethodGet, "/v1/models", nil)
    rec := httptest.NewRecorder()
    srv.handleOpenAIModels(rec, req)

    if rec.Code != http.StatusOK {
        t.Fatalf("status=%d body=%s", rec.Code, rec.Body.String())
    }
    if !strings.Contains(rec.Body.String(), "ggml-small") {
        t.Fatalf("expected model list, got %s", rec.Body.String())
    }
}