Lidarr/src/Lidarr.Api.V1/Albums/AlbumResource.cs

using System;
using System.Collections.Generic;
using System.Linq;
using Lidarr.Api.V1.Artist;
using Lidarr.Http.REST;
using Newtonsoft.Json;
using NzbDrone.Core.MediaCover;
using NzbDrone.Core.Music;

namespace Lidarr.Api.V1.Albums
{
    public class AlbumResource : RestResource
    {
        public string Title { get; set; }
        public string Disambiguation { get; set; }
        public string Overview { get; set; }
        public int ArtistId { get; set; }
        public string ForeignAlbumId { get; set; }
        public bool Monitored { get; set; }
        public bool AnyReleaseOk { get; set; }
        public int ProfileId { get; set; }
        public int Duration { get; set; }
        public string AlbumType { get; set; }
        public List<string> SecondaryTypes { get; set; }
        public int MediumCount
        {
            get
            {
                if (Media == null)
                {
                    return 0;
                }

                return Media.Where(s => s.MediumNumber > 0).Count();
            }
        }

        public Ratings Ratings { get; set; }
        public DateTime? ReleaseDate { get; set; }
        public List<AlbumReleaseResource> Releases { get; set; }
        public List<string> Genres { get; set; }
        public List<MediumResource> Media { get; set; }
        public ArtistResource Artist { get; set; }
        public List<MediaCover> Images { get; set; }
        public List<Links> Links { get; set; }
        public AlbumStatisticsResource Statistics { get; set; }
        public AddAlbumOptions AddOptions { get; set; }
        public string RemoteCover { get; set; }

        // Hiding this so people don't think its usable (only used to set the initial state)
        [JsonProperty(DefaultValueHandling = DefaultValueHandling.Ignore)]
        public bool Grabbed { get; set; }
    }

    public static class AlbumResourceMapper
    {
        public static AlbumResource ToResource(this Album model)
        {
            if (model == null)
            {
                return null;
            }

            var selectedRelease = model.AlbumReleases?.Value.Where(x => x.Monitored).SingleOrDefault();

            return new AlbumResource
            {
                Id = model.Id,
                ArtistId = model.ArtistId,
                ForeignAlbumId = model.ForeignAlbumId,
                ProfileId = model.ProfileId,
                Monitored = model.Monitored,
                AnyReleaseOk = model.AnyReleaseOk,
                ReleaseDate = model.ReleaseDate,
                Genres = model.Genres,
                Title = model.Title,
                Disambiguation = model.Disambiguation,
                Overview = model.Overview,
                Images = model.Images,
                Links = model.Links,
                Ratings = model.Ratings,
                Duration = selectedRelease?.Duration ?? 0,
                AlbumType = model.AlbumType,
                SecondaryTypes = model.SecondaryTypes.Select(s => s.Name).ToList(),
                Releases = model.AlbumReleases?.Value.ToResource() ?? new List<AlbumReleaseResource>(),
                Media = selectedRelease?.Media.ToResource() ?? new List<MediumResource>(),
                Artist = model.Artist?.Value.ToResource()
            };
        }

        public static Album ToModel(this AlbumResource resource)
        {
            if (resource == null)
            {
                return null;
            }

            var artist = resource.Artist?.ToModel() ?? new NzbDrone.Core.Music.Artist();

            return new Album
            {
                Id = resource.Id,
                ForeignAlbumId = resource.ForeignAlbumId,
                Title = resource.Title,
                Disambiguation = resource.Disambiguation,
                Overview = resource.Overview,
                Images = resource.Images,
                AlbumType = resource.AlbumType,
                Monitored = resource.Monitored,
                AnyReleaseOk = resource.AnyReleaseOk,
                AlbumReleases = resource.Releases.ToModel(),
                AddOptions = resource.AddOptions,
                Artist = artist,
                ArtistMetadata = artist.Metadata.Value
            };
        }

        public static Album ToModel(this AlbumResource resource, Album album)
        {
            var updatedAlbum = resource.ToModel();

            album.ApplyChanges(updatedAlbum);
            album.AlbumReleases = updatedAlbum.AlbumReleases;

            return album;
        }

        public static List<AlbumResource> ToResource(this IEnumerable<Album> models)
        {
            return models?.Select(ToResource).ToList();
        }

        public static List<Album> ToModel(this IEnumerable<AlbumResource> resources)
        {
            return resources.Select(ToModel).ToList();
        }
    }
}