package monitoring

import (
	"testing"
	"time"

	"github.com/rcourtman/pulse-go-rewrite/internal/models"
	unifiedresources "github.com/rcourtman/pulse-go-rewrite/internal/unifiedresources"
)

func TestBuildConnectedInfrastructure_AppliesIgnoreStateToActiveSurfaces(t *testing.T) {
	now := time.Unix(1_700_000_000, 0)
	items := buildConnectedInfrastructure([]unifiedresources.Resource{
		{
			ID:       "tower-resource",
			Name:     "Tower",
			LastSeen: now,
			Status:   unifiedresources.StatusOnline,
			Agent: &unifiedresources.AgentData{
				AgentID:      "tower-agent",
				AgentVersion: "1.2.3",
				Hostname:     "tower.local",
				Platform:     "linux",
			},
			Docker: &unifiedresources.DockerData{
				HostSourceID: "tower-docker",
				AgentID:      "tower-agent",
			},
			PBS: &unifiedresources.PBSData{
				Version: "3.4.1",
			},
			Identity: unifiedresources.ResourceIdentity{
				Hostnames: []string{"tower.local"},
			},
		},
	}, models.StateSnapshot{
		RemovedDockerHosts: []models.RemovedDockerHost{
			{ID: "tower-docker", RemovedAt: now},
		},
	})

	if len(items) != 2 {
		t.Fatalf("expected 2 connected infrastructure items, got %d", len(items))
	}

	var active *models.ConnectedInfrastructureItemFrontend
	var ignored *models.ConnectedInfrastructureItemFrontend
	for i := range items {
		item := &items[i]
		switch item.Status {
		case "active":
			active = item
		case "ignored":
			ignored = item
		}
	}

	if active == nil {
		t.Fatal("expected active connected infrastructure item")
	}
	if ignored == nil {
		t.Fatal("expected ignored connected infrastructure item")
	}

	if len(active.Surfaces) != 2 {
		t.Fatalf("expected host-managed active surfaces to remain without docker, got %#v", active.Surfaces)
	}
	for _, surface := range active.Surfaces {
		if surface.Kind == "docker" {
			t.Fatalf("expected docker surface to be removed from active item, got %#v", active.Surfaces)
		}
	}

	if len(ignored.Surfaces) != 1 || ignored.Surfaces[0].Kind != "docker" {
		t.Fatalf("expected ignored docker surface, got %#v", ignored.Surfaces)
	}
}

func TestBuildConnectedInfrastructure_UsesSharedDisplayNameFallback(t *testing.T) {
	now := time.Unix(1_700_000_000, 0)
	items := buildConnectedInfrastructure([]unifiedresources.Resource{
		{
			ID:       "tower-resource",
			LastSeen: now,
			Status:   unifiedresources.StatusOnline,
			Canonical: &unifiedresources.CanonicalIdentity{
				DisplayName: "Canonical Tower",
			},
			Agent: &unifiedresources.AgentData{
				AgentID:      "tower-agent",
				AgentVersion: "1.2.3",
				Hostname:     "tower.local",
				Platform:     "linux",
			},
			Identity: unifiedresources.ResourceIdentity{
				Hostnames: []string{"tower.local"},
			},
		},
	}, models.StateSnapshot{})

	if len(items) != 1 {
		t.Fatalf("expected 1 connected infrastructure item, got %d", len(items))
	}

	item := items[0]
	if item.Name != "Canonical Tower" {
		t.Fatalf("expected canonical display name to drive name, got %q", item.Name)
	}
	if item.DisplayName != "Canonical Tower" {
		t.Fatalf("expected canonical display name to drive displayName, got %q", item.DisplayName)
	}
}

func TestBuildConnectedInfrastructure_UsesSharedTopLevelSystemGrouping(t *testing.T) {
	now := time.Unix(1_700_000_000, 0)
	items := buildConnectedInfrastructure([]unifiedresources.Resource{
		{
			ID:       "tower-agent",
			Name:     "Tower Agent",
			LastSeen: now,
			Status:   unifiedresources.StatusOnline,
			Agent: &unifiedresources.AgentData{
				AgentID:      "tower-agent",
				AgentVersion: "1.2.3",
				Hostname:     "tower.local",
			},
			Identity: unifiedresources.ResourceIdentity{
				Hostnames: []string{"tower.local"},
			},
		},
		{
			ID:       "tower-pbs",
			Name:     "Tower PBS",
			LastSeen: now,
			Status:   unifiedresources.StatusOnline,
			PBS: &unifiedresources.PBSData{
				InstanceID: "pbs-1",
				Hostname:   "tower.local",
				HostURL:    "https://tower.local:8007",
				Version:    "3.4.1",
			},
		},
	}, models.StateSnapshot{})

	if len(items) != 1 {
		t.Fatalf("expected one connected infrastructure item after shared grouping, got %d", len(items))
	}

	item := items[0]
	if len(item.Surfaces) != 2 {
		t.Fatalf("expected host telemetry and PBS surfaces on one item, got %#v", item.Surfaces)
	}
	if item.ID == "resource:tower-pbs" {
		t.Fatalf("expected PBS surface to attach to the shared top-level system group, got %q", item.ID)
	}
}