package service

import "testing"

func TestParseEpisode(t *testing.T) {
	cases := []struct {
		in           string
		wantS, wantE int
	}{
		{"Breaking.Bad.S01E02.1080p.mkv", 1, 2},
		{"breaking.bad.s5e14.mkv", 5, 14},
		{"Friends 1x02.mp4", 1, 2},
		{"Friends 10x24 - The One Where.mkv", 10, 24},
		{"Some Anime - EP05 [1080p].mkv", 1, 5},
		{"Some Anime - E12.mkv", 1, 12},
		{"[MagicStar] 凡人修仙传 年番 - 146 [1080p].mkv", 1, 146},
		{`Some Show/Season 02/Some Show - EP03.mkv`, 2, 3},
		{`Some Show/S02/Some Show - E04.mkv`, 2, 4},
		{`剧集/第2季/剧集 第05集.mkv`, 2, 5},
		{"日剧 第03集.mkv", 1, 3},
		{"日剧 第十集.mkv", 1, 10},
		{"日剧 第二十五话.mkv", 1, 25},
		{"日剧 第12话.mkv", 1, 12},
		{"综艺 第4期下.mkv", 1, 4},
		{`综艺/Season 06/综艺 第17期.mkv`, 6, 17},
		{`动漫/第二季/04.mkv`, 2, 4},
		{`动漫/第十季/第十一集.mkv`, 10, 11},
		{`剧集/S00/剧集 - E01.mkv`, 0, 1},
		{`剧集/Specials/剧集 - 02.mkv`, 0, 2},
		{`剧集/特别篇/03.mkv`, 0, 3},
		{`剧集/剧集 - S00E04.mkv`, 0, 4},
		{`动漫/摇曳露营/OVA/Season 3 [OVA01 [1080p].mkv`, 0, 1},
		{`动漫/示例/OAD/示例.OAD02.mkv`, 0, 2},
		{`动漫/示例/OVD/示例-OVD03.mkv`, 0, 3},
		{`动漫/示例/ONA/示例_ONA04.mkv`, 0, 4},
		{"Movie.2020.1080p.mkv", 0, 0},
		// 分辨率不能被当成季集号：1920x1080 曾匹配出 20x108。
		{"Movie.2020.1920x1080.mkv", 0, 0},
		{"1920x1080.mkv", 0, 0},
		{"[Group][Show][02][3840x2160].mkv", 1, 2},
		// 字幕组方括号集号。
		{"[UHA-WINGS][Peter Grill to Kenja no Jikan][01][BDRIP 1920x1080 HEVC-YUV420P10 FLAC].strm", 1, 1},
		{"[UHA-WINGS][Peter Grill to Kenja no Jikan][12][BDRIP 1920x1080 HEVC-YUV420P10 FLAC].strm", 1, 12},
		// 方括号里的年份/分辨率不是集号。
		{"[Group][Show][2024][1080p].mkv", 0, 0},
	}
	for _, tc := range cases {
		t.Run(tc.in, func(t *testing.T) {
			s, e := ParseEpisode(tc.in)
			if s != tc.wantS || e != tc.wantE {
				t.Errorf("ParseEpisode(%q) = (%d, %d), want (%d, %d)",
					tc.in, s, e, tc.wantS, tc.wantE)
			}
		})
	}
}

func TestEpisodeRefsFromTitleParsesRanges(t *testing.T) {
	tests := []struct {
		name string
		want []episodeRef
	}{
		{
			name: "Archives The Nanyang Mystery 2026 S01E07-S01E08 2160p",
			want: []episodeRef{{Season: 1, Episode: 7}, {Season: 1, Episode: 8}},
		},
		{
			name: "The Heir 2026 S01E39-E42 WEB-DL",
			want: []episodeRef{{Season: 1, Episode: 39}, {Season: 1, Episode: 40}, {Season: 1, Episode: 41}, {Season: 1, Episode: 42}},
		},
	}
	for _, tt := range tests {
		got := episodeRefsFromTitle(tt.name)
		if len(got) != len(tt.want) {
			t.Fatalf("episodeRefsFromTitle(%q) len = %d, want %d: %#v", tt.name, len(got), len(tt.want), got)
		}
		for i := range tt.want {
			if got[i] != tt.want[i] {
				t.Fatalf("episodeRefsFromTitle(%q)[%d] = %#v, want %#v", tt.name, i, got[i], tt.want[i])
			}
		}
	}
}

func TestOnlineEpisodeIdentityFromPathMapsAnimeEpisodeZeroToSpecials(t *testing.T) {
	cases := []struct {
		path        string
		wantSeason  int
		wantEpisode int
	}{
		{`动漫/路人女主/Season 1/S01E00 - 爱与青春的杀必死回.mkv`, 0, 1},
		{`动漫/路人女主/Season 2/S02E00 - 恋爱与纯情的杀必死回.mkv`, 0, 2},
		{`动漫/路人女主/Season 2/S02E03 - 初稿与二稿.mkv`, 2, 3},
		{`动漫/路人女主/Specials/S00E04.mkv`, 0, 4},
	}
	for _, tc := range cases {
		season, episode := onlineEpisodeIdentityFromPath(tc.path)
		if season != tc.wantSeason || episode != tc.wantEpisode {
			t.Errorf("onlineEpisodeIdentityFromPath(%q) = (%d, %d), want (%d, %d)",
				tc.path, season, episode, tc.wantSeason, tc.wantEpisode)
		}
	}
}

func TestResolutionEpisodeArtifact(t *testing.T) {
	cases := []struct {
		path        string
		wantSeason  int
		wantEpisode int
		wantOK      bool
	}{
		{"[UHA-WINGS][Peter Grill to Kenja no Jikan][01][BDRIP 1920x1080 HEVC-YUV420P10 FLAC].strm", 20, 108, true},
		{"今永纱奈/20170601000150_2,00x_3840x2160_amq-13.strm", 40, 216, true},
		{"Movie.2020.1080p.mkv", 0, 0, false},
		{"Show.S01E02.mkv", 0, 0, false},
	}
	for _, tc := range cases {
		season, episode, ok := resolutionEpisodeArtifact(tc.path)
		if season != tc.wantSeason || episode != tc.wantEpisode || ok != tc.wantOK {
			t.Errorf("resolutionEpisodeArtifact(%q) = (%d, %d, %v), want (%d, %d, %v)",
				tc.path, season, episode, ok, tc.wantSeason, tc.wantEpisode, tc.wantOK)
		}
	}
}

func TestDropResolutionArtifactEpisodeIdentity(t *testing.T) {
	// 分辨率伪集号被剔除，并从生成的「第 N 集」标题里清掉。
	polluted := &LocalMetadata{SeasonNum: 20, EpisodeNum: 108, EpisodeTitle: "第 108 集"}
	dropResolutionArtifactEpisodeIdentity(polluted, "[G][Show][01][BDRIP 1920x1080 x].strm")
	if polluted.SeasonNum != 0 || polluted.EpisodeNum != 0 || polluted.EpisodeTitle != "" {
		t.Fatalf("resolution artifact not dropped: %+v", polluted)
	}

	// 真实单集号不受影响。
	real := &LocalMetadata{SeasonNum: 1, EpisodeNum: 3, EpisodeTitle: "本地第三集"}
	dropResolutionArtifactEpisodeIdentity(real, "Show/S01E03 1920x1080.mkv")
	if real.SeasonNum != 1 || real.EpisodeNum != 3 || real.EpisodeTitle != "本地第三集" {
		t.Fatalf("real episode identity was modified: %+v", real)
	}

	// 名称里没有分辨率时不改动任何值。
	plain := &LocalMetadata{SeasonNum: 20, EpisodeNum: 108}
	dropResolutionArtifactEpisodeIdentity(plain, "Show/S20E108.mkv")
	if plain.SeasonNum != 20 || plain.EpisodeNum != 108 {
		t.Fatalf("unrelated identity was cleared: %+v", plain)
	}

	// 文件名里 S20E108 是真实标记时，即使同时含分辨率也必须保留。
	legit := &LocalMetadata{SeasonNum: 20, EpisodeNum: 108}
	dropResolutionArtifactEpisodeIdentity(legit, "Show/Show.S20E108.1920x1080.mkv")
	if legit.SeasonNum != 20 || legit.EpisodeNum != 108 {
		t.Fatalf("legitimate S20E108 identity was cleared: %+v", legit)
	}
}

// S01E11.5 这类「半集」要保留小数部分，才能和真正的第 11 集区分开；
// 同时不能把 S01E11.1080p / S01E05.10bit 这类分辨率、位深读成小数集号。
func TestParseEpisodePartsReadsHalfEpisodes(t *testing.T) {
	cases := []struct {
		name         string
		path         string
		wantSeason   int
		wantEpisode  int
		wantFraction float64
	}{
		{"half episode in anime folder", `动漫/三月的狮子/3月的狮子 S01E11.5.mkv.strm`, 1, 11, 0.5},
		{"half episode bare", `三月的狮子 S01E07.5.mkv`, 1, 7, 0.5},
		{"plain episode has no fraction", `三月的狮子 S01E11.mkv`, 1, 11, 0},
		{"resolution is not a fraction", `Show S01E11.1080p.WEB-DL.mkv`, 1, 11, 0},
		{"bit depth is not a fraction", `Show S01E05.10bit.mkv`, 1, 5, 0},
		{"special naming keeps integral episode", `三月的狮子 S00E11.mkv`, 0, 11, 0},
		{"no episode marker", `movie.mkv`, 0, 0, 0},
	}
	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			season, episode, fraction := ParseEpisodeParts(tc.path)
			if season != tc.wantSeason || episode != tc.wantEpisode || fraction != tc.wantFraction {
				t.Fatalf("ParseEpisodeParts(%q) = (%d, %d, %v), want (%d, %d, %v)",
					tc.path, season, episode, fraction, tc.wantSeason, tc.wantEpisode, tc.wantFraction)
			}
			// ParseEpisode 必须保持原行为（只返回整数季集号）。
			plainSeason, plainEpisode := ParseEpisode(tc.path)
			if plainSeason != tc.wantSeason || plainEpisode != tc.wantEpisode {
				t.Fatalf("ParseEpisode(%q) = (%d, %d), want (%d, %d)",
					tc.path, plainSeason, plainEpisode, tc.wantSeason, tc.wantEpisode)
			}
		})
	}
}

func TestOnlineEpisodeFractionFromPath(t *testing.T) {
	if got := onlineEpisodeFractionFromPath(`动漫/三月的狮子/3月的狮子 S01E11.5.mkv.strm`); got != 0.5 {
		t.Fatalf("half episode fraction = %v, want 0.5", got)
	}
	if got := onlineEpisodeFractionFromPath(`三月的狮子 S01E11.mkv`); got != 0 {
		t.Fatalf("plain episode fraction = %v, want 0", got)
	}
	// S01E00 会被重映射成特别篇 S00E01，此时不能再带上原季的小数规则。
	if got := onlineEpisodeFractionFromPath(`三月的狮子 S01E00.mkv`); got != 0 {
		t.Fatalf("remapped special fraction = %v, want 0", got)
	}
}

func TestFormatEpisodeNumber(t *testing.T) {
	if got := FormatEpisodeNumber(11, 0.5); got != "11.5" {
		t.Fatalf("FormatEpisodeNumber(11, 0.5) = %q, want 11.5", got)
	}
	if got := FormatEpisodeNumber(11, 0); got != "11" {
		t.Fatalf("FormatEpisodeNumber(11, 0) = %q, want 11", got)
	}
}