Backupppert/jellyfin
0
0
Fork 0
mirror of https://github.com/jellyfin/jellyfin.git synced 2024-07-22 05:30:46 +02:00
Code Issues Packages Projects Releases Wiki Activity

finish implementation of opensubtitles downloader

Browse source
This commit is contained in:
Luis Miguel Almánzar 2014-05-05 03:36:26 -04:00
parent b522a11f64
commit 9ef0793b28
3 changed files with 44 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
MediaBrowser.Controller/Providers/ISubtitleProvider.cs
View file

@ -49,7 +49,9 @@ namespace MediaBrowser.Controller.Providers
public string MediaPath { get; set; } public string MediaPath { get; set; }
public string SeriesName { get; set; } public string SeriesName { get; set; }
public string Name { get; set; }
public int? IndexNumber { get; set; } public int? IndexNumber { get; set; }
public int? ParentIndexNumber { get; set; } public int? ParentIndexNumber { get; set; }
public long ImdbId { get; set; }
} }
} }

47
MediaBrowser.Providers/Subtitles/OpenSubtitleDownloader.cs
View file

@ -37,25 +37,31 @@ namespace MediaBrowser.Providers.Subtitles
public Task<SubtitleResponse> GetSubtitles(SubtitleRequest request, CancellationToken cancellationToken) public Task<SubtitleResponse> GetSubtitles(SubtitleRequest request, CancellationToken cancellationToken)
{ {
return request.ContentType == SubtitleMediaType.Episode return GetMediaSubtitleSubtitles(request, cancellationToken);
? GetEpisodeSubtitles(request, cancellationToken)
: GetMovieSubtitles(request, cancellationToken);
} }
public async Task<SubtitleResponse> GetMovieSubtitles(SubtitleRequest request, CancellationToken cancellationToken) public async Task<SubtitleResponse> GetMediaSubtitleSubtitles(SubtitleRequest request, CancellationToken cancellationToken)
{
throw new NotImplementedException();
}
public async Task<SubtitleResponse> GetEpisodeSubtitles(SubtitleRequest request, CancellationToken cancellationToken)
{ {
var response = new SubtitleResponse(); var response = new SubtitleResponse();
if (!request.IndexNumber.HasValue || !request.ParentIndexNumber.HasValue) switch (request.ContentType)
{ {
_logger.Debug("Information Missing"); case SubtitleMediaType.Episode:
return response; if (!request.IndexNumber.HasValue || !request.ParentIndexNumber.HasValue || string.IsNullOrEmpty(request.SeriesName))
{
_logger.Debug("Information Missing");
return response;
}
break;
case SubtitleMediaType.Movie:
if (string.IsNullOrEmpty(request.Name))
{
_logger.Debug("Information Missing");
return response;
}
break;
} }
if (string.IsNullOrEmpty(request.MediaPath)) if (string.IsNullOrEmpty(request.MediaPath))
{ {
_logger.Debug("Path Missing"); _logger.Debug("Path Missing");
@ -76,10 +82,13 @@ namespace MediaBrowser.Providers.Subtitles
var fileInfo = new FileInfo(request.MediaPath); var fileInfo = new FileInfo(request.MediaPath);
var movieByteSize = fileInfo.Length; var movieByteSize = fileInfo.Length;
var subtitleSearchParameters = request.ContentType == SubtitleMediaType.Episode
? new SubtitleSearchParameters(subLanguageId, request.SeriesName, request.ParentIndexNumber.Value.ToString(_usCulture), request.IndexNumber.Value.ToString(_usCulture))
: new SubtitleSearchParameters(subLanguageId, request.Name);
var parms = new List<SubtitleSearchParameters> { var parms = new List<SubtitleSearchParameters> {
new SubtitleSearchParameters(subLanguageId, hash, movieByteSize), new SubtitleSearchParameters(subLanguageId, hash, movieByteSize),
new SubtitleSearchParameters(subLanguageId, request.SeriesName, request.ParentIndexNumber.Value.ToString(_usCulture), request.IndexNumber.Value.ToString(_usCulture)), subtitleSearchParameters
}; };
var result = OpenSubtitles.SearchSubtitles(parms.ToArray()); var result = OpenSubtitles.SearchSubtitles(parms.ToArray());
@ -89,8 +98,14 @@ namespace MediaBrowser.Providers.Subtitles
return null; return null;
} }
Predicate<SubtitleSearchResult> mediaFilter =
x =>
request.ContentType == SubtitleMediaType.Episode
? int.Parse(x.SeriesSeason) == request.ParentIndexNumber && int.Parse(x.SeriesEpisode) == request.IndexNumber
: long.Parse(x.IDMovieImdb) == request.ImdbId;
var results = ((MethodResponseSubtitleSearch)result).Results; var results = ((MethodResponseSubtitleSearch)result).Results;
var bestResult = results.Where(x => x.SubBad == "0" && int.Parse(x.SeriesSeason) == request.ParentIndexNumber && int.Parse(x.SeriesEpisode) == request.IndexNumber) var bestResult = results.Where(x => x.SubBad == "0" && mediaFilter(x))
.OrderBy(x => x.MovieHash == hash) .OrderBy(x => x.MovieHash == hash)
.ThenBy(x => Math.Abs(long.Parse(x.MovieByteSize) - movieByteSize)) .ThenBy(x => Math.Abs(long.Parse(x.MovieByteSize) - movieByteSize))
.ThenByDescending(x => int.Parse(x.SubDownloadsCnt)) .ThenByDescending(x => int.Parse(x.SubDownloadsCnt))
@ -124,7 +139,7 @@ namespace MediaBrowser.Providers.Subtitles
var data = Convert.FromBase64String(res.Data); var data = Convert.FromBase64String(res.Data);
response.HasContent = true; response.HasContent = true;
response.Format = SubtitleFormat.SRT; response.Format = subtitle.SubFormat.ToUpper();
response.Stream = new MemoryStream(Utilities.Decompress(new MemoryStream(data))); response.Stream = new MemoryStream(Utilities.Decompress(new MemoryStream(data)));
return response; return response;
} }

11
OpenSubtitlesHandler/SubtitleTypes/SubtitleSearchParameters.cs
View file

@ -42,6 +42,17 @@ namespace OpenSubtitlesHandler
this._query = ""; this._query = "";
} }
public SubtitleSearchParameters(string subLanguageId, string query)
{
this.subLanguageId = subLanguageId;
this.movieHash = "";
this.movieByteSize = 0;
this.imdbid = "";
this._episode = "";
this._season = "";
this._query = query;
}
public SubtitleSearchParameters(string subLanguageId, string query, string season, string episode) public SubtitleSearchParameters(string subLanguageId, string query, string season, string episode)
{ {
this.subLanguageId = subLanguageId; this.subLanguageId = subLanguageId;