123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166 |
- 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;
- using MediaBrowser.Model.Providers;
- using System;
- using System.Collections.Generic;
- using System.Linq;
- using System.Threading;
- using System.Threading.Tasks;
- using System.IO;
- using MediaBrowser.Controller.Dto;
- using MediaBrowser.Model.Serialization;
- using MediaBrowser.Model.Tasks;
- namespace MediaBrowser.Providers.MediaInfo
- {
- public class SubtitleScheduledTask : IScheduledTask
- {
- private readonly ILibraryManager _libraryManager;
- private readonly IServerConfigurationManager _config;
- private readonly ISubtitleManager _subtitleManager;
- private readonly IMediaSourceManager _mediaSourceManager;
- private readonly ILogger _logger;
- private readonly IJsonSerializer _json;
- public SubtitleScheduledTask(ILibraryManager libraryManager, IJsonSerializer json, IServerConfigurationManager config, ISubtitleManager subtitleManager, ILogger logger, IMediaSourceManager mediaSourceManager)
- {
- _libraryManager = libraryManager;
- _config = config;
- _subtitleManager = subtitleManager;
- _logger = logger;
- _mediaSourceManager = mediaSourceManager;
- _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"; }
- }
- private SubtitleOptions GetOptions()
- {
- return _config.GetConfiguration<SubtitleOptions>("subtitles");
- }
- public async Task Execute(CancellationToken cancellationToken, IProgress<double> progress)
- {
- var options = GetOptions();
- var types = new List<string>();
- if (options.DownloadEpisodeSubtitles)
- {
- types.Add("Episode");
- }
- if (options.DownloadMovieSubtitles)
- {
- types.Add("Movie");
- }
- if (types.Count == 0)
- {
- return;
- }
- var videos = _libraryManager.GetItemList(new InternalItemsQuery
- {
- MediaTypes = new string[] { MediaType.Video },
- IsVirtualItem = false,
- IncludeItemTypes = types.ToArray(),
- DtoOptions = new DtoOptions(true)
- }).OfType<Video>()
- .Where(i => i.LocationType != LocationType.Remote)
- .ToList();
- if (videos.Count == 0)
- {
- return;
- }
- var numComplete = 0;
- foreach (var video in videos)
- {
- try
- {
- await DownloadSubtitles(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);
- }
- }
- private async Task<bool> DownloadSubtitles(Video video, SubtitleOptions options, CancellationToken cancellationToken)
- {
- if ((options.DownloadEpisodeSubtitles &&
- video is Episode) ||
- (options.DownloadMovieSubtitles &&
- video is Movie))
- {
- var mediaStreams = _mediaSourceManager.GetStaticMediaSources(video, false).First().MediaStreams;
- var downloadedLanguages = await new SubtitleDownloader(_logger,
- _subtitleManager)
- .DownloadSubtitles(video,
- mediaStreams,
- options.SkipIfEmbeddedSubtitlesPresent,
- options.SkipIfAudioTrackMatches,
- options.RequirePerfectMatch,
- options.DownloadLanguages,
- cancellationToken).ConfigureAwait(false);
- // Rescan
- if (downloadedLanguages.Count > 0)
- {
- await video.RefreshMetadata(cancellationToken).ConfigureAwait(false);
- return false;
- }
- return true;
- }
- 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"; }
- }
- }
- }
|