package api

import (
	"encoding/json"
	"net/http"
	"net/http/httptest"
	"testing"

	"crypto-miner-server/internal/db"
	"github.com/go-chi/chi/v5"
)

func newTestHandler(t *testing.T) *Handler {
	t.Helper()
	database, err := db.New(t.TempDir())
	if err != nil {
		t.Fatal(err)
	}
	t.Cleanup(func() { _ = database.Close() })
	return NewHandler(database)
}

func TestHealthCheckReturnsOK(t *testing.T) {
	h := newTestHandler(t)
	req := httptest.NewRequest(http.MethodGet, "/api/v1/health", nil)
	rec := httptest.NewRecorder()
	h.HealthCheck(rec, req)
	if rec.Code != http.StatusOK {
		t.Fatalf("status %d", rec.Code)
	}
	var body map[string]string
	if err := json.Unmarshal(rec.Body.Bytes(), &body); err != nil {
		t.Fatal(err)
	}
	if body["status"] != "ok" {
		t.Fatalf("unexpected body: %v", body)
	}
}

func TestListAgentsEmptyArray(t *testing.T) {
	h := newTestHandler(t)
	req := httptest.NewRequest(http.MethodGet, "/api/v1/agents", nil)
	rec := httptest.NewRecorder()
	h.ListAgents(rec, req)
	if rec.Code != http.StatusOK {
		t.Fatalf("status %d body %s", rec.Code, rec.Body.String())
	}
	if rec.Body.String() != "[]\n" && rec.Body.String() != "[]" {
		var agents []json.RawMessage
		if err := json.Unmarshal(rec.Body.Bytes(), &agents); err != nil {
			t.Fatal(err)
		}
		if len(agents) != 0 {
			t.Fatalf("expected empty list, got %d", len(agents))
		}
	}
}

func TestGetAgentStatsLimitCap(t *testing.T) {
	h := newTestHandler(t)
	req := httptest.NewRequest(http.MethodGet, "/agents/missing-agent/stats?limit=5000", nil)
	rec := httptest.NewRecorder()
	r := chi.NewRouter()
	r.Get("/agents/{id}/stats", h.GetAgentStats)
	r.ServeHTTP(rec, req)
	if rec.Code == http.StatusInternalServerError {
		t.Fatalf("limit cap caused 500: %s", rec.Body.String())
	}
}

func TestGetRecentSharesLimitCap(t *testing.T) {
	h := newTestHandler(t)
	req := httptest.NewRequest(http.MethodGet, "/api/v1/shares?limit=999999", nil)
	rec := httptest.NewRecorder()
	h.GetRecentShares(rec, req)
	if rec.Code != http.StatusOK {
		t.Fatalf("status %d body %s", rec.Code, rec.Body.String())
	}
}

func TestGetAgentNotFound(t *testing.T) {
	h := newTestHandler(t)
	req := httptest.NewRequest(http.MethodGet, "/agents/nope", nil)
	rec := httptest.NewRecorder()
	r := chi.NewRouter()
	r.Get("/agents/{id}", h.GetAgent)
	r.ServeHTTP(rec, req)
	if rec.Code != http.StatusNotFound {
		t.Fatalf("expected 404, got %d", rec.Code)
	}
}