Partially implemented ArtistRefreshCommand. In order to move forward, API may need to be switched to Spotify.

2026-04-21 22:04:31 -04:00 · 2017-05-07 11:58:24 -05:00
parent fa80bca2c9
commit b481bc6e45
14 changed files with 420 additions and 33 deletions
@@ -0,0 +1,125 @@
+using NLog;
+using NzbDrone.Common.Extensions;
+using NzbDrone.Core.Messaging.Events;
+using NzbDrone.Core.Music.Events;
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+
+namespace NzbDrone.Core.Music
+{
+    public interface IRefreshTrackService
+    {
+        void RefreshTrackInfo(Artist artist, IEnumerable<Track> remoteTracks);
+    }
+
+    public class RefreshTrackService : IRefreshTrackService
+    {
+        private readonly ITrackService _trackService;
+        private readonly IEventAggregator _eventAggregator;
+        private readonly Logger _logger;
+
+        public RefreshTrackService(ITrackService trackService, IEventAggregator eventAggregator, Logger logger)
+        {
+            _trackService = trackService;
+            _eventAggregator = eventAggregator;
+            _logger = logger;
+        }
+
+        public void RefreshTrackInfo(Artist artist, IEnumerable<Track> remoteTracks)
+        {
+            _logger.Info("Starting track info refresh for: {0}", artist);
+            var successCount = 0;
+            var failCount = 0;
+
+            var existingTracks = _trackService.GetTrackByArtist(artist.ItunesId);
+            var albums = artist.Albums;
+
+            var updateList = new List<Track>();
+            var newList = new List<Track>();
+            var dupeFreeRemoteTracks = remoteTracks.DistinctBy(m => new { m.AlbumId, m.TrackNumber }).ToList();
+
+            foreach (var track in OrderTracks(artist, dupeFreeRemoteTracks))
+            {
+                try
+                {
+                    var trackToUpdate = GetTrackToUpdate(artist, track, existingTracks);
+
+                    if (trackToUpdate != null)
+                    {
+                        existingTracks.Remove(trackToUpdate);
+                        updateList.Add(trackToUpdate);
+                    }
+                    else
+                    {
+                        trackToUpdate = new Track();
+                        trackToUpdate.Monitored = GetMonitoredStatus(track, albums);
+                        newList.Add(trackToUpdate);
+                    }
+                    trackToUpdate.ArtistId = artist.ItunesId; // TODO: Ensure LazyLoaded<Artist> field gets updated.
+                    trackToUpdate.TrackNumber = track.TrackNumber;
+                    trackToUpdate.Title = track.Title ?? "Unknown";
+                    
+                    // TODO: Implement rest of [RefreshTrackService] fields
+                    
+                    
+
+                    successCount++;
+                }
+                catch (Exception e)
+                {
+                    _logger.Fatal(e, "An error has occurred while updating track info for artist {0}. {1}", artist, track);
+                    failCount++;
+                }
+            }
+
+            var allTracks = new List<Track>();
+            allTracks.AddRange(newList);
+            allTracks.AddRange(updateList);
+
+            // TODO: See if anything needs to be done here
+            //AdjustMultiEpisodeAirTime(artist, allTracks);
+            //AdjustDirectToDvdAirDate(artist, allTracks);
+
+            _trackService.DeleteMany(existingTracks);
+            _trackService.UpdateMany(updateList);
+            _trackService.InsertMany(newList);
+
+            _eventAggregator.PublishEvent(new TrackInfoRefreshedEvent(artist, newList, updateList));
+
+            if (failCount != 0)
+            {
+                _logger.Info("Finished track refresh for artist: {0}. Successful: {1} - Failed: {2} ",
+                    artist.ArtistName, successCount, failCount);
+            }
+            else
+            {
+                _logger.Info("Finished track refresh for artist: {0}.", artist);
+            }
+        }
+
+        private bool GetMonitoredStatus(Track track, IEnumerable<Album> albums)
+        {
+            if (track.TrackNumber == 0 /*&& track.AlbumId != 1*/)
+            {
+                return false;
+            }
+
+            var album = albums.SingleOrDefault(c => c.AlbumId == track.AlbumId);
+            return album == null || album.Monitored;
+        }
+
+
+        private Track GetTrackToUpdate(Artist artist, Track track, List<Track> existingTracks)
+        {
+            return existingTracks.FirstOrDefault(e => e.AlbumId == track.AlbumId && e.TrackNumber == track.TrackNumber);
+        }
+
+        private IEnumerable<Track> OrderTracks(Artist artist, List<Track> tracks)
+        {
+            return tracks.OrderBy(e => e.AlbumId).ThenBy(e => e.TrackNumber);
+        }
+    }
+}
+