You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
139 lines
5.6 KiB
139 lines
5.6 KiB
using System;
|
|
using System.Collections.Generic;
|
|
using System.Globalization;
|
|
using System.Linq;
|
|
using System.Net;
|
|
using NzbDrone.Common.Extensions;
|
|
using NzbDrone.Common.Http;
|
|
using NzbDrone.Common.Serializer;
|
|
using NzbDrone.Core.Indexers.Exceptions;
|
|
using NzbDrone.Core.Parser;
|
|
using NzbDrone.Core.Parser.Model;
|
|
|
|
namespace NzbDrone.Core.Indexers.Definitions.Avistaz
|
|
{
|
|
public class AvistazParserBase : IParseIndexerResponse
|
|
{
|
|
protected virtual string TimezoneOffset => "-04:00"; // Avistaz does not specify a timezone & returns server time
|
|
private readonly HashSet<string> _hdResolutions = new () { "1080p", "1080i", "720p" };
|
|
|
|
public Action<IDictionary<string, string>, DateTime?> CookiesUpdater { get; set; }
|
|
|
|
public IList<ReleaseInfo> ParseResponse(IndexerResponse indexerResponse)
|
|
{
|
|
var releaseInfos = new List<ReleaseInfo>();
|
|
|
|
if (indexerResponse.HttpResponse.StatusCode == HttpStatusCode.NotFound)
|
|
{
|
|
return releaseInfos.ToArray();
|
|
}
|
|
|
|
if (indexerResponse.HttpResponse.StatusCode == HttpStatusCode.TooManyRequests)
|
|
{
|
|
throw new RequestLimitReachedException(indexerResponse, "API Request Limit Reached");
|
|
}
|
|
|
|
if (indexerResponse.HttpResponse.StatusCode != HttpStatusCode.OK)
|
|
{
|
|
throw new IndexerException(indexerResponse, $"Unexpected response status {indexerResponse.HttpResponse.StatusCode} code from indexer request");
|
|
}
|
|
|
|
if (!indexerResponse.HttpResponse.Headers.ContentType.Contains(HttpAccept.Json.Value))
|
|
{
|
|
throw new IndexerException(indexerResponse, $"Unexpected response header {indexerResponse.HttpResponse.Headers.ContentType} from indexer request, expected {HttpAccept.Json.Value}");
|
|
}
|
|
|
|
var jsonResponse = STJson.Deserialize<AvistazResponse>(indexerResponse.HttpResponse.Content);
|
|
|
|
foreach (var row in jsonResponse.Data)
|
|
{
|
|
var details = row.Url;
|
|
var link = row.Download;
|
|
|
|
var cats = ParseCategories(row);
|
|
|
|
var release = new TorrentInfo
|
|
{
|
|
Title = row.FileName,
|
|
DownloadUrl = link,
|
|
InfoHash = row.InfoHash,
|
|
InfoUrl = details,
|
|
Guid = details,
|
|
Categories = cats,
|
|
PublishDate = DateTime.Parse($"{row.CreatedAt} {TimezoneOffset}", CultureInfo.InvariantCulture, DateTimeStyles.AdjustToUniversal),
|
|
Size = row.FileSize,
|
|
Files = row.FileCount,
|
|
Grabs = row.Completed,
|
|
Seeders = row.Seed,
|
|
Peers = row.Leech + row.Seed,
|
|
DownloadVolumeFactor = row.DownloadMultiply,
|
|
UploadVolumeFactor = row.UploadMultiply,
|
|
MinimumRatio = 1,
|
|
MinimumSeedTime = 259200, // 72 hours
|
|
Languages = row.Audio?.Select(x => x.Language).ToList() ?? new List<string>(),
|
|
Subs = row.Subtitle?.Select(x => x.Language).ToList() ?? new List<string>()
|
|
};
|
|
|
|
if (row.FileSize is > 0)
|
|
{
|
|
var sizeGigabytes = row.FileSize.Value / Math.Pow(1024, 3);
|
|
|
|
release.MinimumSeedTime = sizeGigabytes switch
|
|
{
|
|
> 50.0 => (long)((100 * Math.Log(sizeGigabytes)) - 219.2023) * 3600,
|
|
_ => 259200 + (long)(sizeGigabytes * 7200)
|
|
};
|
|
}
|
|
|
|
if (row.MovieTvinfo != null)
|
|
{
|
|
release.ImdbId = ParseUtil.GetImdbId(row.MovieTvinfo.Imdb).GetValueOrDefault();
|
|
release.TmdbId = row.MovieTvinfo.Tmdb.IsNullOrWhiteSpace() ? 0 : ParseUtil.TryCoerceInt(row.MovieTvinfo.Tmdb, out var tmdbResult) ? tmdbResult : 0;
|
|
release.TvdbId = row.MovieTvinfo.Tvdb.IsNullOrWhiteSpace() ? 0 : ParseUtil.TryCoerceInt(row.MovieTvinfo.Tvdb, out var tvdbResult) ? tvdbResult : 0;
|
|
}
|
|
|
|
releaseInfos.Add(release);
|
|
}
|
|
|
|
// order by date
|
|
return releaseInfos
|
|
.OrderByDescending(o => o.PublishDate)
|
|
.ToArray();
|
|
}
|
|
|
|
// hook to adjust category parsing
|
|
protected virtual List<IndexerCategory> ParseCategories(AvistazRelease row)
|
|
{
|
|
var cats = new List<IndexerCategory>();
|
|
var resolution = row.VideoQuality;
|
|
|
|
switch (row.Type)
|
|
{
|
|
case "Movie":
|
|
cats.Add(resolution switch
|
|
{
|
|
var res when _hdResolutions.Contains(res) => NewznabStandardCategory.MoviesHD,
|
|
"2160p" => NewznabStandardCategory.MoviesUHD,
|
|
_ => NewznabStandardCategory.MoviesSD
|
|
});
|
|
break;
|
|
case "TV-Show":
|
|
cats.Add(resolution switch
|
|
{
|
|
var res when _hdResolutions.Contains(res) => NewznabStandardCategory.TVHD,
|
|
"2160p" => NewznabStandardCategory.TVUHD,
|
|
_ => NewznabStandardCategory.TVSD
|
|
});
|
|
break;
|
|
case "Music":
|
|
cats.Add(NewznabStandardCategory.Audio);
|
|
break;
|
|
default:
|
|
throw new Exception($"Error parsing Avistaz category type {row.Type}");
|
|
}
|
|
|
|
return cats;
|
|
}
|
|
}
|
|
}
|