Add Metadata Profiles (#132)

* Add Metadata Profiles * fixup! Codacy
2026-04-20 21:54:25 -04:00 · 2017-11-25 22:51:37 -05:00
parent dd11f74073
commit 5b7339cd73
92 changed files with 2611 additions and 145 deletions
@@ -0,0 +1,102 @@
+using NLog;
+using NzbDrone.Core.Lifecycle;
+using NzbDrone.Core.Messaging.Events;
+using NzbDrone.Core.Music;
+using System.Collections.Generic;
+using System.Linq;
+
+namespace NzbDrone.Core.Profiles.Metadata
+{
+    public interface IMetadataProfileService
+    {
+        MetadataProfile Add(MetadataProfile profile);
+        void Update(MetadataProfile profile);
+        void Delete(int id);
+        List<MetadataProfile> All();
+        MetadataProfile Get(int id);
+        bool Exists(int id);
+    }
+
+    public class MetadataProfileService : IMetadataProfileService, IHandle<ApplicationStartedEvent>
+    {
+        private readonly IMetadataProfileRepository _profileRepository;
+        private readonly IArtistService _artistService;
+        private readonly Logger _logger;
+
+        public MetadataProfileService(IMetadataProfileRepository profileRepository, IArtistService artistService, Logger logger)
+        {
+            _profileRepository = profileRepository;
+            _artistService = artistService;
+            _logger = logger;
+        }
+
+        public MetadataProfile Add(MetadataProfile profile)
+        {
+            return _profileRepository.Insert(profile);
+        }
+
+        public void Update(MetadataProfile profile)
+        {
+            _profileRepository.Update(profile);
+        }
+
+        public void Delete(int id)
+        {
+            if (_artistService.GetAllArtists().Any(c => c.MetadataProfileId == id))
+            {
+                throw new MetadataProfileInUseException(id);
+            }
+
+            _profileRepository.Delete(id);
+        }
+
+        public List<MetadataProfile> All()
+        {
+            return _profileRepository.All().ToList();
+        }
+
+        public MetadataProfile Get(int id)
+        {
+            return _profileRepository.Get(id);
+        }
+
+        public bool Exists(int id)
+        {
+            return _profileRepository.Exists(id);
+        }
+
+        private void AddDefaultProfile(string name, List<PrimaryAlbumType> primAllowed, List<SecondaryAlbumType> secAllowed)
+        {
+            var primaryTypes = PrimaryAlbumType.All
+                                    .OrderByDescending(l => l.Name)
+                                    .Select(v => new ProfilePrimaryAlbumTypeItem { PrimaryAlbumType = v, Allowed = primAllowed.Contains(v) })
+                                    .ToList();
+
+            var secondaryTypes = SecondaryAlbumType.All
+                                    .OrderByDescending(l => l.Name)
+                                    .Select(v => new ProfileSecondaryAlbumTypeItem { SecondaryAlbumType = v, Allowed = secAllowed.Contains(v) })
+                                    .ToList();
+
+            var profile = new MetadataProfile
+            {
+                Name = name,
+                PrimaryAlbumTypes = primaryTypes,
+                SecondaryAlbumTypes = secondaryTypes
+            };
+
+            Add(profile);
+        }
+
+        public void Handle(ApplicationStartedEvent message)
+        {
+            if (All().Any())
+            {
+                return;
+            }
+
+            _logger.Info("Setting up default metadata profile");
+
+            AddDefaultProfile("Standard", new List<PrimaryAlbumType>{PrimaryAlbumType.Album}, new List<SecondaryAlbumType>{ SecondaryAlbumType.Studio });
+        }
+    }
+}