jellyfin/MediaBrowser.Providers/MediaInfo/SubtitleScheduledTask.cs

197 lines
6.3 KiB
C#
Raw Normal View History

2014-08-10 22:13:17 +00:00
using MediaBrowser.Common.Configuration;
using MediaBrowser.Controller.Configuration;
using MediaBrowser.Controller.Entities;
using MediaBrowser.Controller.Entities.Movies;
using MediaBrowser.Controller.Entities.TV;
using MediaBrowser.Controller.Library;
using MediaBrowser.Controller.Subtitles;
using MediaBrowser.Model.Entities;
using MediaBrowser.Model.Logging;
2015-03-31 16:24:16 +00:00
using MediaBrowser.Model.Providers;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
2016-10-05 07:15:29 +00:00
using System.IO;
2017-05-21 07:25:49 +00:00
using MediaBrowser.Controller.Dto;
2016-10-05 07:15:29 +00:00
using MediaBrowser.Model.Serialization;
using MediaBrowser.Model.Tasks;
using MediaBrowser.Model.Extensions;
namespace MediaBrowser.Providers.MediaInfo
{
public class SubtitleScheduledTask : IScheduledTask
{
private readonly ILibraryManager _libraryManager;
private readonly IServerConfigurationManager _config;
private readonly ISubtitleManager _subtitleManager;
2015-03-31 16:24:16 +00:00
private readonly IMediaSourceManager _mediaSourceManager;
private readonly ILogger _logger;
2016-10-22 14:50:45 +00:00
private readonly IJsonSerializer _json;
2016-10-05 07:15:29 +00:00
public SubtitleScheduledTask(ILibraryManager libraryManager, IJsonSerializer json, IServerConfigurationManager config, ISubtitleManager subtitleManager, ILogger logger, IMediaSourceManager mediaSourceManager)
{
_libraryManager = libraryManager;
_config = config;
_subtitleManager = subtitleManager;
_logger = logger;
2015-03-31 16:24:16 +00:00
_mediaSourceManager = mediaSourceManager;
2016-10-05 07:15:29 +00:00
_json = json;
}
public string Name
{
get { return "Download missing subtitles"; }
}
public string Description
{
get { return "Searches the internet for missing subtitles based on metadata configuration."; }
}
public string Category
{
get { return "Library"; }
}
2014-08-10 22:13:17 +00:00
private SubtitleOptions GetOptions()
{
return _config.GetConfiguration<SubtitleOptions>("subtitles");
}
public async Task Execute(CancellationToken cancellationToken, IProgress<double> progress)
{
2014-08-10 22:13:17 +00:00
var options = GetOptions();
2016-10-05 07:15:29 +00:00
var types = new List<string>();
2015-01-25 06:34:50 +00:00
2016-10-05 07:15:29 +00:00
if (options.DownloadEpisodeSubtitles)
{
types.Add("Episode");
}
if (options.DownloadMovieSubtitles)
{
types.Add("Movie");
}
2016-10-05 07:15:29 +00:00
if (types.Count == 0)
{
return;
}
2017-09-10 03:18:23 +00:00
var dict = new Dictionary<Guid, BaseItem>();
foreach (var lang in options.DownloadLanguages)
2016-10-05 07:15:29 +00:00
{
2017-09-10 03:18:23 +00:00
var query = new InternalItemsQuery
{
MediaTypes = new string[] {MediaType.Video},
IsVirtualItem = false,
IncludeItemTypes = types.ToArray(types.Count),
DtoOptions = new DtoOptions(true),
SourceTypes = new[] {SourceType.Library}
};
if (options.SkipIfAudioTrackMatches)
{
query.HasNoAudioTrackWithLanguage = lang;
}
2016-10-05 07:15:29 +00:00
2017-09-10 03:18:23 +00:00
if (options.SkipIfEmbeddedSubtitlesPresent)
{
// Exclude if it already has any subtitles of the same language
query.HasNoSubtitleTrackWithLanguage = lang;
}
else
{
// Exclude if it already has external subtitles of the same language
query.HasNoExternalSubtitleTrackWithLanguage = lang;
}
2017-09-10 03:18:23 +00:00
var videosByLanguage = _libraryManager.GetItemList(query);
foreach (var video in videosByLanguage)
{
dict[video.Id] = video;
}
}
var videos = dict.Values.ToList();
2016-10-25 19:02:04 +00:00
if (videos.Count == 0)
{
return;
}
var numComplete = 0;
foreach (var video in videos)
{
2017-09-10 03:18:23 +00:00
cancellationToken.ThrowIfCancellationRequested();
try
{
2017-09-10 03:18:23 +00:00
await DownloadSubtitles(video as Video, options, cancellationToken).ConfigureAwait(false);
}
catch (Exception ex)
{
_logger.ErrorException("Error downloading subtitles for {0}", ex, video.Path);
}
// Update progress
numComplete++;
double percent = numComplete;
percent /= videos.Count;
progress.Report(100 * percent);
}
2016-10-05 07:15:29 +00:00
}
private async Task<bool> DownloadSubtitles(Video video, SubtitleOptions options, CancellationToken cancellationToken)
{
2014-08-10 22:13:17 +00:00
if ((options.DownloadEpisodeSubtitles &&
video is Episode) ||
2014-08-10 22:13:17 +00:00
(options.DownloadMovieSubtitles &&
video is Movie))
{
2015-03-31 16:24:16 +00:00
var mediaStreams = _mediaSourceManager.GetStaticMediaSources(video, false).First().MediaStreams;
var downloadedLanguages = await new SubtitleDownloader(_logger,
_subtitleManager)
.DownloadSubtitles(video,
mediaStreams,
2016-03-19 22:31:00 +00:00
options.SkipIfEmbeddedSubtitlesPresent,
2014-08-10 22:13:17 +00:00
options.SkipIfAudioTrackMatches,
options.RequirePerfectMatch,
2014-08-10 22:13:17 +00:00
options.DownloadLanguages,
cancellationToken).ConfigureAwait(false);
// Rescan
if (downloadedLanguages.Count > 0)
{
await video.RefreshMetadata(cancellationToken).ConfigureAwait(false);
2016-10-05 07:15:29 +00:00
return false;
}
2016-10-05 07:15:29 +00:00
return true;
}
2016-10-05 07:15:29 +00:00
return false;
}
public IEnumerable<TaskTriggerInfo> GetDefaultTriggers()
{
return new[] {
// Every so often
new TaskTriggerInfo { Type = TaskTriggerInfo.TriggerInterval, IntervalTicks = TimeSpan.FromHours(24).Ticks}
};
}
public string Key
{
get { return "DownloadSubtitles"; }
}
}
}