jellyfin/MediaBrowser.Server.Implementations/ScheduledTasks/PluginUpdateTask.cs

135 lines
4.2 KiB
C#
Raw Normal View History

2014-09-06 20:09:35 +00:00
using MediaBrowser.Common;
using MediaBrowser.Common.ScheduledTasks;
using MediaBrowser.Common.Updates;
2013-02-23 07:57:11 +00:00
using MediaBrowser.Model.Logging;
2013-02-21 01:33:05 +00:00
using MediaBrowser.Model.Net;
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
2014-09-06 20:09:35 +00:00
namespace MediaBrowser.Server.Implementations.ScheduledTasks
2013-02-21 01:33:05 +00:00
{
/// <summary>
/// Plugin Update Task
/// </summary>
public class PluginUpdateTask : IScheduledTask
2013-02-21 01:33:05 +00:00
{
/// <summary>
/// The _logger
/// </summary>
private readonly ILogger _logger;
private readonly IInstallationManager _installationManager;
private readonly IApplicationHost _appHost;
public PluginUpdateTask(ILogger logger, IInstallationManager installationManager, IApplicationHost appHost)
2013-02-23 07:57:11 +00:00
{
_logger = logger;
_installationManager = installationManager;
_appHost = appHost;
2013-02-23 07:57:11 +00:00
}
2013-02-21 01:33:05 +00:00
/// <summary>
/// Creates the triggers that define when the task will run
/// </summary>
/// <returns>IEnumerable{BaseTaskTrigger}.</returns>
public IEnumerable<ITaskTrigger> GetDefaultTriggers()
2013-02-21 01:33:05 +00:00
{
2013-02-24 21:53:54 +00:00
return new ITaskTrigger[] {
2013-02-21 01:33:05 +00:00
2013-08-16 18:43:11 +00:00
// At startup
2014-08-11 23:41:11 +00:00
new StartupTrigger(),
2013-02-21 01:33:05 +00:00
2013-08-16 18:43:11 +00:00
// Every so often
new IntervalTrigger { Interval = TimeSpan.FromHours(24)}
2013-02-21 01:33:05 +00:00
};
}
/// <summary>
/// Update installed plugins
/// </summary>
/// <param name="cancellationToken">The cancellation token.</param>
/// <param name="progress">The progress.</param>
/// <returns>Task.</returns>
public async Task Execute(CancellationToken cancellationToken, IProgress<double> progress)
2013-02-21 01:33:05 +00:00
{
2013-02-22 04:23:06 +00:00
progress.Report(0);
2013-02-21 01:33:05 +00:00
var packagesToInstall = (await _installationManager.GetAvailablePluginUpdates(_appHost.ApplicationVersion, true, cancellationToken).ConfigureAwait(false)).ToList();
2013-02-21 01:33:05 +00:00
2013-02-22 04:23:06 +00:00
progress.Report(10);
2013-02-21 01:33:05 +00:00
var numComplete = 0;
// Create tasks for each one
var tasks = packagesToInstall.Select(i => Task.Run(async () =>
{
cancellationToken.ThrowIfCancellationRequested();
try
{
await _installationManager.InstallPackage(i, new Progress<double>(), cancellationToken).ConfigureAwait(false);
2013-02-21 01:33:05 +00:00
}
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);
2013-02-21 01:33:05 +00:00
}
catch (IOException ex)
{
_logger.ErrorException("Error updating {0}", ex, i.name);
2013-02-21 01:33:05 +00:00
}
// Update progress
lock (progress)
{
numComplete++;
double percent = numComplete;
percent /= packagesToInstall.Count;
2013-02-22 04:23:06 +00:00
progress.Report((90 * percent) + 10);
2013-02-21 01:33:05 +00:00
}
}));
cancellationToken.ThrowIfCancellationRequested();
await Task.WhenAll(tasks).ConfigureAwait(false);
2013-02-22 04:23:06 +00:00
progress.Report(100);
2013-02-21 01:33:05 +00:00
}
/// <summary>
/// Gets the name of the task
/// </summary>
/// <value>The name.</value>
public string Name
2013-02-21 01:33:05 +00:00
{
get { return "Check for plugin updates"; }
}
/// <summary>
/// Gets the description.
/// </summary>
/// <value>The description.</value>
public string Description
2013-02-21 01:33:05 +00:00
{
get { return "Downloads and installs updates for plugins that are configured to update automatically."; }
}
public string Category
{
get { return "Application"; }
}
2013-02-21 01:33:05 +00:00
}
}