|
@@ -1,8 +1,10 @@
|
|
|
-using MediaBrowser.Common.Net;
|
|
|
+using MediaBrowser.Common.Extensions;
|
|
|
+using MediaBrowser.Common.Net;
|
|
|
using MediaBrowser.Controller.Providers;
|
|
|
+using MediaBrowser.Controller.Subtitles;
|
|
|
using MediaBrowser.Model.Entities;
|
|
|
using MediaBrowser.Model.Logging;
|
|
|
-using MediaBrowser.Model.MediaInfo;
|
|
|
+using MediaBrowser.Model.Providers;
|
|
|
using OpenSubtitlesHandler;
|
|
|
using System;
|
|
|
using System.Collections.Generic;
|
|
@@ -20,9 +22,9 @@ namespace MediaBrowser.Providers.Subtitles
|
|
|
private readonly IHttpClient _httpClient;
|
|
|
private readonly CultureInfo _usCulture = new CultureInfo("en-US");
|
|
|
|
|
|
- public OpenSubtitleDownloader(ILogger logger, IHttpClient httpClient)
|
|
|
+ public OpenSubtitleDownloader(ILogManager logManager, IHttpClient httpClient)
|
|
|
{
|
|
|
- _logger = logger;
|
|
|
+ _logger = logManager.GetLogger(GetType().Name);
|
|
|
_httpClient = httpClient;
|
|
|
}
|
|
|
|
|
@@ -36,39 +38,71 @@ namespace MediaBrowser.Providers.Subtitles
|
|
|
get { return new[] { SubtitleMediaType.Episode, SubtitleMediaType.Movie }; }
|
|
|
}
|
|
|
|
|
|
- public Task<SubtitleResponse> GetSubtitles(SubtitleRequest request, CancellationToken cancellationToken)
|
|
|
+ public Task<SubtitleResponse> GetSubtitles(string id, CancellationToken cancellationToken)
|
|
|
{
|
|
|
- return GetSubtitlesInternal(request, cancellationToken);
|
|
|
+ return GetSubtitlesInternal(id, cancellationToken);
|
|
|
}
|
|
|
|
|
|
- private async Task<SubtitleResponse> GetSubtitlesInternal(SubtitleRequest request,
|
|
|
+ private async Task<SubtitleResponse> GetSubtitlesInternal(string id,
|
|
|
CancellationToken cancellationToken)
|
|
|
{
|
|
|
- var response = new SubtitleResponse();
|
|
|
+ if (string.IsNullOrWhiteSpace(id))
|
|
|
+ {
|
|
|
+ throw new ArgumentNullException("id");
|
|
|
+ }
|
|
|
+
|
|
|
+ var idParts = id.Split(new[] { '-' }, 3);
|
|
|
+
|
|
|
+ var format = idParts[0];
|
|
|
+ var language = idParts[1];
|
|
|
+ var ossId = idParts[2];
|
|
|
+
|
|
|
+ var downloadsList = new[] { int.Parse(ossId, _usCulture) };
|
|
|
+
|
|
|
+ var resultDownLoad = OpenSubtitles.DownloadSubtitles(downloadsList);
|
|
|
+ if (!(resultDownLoad is MethodResponseSubtitleDownload))
|
|
|
+ {
|
|
|
+ throw new ApplicationException("Invalid response type");
|
|
|
+ }
|
|
|
|
|
|
+ var res = ((MethodResponseSubtitleDownload)resultDownLoad).Results.First();
|
|
|
+ var data = Convert.FromBase64String(res.Data);
|
|
|
+
|
|
|
+ return new SubtitleResponse
|
|
|
+ {
|
|
|
+ Format = format,
|
|
|
+ Language = language,
|
|
|
+
|
|
|
+ Stream = new MemoryStream(Utilities.Decompress(new MemoryStream(data)))
|
|
|
+ };
|
|
|
+ }
|
|
|
+
|
|
|
+ public async Task<IEnumerable<RemoteSubtitleInfo>> SearchSubtitles(SubtitleSearchRequest request, CancellationToken cancellationToken)
|
|
|
+ {
|
|
|
var imdbIdText = request.GetProviderId(MetadataProviders.Imdb);
|
|
|
long imdbId;
|
|
|
|
|
|
if (string.IsNullOrWhiteSpace(imdbIdText) ||
|
|
|
- long.TryParse(imdbIdText.TrimStart('t'), NumberStyles.Any, _usCulture, out imdbId))
|
|
|
+ !long.TryParse(imdbIdText.TrimStart('t'), NumberStyles.Any, _usCulture, out imdbId))
|
|
|
{
|
|
|
- return response;
|
|
|
+ _logger.Debug("Imdb id missing");
|
|
|
+ return new List<RemoteSubtitleInfo>();
|
|
|
}
|
|
|
-
|
|
|
+
|
|
|
switch (request.ContentType)
|
|
|
{
|
|
|
case SubtitleMediaType.Episode:
|
|
|
if (!request.IndexNumber.HasValue || !request.ParentIndexNumber.HasValue || string.IsNullOrEmpty(request.SeriesName))
|
|
|
{
|
|
|
- _logger.Debug("Information Missing");
|
|
|
- return response;
|
|
|
+ _logger.Debug("Episode information missing");
|
|
|
+ return new List<RemoteSubtitleInfo>();
|
|
|
}
|
|
|
break;
|
|
|
case SubtitleMediaType.Movie:
|
|
|
if (string.IsNullOrEmpty(request.Name))
|
|
|
{
|
|
|
- _logger.Debug("Information Missing");
|
|
|
- return response;
|
|
|
+ _logger.Debug("Movie name missing");
|
|
|
+ return new List<RemoteSubtitleInfo>();
|
|
|
}
|
|
|
break;
|
|
|
}
|
|
@@ -76,16 +110,18 @@ namespace MediaBrowser.Providers.Subtitles
|
|
|
if (string.IsNullOrEmpty(request.MediaPath))
|
|
|
{
|
|
|
_logger.Debug("Path Missing");
|
|
|
- return response;
|
|
|
+ return new List<RemoteSubtitleInfo>();
|
|
|
}
|
|
|
|
|
|
Utilities.HttpClient = _httpClient;
|
|
|
OpenSubtitles.SetUserAgent("OS Test User Agent");
|
|
|
- var loginResponse = OpenSubtitles.LogIn("", "", "en");
|
|
|
+
|
|
|
+ var loginResponse = await OpenSubtitles.LogInAsync("", "", "en", cancellationToken).ConfigureAwait(false);
|
|
|
+
|
|
|
if (!(loginResponse is MethodResponseLogIn))
|
|
|
{
|
|
|
_logger.Debug("Login error");
|
|
|
- return response;
|
|
|
+ return new List<RemoteSubtitleInfo>();
|
|
|
}
|
|
|
|
|
|
var subLanguageId = request.Language;
|
|
@@ -105,54 +141,42 @@ namespace MediaBrowser.Providers.Subtitles
|
|
|
var result = OpenSubtitles.SearchSubtitles(parms.ToArray());
|
|
|
if (!(result is MethodResponseSubtitleSearch))
|
|
|
{
|
|
|
- _logger.Debug("invalid response type");
|
|
|
- return null;
|
|
|
+ _logger.Debug("Invalid response type");
|
|
|
+ return new List<RemoteSubtitleInfo>();
|
|
|
}
|
|
|
|
|
|
Predicate<SubtitleSearchResult> mediaFilter =
|
|
|
x =>
|
|
|
request.ContentType == SubtitleMediaType.Episode
|
|
|
- ? int.Parse(x.SeriesSeason) == request.ParentIndexNumber && int.Parse(x.SeriesEpisode) == request.IndexNumber
|
|
|
- : long.Parse(x.IDMovieImdb) == imdbId;
|
|
|
+ ? int.Parse(x.SeriesSeason, _usCulture) == request.ParentIndexNumber && int.Parse(x.SeriesEpisode, _usCulture) == request.IndexNumber
|
|
|
+ : long.Parse(x.IDMovieImdb, _usCulture) == imdbId;
|
|
|
|
|
|
var results = ((MethodResponseSubtitleSearch)result).Results;
|
|
|
- var bestResult = results.Where(x => x.SubBad == "0" && mediaFilter(x))
|
|
|
- .OrderBy(x => x.MovieHash == hash)
|
|
|
- .ThenBy(x => Math.Abs(long.Parse(x.MovieByteSize) - movieByteSize))
|
|
|
- .ThenByDescending(x => int.Parse(x.SubDownloadsCnt))
|
|
|
- .ThenByDescending(x => double.Parse(x.SubRating))
|
|
|
- .ToList();
|
|
|
-
|
|
|
- if (!bestResult.Any())
|
|
|
- {
|
|
|
- _logger.Debug("No Subtitles");
|
|
|
- return response;
|
|
|
- }
|
|
|
-
|
|
|
- _logger.Debug("Found " + bestResult.Count + " subtitles.");
|
|
|
|
|
|
- var subtitle = bestResult.First();
|
|
|
- var downloadsList = new[] { int.Parse(subtitle.IDSubtitleFile) };
|
|
|
+ // Avoid implicitly captured closure
|
|
|
+ var hasCopy = hash;
|
|
|
|
|
|
- var resultDownLoad = OpenSubtitles.DownloadSubtitles(downloadsList);
|
|
|
- if (!(resultDownLoad is MethodResponseSubtitleDownload))
|
|
|
- {
|
|
|
- _logger.Debug("invalid response type");
|
|
|
- return response;
|
|
|
- }
|
|
|
- if (!((MethodResponseSubtitleDownload)resultDownLoad).Results.Any())
|
|
|
- {
|
|
|
- _logger.Debug("No Subtitle Downloads");
|
|
|
- return response;
|
|
|
- }
|
|
|
-
|
|
|
- var res = ((MethodResponseSubtitleDownload)resultDownLoad).Results.First();
|
|
|
- var data = Convert.FromBase64String(res.Data);
|
|
|
-
|
|
|
- response.HasContent = true;
|
|
|
- response.Format = subtitle.SubFormat.ToUpper();
|
|
|
- response.Stream = new MemoryStream(Utilities.Decompress(new MemoryStream(data)));
|
|
|
- return response;
|
|
|
+ return results.Where(x => x.SubBad == "0" && mediaFilter(x))
|
|
|
+ .OrderBy(x => x.MovieHash == hash)
|
|
|
+ .ThenBy(x => Math.Abs(long.Parse(x.MovieByteSize, _usCulture) - movieByteSize))
|
|
|
+ .ThenByDescending(x => int.Parse(x.SubDownloadsCnt, _usCulture))
|
|
|
+ .ThenByDescending(x => double.Parse(x.SubRating, _usCulture))
|
|
|
+ .Select(i => new RemoteSubtitleInfo
|
|
|
+ {
|
|
|
+ Author = i.UserNickName,
|
|
|
+ Comment = i.SubAuthorComment,
|
|
|
+ CommunityRating = float.Parse(i.SubRating, _usCulture),
|
|
|
+ DownloadCount = int.Parse(i.SubDownloadsCnt, _usCulture),
|
|
|
+ Format = i.SubFormat,
|
|
|
+ ProviderName = Name,
|
|
|
+ Language = i.SubLanguageID,
|
|
|
+
|
|
|
+ Id = i.SubFormat + "-" + i.SubLanguageID + "-" + i.IDSubtitle,
|
|
|
+
|
|
|
+ Name = i.SubFileName,
|
|
|
+ DateCreated = DateTime.Parse(i.SubAddDate, _usCulture),
|
|
|
+ IsHashMatch = i.MovieHash == hasCopy
|
|
|
+ });
|
|
|
}
|
|
|
}
|
|
|
}
|