Radarr/NzbDrone.Core/Jobs/AppUpdateJob.cs

93 lines
3.9 KiB
C#
Raw Normal View History

using System;
using System.Linq;
2011-12-02 01:33:17 +00:00
using System.Diagnostics;
2011-11-21 02:59:42 +00:00
using System.IO;
using NLog;
using NzbDrone.Common;
using NzbDrone.Core.Model.Notification;
2011-12-02 01:33:17 +00:00
using NzbDrone.Core.Providers;
2011-11-21 02:59:42 +00:00
using NzbDrone.Core.Providers.Core;
2011-11-14 02:54:09 +00:00
2011-12-02 01:33:17 +00:00
namespace NzbDrone.Core.Jobs
2011-11-14 02:54:09 +00:00
{
public class AppUpdateJob : IJob
{
private readonly UpdateProvider _updateProvider;
private readonly EnvironmentProvider _environmentProvider;
2011-11-21 02:59:42 +00:00
private readonly DiskProvider _diskProvider;
private readonly HttpProvider _httpProvider;
private readonly ProcessProvider _processProvider;
private readonly ArchiveProvider _archiveProvider;
private readonly ConfigFileProvider _configFileProvider;
2011-11-14 02:54:09 +00:00
2011-11-21 02:59:42 +00:00
private static readonly Logger logger = LogManager.GetCurrentClassLogger();
public AppUpdateJob(UpdateProvider updateProvider, EnvironmentProvider environmentProvider, DiskProvider diskProvider,
2011-11-21 02:59:42 +00:00
HttpProvider httpProvider, ProcessProvider processProvider, ArchiveProvider archiveProvider, ConfigFileProvider configFileProvider)
2011-11-14 02:54:09 +00:00
{
_updateProvider = updateProvider;
_environmentProvider = environmentProvider;
2011-11-21 02:59:42 +00:00
_diskProvider = diskProvider;
_httpProvider = httpProvider;
_processProvider = processProvider;
_archiveProvider = archiveProvider;
_configFileProvider = configFileProvider;
2011-11-14 02:54:09 +00:00
}
public string Name
{
get { return "Update Application Job"; }
}
public TimeSpan DefaultInterval
2011-11-14 02:54:09 +00:00
{
get { return TimeSpan.FromDays(2); }
2011-11-14 02:54:09 +00:00
}
2012-09-10 19:04:17 +00:00
public virtual void Start(ProgressNotification notification, dynamic options)
{
2012-01-27 05:26:39 +00:00
notification.CurrentMessage = "Checking for updates";
2011-11-14 02:54:09 +00:00
var updatePackage = _updateProvider.GetAvilableUpdate(_environmentProvider.Version);
2011-11-14 02:54:09 +00:00
//No updates available
if (updatePackage == null)
return;
var packageDestination = Path.Combine(_environmentProvider.GetUpdateSandboxFolder(), updatePackage.FileName);
2011-11-21 02:59:42 +00:00
if (_diskProvider.FolderExists(_environmentProvider.GetUpdateSandboxFolder()))
2011-11-21 02:59:42 +00:00
{
logger.Info("Deleting old update files");
_diskProvider.DeleteFolder(_environmentProvider.GetUpdateSandboxFolder(), true);
2011-11-21 02:59:42 +00:00
}
logger.Info("Downloading update package from [{0}] to [{1}]", updatePackage.Url, packageDestination);
notification.CurrentMessage = "Downloading Update " + updatePackage.Version;
_httpProvider.DownloadFile(updatePackage.Url, packageDestination);
logger.Info("Download completed for update package from [{0}]", updatePackage.FileName);
logger.Info("Extracting Update package");
notification.CurrentMessage = "Extracting Update";
_archiveProvider.ExtractArchive(packageDestination, _environmentProvider.GetUpdateSandboxFolder());
2011-11-21 02:59:42 +00:00
logger.Info("Update package extracted successfully");
logger.Info("Preparing client");
notification.CurrentMessage = "Preparing to start Update";
_diskProvider.MoveDirectory(_environmentProvider.GetUpdateClientFolder(), _environmentProvider.GetUpdateSandboxFolder());
2011-11-21 02:59:42 +00:00
logger.Info("Starting update client");
var startInfo = new ProcessStartInfo
{
FileName = _environmentProvider.GetUpdateClientExePath(),
Arguments = string.Format("{0} {1}", _environmentProvider.NzbDroneProcessIdFromEnviroment, _configFileProvider.Guid)
};
2011-11-21 02:59:42 +00:00
var process = _processProvider.Start(startInfo);
2011-11-21 06:21:36 +00:00
notification.CurrentMessage = "Update in progress. NzbDrone will restart shortly.";
_processProvider.WaitForExit(process);
2011-11-14 02:54:09 +00:00
}
}
}