New: Lidarr to Readarr

2026-04-23 22:25:09 -04:00 · 2020-02-29 15:51:29 -05:00
parent 7359c2a9fa
commit 3b7eb01918
565 changed files with 1669 additions and 4272 deletions
@@ -0,0 +1,90 @@
+using System.Collections.Generic;
+using System.Linq;
+using Nancy;
+using NLog;
+using NzbDrone.Core.MediaFiles;
+using NzbDrone.Core.MediaFiles.TrackImport.Manual;
+using NzbDrone.Core.Music;
+using NzbDrone.Core.Qualities;
+using Readarr.Http;
+using Readarr.Http.Extensions;
+
+namespace Readarr.Api.V1.ManualImport
+{
+    public class ManualImportModule : ReadarrRestModule<ManualImportResource>
+    {
+        private readonly IArtistService _artistService;
+        private readonly IAlbumService _albumService;
+        private readonly IReleaseService _releaseService;
+        private readonly IManualImportService _manualImportService;
+        private readonly Logger _logger;
+
+        public ManualImportModule(IManualImportService manualImportService,
+                                  IArtistService artistService,
+                                  IAlbumService albumService,
+                                  IReleaseService releaseService,
+                                  Logger logger)
+        {
+            _artistService = artistService;
+            _albumService = albumService;
+            _releaseService = releaseService;
+            _manualImportService = manualImportService;
+            _logger = logger;
+
+            GetResourceAll = GetMediaFiles;
+
+            Put("/", options =>
+                {
+                    var resource = Request.Body.FromJson<List<ManualImportResource>>();
+                    return ResponseWithCode(UpdateImportItems(resource), HttpStatusCode.Accepted);
+                });
+        }
+
+        private List<ManualImportResource> GetMediaFiles()
+        {
+            var folder = (string)Request.Query.folder;
+            var downloadId = (string)Request.Query.downloadId;
+            var filter = Request.GetBooleanQueryParameter("filterExistingFiles", true) ? FilterFilesType.Matched : FilterFilesType.None;
+            var replaceExistingFiles = Request.GetBooleanQueryParameter("replaceExistingFiles", true);
+
+            return _manualImportService.GetMediaFiles(folder, downloadId, filter, replaceExistingFiles).ToResource().Select(AddQualityWeight).ToList();
+        }
+
+        private ManualImportResource AddQualityWeight(ManualImportResource item)
+        {
+            if (item.Quality != null)
+            {
+                item.QualityWeight = Quality.DefaultQualityDefinitions.Single(q => q.Quality == item.Quality.Quality).Weight;
+                item.QualityWeight += item.Quality.Revision.Real * 10;
+                item.QualityWeight += item.Quality.Revision.Version;
+            }
+
+            return item;
+        }
+
+        private List<ManualImportResource> UpdateImportItems(List<ManualImportResource> resources)
+        {
+            var items = new List<ManualImportItem>();
+            foreach (var resource in resources)
+            {
+                items.Add(new ManualImportItem
+                {
+                    Id = resource.Id,
+                    Path = resource.Path,
+                    Name = resource.Name,
+                    Size = resource.Size,
+                    Artist = resource.Artist == null ? null : _artistService.GetArtist(resource.Artist.Id),
+                    Album = resource.Album == null ? null : _albumService.GetAlbum(resource.Album.Id),
+                    Release = resource.AlbumReleaseId == 0 ? null : _releaseService.GetRelease(resource.AlbumReleaseId),
+                    Quality = resource.Quality,
+                    DownloadId = resource.DownloadId,
+                    AdditionalFile = resource.AdditionalFile,
+                    ReplaceExistingFiles = resource.ReplaceExistingFiles,
+                    DisableReleaseSwitching = resource.DisableReleaseSwitching
+                });
+            }
+
+            return _manualImportService.UpdateItems(items).Select(x => x.ToResource()).ToList();
+        }
+    }
+}
@@ -0,0 +1,69 @@
+using System.Collections.Generic;
+using System.Linq;
+using NzbDrone.Core.DecisionEngine;
+using NzbDrone.Core.MediaFiles.TrackImport.Manual;
+using NzbDrone.Core.Parser.Model;
+using NzbDrone.Core.Qualities;
+using Readarr.Api.V1.Albums;
+using Readarr.Api.V1.Artist;
+using Readarr.Api.V1.Tracks;
+using Readarr.Http.REST;
+
+namespace Readarr.Api.V1.ManualImport
+{
+    public class ManualImportResource : RestResource
+    {
+        public string Path { get; set; }
+        public string Name { get; set; }
+        public long Size { get; set; }
+        public ArtistResource Artist { get; set; }
+        public AlbumResource Album { get; set; }
+        public int AlbumReleaseId { get; set; }
+        public List<TrackResource> Tracks { get; set; }
+        public QualityModel Quality { get; set; }
+        public int QualityWeight { get; set; }
+        public string DownloadId { get; set; }
+        public IEnumerable<Rejection> Rejections { get; set; }
+        public ParsedTrackInfo AudioTags { get; set; }
+        public bool AdditionalFile { get; set; }
+        public bool ReplaceExistingFiles { get; set; }
+        public bool DisableReleaseSwitching { get; set; }
+    }
+
+    public static class ManualImportResourceMapper
+    {
+        public static ManualImportResource ToResource(this ManualImportItem model)
+        {
+            if (model == null)
+            {
+                return null;
+            }
+
+            return new ManualImportResource
+            {
+                Id = model.Id,
+                Path = model.Path,
+                Name = model.Name,
+                Size = model.Size,
+                Artist = model.Artist.ToResource(),
+                Album = model.Album.ToResource(),
+                AlbumReleaseId = model.Release?.Id ?? 0,
+                Tracks = model.Tracks.ToResource(),
+                Quality = model.Quality,
+
+                //QualityWeight
+                DownloadId = model.DownloadId,
+                Rejections = model.Rejections,
+                AudioTags = model.Tags,
+                AdditionalFile = model.AdditionalFile,
+                ReplaceExistingFiles = model.ReplaceExistingFiles,
+                DisableReleaseSwitching = model.DisableReleaseSwitching
+            };
+        }
+
+        public static List<ManualImportResource> ToResource(this IEnumerable<ManualImportItem> models)
+        {
+            return models.Select(ToResource).ToList();
+        }
+    }
+}