package service

import (
	"reflect"
	"testing"

	"vidarchive/internal/models"
)

func TestGetEngagement(t *testing.T) {
	svc, dir := newLibrary(t)
	info := `{"comments":[{"author":"Alice","text":"hello <world>","timestamp":1700000000,"like_count":3,"author_is_uploader":true},{"channel":"Bob","text":"second","time_text":"yesterday"}],"heatmap":[{"start_time":0,"end_time":10,"value":0.5},{"start_time":10,"end_time":20,"value":1.0}]}`
	writeItem(t, dir, "engagement", `name = "Engagement"`, map[string]string{
		"video.mp4": "dummy",
		"info.json": info,
	})

	comments, heatmap, err := svc.GetEngagement("engagement")
	if err != nil {
		t.Fatalf("GetEngagement: %v", err)
	}
	if len(comments) != 2 || comments[0].Author != "Alice" || comments[0].LikeCount != 3 || !comments[0].AuthorIsUploader {
		t.Fatalf("unexpected comments: %+v", comments)
	}
	if comments[0].TimeText != "2023-11-14 22:13" {
		t.Errorf("timestamp fallback = %q, want UTC time", comments[0].TimeText)
	}
	if comments[1].Author != "Bob" || comments[1].TimeText != "yesterday" {
		t.Fatalf("fallback comment fields not parsed: %+v", comments[1])
	}
	if len(heatmap) != 2 || heatmap[0].Width != 50 || heatmap[1].Height != 100 {
		t.Fatalf("unexpected heatmap: %+v", heatmap)
	}
}

func TestGetEngagementFiltersAndSortsHeatmap(t *testing.T) {
	svc, dir := newLibrary(t)
	writeItem(t, dir, "heatmap-edge", `name = "Heatmap edge"`, map[string]string{
		"video.mp4": "dummy",
		"info.json": `{"heatmap":[{"start_time":10,"end_time":20,"value":0.5},{"start_time":2,"end_time":4,"value":1},{"start_time":5,"end_time":4,"value":1},{"start_time":20,"end_time":21,"value":-1}]}`,
	})
	_, heatmap, err := svc.GetEngagement("heatmap-edge")
	if err != nil {
		t.Fatalf("GetEngagement: %v", err)
	}
	if len(heatmap) != 2 {
		t.Fatalf("valid heatmap segment count = %d, want 2: %+v", len(heatmap), heatmap)
	}
	if heatmap[0].StartTime != 2 || heatmap[1].StartTime != 10 {
		t.Errorf("heatmap order = %v, want ascending start time", heatmap)
	}
	for _, segment := range heatmap {
		if segment.Width <= 0 || segment.Height < 8 || segment.Height > 100 {
			t.Errorf("invalid normalized heatmap segment: %+v", segment)
		}
	}
}

func TestOrderCommentsNestsRepliesAndCapsDepth(t *testing.T) {
	comments := []models.Comment{
		{ID: "root", Parent: "root", Author: "root"},
		{ID: "reply", Parent: "root", Author: "reply"},
		{ID: "deep-1", Parent: "reply", Author: "deep-1"},
		{ID: "deep-2", Parent: "deep-1", Author: "deep-2"},
		{ID: "deep-3", Parent: "deep-2", Author: "deep-3"},
		{ID: "deep-4", Parent: "deep-3", Author: "deep-4"},
		{ID: "deep-5", Parent: "deep-4", Author: "deep-5"},
	}

	ordered := orderComments(comments)
	if len(ordered) != len(comments) {
		t.Fatalf("orderComments dropped comments: %d of %d", len(ordered), len(comments))
	}
	for i, want := range []string{"root", "reply", "deep-1", "deep-2", "deep-3", "deep-4", "deep-5"} {
		if ordered[i].ID != want {
			t.Errorf("comment %d = %q, want %q", i, ordered[i].ID, want)
		}
	}
	if ordered[1].ReplyTo != "root" || ordered[len(ordered)-1].ReplyTo != "deep-4" {
		t.Errorf("reply targets = %q, %q; want root and deep-4", ordered[1].ReplyTo, ordered[len(ordered)-1].ReplyTo)
	}
	if ordered[len(ordered)-1].Depth != 4 {
		t.Errorf("deep reply depth = %d, want capped depth 4", ordered[len(ordered)-1].Depth)
	}
}

func TestOrderCommentsHandlesMissingParentsCyclesAndSiblings(t *testing.T) {
	comments := []models.Comment{
		{ID: "root", Parent: "root", Author: "root"},
		{ID: "second", Parent: " root ", Author: "second"},
		{ID: "first", Parent: "root", Author: "first"},
		{ID: "orphan", Parent: "missing", Author: "orphan"},
		{ID: "cycle-a", Parent: "cycle-b", Author: "a"},
		{ID: "cycle-b", Parent: "cycle-a", Author: "b"},
	}

	ordered := orderComments(comments)
	if len(ordered) != len(comments) {
		t.Fatalf("orderComments dropped malformed-tree comments: %d of %d", len(ordered), len(comments))
	}
	var ids []string
	for _, comment := range ordered {
		ids = append(ids, comment.ID)
	}
	if !reflect.DeepEqual(ids[:3], []string{"root", "second", "first"}) {
		t.Errorf("sibling/root order = %v, want root then source-order siblings", ids)
	}
	for _, comment := range ordered {
		if comment.ID == "orphan" && comment.Depth != 0 {
			t.Errorf("orphan depth = %d, want top-level", comment.Depth)
		}
	}
}

func TestGetEngagementMalformedSidecarReturnsError(t *testing.T) {
	svc, dir := newLibrary(t)
	writeItem(t, dir, "bad-engagement", `name = "Bad engagement"`, map[string]string{
		"video.mp4": "dummy",
		"info.json": `{"comments":[`,
	})
	if _, _, err := svc.GetEngagement("bad-engagement"); err == nil {
		t.Fatal("expected malformed engagement JSON to return an error")
	}
}

func TestGetEngagementMissingSidecarIsEmpty(t *testing.T) {
	svc, dir := newLibrary(t)
	writeItem(t, dir, "no-engagement", `name = "No engagement"`, map[string]string{"video.mp4": "dummy"})
	comments, heatmap, err := svc.GetEngagement("no-engagement")
	if err != nil {
		t.Fatalf("GetEngagement: %v", err)
	}
	if len(comments) != 0 || len(heatmap) != 0 {
		t.Fatalf("expected empty engagement, got comments=%v heatmap=%v", comments, heatmap)
	}
}