shpaq/Prowlarr-Prowlarr
1
0
Fork 0
mirror of https://github.com/Prowlarr/Prowlarr.git synced 2025-10-03 09:09:42 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fixed: (TorrentDay) Add freeleech only setting

Browse Source
This commit is contained in:
Bogdan
2023-01-18 05:36:49 +02:00
committed by Qstick
parent a0d11e7e33
commit edf9473e9a
2 changed files with 44 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/NzbDrone.Core/Datastore/Migration/026_torrentday_cookiesettings_to_torrentdaysettings.cs Normal file
View File

@@ -0,0 +1,14 @@
using FluentMigrator;
using NzbDrone.Core.Datastore.Migration.Framework;
namespace NzbDrone.Core.Datastore.Migration
{
[Migration(026)]
public class torrentday_cookiesettings_to_torrentdaysettings : NzbDroneMigrationBase
{
protected override void MainDbUpgrade()
{
Update.Table("Indexers").Set(new { ConfigContract = "TorrentDaySettings" }).Where(new { Implementation = "TorrentDay" });
}
}
}

55
src/NzbDrone.Core/Indexers/Definitions/TorrentDay.cs
View File

@@ -1,7 +1,6 @@
using System; using System;
using System.Collections.Generic; using System.Collections.Generic;
using System.Text; using System.Text;
using FluentValidation;
using Newtonsoft.Json; using Newtonsoft.Json;
using NLog; using NLog;
using NzbDrone.Common.Extensions; using NzbDrone.Common.Extensions;
@@ -13,15 +12,13 @@ using NzbDrone.Core.IndexerSearch.Definitions;
using NzbDrone.Core.Messaging.Events; using NzbDrone.Core.Messaging.Events;
using NzbDrone.Core.Parser; using NzbDrone.Core.Parser;
using NzbDrone.Core.Parser.Model; using NzbDrone.Core.Parser.Model;
using NzbDrone.Core.Validation;
namespace NzbDrone.Core.Indexers.Definitions namespace NzbDrone.Core.Indexers.Definitions
{ {
public class TorrentDay : TorrentIndexerBase<CookieTorrentBaseSettings> public class TorrentDay : TorrentIndexerBase<TorrentDaySettings>
{ {
public override string Name => "TorrentDay"; public override string Name => "TorrentDay";
public override string[] IndexerUrls => new[]
public override string[] IndexerUrls => new string[]
{ {
"https://torrentday.cool/", "https://torrentday.cool/",
"https://tday.love/", "https://tday.love/",
@@ -46,7 +43,7 @@ namespace NzbDrone.Core.Indexers.Definitions
public override IIndexerRequestGenerator GetRequestGenerator() public override IIndexerRequestGenerator GetRequestGenerator()
{ {
return new TorrentDayRequestGenerator() { Settings = Settings, Capabilities = Capabilities }; return new TorrentDayRequestGenerator { Settings = Settings, Capabilities = Capabilities };
} }
public override IParseIndexerResponse GetParser() public override IParseIndexerResponse GetParser()
@@ -64,21 +61,21 @@ namespace NzbDrone.Core.Indexers.Definitions
var caps = new IndexerCapabilities var caps = new IndexerCapabilities
{ {
TvSearchParams = new List<TvSearchParam> TvSearchParams = new List<TvSearchParam>
{ {
TvSearchParam.Q, TvSearchParam.Season, TvSearchParam.Ep, TvSearchParam.ImdbId TvSearchParam.Q, TvSearchParam.Season, TvSearchParam.Ep, TvSearchParam.ImdbId
}, },
MovieSearchParams = new List<MovieSearchParam> MovieSearchParams = new List<MovieSearchParam>
{ {
MovieSearchParam.Q, MovieSearchParam.ImdbId MovieSearchParam.Q, MovieSearchParam.ImdbId
}, },
MusicSearchParams = new List<MusicSearchParam> MusicSearchParams = new List<MusicSearchParam>
{ {
MusicSearchParam.Q MusicSearchParam.Q
}, },
BookSearchParams = new List<BookSearchParam> BookSearchParams = new List<BookSearchParam>
{ {
BookSearchParam.Q BookSearchParam.Q
} }
}; };
caps.Categories.AddCategoryMapping(29, NewznabStandardCategory.TVAnime, "Anime"); caps.Categories.AddCategoryMapping(29, NewznabStandardCategory.TVAnime, "Anime");
@@ -135,13 +132,9 @@ namespace NzbDrone.Core.Indexers.Definitions
public class TorrentDayRequestGenerator : IIndexerRequestGenerator public class TorrentDayRequestGenerator : IIndexerRequestGenerator
{ {
public CookieTorrentBaseSettings Settings { get; set; } public TorrentDaySettings Settings { get; set; }
public IndexerCapabilities Capabilities { get; set; } public IndexerCapabilities Capabilities { get; set; }
public TorrentDayRequestGenerator()
{
}
private IEnumerable<IndexerRequest> GetPagedRequests(string term, int[] categories, string imdbId = null) private IEnumerable<IndexerRequest> GetPagedRequests(string term, int[] categories, string imdbId = null)
{ {
var searchUrl = Settings.BaseUrl + "t.json"; var searchUrl = Settings.BaseUrl + "t.json";
@@ -154,6 +147,12 @@ namespace NzbDrone.Core.Indexers.Definitions
var catStr = string.Join(";", cats); var catStr = string.Join(";", cats);
searchUrl = searchUrl + "?" + catStr; searchUrl = searchUrl + "?" + catStr;
if (Settings.FreeLeechOnly)
{
searchUrl += ";free";
}
searchUrl += ";q="; searchUrl += ";q=";
if (imdbId.IsNotNullOrWhiteSpace()) if (imdbId.IsNotNullOrWhiteSpace())
@@ -219,10 +218,10 @@ namespace NzbDrone.Core.Indexers.Definitions
public class TorrentDayParser : IParseIndexerResponse public class TorrentDayParser : IParseIndexerResponse
{ {
private readonly CookieTorrentBaseSettings _settings; private readonly TorrentDaySettings _settings;
private readonly IndexerCapabilitiesCategories _categories; private readonly IndexerCapabilitiesCategories _categories;
public TorrentDayParser(CookieTorrentBaseSettings settings, IndexerCapabilitiesCategories categories) public TorrentDayParser(TorrentDaySettings settings, IndexerCapabilitiesCategories categories)
{ {
_settings = settings; _settings = settings;
_categories = categories; _categories = categories;
@@ -275,4 +274,10 @@ namespace NzbDrone.Core.Indexers.Definitions
public Action<IDictionary<string, string>, DateTime?> CookiesUpdater { get; set; } public Action<IDictionary<string, string>, DateTime?> CookiesUpdater { get; set; }
} }
public class TorrentDaySettings : CookieTorrentBaseSettings
{
[FieldDefinition(3, Label = "FreeLeech Only", Type = FieldType.Checkbox, HelpText = "Search Freeleech torrents only")]
public bool FreeLeechOnly { get; set; }
}
} }