ErsatzTV/ErsatzTV.Core/Metadata/SmartCollectionBuilder.cs

using System.Collections.Generic;
using System.Linq;
using System.Threading.Tasks;
using ErsatzTV.Core.Domain;
using ErsatzTV.Core.Interfaces.Metadata;
using ErsatzTV.Core.Interfaces.Repositories;
using LanguageExt;
using static LanguageExt.Prelude;

namespace ErsatzTV.Core.Metadata
{
    public class SmartCollectionBuilder : ISmartCollectionBuilder
    {
        private readonly IMediaCollectionRepository _mediaCollectionRepository;

        public SmartCollectionBuilder(IMediaCollectionRepository mediaCollectionRepository) =>
            _mediaCollectionRepository = mediaCollectionRepository;

        public async Task RefreshSmartCollections(MediaItem mediaItem)
        {
            foreach (TelevisionMediaCollection collection in GetTelevisionCollections(mediaItem))
            {
                await _mediaCollectionRepository.InsertOrIgnore(collection);
            }
        }

        private IEnumerable<TelevisionMediaCollection> GetTelevisionCollections(MediaItem mediaItem)
        {
            IList<MediaItem> televisionMediaItems = new[] { mediaItem }
                .Where(c => c.Metadata.MediaType == MediaType.TvShow)
                .ToList();

            IEnumerable<TelevisionMediaCollection> televisionShowCollections = televisionMediaItems
                .Map(c => c.Metadata.Title)
                .Distinct().Map(
                    t => new TelevisionMediaCollection
                    {
                        Name = $"{t} - All Seasons",
                        ShowTitle = t,
                        SeasonNumber = null
                    });

            IEnumerable<TelevisionMediaCollection> televisionShowSeasonCollections = televisionMediaItems
                .Map(c => new { c.Metadata.Title, c.Metadata.SeasonNumber }).Distinct()
                .Map(
                    ts =>
                    {
                        return Optional(ts.SeasonNumber).Map(
                            sn => new TelevisionMediaCollection
                            {
                                Name = $"{ts.Title} - Season {sn:00}",
                                ShowTitle = ts.Title,
                                SeasonNumber = sn
                            });
                    })
                .Sequence().Flatten();

            return Seq(televisionShowCollections, televisionShowSeasonCollections).Flatten();
        }
    }
}