package gobackend

import (
	"context"
	"fmt"
	"testing"
)

func TestSetExtensionFallbackProviderIDsJSONEmptyStringResetsDefault(t *testing.T) {
	original := GetExtensionFallbackProviderIDs()
	defer SetExtensionFallbackProviderIDs(original)

	SetExtensionFallbackProviderIDs([]string{"custom-ext"})

	if err := SetExtensionFallbackProviderIDsJSON(""); err != nil {
		t.Fatalf("SetExtensionFallbackProviderIDsJSON returned error: %v", err)
	}

	if got := GetExtensionFallbackProviderIDs(); got != nil {
		t.Fatalf("expected nil fallback provider list after reset, got %v", got)
	}
}

func TestBuildDownloadSuccessResponsePrefersRequestedAlbumMetadata(t *testing.T) {
	req := DownloadRequest{
		TrackName:   "Bonus Track",
		ArtistName:  "Artist",
		AlbumName:   "Album (Deluxe)",
		AlbumArtist: "Artist",
		ReleaseDate: "2024-01-01",
		TrackNumber: 14,
		DiscNumber:  1,
		ISRC:        "REQ123",
		CoverURL:    "https://example.com/cover.jpg",
		Genre:       "Pop",
		Label:       "Label",
		Copyright:   "Copyright",
	}

	result := DownloadResult{
		Title:       "Bonus Track",
		Artist:      "Artist",
		Album:       "Album",
		ReleaseDate: "2023-12-01",
		TrackNumber: 2,
		DiscNumber:  9,
		ISRC:        "RES456",
	}

	resp := buildDownloadSuccessResponse(
		req,
		result,
		"tidal",
		"ok",
		"/tmp/test.flac",
		false,
	)

	if resp.Album != req.AlbumName {
		t.Fatalf("album = %q, want %q", resp.Album, req.AlbumName)
	}
	if resp.ReleaseDate != req.ReleaseDate {
		t.Fatalf("release date = %q, want %q", resp.ReleaseDate, req.ReleaseDate)
	}
	if resp.TrackNumber != req.TrackNumber {
		t.Fatalf("track number = %d, want %d", resp.TrackNumber, req.TrackNumber)
	}
	if resp.DiscNumber != req.DiscNumber {
		t.Fatalf("disc number = %d, want %d", resp.DiscNumber, req.DiscNumber)
	}
	if resp.Artist != result.Artist {
		t.Fatalf("artist = %q, want provider artist %q", resp.Artist, result.Artist)
	}
	if resp.ISRC != result.ISRC {
		t.Fatalf("isrc = %q, want provider isrc %q", resp.ISRC, result.ISRC)
	}
}

func TestPreferredReleaseMetadataPrefersRequestValues(t *testing.T) {
	album, releaseDate, trackNumber, discNumber := preferredReleaseMetadata(
		DownloadRequest{
			AlbumName:   "Album (Deluxe Edition)",
			ReleaseDate: "2024-01-01",
			TrackNumber: 13,
			DiscNumber:  2,
		},
		"Album",
		"2023-01-01",
		3,
		1,
	)

	if album != "Album (Deluxe Edition)" {
		t.Fatalf("album = %q", album)
	}
	if releaseDate != "2024-01-01" {
		t.Fatalf("release date = %q", releaseDate)
	}
	if trackNumber != 13 {
		t.Fatalf("track number = %d", trackNumber)
	}
	if discNumber != 2 {
		t.Fatalf("disc number = %d", discNumber)
	}
}

func TestBuildDownloadSuccessResponsePrefersProviderCoverURL(t *testing.T) {
	req := DownloadRequest{
		TrackName:   "Track",
		ArtistName:  "Artist",
		AlbumName:   "Album",
		AlbumArtist: "Artist",
	}

	result := DownloadResult{
		Title:    "Track",
		Artist:   "Artist",
		Album:    "Album",
		CoverURL: "https://cdn.qobuz.test/cover.jpg",
	}

	resp := buildDownloadSuccessResponse(
		req,
		result,
		"qobuz",
		"ok",
		"/tmp/test.flac",
		false,
	)

	if resp.CoverURL != result.CoverURL {
		t.Fatalf("cover url = %q, want %q", resp.CoverURL, result.CoverURL)
	}
}

func TestBuildDownloadSuccessResponseNormalizesDecryptionDescriptor(t *testing.T) {
	req := DownloadRequest{
		TrackName:  "Track",
		ArtistName: "Artist",
	}

	result := DownloadResult{
		Title:         "Track",
		Artist:        "Artist",
		DecryptionKey: "00112233",
	}

	resp := buildDownloadSuccessResponse(
		req,
		result,
		"amazon",
		"ok",
		"/tmp/test.m4a",
		false,
	)

	if resp.Decryption == nil {
		t.Fatal("expected decryption descriptor to be present")
	}
	if resp.Decryption.Strategy != genericFFmpegMOVDecryptionStrategy {
		t.Fatalf("strategy = %q", resp.Decryption.Strategy)
	}
	if resp.Decryption.Key != result.DecryptionKey {
		t.Fatalf("key = %q, want %q", resp.Decryption.Key, result.DecryptionKey)
	}
}

func TestFormatMusicBrainzGenrePrefersHighestCountTag(t *testing.T) {
	got := formatMusicBrainzGenre([]musicBrainzTag{
		{Name: "art pop", Count: 3},
		{Name: "pop", Count: 8},
		{Name: "dance pop", Count: 5},
	})

	if got != "Pop" {
		t.Fatalf("genre = %q, want %q", got, "Pop")
	}
}

func TestSelectMusicBrainzAlbumArtistPrefersMatchingRelease(t *testing.T) {
	releases := []musicBrainzRelease{
		{
			Title: "Other Album",
			ArtistCredit: []musicBrainzArtistCredit{
				{Name: "Wrong Artist"},
			},
		},
		{
			Title: "Target Album",
			ArtistCredit: []musicBrainzArtistCredit{
				{Name: "Artist A", JoinPhrase: " & "},
				{Name: "Artist B"},
			},
		},
	}

	got := selectMusicBrainzAlbumArtist(releases, "Target Album")
	if got != "Artist A & Artist B" {
		t.Fatalf("album artist = %q, want matching release artist credit", got)
	}
}

func TestEnrichRequestExtendedMetadataUsesMusicBrainzAlbumArtist(t *testing.T) {
	origDeezerFetcher := fetchDeezerExtendedMetadataByISRC
	origMusicBrainzGenreFetcher := fetchMusicBrainzGenreByISRC
	origMusicBrainzAlbumArtistFetcher := fetchMusicBrainzAlbumArtistByISRC
	defer func() {
		fetchDeezerExtendedMetadataByISRC = origDeezerFetcher
		fetchMusicBrainzGenreByISRC = origMusicBrainzGenreFetcher
		fetchMusicBrainzAlbumArtistByISRC = origMusicBrainzAlbumArtistFetcher
	}()

	fetchDeezerExtendedMetadataByISRC = func(ctx context.Context, isrc string) (*AlbumExtendedMetadata, error) {
		return &AlbumExtendedMetadata{}, nil
	}
	fetchMusicBrainzGenreByISRC = func(isrc string) (string, error) {
		return "", fmt.Errorf("no genre")
	}
	fetchMusicBrainzAlbumArtistByISRC = func(isrc string, albumName string) (string, error) {
		if isrc != "TESTISRC" || albumName != "Target Album" {
			t.Fatalf("unexpected MusicBrainz args: %q / %q", isrc, albumName)
		}
		return "MusicBrainz Album Artist", nil
	}

	req := DownloadRequest{
		ISRC:       "TESTISRC",
		ArtistName: "Track Artist",
		AlbumName:  "Target Album",
	}

	enrichRequestExtendedMetadata(&req)

	if req.AlbumArtist != "MusicBrainz Album Artist" {
		t.Fatalf("album artist = %q, want MusicBrainz value", req.AlbumArtist)
	}
}

func TestEnrichRequestExtendedMetadataDoesNotFallbackAlbumArtistToTrackArtist(t *testing.T) {
	origDeezerFetcher := fetchDeezerExtendedMetadataByISRC
	origMusicBrainzGenreFetcher := fetchMusicBrainzGenreByISRC
	origMusicBrainzAlbumArtistFetcher := fetchMusicBrainzAlbumArtistByISRC
	defer func() {
		fetchDeezerExtendedMetadataByISRC = origDeezerFetcher
		fetchMusicBrainzGenreByISRC = origMusicBrainzGenreFetcher
		fetchMusicBrainzAlbumArtistByISRC = origMusicBrainzAlbumArtistFetcher
	}()

	fetchDeezerExtendedMetadataByISRC = func(ctx context.Context, isrc string) (*AlbumExtendedMetadata, error) {
		return &AlbumExtendedMetadata{}, nil
	}
	fetchMusicBrainzGenreByISRC = func(isrc string) (string, error) {
		return "", fmt.Errorf("no genre")
	}
	fetchMusicBrainzAlbumArtistByISRC = func(isrc string, albumName string) (string, error) {
		return "", fmt.Errorf("no album artist")
	}

	req := DownloadRequest{
		ISRC:       "TESTISRC",
		ArtistName: "Track Artist",
		AlbumName:  "Target Album",
	}

	enrichRequestExtendedMetadata(&req)

	if req.AlbumArtist != "" {
		t.Fatalf("album artist = %q, want empty when MusicBrainz has no value", req.AlbumArtist)
	}
}

func TestEnrichExtraMetadataByISRCFallsBackToMusicBrainzGenre(t *testing.T) {
	origDeezerFetcher := fetchDeezerExtendedMetadataByISRC
	origMusicBrainzFetcher := fetchMusicBrainzGenreByISRC
	defer func() {
		fetchDeezerExtendedMetadataByISRC = origDeezerFetcher
		fetchMusicBrainzGenreByISRC = origMusicBrainzFetcher
	}()

	fetchDeezerExtendedMetadataByISRC = func(ctx context.Context, isrc string) (*AlbumExtendedMetadata, error) {
		return nil, nil
	}
	fetchMusicBrainzGenreByISRC = func(isrc string) (string, error) {
		if isrc != "TEST123" {
			t.Fatalf("unexpected isrc: %q", isrc)
		}
		return "Alternative Rock", nil
	}

	genre := ""
	label := ""
	copyright := ""
	enrichExtraMetadataByISRC("DownloadWithFallback", "TEST123", &genre, &label, &copyright)

	if genre != "Alternative Rock" {
		t.Fatalf("genre = %q, want fallback genre", genre)
	}
	if label != "" {
		t.Fatalf("label = %q, want empty", label)
	}
	if copyright != "" {
		t.Fatalf("copyright = %q, want empty", copyright)
	}
}

func TestEnrichExtraMetadataByISRCPrefersDeezerGenre(t *testing.T) {
	origDeezerFetcher := fetchDeezerExtendedMetadataByISRC
	origMusicBrainzFetcher := fetchMusicBrainzGenreByISRC
	defer func() {
		fetchDeezerExtendedMetadataByISRC = origDeezerFetcher
		fetchMusicBrainzGenreByISRC = origMusicBrainzFetcher
	}()

	musicBrainzCalled := false
	fetchDeezerExtendedMetadataByISRC = func(ctx context.Context, isrc string) (*AlbumExtendedMetadata, error) {
		return &AlbumExtendedMetadata{
			Genre:     "Synthpop",
			Label:     "EMI",
			Copyright: "(C) Test",
		}, nil
	}
	fetchMusicBrainzGenreByISRC = func(isrc string) (string, error) {
		musicBrainzCalled = true
		return "Rock", nil
	}

	genre := ""
	label := ""
	copyright := ""
	enrichExtraMetadataByISRC("DownloadWithFallback", "TEST456", &genre, &label, &copyright)

	if genre != "Synthpop" {
		t.Fatalf("genre = %q, want Deezer genre", genre)
	}
	if label != "EMI" {
		t.Fatalf("label = %q, want Deezer label", label)
	}
	if copyright != "(C) Test" {
		t.Fatalf("copyright = %q, want Deezer copyright", copyright)
	}
	if musicBrainzCalled {
		t.Fatal("expected MusicBrainz not to be called when Deezer already provides genre")
	}
}

func TestApplyReEnrichTrackMetadataPreservesExistingReleaseDateWhenCandidateMissing(t *testing.T) {
	req := reEnrichRequest{
		SpotifyID:   "spotify-track-id",
		AlbumName:   "Original Album",
		ReleaseDate: "2024-01-01",
		ISRC:        "REQ123",
	}

	applyReEnrichTrackMetadata(&req, ExtTrackMetadata{
		AlbumName:   "Resolved Album",
		ReleaseDate: "",
		ISRC:        "",
	})

	if req.ReleaseDate != "2024-01-01" {
		t.Fatalf("release date = %q, want existing value preserved", req.ReleaseDate)
	}
	if req.AlbumName != "Resolved Album" {
		t.Fatalf("album = %q, want updated album", req.AlbumName)
	}
	if req.ISRC != "REQ123" {
		t.Fatalf("isrc = %q, want existing value preserved", req.ISRC)
	}
}

func TestSelectBestReEnrichTrackPrefersCandidateWithReleaseDate(t *testing.T) {
	req := reEnrichRequest{
		TrackName:   "Song Title",
		ArtistName:  "Artist Name",
		AlbumName:   "Album Name",
		ReleaseDate: "",
		DurationMs:  180000,
	}

	tracks := []ExtTrackMetadata{
		{
			ID:          "first",
			Name:        "Song Title",
			Artists:     "Artist Name",
			AlbumName:   "Album Name",
			DurationMS:  180000,
			ReleaseDate: "",
			ProviderID:  "spotify",
		},
		{
			ID:          "second",
			Name:        "Song Title",
			Artists:     "Artist Name",
			AlbumName:   "Album Name",
			DurationMS:  180000,
			ReleaseDate: "2024-03-09",
			ProviderID:  "deezer",
		},
	}

	best := selectBestReEnrichTrack(req, tracks)
	if best == nil {
		t.Fatal("expected a selected track")
	}
	if best.ID != "second" {
		t.Fatalf("selected track = %q, want candidate with release date", best.ID)
	}
}

func TestSelectBestReEnrichTrackRejectsMismatchedSearchResults(t *testing.T) {
	req := reEnrichRequest{
		TrackName:  "Song Title",
		ArtistName: "Artist Name",
		AlbumName:  "Album Name",
		DurationMs: 180000,
	}

	tracks := []ExtTrackMetadata{
		{
			ID:          "wrong-rich-metadata",
			Name:        "Different Song",
			Artists:     "Different Artist",
			AlbumName:   "Album Name",
			DurationMS:  180000,
			ReleaseDate: "2024-03-09",
			TrackNumber: 4,
			DiscNumber:  1,
			ISRC:        "WRONG1234567",
			ProviderID:  "deezer",
		},
	}

	if best := selectBestReEnrichTrack(req, tracks); best != nil {
		t.Fatalf("selected track = %q, want no match", best.ID)
	}
}

func TestSelectBestReEnrichTrackAllowsExactISRCDespiteMetadataMismatch(t *testing.T) {
	req := reEnrichRequest{
		TrackName:  "Song Title",
		ArtistName: "Artist Name",
		ISRC:       "USRC17607839",
		DurationMs: 999999000,
	}

	tracks := []ExtTrackMetadata{
		{
			ID:         "same-isrc",
			Name:       "Different Song",
			Artists:    "Different Artist",
			DurationMS: 180000,
			ISRC:       "USRC17607839",
			ProviderID: "deezer",
		},
	}

	best := selectBestReEnrichTrack(req, tracks)
	if best == nil {
		t.Fatal("expected exact ISRC candidate to be selected")
	}
	if best.ID != "same-isrc" {
		t.Fatalf("selected track = %q, want exact ISRC candidate", best.ID)
	}
}

func TestSelectBestReEnrichTrackPlaceholderFallsBackToAlbum(t *testing.T) {
	req := reEnrichRequest{
		TrackName:  "Unknown Title",
		ArtistName: "Unknown Artist",
		AlbumName:  "Harry Styles",
		DurationMs: 180000,
	}

	tracks := []ExtTrackMetadata{
		{
			ID:         "album-match",
			Name:       "Sign of the Times",
			Artists:    "Harry Styles",
			AlbumName:  "Harry Styles",
			DurationMS: 180000,
			ProviderID: "deezer",
		},
	}

	best := selectBestReEnrichTrack(req, tracks)
	if best == nil {
		t.Fatal("expected album-matching candidate to be selected when title/artist are placeholders")
	}
	if best.ID != "album-match" {
		t.Fatalf("selected track = %q, want album-match", best.ID)
	}
}

func TestBuildReEnrichFFmpegMetadataOmitsEmptyFields(t *testing.T) {
	req := reEnrichRequest{
		TrackName:   "Song",
		ArtistName:  "Artist",
		AlbumName:   "Album",
		AlbumArtist: "",
		ReleaseDate: "",
		TrackNumber: 0,
		DiscNumber:  0,
		ISRC:        "",
		Genre:       "",
		Label:       "",
		Copyright:   "",
	}

	metadata := buildReEnrichFFmpegMetadata(&req, "")

	if metadata["TITLE"] != "Song" {
		t.Fatalf("title = %q", metadata["TITLE"])
	}
	if metadata["ARTIST"] != "Artist" {
		t.Fatalf("artist = %q", metadata["ARTIST"])
	}
	if metadata["ALBUM"] != "Album" {
		t.Fatalf("album = %q", metadata["ALBUM"])
	}

	for _, key := range []string{
		"ALBUMARTIST",
		"DATE",
		"TRACKNUMBER",
		"DISCNUMBER",
		"ISRC",
		"GENRE",
		"ORGANIZATION",
		"COPYRIGHT",
		"LYRICS",
		"UNSYNCEDLYRICS",
	} {
		if _, exists := metadata[key]; exists {
			t.Fatalf("did not expect key %s in metadata: %#v", key, metadata)
		}
	}
}

func TestBuildReEnrichSearchQuerySkipsPlaceholderArtist(t *testing.T) {
	req := reEnrichRequest{
		TrackName:  "Sign of the Times",
		ArtistName: "Unknown Artist",
		AlbumName:  "Harry Styles",
	}

	query := buildReEnrichSearchQuery(req)
	if query != "Sign of the Times" {
		t.Fatalf("query = %q", query)
	}

	req = reEnrichRequest{
		TrackName:  "Unknown Title",
		ArtistName: "Unknown Artist",
		AlbumName:  "Harry Styles",
	}
	query = buildReEnrichSearchQuery(req)
	if query != "Harry Styles" {
		t.Fatalf("fallback album query = %q", query)
	}
}

func TestApplyReEnrichTrackMetadataCopiesComposerAndTotals(t *testing.T) {
	req := reEnrichRequest{}

	applyReEnrichTrackMetadata(&req, ExtTrackMetadata{
		Name:        "Resolved Song",
		Artists:     "Resolved Artist",
		TrackNumber: 7,
		TotalTracks: 12,
		DiscNumber:  2,
		TotalDiscs:  3,
		Composer:    "Composer",
	})

	if req.TrackNumber != 7 || req.TotalTracks != 12 {
		t.Fatalf("track metadata = %d/%d", req.TrackNumber, req.TotalTracks)
	}
	if req.DiscNumber != 2 || req.TotalDiscs != 3 {
		t.Fatalf("disc metadata = %d/%d", req.DiscNumber, req.TotalDiscs)
	}
	if req.TrackName != "Resolved Song" || req.ArtistName != "Resolved Artist" {
		t.Fatalf("basic tags = %q / %q", req.TrackName, req.ArtistName)
	}
	if req.Composer != "Composer" {
		t.Fatalf("composer = %q", req.Composer)
	}
}

func TestBuildReEnrichFFmpegMetadataFormatsTotalsAndComposer(t *testing.T) {
	req := reEnrichRequest{
		TrackNumber: 7,
		TotalTracks: 12,
		DiscNumber:  2,
		TotalDiscs:  3,
		Composer:    "Composer",
	}

	metadata := buildReEnrichFFmpegMetadata(&req, "")

	if metadata["TRACKNUMBER"] != "7/12" {
		t.Fatalf("TRACKNUMBER = %q", metadata["TRACKNUMBER"])
	}
	if metadata["DISCNUMBER"] != "2/3" {
		t.Fatalf("DISCNUMBER = %q", metadata["DISCNUMBER"])
	}
	if metadata["COMPOSER"] != "Composer" {
		t.Fatalf("COMPOSER = %q", metadata["COMPOSER"])
	}
}