aboutsummaryrefslogtreecommitdiff
path: root/MediaBrowser.Controller/ScheduledTasks/ImageCleanupTask.cs
diff options
context:
space:
mode:
authorLukePulverenti <luke.pulverenti@gmail.com>2013-02-25 22:43:04 -0500
committerLukePulverenti <luke.pulverenti@gmail.com>2013-02-25 22:43:04 -0500
commit2d06095447b972c8c7239277428e2c67c8b7ca86 (patch)
tree14278bd4c0732ee962b73ff4845e5022e157a0a3 /MediaBrowser.Controller/ScheduledTasks/ImageCleanupTask.cs
parent364fbb9e0c7586afa296ddd7d739df086f4c3533 (diff)
plugin security fixes and other abstractions
Diffstat (limited to 'MediaBrowser.Controller/ScheduledTasks/ImageCleanupTask.cs')
-rw-r--r--MediaBrowser.Controller/ScheduledTasks/ImageCleanupTask.cs210
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";
- }
- }
- }
-}