jellyfin/MediaBrowser.Providers/MediaInfo/SubtitleDownloader.cs

180 lines
6.1 KiB
C#
Raw Normal View History

2014-05-07 02:28:19 +00:00
using MediaBrowser.Controller.Entities;
using MediaBrowser.Controller.Entities.Movies;
using MediaBrowser.Controller.Entities.TV;
2014-05-11 22:38:10 +00:00
using MediaBrowser.Controller.Providers;
2014-05-07 02:28:19 +00:00
using MediaBrowser.Controller.Subtitles;
using MediaBrowser.Model.Entities;
using MediaBrowser.Model.Logging;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
using MediaBrowser.Common.Extensions;
2014-05-07 02:28:19 +00:00
namespace MediaBrowser.Providers.MediaInfo
{
public class SubtitleDownloader
{
private readonly ILogger _logger;
private readonly ISubtitleManager _subtitleManager;
public SubtitleDownloader(ILogger logger, ISubtitleManager subtitleManager)
{
_logger = logger;
_subtitleManager = subtitleManager;
}
public async Task<List<string>> DownloadSubtitles(Video video,
List<MediaStream> mediaStreams,
2016-03-19 22:31:00 +00:00
bool skipIfEmbeddedSubtitlesPresent,
2014-05-08 02:16:11 +00:00
bool skipIfAudioTrackMatches,
bool requirePerfectMatch,
2014-05-07 02:28:19 +00:00
IEnumerable<string> languages,
CancellationToken cancellationToken)
2017-09-10 03:18:23 +00:00
{
var downloadedLanguages = new List<string>();
foreach (var lang in languages)
{
var downloaded = await DownloadSubtitles(video, mediaStreams, skipIfEmbeddedSubtitlesPresent,
skipIfAudioTrackMatches, requirePerfectMatch, lang, cancellationToken).ConfigureAwait(false);
if (downloaded)
{
downloadedLanguages.Add(lang);
}
}
return downloadedLanguages;
}
public Task<bool> DownloadSubtitles(Video video,
List<MediaStream> mediaStreams,
bool skipIfEmbeddedSubtitlesPresent,
bool skipIfAudioTrackMatches,
bool requirePerfectMatch,
string lang,
CancellationToken cancellationToken)
2014-05-07 02:28:19 +00:00
{
if (video.LocationType != LocationType.FileSystem ||
video.VideoType != VideoType.VideoFile)
{
2017-09-10 03:18:23 +00:00
return Task.FromResult(false);
2014-05-07 02:28:19 +00:00
}
2017-08-26 19:50:02 +00:00
if (!video.IsCompleteMedia)
{
2017-09-10 03:18:23 +00:00
return Task.FromResult(false);
2017-08-26 19:50:02 +00:00
}
2014-05-11 22:38:10 +00:00
VideoContentType mediaType;
2014-05-07 02:28:19 +00:00
if (video is Episode)
{
2014-05-11 22:38:10 +00:00
mediaType = VideoContentType.Episode;
2014-05-07 02:28:19 +00:00
}
else if (video is Movie)
{
2014-05-11 22:38:10 +00:00
mediaType = VideoContentType.Movie;
2014-05-07 02:28:19 +00:00
}
else
{
// These are the only supported types
2017-09-10 03:18:23 +00:00
return Task.FromResult(false);
2014-05-07 02:28:19 +00:00
}
2017-09-10 03:18:23 +00:00
return DownloadSubtitles(video, mediaStreams, skipIfEmbeddedSubtitlesPresent, skipIfAudioTrackMatches,
requirePerfectMatch, lang, mediaType, cancellationToken);
2014-05-07 02:28:19 +00:00
}
private async Task<bool> DownloadSubtitles(Video video,
List<MediaStream> mediaStreams,
2016-03-19 22:31:00 +00:00
bool skipIfEmbeddedSubtitlesPresent,
2014-05-08 02:16:11 +00:00
bool skipIfAudioTrackMatches,
bool requirePerfectMatch,
2014-05-07 02:28:19 +00:00
string language,
2014-05-11 22:38:10 +00:00
VideoContentType mediaType,
2014-05-07 02:28:19 +00:00
CancellationToken cancellationToken)
{
// There's already subtitles for this language
if (mediaStreams.Any(i => i.Type == MediaStreamType.Subtitle && i.IsTextSubtitleStream && string.Equals(i.Language, language, StringComparison.OrdinalIgnoreCase)))
2014-05-07 02:28:19 +00:00
{
return false;
}
var audioStreams = mediaStreams.Where(i => i.Type == MediaStreamType.Audio).ToList();
2014-05-08 20:09:53 +00:00
var defaultAudioStreams = audioStreams.Where(i => i.IsDefault).ToList();
// If none are marked as default, just take a guess
if (defaultAudioStreams.Count == 0)
{
defaultAudioStreams = audioStreams.Take(1).ToList();
}
2014-05-08 05:04:39 +00:00
// There's already a default audio stream for this language
2014-05-08 02:16:11 +00:00
if (skipIfAudioTrackMatches &&
2014-05-08 20:09:53 +00:00
defaultAudioStreams.Any(i => string.Equals(i.Language, language, StringComparison.OrdinalIgnoreCase)))
2014-05-07 20:28:28 +00:00
{
return false;
}
2014-05-07 20:24:34 +00:00
2014-05-07 02:28:19 +00:00
// There's an internal subtitle stream for this language
2016-03-19 22:31:00 +00:00
if (skipIfEmbeddedSubtitlesPresent &&
mediaStreams.Any(i => i.Type == MediaStreamType.Subtitle && !i.IsExternal && string.Equals(i.Language, language, StringComparison.OrdinalIgnoreCase)))
2014-05-07 02:28:19 +00:00
{
return false;
}
var request = new SubtitleSearchRequest
{
ContentType = mediaType,
IndexNumber = video.IndexNumber,
Language = language,
MediaPath = video.Path,
Name = video.Name,
ParentIndexNumber = video.ParentIndexNumber,
ProductionYear = video.ProductionYear,
2014-05-17 04:24:10 +00:00
ProviderIds = video.ProviderIds,
// Stop as soon as we find something
SearchAllProviders = false,
IsPerfectMatch = requirePerfectMatch
2014-05-07 02:28:19 +00:00
};
var episode = video as Episode;
if (episode != null)
{
request.IndexNumberEnd = episode.IndexNumberEnd;
request.SeriesName = episode.SeriesName;
}
2014-05-07 02:28:19 +00:00
try
{
var searchResults = await _subtitleManager.SearchSubtitles(request, cancellationToken).ConfigureAwait(false);
var result = searchResults.FirstOrDefault();
if (result != null)
{
2014-05-17 04:24:10 +00:00
await _subtitleManager.DownloadSubtitles(video, result.Id, cancellationToken)
2014-05-07 02:28:19 +00:00
.ConfigureAwait(false);
return true;
}
}
catch (RateLimitExceededException)
{
}
2014-05-07 02:28:19 +00:00
catch (Exception ex)
{
_logger.ErrorException("Error downloading subtitles", ex);
}
return false;
}
}
}