package service

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

	"metis/pkg/facts"
	"metis/pkg/sentinel"
)

func TestScratchHealthAnnotations(t *testing.T) {
	observed := time.Date(2026, 4, 22, 6, 45, 0, 0, time.UTC)
	annotations := scratchHealthAnnotations(&facts.USBScratch{
		Mountpoint:   "/mnt/astraios",
		UUID:         "usb-1",
		FS:           "ext4",
		MountHealthy: true,
		UUIDHealthy:  true,
		BindHealthy:  true,
		BindTargets:  []facts.USBBindTarget{{Path: "/var/log/pods", Healthy: true}, {Path: "/var/tmp", Healthy: true}},
	}, observed)
	if annotations["maintenance.bstein.dev/usb-scratch-status"] != "ok" || annotations["maintenance.bstein.dev/astraios-detail"] != "healthy" {
		t.Fatalf("unexpected healthy annotations: %#v", annotations)
	}
	if annotations["maintenance.bstein.dev/usb-scratch-selector"] != "UUID=usb-1" {
		t.Fatalf("selector annotation missing: %#v", annotations)
	}
	if annotations["maintenance.bstein.dev/astraios-managed-paths"] != "/var/log/pods_/var/tmp" {
		t.Fatalf("managed paths annotation mismatch: %#v", annotations)
	}

	status, detail := scratchStatusDetail(&facts.USBScratch{MountHealthy: false, BindHealthy: false})
	if status != "error" || !strings.Contains(detail, "mount-unhealthy") || !strings.Contains(detail, "bind-mount-incomplete") {
		t.Fatalf("unexpected unhealthy detail: %s %s", status, detail)
	}
}

func TestStoreSnapshotPatchesNodeAnnotations(t *testing.T) {
	var patchPath string
	var patchContentType string
	var patchBody map[string]any
	kube := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		if r.Method != http.MethodPatch || r.URL.Path != "/api/v1/nodes/titan-04" {
			http.NotFound(w, r)
			return
		}
		patchPath = r.URL.Path
		patchContentType = r.Header.Get("Content-Type")
		if err := json.NewDecoder(r.Body).Decode(&patchBody); err != nil {
			t.Fatalf("decode patch body: %v", err)
		}
		_ = json.NewEncoder(w).Encode(map[string]any{"status": "ok"})
	}))
	defer kube.Close()

	origFactory := kubeClientFactory
	kubeClientFactory = func() (*kubeClient, error) {
		return kubeClientFactoryForURL(kube.URL, kube.Client()), nil
	}
	t.Cleanup(func() { kubeClientFactory = origFactory })

	app := newTestApp(t)
	if err := app.StoreSnapshot(SnapshotRecord{
		Node:        "titan-04",
		CollectedAt: time.Date(2026, 4, 22, 6, 50, 0, 0, time.UTC),
		Snapshot: sentinel.Snapshot{
			Hostname: "titan-04",
			USBScratch: &facts.USBScratch{
				Mountpoint:   "/mnt/astraios",
				UUID:         "usb-1",
				MountHealthy: true,
				UUIDHealthy:  true,
				BindHealthy:  true,
				BindTargets:  []facts.USBBindTarget{{Path: "/var/log/pods", Healthy: true}},
			},
		},
	}); err != nil {
		t.Fatalf("StoreSnapshot: %v", err)
	}
	if patchPath != "/api/v1/nodes/titan-04" || patchContentType != "application/merge-patch+json" {
		t.Fatalf("patch request mismatch: path=%q content-type=%q", patchPath, patchContentType)
	}
	metadata := patchBody["metadata"].(map[string]any)
	annotations := metadata["annotations"].(map[string]any)
	if annotations["maintenance.bstein.dev/usb-scratch-status"] != "ok" || annotations["maintenance.bstein.dev/astraios-selector"] != "UUID=usb-1" {
		t.Fatalf("annotation patch mismatch: %#v", annotations)
	}
}