diff options
| author | LukePulverenti <luke.pulverenti@gmail.com> | 2013-02-25 22:43:04 -0500 |
|---|---|---|
| committer | LukePulverenti <luke.pulverenti@gmail.com> | 2013-02-25 22:43:04 -0500 |
| commit | 2d06095447b972c8c7239277428e2c67c8b7ca86 (patch) | |
| tree | 14278bd4c0732ee962b73ff4845e5022e157a0a3 /MediaBrowser.Controller/ScheduledTasks/ImageCleanupTask.cs | |
| parent | 364fbb9e0c7586afa296ddd7d739df086f4c3533 (diff) | |
plugin security fixes and other abstractions
Diffstat (limited to 'MediaBrowser.Controller/ScheduledTasks/ImageCleanupTask.cs')
| -rw-r--r-- | MediaBrowser.Controller/ScheduledTasks/ImageCleanupTask.cs | 210 |
1 files changed, 0 insertions, 210 deletions
diff --git a/MediaBrowser.Controller/ScheduledTasks/ImageCleanupTask.cs b/MediaBrowser.Controller/ScheduledTasks/ImageCleanupTask.cs deleted file mode 100644 index 7e0094a67..000000000 --- a/MediaBrowser.Controller/ScheduledTasks/ImageCleanupTask.cs +++ /dev/null @@ -1,210 +0,0 @@ -using MediaBrowser.Common.ScheduledTasks; -using MediaBrowser.Controller.Entities; -using MediaBrowser.Controller.Entities.Movies; -using MediaBrowser.Model.Logging; -using System; -using System.Collections.Generic; -using System.IO; -using System.Linq; -using System.Threading; -using System.Threading.Tasks; - -namespace MediaBrowser.Controller.ScheduledTasks -{ - /// <summary> - /// Class ImageCleanupTask - /// </summary> - public class ImageCleanupTask : BaseScheduledTask<Kernel> - { - /// <summary> - /// Initializes a new instance of the <see cref="ImageCleanupTask" /> class. - /// </summary> - /// <param name="kernel">The kernel.</param> - /// <param name="taskManager">The task manager.</param> - /// <param name="logger">The logger.</param> - public ImageCleanupTask(Kernel kernel, ITaskManager taskManager, ILogger logger) - : base(kernel, taskManager, logger) - { - } - - /// <summary> - /// Creates the triggers that define when the task will run - /// </summary> - /// <returns>IEnumerable{BaseTaskTrigger}.</returns> - public override IEnumerable<ITaskTrigger> GetDefaultTriggers() - { - return new ITaskTrigger[] - { - 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<double> 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(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"; - } - } - } -} |
