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 { /// /// Plugin Update Task /// [Export(typeof(IScheduledTask))] public class PluginUpdateTask : BaseScheduledTask { /// /// Creates the triggers that define when the task will run /// /// IEnumerable{BaseTaskTrigger}. protected override IEnumerable GetDefaultTriggers() { return new BaseTaskTrigger[] { // 1:30am new DailyTrigger { TimeOfDay = TimeSpan.FromHours(1.5) }, new IntervalTrigger { Interval = TimeSpan.FromHours(2)} }; } /// /// Update installed plugins /// /// The cancellation token. /// The progress. /// Task. protected override async Task ExecuteInternal(CancellationToken cancellationToken, IProgress progress) { progress.Report(0); var packagesToInstall = (await Kernel.InstallationManager.GetAvailablePluginUpdates(true, cancellationToken).ConfigureAwait(false)).ToList(); progress.Report(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 { }, 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((90 * percent) + 10); } })); cancellationToken.ThrowIfCancellationRequested(); await Task.WhenAll(tasks).ConfigureAwait(false); progress.Report(100); } /// /// Gets the name of the task /// /// The name. public override string Name { get { return "Check for plugin updates"; } } /// /// Gets the description. /// /// The description. public override string Description { get { return "Downloads and installs updates for plugins that are configured to update automatically."; } } } }