package model_test

import (
	"time"

	"github.com/navidrome/navidrome/conf"
	"github.com/navidrome/navidrome/conf/configtest"
	"github.com/navidrome/navidrome/consts"
	. "github.com/navidrome/navidrome/model"
	. "github.com/onsi/ginkgo/v2"
	. "github.com/onsi/gomega"
)

var _ = Describe("MediaFiles", func() {
	var mfs MediaFiles

	Context("Simple attributes", func() {
		BeforeEach(func() {
			mfs = MediaFiles{
				{
					ID: "1", AlbumID: "AlbumID", Album: "Album", ArtistID: "ArtistID", Artist: "Artist", AlbumArtistID: "AlbumArtistID", AlbumArtist: "AlbumArtist",
					SortAlbumName: "SortAlbumName", SortArtistName: "SortArtistName", SortAlbumArtistName: "SortAlbumArtistName",
					OrderAlbumName: "OrderAlbumName", OrderAlbumArtistName: "OrderAlbumArtistName",
					MbzAlbumArtistID: "MbzAlbumArtistID", MbzAlbumType: "MbzAlbumType", MbzAlbumComment: "MbzAlbumComment",
					Compilation: false, CatalogNum: "", Path: "/music1/file1.mp3",
				},
				{
					ID: "2", Album: "Album", ArtistID: "ArtistID", Artist: "Artist", AlbumArtistID: "AlbumArtistID", AlbumArtist: "AlbumArtist", AlbumID: "AlbumID",
					SortAlbumName: "SortAlbumName", SortArtistName: "SortArtistName", SortAlbumArtistName: "SortAlbumArtistName",
					OrderAlbumName: "OrderAlbumName", OrderArtistName: "OrderArtistName", OrderAlbumArtistName: "OrderAlbumArtistName",
					MbzAlbumArtistID: "MbzAlbumArtistID", MbzAlbumType: "MbzAlbumType", MbzAlbumComment: "MbzAlbumComment",
					Compilation: true, CatalogNum: "CatalogNum", HasCoverArt: true, Path: "/music2/file2.mp3",
				},
			}
		})

		It("sets the single values correctly", func() {
			album := mfs.ToAlbum()
			Expect(album.ID).To(Equal("AlbumID"))
			Expect(album.Name).To(Equal("Album"))
			Expect(album.Artist).To(Equal("Artist"))
			Expect(album.ArtistID).To(Equal("ArtistID"))
			Expect(album.AlbumArtist).To(Equal("AlbumArtist"))
			Expect(album.AlbumArtistID).To(Equal("AlbumArtistID"))
			Expect(album.SortAlbumName).To(Equal("SortAlbumName"))
			Expect(album.SortArtistName).To(Equal("SortArtistName"))
			Expect(album.SortAlbumArtistName).To(Equal("SortAlbumArtistName"))
			Expect(album.OrderAlbumName).To(Equal("OrderAlbumName"))
			Expect(album.OrderAlbumArtistName).To(Equal("OrderAlbumArtistName"))
			Expect(album.MbzAlbumArtistID).To(Equal("MbzAlbumArtistID"))
			Expect(album.MbzAlbumType).To(Equal("MbzAlbumType"))
			Expect(album.MbzAlbumComment).To(Equal("MbzAlbumComment"))
			Expect(album.CatalogNum).To(Equal("CatalogNum"))
			Expect(album.Compilation).To(BeTrue())
			Expect(album.EmbedArtPath).To(Equal("/music2/file2.mp3"))
			Expect(album.Paths).To(Equal("/music1" + consts.Zwsp + "/music2"))
		})
	})
	Context("Aggregated attributes", func() {
		When("we have only one song", func() {
			BeforeEach(func() {
				mfs = MediaFiles{
					{Duration: 100.2, Size: 1024, Year: 1985, Date: "1985-01-02", UpdatedAt: t("2022-12-19 09:30"), CreatedAt: t("2022-12-19 08:30")},
				}
			})
			It("calculates the aggregates correctly", func() {
				album := mfs.ToAlbum()
				Expect(album.Duration).To(Equal(float32(100.2)))
				Expect(album.Size).To(Equal(int64(1024)))
				Expect(album.MinYear).To(Equal(1985))
				Expect(album.MaxYear).To(Equal(1985))
				Expect(album.Date).To(Equal("1985-01-02"))
				Expect(album.UpdatedAt).To(Equal(t("2022-12-19 09:30")))
				Expect(album.CreatedAt).To(Equal(t("2022-12-19 08:30")))
			})
		})

		When("we have multiple songs with different dates", func() {
			BeforeEach(func() {
				mfs = MediaFiles{
					{Duration: 100.2, Size: 1024, Year: 1985, Date: "1985-01-02", UpdatedAt: t("2022-12-19 09:30"), CreatedAt: t("2022-12-19 08:30")},
					{Duration: 200.2, Size: 2048, Year: 0, Date: "", UpdatedAt: t("2022-12-19 09:45"), CreatedAt: t("2022-12-19 08:30")},
					{Duration: 150.6, Size: 1000, Year: 1986, Date: "1986-01-02", UpdatedAt: t("2022-12-19 09:45"), CreatedAt: t("2022-12-19 07:30")},
				}
			})
			It("calculates the aggregates correctly", func() {
				album := mfs.ToAlbum()
				Expect(album.Duration).To(Equal(float32(451.0)))
				Expect(album.Size).To(Equal(int64(4072)))
				Expect(album.MinYear).To(Equal(1985))
				Expect(album.MaxYear).To(Equal(1986))
				Expect(album.Date).To(BeEmpty())
				Expect(album.UpdatedAt).To(Equal(t("2022-12-19 09:45")))
				Expect(album.CreatedAt).To(Equal(t("2022-12-19 07:30")))
			})
			Context("MinYear", func() {
				It("returns 0 when all values are 0", func() {
					mfs = MediaFiles{{Year: 0}, {Year: 0}, {Year: 0}}
					a := mfs.ToAlbum()
					Expect(a.MinYear).To(Equal(0))
				})
				It("returns the smallest value from the list, not counting 0", func() {
					mfs = MediaFiles{{Year: 2000}, {Year: 0}, {Year: 1999}}
					a := mfs.ToAlbum()
					Expect(a.MinYear).To(Equal(1999))
				})
			})
		})
		When("we have multiple songs with same dates", func() {
			BeforeEach(func() {
				mfs = MediaFiles{
					{Duration: 100.2, Size: 1024, Year: 1985, Date: "1985-01-02", UpdatedAt: t("2022-12-19 09:30"), CreatedAt: t("2022-12-19 08:30")},
					{Duration: 200.2, Size: 2048, Year: 1985, Date: "1985-01-02", UpdatedAt: t("2022-12-19 09:45"), CreatedAt: t("2022-12-19 08:30")},
					{Duration: 150.6, Size: 1000, Year: 1985, Date: "1985-01-02", UpdatedAt: t("2022-12-19 09:45"), CreatedAt: t("2022-12-19 07:30")},
				}
			})
			It("sets the date field correctly", func() {
				album := mfs.ToAlbum()
				Expect(album.Date).To(Equal("1985-01-02"))
				Expect(album.MinYear).To(Equal(1985))
				Expect(album.MaxYear).To(Equal(1985))
			})
		})
	})
	Context("Calculated attributes", func() {
		Context("Discs", func() {
			When("we have no discs", func() {
				BeforeEach(func() {
					mfs = MediaFiles{{Album: "Album1"}, {Album: "Album1"}, {Album: "Album1"}}
				})
				It("sets the correct Discs", func() {
					album := mfs.ToAlbum()
					Expect(album.Discs).To(BeEmpty())
				})
			})
			When("we have only one disc", func() {
				BeforeEach(func() {
					mfs = MediaFiles{{DiscNumber: 1, DiscSubtitle: "DiscSubtitle"}}
				})
				It("sets the correct Discs", func() {
					album := mfs.ToAlbum()
					Expect(album.Discs).To(Equal(Discs{1: "DiscSubtitle"}))
				})
			})
			When("we have multiple discs", func() {
				BeforeEach(func() {
					mfs = MediaFiles{{DiscNumber: 1, DiscSubtitle: "DiscSubtitle"}, {DiscNumber: 2, DiscSubtitle: "DiscSubtitle2"}, {DiscNumber: 1, DiscSubtitle: "DiscSubtitle"}}
				})
				It("sets the correct Discs", func() {
					album := mfs.ToAlbum()
					Expect(album.Discs).To(Equal(Discs{1: "DiscSubtitle", 2: "DiscSubtitle2"}))
				})
			})
		})

		Context("Genres", func() {
			When("we have only one Genre", func() {
				BeforeEach(func() {
					mfs = MediaFiles{{Genres: Genres{{ID: "g1", Name: "Rock"}}}}
				})
				It("sets the correct Genre", func() {
					album := mfs.ToAlbum()
					Expect(album.Genre).To(Equal("Rock"))
					Expect(album.Genres).To(ConsistOf(Genre{ID: "g1", Name: "Rock"}))
				})
			})
			When("we have multiple Genres", func() {
				BeforeEach(func() {
					mfs = MediaFiles{{Genres: Genres{{ID: "g1", Name: "Rock"}, {ID: "g2", Name: "Punk"}, {ID: "g3", Name: "Alternative"}}}}
				})
				It("sets the correct Genre", func() {
					album := mfs.ToAlbum()
					Expect(album.Genre).To(Equal("Rock"))
					Expect(album.Genres).To(Equal(Genres{{ID: "g1", Name: "Rock"}, {ID: "g2", Name: "Punk"}, {ID: "g3", Name: "Alternative"}}))
				})
			})
			When("we have one predominant Genre", func() {
				var album Album
				BeforeEach(func() {
					mfs = MediaFiles{{Genres: Genres{{ID: "g2", Name: "Punk"}, {ID: "g1", Name: "Rock"}, {ID: "g2", Name: "Punk"}}}}
					album = mfs.ToAlbum()
				})
				It("sets the correct Genre", func() {
					Expect(album.Genre).To(Equal("Punk"))
				})
				It("removes duplications from Genres", func() {
					Expect(album.Genres).To(Equal(Genres{{ID: "g1", Name: "Rock"}, {ID: "g2", Name: "Punk"}}))
				})
			})
		})
		Context("Comments", func() {
			When("we have only one Comment", func() {
				BeforeEach(func() {
					mfs = MediaFiles{{Comment: "comment1"}}
				})
				It("sets the correct Comment", func() {
					album := mfs.ToAlbum()
					Expect(album.Comment).To(Equal("comment1"))
				})
			})
			When("we have multiple equal comments", func() {
				BeforeEach(func() {
					mfs = MediaFiles{{Comment: "comment1"}, {Comment: "comment1"}, {Comment: "comment1"}}
				})
				It("sets the correct Comment", func() {
					album := mfs.ToAlbum()
					Expect(album.Comment).To(Equal("comment1"))
				})
			})
			When("we have different comments", func() {
				BeforeEach(func() {
					mfs = MediaFiles{{Comment: "comment1"}, {Comment: "not the same"}, {Comment: "comment1"}}
				})
				It("sets the correct Genre", func() {
					album := mfs.ToAlbum()
					Expect(album.Comment).To(BeEmpty())
				})
			})
		})
		Context("AllArtistIds", func() {
			BeforeEach(func() {
				mfs = MediaFiles{
					{AlbumArtistID: "22", ArtistID: "11"},
					{AlbumArtistID: "22", ArtistID: "33"},
					{AlbumArtistID: "22", ArtistID: "11"},
				}
			})
			It("removes duplications", func() {
				album := mfs.ToAlbum()
				Expect(album.AllArtistIDs).To(Equal("11 22 33"))
			})
		})
		Context("FullText", func() {
			BeforeEach(func() {
				mfs = MediaFiles{
					{
						Album: "Album1", AlbumArtist: "AlbumArtist1", Artist: "Artist1", DiscSubtitle: "DiscSubtitle1",
						SortAlbumName: "SortAlbumName1", SortAlbumArtistName: "SortAlbumArtistName1", SortArtistName: "SortArtistName1",
					},
					{
						Album: "Album1", AlbumArtist: "AlbumArtist1", Artist: "Artist2", DiscSubtitle: "DiscSubtitle2",
						SortAlbumName: "SortAlbumName1", SortAlbumArtistName: "SortAlbumArtistName1", SortArtistName: "SortArtistName2",
					},
				}
			})
			It("fills the fullText attribute correctly", func() {
				album := mfs.ToAlbum()
				Expect(album.FullText).To(Equal(" album1 albumartist1 artist1 artist2 discsubtitle1 discsubtitle2 sortalbumartistname1 sortalbumname1 sortartistname1 sortartistname2"))
			})
		})
		Context("MbzAlbumID", func() {
			When("we have only one MbzAlbumID", func() {
				BeforeEach(func() {
					mfs = MediaFiles{{MbzAlbumID: "id1"}}
				})
				It("sets the correct MbzAlbumID", func() {
					album := mfs.ToAlbum()
					Expect(album.MbzAlbumID).To(Equal("id1"))
				})
			})
			When("we have multiple MbzAlbumID", func() {
				BeforeEach(func() {
					mfs = MediaFiles{{MbzAlbumID: "id1"}, {MbzAlbumID: "id2"}, {MbzAlbumID: "id1"}}
				})
				It("sets the correct MbzAlbumID", func() {
					album := mfs.ToAlbum()
					Expect(album.MbzAlbumID).To(Equal("id1"))
				})
			})
		})
	})
})

var _ = Describe("MediaFile", func() {
	BeforeEach(func() {
		DeferCleanup(configtest.SetupConfig())
		conf.Server.EnableMediaFileCoverArt = true
	})
	Describe(".CoverArtId()", func() {
		It("returns its own id if it HasCoverArt", func() {
			mf := MediaFile{ID: "111", AlbumID: "1", HasCoverArt: true}
			id := mf.CoverArtID()
			Expect(id.Kind).To(Equal(KindMediaFileArtwork))
			Expect(id.ID).To(Equal(mf.ID))
		})
		It("returns its album id if HasCoverArt is false", func() {
			mf := MediaFile{ID: "111", AlbumID: "1", HasCoverArt: false}
			id := mf.CoverArtID()
			Expect(id.Kind).To(Equal(KindAlbumArtwork))
			Expect(id.ID).To(Equal(mf.AlbumID))
		})
		It("returns its album id if EnableMediaFileCoverArt is disabled", func() {
			conf.Server.EnableMediaFileCoverArt = false
			mf := MediaFile{ID: "111", AlbumID: "1", HasCoverArt: true}
			id := mf.CoverArtID()
			Expect(id.Kind).To(Equal(KindAlbumArtwork))
			Expect(id.ID).To(Equal(mf.AlbumID))
		})
	})
})

func t(v string) time.Time {
	var timeFormats = []string{"2006-01-02", "2006-01-02 15:04", "2006-01-02 15:04:05", "2006-01-02T15:04:05", "2006-01-02T15:04", "2006-01-02 15:04:05.999999999 -0700 MST"}
	for _, f := range timeFormats {
		t, err := time.ParseInLocation(f, v, time.UTC)
		if err == nil {
			return t.UTC()
		}
	}
	return time.Time{}
}