package api

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

	fleetai "crypto-miner-server/internal/ai"
	"crypto-miner-server/internal/db"
)

type stubFleetAIConfig struct {
	view FleetAIConfigView
}

func (s *stubFleetAIConfig) GetFleetAIConfig() FleetAIConfigView { return s.view }
func (s *stubFleetAIConfig) UpdateFleetAIConfig(v FleetAIConfigView) error {
	s.view = v
	return nil
}

func TestFleetAIHandlerGetPutConfig(t *testing.T) {
	cfg := &stubFleetAIConfig{view: FleetAIConfigView{
		AIControlEnabled: false, AIEndpoint: "http://127.0.0.1:11434/v1", AIDecisionIntervalSec: 60,
	}}
	h := NewFleetAIHandler(cfg, nil)

	req := httptest.NewRequest(http.MethodGet, "/api/v1/ai/config", nil)
	rec := httptest.NewRecorder()
	h.GetConfig(rec, req)
	if rec.Code != http.StatusOK {
		t.Fatalf("status %d", rec.Code)
	}

	body := `{"ai_control_enabled":true,"ai_endpoint":"http://127.0.0.1:11434/v1","ai_model":"llama3.2","ai_no_context":true,"ai_decision_interval_sec":45}`
	req = httptest.NewRequest(http.MethodPut, "/api/v1/ai/config", strings.NewReader(body))
	rec = httptest.NewRecorder()
	h.PutConfig(rec, req)
	if rec.Code != http.StatusOK {
		t.Fatalf("put status %d body %s", rec.Code, rec.Body.String())
	}
	if !cfg.view.AIControlEnabled || cfg.view.AIModel != "llama3.2" {
		t.Fatalf("config not updated: %+v", cfg.view)
	}
}

func TestFleetAIHandlerGetDecisions(t *testing.T) {
	database, err := db.New(t.TempDir())
	if err != nil {
		t.Fatal(err)
	}
	t.Cleanup(func() { database.Close() })
	_ = database.InsertAIDecision("agent-x", "abc", `{"commands":[]}`, "noop:ok")

	h := NewFleetAIHandler(nil, database)
	req := httptest.NewRequest(http.MethodGet, "/api/v1/ai/decisions?agent_id=agent-x", nil)
	rec := httptest.NewRecorder()
	h.GetDecisions(rec, req)
	if rec.Code != http.StatusOK {
		t.Fatalf("status %d", rec.Code)
	}
	var rows []map[string]interface{}
	if err := json.Unmarshal(rec.Body.Bytes(), &rows); err != nil {
		t.Fatal(err)
	}
	if len(rows) != 1 {
		t.Fatalf("rows: %v", rows)
	}
}

func TestFleetAIExecutorRestartMining(t *testing.T) {
	hub := NewWSHub(nil)
	exec := &FleetAIExecutor{Hub: hub}
	_, err := exec.Execute("missing", fleetai.Command{Type: fleetai.CmdRestartMining})
	if err == nil {
		t.Fatal("expected error for disconnected agent")
	}
}