package ollama

import (
	"net/http"
	"net/http/httptest"
	"testing"
	"strings"
	"github.com/go-chi/chi/v5"
)

type ollamaTestSurface struct {
	Store       ConfigReader
	handler     *Handler
}

func registerOllamaTestRoutes(r chi.Router, h *ollamaTestSurface) {
	r.Get("/api/version", h.handler().GetVersion)
	r.Get("/api/tags", h.handler().ListOllamaModels)
	r.Post("/api/show", h.handler().GetOllamaModel)
}


func TestGetOllamaVersionRoute(t *testing.T) {
	h := &ollamaTestSurface{}
	r := chi.NewRouter()
	registerOllamaTestRoutes(r, h)
    req := httptest.NewRequest(http.MethodGet, "/api/version", nil)    
	rec := httptest.NewRecorder()
	r.ServeHTTP(rec, req)
	if rec.Code != http.StatusOK {
		t.Fatalf("expected 200, got %d body=%s", rec.Code, rec.Body.String())
	}
}


func TestGetOllamaModelsRoute(t *testing.T) {
	h := &ollamaTestSurface{}
	r := chi.NewRouter()
	registerOllamaTestRoutes(r, h)
    req := httptest.NewRequest(http.MethodGet, "/api/tags", nil)    
	rec := httptest.NewRecorder()
	r.ServeHTTP(rec, req)
	if rec.Code != http.StatusOK {
		t.Fatalf("expected 200, got %d body=%s", rec.Code, rec.Body.String())
	}
}


func TestGetOllamaModelRoute(t *testing.T) {
	h := &ollamaTestSurface{}
	r := chi.NewRouter()
	registerOllamaTestRoutes(r, h)

	t.Run("direct", func(t *testing.T) {
		body := `{"model":"deepseek-v4-flash"}`
    	req := httptest.NewRequest(http.MethodPost, "/api/show", strings.NewReader(body))    
		req.Header.Set("Content-Type", "application/json")
		rec := httptest.NewRecorder()
		r.ServeHTTP(rec, req)
		if rec.Code != http.StatusOK {
			t.Fatalf("expected 200, got %d body=%s", rec.Code, rec.Body.String())
		}
	})

	t.Run("direct_nothinking", func(t *testing.T) {
		body := `{"model":"deepseek-v4-flash-nothinking"}`
    	req := httptest.NewRequest(http.MethodPost, "/api/show", strings.NewReader(body))    
		req.Header.Set("Content-Type", "application/json")
		rec := httptest.NewRecorder()
		r.ServeHTTP(rec, req)
		if rec.Code != http.StatusOK {
			t.Fatalf("expected 200, got %d body=%s", rec.Code, rec.Body.String())
		}
	})

	t.Run("direct_expert", func(t *testing.T) {
		body := `{"model":"models/deepseek-v4-pro"}`
    	req := httptest.NewRequest(http.MethodPost, "/api/show", strings.NewReader(body))    
		req.Header.Set("Content-Type", "application/json")
		rec := httptest.NewRecorder()
		r.ServeHTTP(rec, req)
		if rec.Code != http.StatusOK {
			t.Fatalf("expected 200, got %d body=%s", rec.Code, rec.Body.String())
		}
	})

	t.Run("direct_vision", func(t *testing.T) {
		body := `{"model":"deepseek-v4-vision"}`
    	req := httptest.NewRequest(http.MethodPost, "/api/show", strings.NewReader(body))    
		req.Header.Set("Content-Type", "application/json")
		rec := httptest.NewRecorder()
		r.ServeHTTP(rec, req)
		if rec.Code != http.StatusOK {
			t.Fatalf("expected 200, got %d body=%s", rec.Code, rec.Body.String())
		}
	})
}

func TestGetOllamaModelRouteNotFound(t *testing.T) {
	h := &ollamaTestSurface{}
	r := chi.NewRouter()
	registerOllamaTestRoutes(r, h)

	body := `{"model":"not-exists"}`
    req := httptest.NewRequest(http.MethodPost, "/api/show", strings.NewReader(body))    
	req.Header.Set("Content-Type", "application/json")
	rec := httptest.NewRecorder()
	r.ServeHTTP(rec, req)
	if rec.Code != http.StatusNotFound {
		t.Fatalf("expected 404, got %d body=%s", rec.Code, rec.Body.String())
	}
}