using MediaBrowser.Common.Kernel;
using System;
using System.Collections.Generic;
using System.ComponentModel.Composition;
using System.Threading;
using System.Threading.Tasks;
namespace MediaBrowser.Common.ScheduledTasks.Tasks
{
///
/// Plugin Update Task
///
[Export(typeof(IScheduledTask))]
public class SystemUpdateTask : BaseScheduledTask
{
///
/// The _app host
///
private readonly IApplicationHost _appHost;
///
/// Initializes a new instance of the class.
///
/// The app host.
[ImportingConstructor]
public SystemUpdateTask([Import("appHost")] IApplicationHost appHost)
{
_appHost = appHost;
}
///
/// Creates the triggers that define when the task will run
///
/// IEnumerable{BaseTaskTrigger}.
protected override IEnumerable GetDefaultTriggers()
{
return new BaseTaskTrigger[] {
// 1am
new DailyTrigger { TimeOfDay = TimeSpan.FromHours(1) },
new IntervalTrigger { Interval = TimeSpan.FromHours(2)}
};
}
///
/// Returns the task to be executed
///
/// The cancellation token.
/// The progress.
/// Task.
protected override async Task ExecuteInternal(CancellationToken cancellationToken, IProgress progress)
{
if (!_appHost.CanSelfUpdate) return;
EventHandler innerProgressHandler = (sender, e) => progress.Report(e * .1);
// Create a progress object for the update check
var innerProgress = new Progress();
innerProgress.ProgressChanged += innerProgressHandler;
var updateInfo = await _appHost.CheckForApplicationUpdate(cancellationToken, innerProgress).ConfigureAwait(false);
// Release the event handler
innerProgress.ProgressChanged -= innerProgressHandler;
progress.Report(10);
if (!updateInfo.IsUpdateAvailable)
{
progress.Report(100);
return;
}
cancellationToken.ThrowIfCancellationRequested();
if (Kernel.Configuration.EnableAutoUpdate)
{
Logger.Info("Update Revision {0} available. Updating...", updateInfo.AvailableVersion);
innerProgressHandler = (sender, e) => progress.Report((e * .9) + .1);
innerProgress = new Progress();
innerProgress.ProgressChanged += innerProgressHandler;
await _appHost.UpdateApplication(cancellationToken, innerProgress).ConfigureAwait(false);
// Release the event handler
innerProgress.ProgressChanged -= innerProgressHandler;
Kernel.OnApplicationUpdated(updateInfo.AvailableVersion);
}
else
{
Logger.Info("A new version of Media Browser is available.");
}
progress.Report(100);
}
///
/// Gets the name of the task
///
/// The name.
public override string Name
{
get { return "Check for application updates"; }
}
///
/// Gets the description.
///
/// The description.
public override string Description
{
get { return "Downloads and installs application updates."; }
}
}
}