package service

import (
	"strings"
	"testing"
)

func TestStoryboardCinematicQualityContractIncludesRequestedCountAndDuration(t *testing.T) {
	contract := storyboardCinematicQualityContract(30, 7)
	if !strings.Contains(contract, "严格等于 30 秒") {
		t.Fatalf("contract does not include requested duration: %s", contract)
	}
	if !strings.Contains(contract, "严格输出 7 个镜头") {
		t.Fatalf("contract does not include requested shot count: %s", contract)
	}
}

func TestStoryboardCinematicQualityContractIncludesCameraLanguageGuide(t *testing.T) {
	contract := storyboardCinematicQualityContract(0, 0)
	for _, term := range []string{"先确定观众此刻跟随谁", "大远景只承担空间", "一个镜头只保留一个主运镜", "明确人物左右位置", "避免每镜都推进"} {
		if !strings.Contains(contract, term) {
			t.Fatalf("camera language guide is missing %q: %s", term, contract)
		}
	}
}

func TestStoryboardPromptsLeaveAspectRatioToVideoNode(t *testing.T) {
	contract := storyboardCinematicQualityContract(0, 0)
	if !strings.Contains(contract, "不要讨论画幅配置") {
		t.Fatalf("contract does not delegate aspect ratio: %s", contract)
	}
	if strings.Contains(defaultStoryboardPromptTemplate(), "2.39:1") {
		t.Fatal("default storyboard prompt still hard-codes 2.39:1")
	}
	prompt := buildStoryboardVideoPrompt("【风格组合】仙侠·宏大叙事·3D 动漫", "高清、明亮、空间层次清楚", agentStoryboardShot{
		Duration: 8, Description: "剑仙踏云进入画面", VisualPrompt: "剑仙位于画面左侧", VideoPrompt: "向前踏出一步后停住", Camera: "平视中景", Motion: "缓慢推近后停住", TimeBeats: "0-4秒：进入；4-8秒：停住",
	})
	if strings.Contains(prompt, "2.39:1") || strings.Contains(prompt, "画幅") {
		t.Fatal("compiled storyboard video prompt still mentions output aspect ratio")
	}
}

func TestRemoveFixedMediaRestrictionsKeepsCurrentProjectMedium(t *testing.T) {
	template := "保留这一行\n- 不要在 visualPrompt 或 videoPrompt 中使用 3D动漫、动画、二次元、游戏CG、卡通渲染。\n继续保留"
	filtered := removeFixedMediaRestrictions(template)
	if strings.Contains(filtered, "不要在 visualPrompt") {
		t.Fatal("fixed 3D/cartoon restriction was not removed")
	}
	if !strings.Contains(filtered, "保留这一行") || !strings.Contains(filtered, "继续保留") {
		t.Fatal("unrelated template content was removed")
	}
}

func TestValidateStoryboardShotCount(t *testing.T) {
	plan := agentStoryboardPlan{Shots: make([]agentStoryboardShot, 3)}
	if err := validateStoryboardShotCount(plan, 3); err != nil {
		t.Fatalf("expected matching shot count to pass: %v", err)
	}
	if err := validateStoryboardShotCount(plan, 2); err == nil {
		t.Fatal("expected mismatched shot count to fail")
	}
	if err := validateStoryboardShotCount(plan, 0); err != nil {
		t.Fatalf("expected automatic shot count to pass: %v", err)
	}
}

func TestValidateStoryboardComplexityRejectsOverloadedShot(t *testing.T) {
	plan := agentStoryboardPlan{Shots: []agentStoryboardShot{{
		Duration: 5, CharacterIDs: []string{"a", "b", "c"}, MustHave: []string{"一", "二", "三", "四"}, TimeBeats: "0-1秒：起，1-2秒：承，2-4秒：转，4-5秒：合", Motion: "航拍推进并升降",
	}}}
	if err := validateStoryboardComplexity(plan); err == nil {
		t.Fatal("expected overloaded shot to fail complexity validation")
	}
}

func TestValidateStoryboardPlanTreatsComplexityAsAdvisory(t *testing.T) {
	plan := agentStoryboardPlan{Shots: []agentStoryboardShot{{
		Duration:     8,
		CharacterIDs: []string{"a", "b", "c"},
		MustHave:     []string{"一", "二", "三", "四"},
		Motion:       "航拍推进并升降",
	}}}
	characters := []storyboardCharacterCard{{AssetID: "a"}, {AssetID: "b"}, {AssetID: "c"}}

	if err := validateStoryboardPlan(plan, 0, 0, characters); err != nil {
		t.Fatalf("complexity should be advisory for an otherwise valid plan: %v", err)
	}
}

func TestValidateStoryboardComplexityAcceptsSingleDirectedMove(t *testing.T) {
	plan := agentStoryboardPlan{Shots: []agentStoryboardShot{{
		Duration: 8, CharacterIDs: []string{"a", "b"}, MustHave: []string{"身份稳定", "动作落点", "结尾状态"}, TimeBeats: "0-2秒：建立；2-6秒：动作；6-8秒：反应", Motion: "缓慢推近后停住", Dialogue: "你终于来了。",
	}}}
	if err := validateStoryboardComplexity(plan); err != nil {
		t.Fatalf("expected focused shot to pass complexity validation: %v", err)
	}
}

func TestNormalizeAutomaticStoryboardDurationsPreservesLongDialogue(t *testing.T) {
	plan := agentStoryboardPlan{Shots: []agentStoryboardShot{{
		Duration: 8,
		Dialogue: strings.Repeat("字", 50),
	}}}

	normalizeAutomaticStoryboardDurations(&plan, 0)

	if plan.Shots[0].Duration < 10 {
		t.Fatalf("duration = %d, want at least 10", plan.Shots[0].Duration)
	}
	if err := validateStoryboardComplexity(plan); err != nil {
		t.Fatalf("normalized automatic duration should pass: %v", err)
	}
}

func TestMatchStoryboardAssetsForShotPrioritizesCurrentCharacterCards(t *testing.T) {
	assets := []storyboardAsset{
		{ID: "environment", Title: "天宫", Tags: []string{"场景:天宫"}},
		{ID: "character-node", Title: "青莲剑仙", CharacterAssetID: "character-1", CharacterVersionID: "version-3"},
	}
	matched := matchStoryboardAssetsForShot(assets, agentStoryboardShot{CharacterIDs: []string{"character-1"}, AssetTags: []string{"场景:天宫"}})
	if len(matched) != 2 || matched[0].ID != "character-node" {
		t.Fatalf("expected current character card before optional tagged assets: %#v", matched)
	}
}