Lidarr/src/Lidarr.Api.V1/Tracks/TrackResource.cs

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

namespace Lidarr.Api.V1.Tracks
{
    public class TrackResource : RestResource
    {
        public int ArtistId { get; set; }
        public int TrackFileId { get; set; }
        public int AlbumId { get; set; }
        public bool Explicit { get; set; }
        public int AbsoluteTrackNumber { get; set; }
        public string TrackNumber { get; set; }
        public string Title { get; set; }
        public int Duration { get; set; }
        public TrackFileResource TrackFile { get; set; }
        public int MediumNumber { get; set; }
        public bool HasFile { get; set; }

        public ArtistResource Artist { get; set; }
        public Ratings Ratings { 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 TrackResourceMapper
    {
        public static TrackResource ToResource(this Track model)
        {
            if (model == null) return null;

            return new TrackResource
            {
                Id = model.Id,

                ArtistId = model.Artist.Value.Id,
                TrackFileId = model.TrackFileId,
                AlbumId = model.AlbumId,
                Explicit = model.Explicit,
                AbsoluteTrackNumber = model.AbsoluteTrackNumber,
                TrackNumber = model.TrackNumber,
                Title = model.Title,
                Duration = model.Duration,
                MediumNumber = model.MediumNumber,
                HasFile = model.HasFile,
                Ratings = model.Ratings,
            };
        }

        public static List<TrackResource> ToResource(this IEnumerable<Track> models)
        {
            if (models == null) return null;

            return models.Select(ToResource).ToList();
        }
    }
}