diff options
Diffstat (limited to 'MediaBrowser.Controller/ScheduledTasks')
5 files changed, 565 insertions, 0 deletions
diff --git a/MediaBrowser.Controller/ScheduledTasks/ChapterImagesTask.cs b/MediaBrowser.Controller/ScheduledTasks/ChapterImagesTask.cs new file mode 100644 index 0000000000..21f1bce5ac --- /dev/null +++ b/MediaBrowser.Controller/ScheduledTasks/ChapterImagesTask.cs @@ -0,0 +1,100 @@ +using MediaBrowser.Common.ScheduledTasks; +using MediaBrowser.Controller.Entities; +using MediaBrowser.Model.Tasks; +using System; +using System.Collections.Generic; +using System.ComponentModel.Composition; +using System.Linq; +using System.Threading; +using System.Threading.Tasks; + +namespace MediaBrowser.Controller.ScheduledTasks +{ + [Export(typeof(IScheduledTask))] + class ChapterImagesTask : BaseScheduledTask<Kernel> + { + /// <summary> + /// Creates the triggers that define when the task will run + /// </summary> + /// <returns>IEnumerable{BaseTaskTrigger}.</returns> + protected override IEnumerable<BaseTaskTrigger> GetDefaultTriggers() + { + return new BaseTaskTrigger[] + { + new DailyTrigger { TimeOfDay = TimeSpan.FromHours(4) } + }; + } + + /// <summary> + /// Returns the task to be executed + /// </summary> + /// <param name="cancellationToken">The cancellation token.</param> + /// <param name="progress">The progress.</param> + /// <returns>Task.</returns> + protected override Task ExecuteInternal(CancellationToken cancellationToken, IProgress<TaskProgress> progress) + { + var videos = Kernel.RootFolder.RecursiveChildren.OfType<Video>().Where(v => v.Chapters != null).ToList(); + + var numComplete = 0; + + var tasks = videos.Select(v => Task.Run(async () => + { + try + { + await Kernel.FFMpegManager.PopulateChapterImages(v, cancellationToken, true, true); + } + catch (OperationCanceledException) + { + throw; + } + catch (Exception ex) + { + Logger.ErrorException("Error creating chapter images for {0}", ex, v.Name); + } + finally + { + lock (progress) + { + numComplete++; + double percent = numComplete; + percent /= videos.Count; + + progress.Report(new TaskProgress { PercentComplete = 100 * percent }); + } + } + })); + + return Task.WhenAll(tasks); + } + + /// <summary> + /// Gets the name of the task + /// </summary> + /// <value>The name.</value> + public override string Name + { + get { return "Create video chapter thumbnails"; } + } + + /// <summary> + /// Gets the description. + /// </summary> + /// <value>The description.</value> + public override string Description + { + get { return "Creates thumbnails for videos that have chapters."; } + } + + /// <summary> + /// Gets the category. + /// </summary> + /// <value>The category.</value> + public override string Category + { + get + { + return "Library"; + } + } + } +} diff --git a/MediaBrowser.Controller/ScheduledTasks/ImageCleanupTask.cs b/MediaBrowser.Controller/ScheduledTasks/ImageCleanupTask.cs new file mode 100644 index 0000000000..c16f714dbc --- /dev/null +++ b/MediaBrowser.Controller/ScheduledTasks/ImageCleanupTask.cs @@ -0,0 +1,202 @@ +using MediaBrowser.Common.Extensions; +using MediaBrowser.Common.ScheduledTasks; +using MediaBrowser.Controller.Entities; +using MediaBrowser.Controller.Entities.Movies; +using MediaBrowser.Model.Tasks; +using System; +using System.Collections.Generic; +using System.ComponentModel.Composition; +using System.IO; +using System.Linq; +using System.Threading; +using System.Threading.Tasks; + +namespace MediaBrowser.Controller.ScheduledTasks +{ + /// <summary> + /// Class ImageCleanupTask + /// </summary> + [Export(typeof(IScheduledTask))] + public class ImageCleanupTask : BaseScheduledTask<Kernel> + { + /// <summary> + /// Creates the triggers that define when the task will run + /// </summary> + /// <returns>IEnumerable{BaseTaskTrigger}.</returns> + protected override IEnumerable<BaseTaskTrigger> GetDefaultTriggers() + { + return new BaseTaskTrigger[] + { + new DailyTrigger { TimeOfDay = TimeSpan.FromHours(2) } + }; + } + + /// <summary> + /// Returns the task to be executed + /// </summary> + /// <param name="cancellationToken">The cancellation token.</param> + /// <param name="progress">The progress.</param> + /// <returns>Task.</returns> + protected override async Task ExecuteInternal(CancellationToken cancellationToken, IProgress<TaskProgress> progress) + { + await EnsureChapterImages(cancellationToken).ConfigureAwait(false); + + // First gather all image files + var files = GetFiles(Kernel.FFMpegManager.AudioImagesDataPath) + .Concat(GetFiles(Kernel.FFMpegManager.VideoImagesDataPath)) + .Concat(GetFiles(Kernel.ProviderManager.ImagesDataPath)) + .ToList(); + + // Now gather all items + var items = Kernel.RootFolder.RecursiveChildren.ToList(); + items.Add(Kernel.RootFolder); + + // Determine all possible image paths + var pathsInUse = items.SelectMany(GetPathsInUse) + .Distinct(StringComparer.OrdinalIgnoreCase) + .ToDictionary(p => p, StringComparer.OrdinalIgnoreCase); + + var numComplete = 0; + + var tasks = files.Select(file => Task.Run(() => + { + cancellationToken.ThrowIfCancellationRequested(); + + if (!pathsInUse.ContainsKey(file)) + { + cancellationToken.ThrowIfCancellationRequested(); + + try + { + File.Delete(file); + } + catch (IOException ex) + { + Logger.ErrorException("Error deleting {0}", ex, file); + } + } + + // Update progress + lock (progress) + { + numComplete++; + double percent = numComplete; + percent /= files.Count; + + progress.Report(new TaskProgress { PercentComplete = 100 * percent }); + } + })); + + await Task.WhenAll(tasks).ConfigureAwait(false); + } + + /// <summary> + /// Ensures the chapter images. + /// </summary> + /// <param name="cancellationToken">The cancellation token.</param> + /// <returns>Task.</returns> + private Task EnsureChapterImages(CancellationToken cancellationToken) + { + var videos = Kernel.RootFolder.RecursiveChildren.OfType<Video>().Where(v => v.Chapters != null).ToList(); + + var tasks = videos.Select(v => Task.Run(async () => + { + await Kernel.FFMpegManager.PopulateChapterImages(v, cancellationToken, false, true); + })); + + return Task.WhenAll(tasks); + } + + /// <summary> + /// Gets the paths in use. + /// </summary> + /// <param name="item">The item.</param> + /// <returns>IEnumerable{System.String}.</returns> + private IEnumerable<string> GetPathsInUse(BaseItem item) + { + IEnumerable<string> images = new List<string> { }; + + if (item.Images != null) + { + images = images.Concat(item.Images.Values); + } + + if (item.BackdropImagePaths != null) + { + images = images.Concat(item.BackdropImagePaths); + } + + if (item.ScreenshotImagePaths != null) + { + images = images.Concat(item.ScreenshotImagePaths); + } + + var video = item as Video; + + if (video != null && video.Chapters != null) + { + images = images.Concat(video.Chapters.Where(i => !string.IsNullOrEmpty(i.ImagePath)).Select(i => i.ImagePath)); + } + + if (item.LocalTrailers != null) + { + foreach (var subItem in item.LocalTrailers) + { + images = images.Concat(GetPathsInUse(subItem)); + } + } + + var movie = item as Movie; + + if (movie != null && movie.SpecialFeatures != null) + { + foreach (var subItem in movie.SpecialFeatures) + { + images = images.Concat(GetPathsInUse(subItem)); + } + } + + return images; + } + + /// <summary> + /// Gets the files. + /// </summary> + /// <param name="path">The path.</param> + /// <returns>IEnumerable{System.String}.</returns> + private IEnumerable<string> GetFiles(string path) + { + return Directory.EnumerateFiles(path, "*.jpg", SearchOption.AllDirectories).Concat(Directory.EnumerateFiles(path, "*.png", SearchOption.AllDirectories)); + } + + /// <summary> + /// Gets the name of the task + /// </summary> + /// <value>The name.</value> + public override string Name + { + get { return "Images cleanup"; } + } + + /// <summary> + /// Gets the description. + /// </summary> + /// <value>The description.</value> + public override string Description + { + get { return "Deletes downloaded and extracted images that are no longer being used."; } + } + + /// <summary> + /// Gets the category. + /// </summary> + /// <value>The category.</value> + public override string Category + { + get + { + return "Maintenance"; + } + } + } +} diff --git a/MediaBrowser.Controller/ScheduledTasks/PeopleValidationTask.cs b/MediaBrowser.Controller/ScheduledTasks/PeopleValidationTask.cs new file mode 100644 index 0000000000..a5f87f212b --- /dev/null +++ b/MediaBrowser.Controller/ScheduledTasks/PeopleValidationTask.cs @@ -0,0 +1,72 @@ +using MediaBrowser.Common.ScheduledTasks; +using MediaBrowser.Model.Tasks; +using System; +using System.Collections.Generic; +using System.ComponentModel.Composition; +using System.Threading; +using System.Threading.Tasks; + +namespace MediaBrowser.Controller.ScheduledTasks +{ + /// <summary> + /// Class PeopleValidationTask + /// </summary> + [Export(typeof(IScheduledTask))] + public class PeopleValidationTask : BaseScheduledTask<Kernel> + { + /// <summary> + /// Creates the triggers that define when the task will run + /// </summary> + /// <returns>IEnumerable{BaseTaskTrigger}.</returns> + protected override IEnumerable<BaseTaskTrigger> GetDefaultTriggers() + { + return new BaseTaskTrigger[] + { + new DailyTrigger { TimeOfDay = TimeSpan.FromHours(2) }, + + new IntervalTrigger{ Interval = TimeSpan.FromHours(12)} + }; + } + + /// <summary> + /// Returns the task to be executed + /// </summary> + /// <param name="cancellationToken">The cancellation token.</param> + /// <param name="progress">The progress.</param> + /// <returns>Task.</returns> + protected override Task ExecuteInternal(CancellationToken cancellationToken, IProgress<TaskProgress> progress) + { + return Kernel.LibraryManager.ValidatePeople(cancellationToken, progress); + } + + /// <summary> + /// Gets the name of the task + /// </summary> + /// <value>The name.</value> + public override string Name + { + get { return "Refresh people"; } + } + + /// <summary> + /// Gets the description. + /// </summary> + /// <value>The description.</value> + public override string Description + { + get { return "Updates metadata for actors, artists and directors in your media library."; } + } + + /// <summary> + /// Gets the category. + /// </summary> + /// <value>The category.</value> + public override string Category + { + get + { + return "Library"; + } + } + } +} diff --git a/MediaBrowser.Controller/ScheduledTasks/PluginUpdateTask.cs b/MediaBrowser.Controller/ScheduledTasks/PluginUpdateTask.cs new file mode 100644 index 0000000000..3a700d94d4 --- /dev/null +++ b/MediaBrowser.Controller/ScheduledTasks/PluginUpdateTask.cs @@ -0,0 +1,113 @@ +using MediaBrowser.Common.ScheduledTasks; +using MediaBrowser.Model.Net; +using MediaBrowser.Model.Tasks; +using System; +using System.Collections.Generic; +using System.ComponentModel.Composition; +using System.IO; +using System.Linq; +using System.Threading; +using System.Threading.Tasks; + +namespace MediaBrowser.Controller.ScheduledTasks +{ + /// <summary> + /// Plugin Update Task + /// </summary> + [Export(typeof(IScheduledTask))] + public class PluginUpdateTask : BaseScheduledTask<Kernel> + { + /// <summary> + /// Creates the triggers that define when the task will run + /// </summary> + /// <returns>IEnumerable{BaseTaskTrigger}.</returns> + protected override IEnumerable<BaseTaskTrigger> GetDefaultTriggers() + { + return new BaseTaskTrigger[] { + + // 1:30am + new DailyTrigger { TimeOfDay = TimeSpan.FromHours(1.5) }, + + new IntervalTrigger { Interval = TimeSpan.FromHours(2)} + }; + } + + /// <summary> + /// Update installed plugins + /// </summary> + /// <param name="cancellationToken">The cancellation token.</param> + /// <param name="progress">The progress.</param> + /// <returns>Task.</returns> + protected override async Task ExecuteInternal(CancellationToken cancellationToken, IProgress<TaskProgress> progress) + { + progress.Report(new TaskProgress { Description = "Checking for plugin updates", PercentComplete = 0 }); + + var packagesToInstall = (await Kernel.InstallationManager.GetAvailablePluginUpdates(true, cancellationToken).ConfigureAwait(false)).ToList(); + + progress.Report(new TaskProgress { PercentComplete = 10 }); + + var numComplete = 0; + + // Create tasks for each one + var tasks = packagesToInstall.Select(i => Task.Run(async () => + { + cancellationToken.ThrowIfCancellationRequested(); + + try + { + await Kernel.InstallationManager.InstallPackage(i, new Progress<double> { }, cancellationToken).ConfigureAwait(false); + } + catch (OperationCanceledException) + { + // InstallPackage has it's own inner cancellation token, so only throw this if it's ours + if (cancellationToken.IsCancellationRequested) + { + throw; + } + } + catch (HttpException ex) + { + Logger.ErrorException("Error downloading {0}", ex, i.name); + } + catch (IOException ex) + { + Logger.ErrorException("Error updating {0}", ex, i.name); + } + + // Update progress + lock (progress) + { + numComplete++; + double percent = numComplete; + percent /= packagesToInstall.Count; + + progress.Report(new TaskProgress { PercentComplete = (90 * percent) + 10 }); + } + })); + + cancellationToken.ThrowIfCancellationRequested(); + + await Task.WhenAll(tasks).ConfigureAwait(false); + + progress.Report(new TaskProgress { PercentComplete = 100 }); + } + + /// <summary> + /// Gets the name of the task + /// </summary> + /// <value>The name.</value> + public override string Name + { + get { return "Check for plugin updates"; } + } + + /// <summary> + /// Gets the description. + /// </summary> + /// <value>The description.</value> + public override string Description + { + get { return "Downloads and installs updates for plugins that are configured to update automatically."; } + } + } +}
\ No newline at end of file diff --git a/MediaBrowser.Controller/ScheduledTasks/RefreshMediaLibraryTask.cs b/MediaBrowser.Controller/ScheduledTasks/RefreshMediaLibraryTask.cs new file mode 100644 index 0000000000..b01019fce3 --- /dev/null +++ b/MediaBrowser.Controller/ScheduledTasks/RefreshMediaLibraryTask.cs @@ -0,0 +1,78 @@ +using MediaBrowser.Common.ScheduledTasks; +using MediaBrowser.Model.Tasks; +using System; +using System.Collections.Generic; +using System.ComponentModel.Composition; +using System.Threading; +using System.Threading.Tasks; + +namespace MediaBrowser.Controller.ScheduledTasks +{ + /// <summary> + /// Class RefreshMediaLibraryTask + /// </summary> + [Export(typeof(IScheduledTask))] + public class RefreshMediaLibraryTask : BaseScheduledTask<Kernel> + { + /// <summary> + /// Gets the default triggers. + /// </summary> + /// <returns>IEnumerable{BaseTaskTrigger}.</returns> + protected override IEnumerable<BaseTaskTrigger> GetDefaultTriggers() + { + return new BaseTaskTrigger[] { + + new StartupTrigger(Kernel), + + new SystemEventTrigger{ SystemEvent = SystemEvent.WakeFromSleep}, + + new IntervalTrigger{ Interval = TimeSpan.FromHours(2)} + }; + } + + /// <summary> + /// Executes the internal. + /// </summary> + /// <param name="cancellationToken">The cancellation token.</param> + /// <param name="progress">The progress.</param> + /// <returns>Task.</returns> + protected override Task ExecuteInternal(CancellationToken cancellationToken, IProgress<TaskProgress> progress) + { + cancellationToken.ThrowIfCancellationRequested(); + + progress.Report(new TaskProgress { PercentComplete = 0 }); + + return Kernel.LibraryManager.ValidateMediaLibrary(progress, cancellationToken); + } + + /// <summary> + /// Gets the name. + /// </summary> + /// <value>The name.</value> + public override string Name + { + get { return "Scan media library"; } + } + + /// <summary> + /// Gets the description. + /// </summary> + /// <value>The description.</value> + public override string Description + { + get { return "Scans your media library and refreshes metatata based on configuration."; } + } + + /// <summary> + /// Gets the category. + /// </summary> + /// <value>The category.</value> + public override string Category + { + get + { + return "Library"; + } + } + } +} |