From 8d4f4d8245dc4fe69358a48ab6231ca346081b75 Mon Sep 17 00:00:00 2001 From: Mark McDowall Date: Fri, 26 May 2023 17:13:58 -0700 Subject: [PATCH] New: Filter SABnzbd queue by category (cherry picked from commit 49fd9c4462741e756e0cd43c287939f929de11b2) Closes #3737 --- .../SabnzbdTests/SabnzbdFixture.cs | 2 +- .../Download/Clients/Sabnzbd/Sabnzbd.cs | 2 +- .../Download/Clients/Sabnzbd/SabnzbdProxy.cs | 15 ++++++++++----- 3 files changed, 12 insertions(+), 7 deletions(-) diff --git a/src/NzbDrone.Core.Test/Download/DownloadClientTests/SabnzbdTests/SabnzbdFixture.cs b/src/NzbDrone.Core.Test/Download/DownloadClientTests/SabnzbdTests/SabnzbdFixture.cs index 905d70496..5d814a7cb 100644 --- a/src/NzbDrone.Core.Test/Download/DownloadClientTests/SabnzbdTests/SabnzbdFixture.cs +++ b/src/NzbDrone.Core.Test/Download/DownloadClientTests/SabnzbdTests/SabnzbdFixture.cs @@ -175,7 +175,7 @@ namespace NzbDrone.Core.Test.Download.DownloadClientTests.SabnzbdTests } Mocker.GetMock() - .Setup(s => s.GetHistory(It.IsAny(), It.IsAny(), It.IsAny(), It.IsAny())) + .Setup(s => s.GetHistory(It.IsAny(), It.IsAny(), It.IsAny())) .Returns(history); } diff --git a/src/NzbDrone.Core/Download/Clients/Sabnzbd/Sabnzbd.cs b/src/NzbDrone.Core/Download/Clients/Sabnzbd/Sabnzbd.cs index f74515a9c..e73310d01 100644 --- a/src/NzbDrone.Core/Download/Clients/Sabnzbd/Sabnzbd.cs +++ b/src/NzbDrone.Core/Download/Clients/Sabnzbd/Sabnzbd.cs @@ -105,7 +105,7 @@ namespace NzbDrone.Core.Download.Clients.Sabnzbd private IEnumerable GetHistory() { - var sabHistory = _proxy.GetHistory(0, _configService.DownloadClientHistoryLimit, Settings.MusicCategory, Settings); + var sabHistory = _proxy.GetHistory(0, _configService.DownloadClientHistoryLimit, Settings); var historyItems = new List(); diff --git a/src/NzbDrone.Core/Download/Clients/Sabnzbd/SabnzbdProxy.cs b/src/NzbDrone.Core/Download/Clients/Sabnzbd/SabnzbdProxy.cs index 6b75df782..1c35a6e1c 100644 --- a/src/NzbDrone.Core/Download/Clients/Sabnzbd/SabnzbdProxy.cs +++ b/src/NzbDrone.Core/Download/Clients/Sabnzbd/SabnzbdProxy.cs @@ -19,7 +19,7 @@ namespace NzbDrone.Core.Download.Clients.Sabnzbd SabnzbdConfig GetConfig(SabnzbdSettings settings); SabnzbdFullStatus GetFullStatus(SabnzbdSettings settings); SabnzbdQueue GetQueue(int start, int limit, SabnzbdSettings settings); - SabnzbdHistory GetHistory(int start, int limit, string category, SabnzbdSettings settings); + SabnzbdHistory GetHistory(int start, int limit, SabnzbdSettings settings); string RetryDownload(string id, SabnzbdSettings settings); } @@ -46,7 +46,7 @@ namespace NzbDrone.Core.Download.Clients.Sabnzbd { var request = BuildRequest("addfile", settings).Post(); - request.AddQueryParam("cat", category); + request.AddQueryParam("cat", settings.MusicCategory); request.AddQueryParam("priority", priority); request.AddFormUpload("name", filename, nzbData, "application/x-nzb"); @@ -111,20 +111,25 @@ namespace NzbDrone.Core.Download.Clients.Sabnzbd request.AddQueryParam("start", start); request.AddQueryParam("limit", limit); + if (settings.MusicCategory.IsNotNullOrWhiteSpace()) + { + request.AddQueryParam("category", settings.MusicCategory); + } + var response = ProcessRequest(request, settings); return Json.Deserialize(JObject.Parse(response).SelectToken("queue").ToString()); } - public SabnzbdHistory GetHistory(int start, int limit, string category, SabnzbdSettings settings) + public SabnzbdHistory GetHistory(int start, int limit, SabnzbdSettings settings) { var request = BuildRequest("history", settings); request.AddQueryParam("start", start); request.AddQueryParam("limit", limit); - if (category.IsNotNullOrWhiteSpace()) + if (settings.MusicCategory.IsNotNullOrWhiteSpace()) { - request.AddQueryParam("category", category); + request.AddQueryParam("category", settings.MusicCategory); } var response = ProcessRequest(request, settings);