Sonarr/NzbDrone.Update/UpdateApp.cs

75 lines
2.3 KiB
C#
Raw Normal View History

using System;
2011-11-14 03:09:34 +00:00
using System.IO;
2011-11-13 20:31:02 +00:00
using NLog;
using NzbDrone.Common;
2013-05-10 23:53:50 +00:00
using NzbDrone.Common.Composition;
using NzbDrone.Common.EnvironmentInfo;
2013-06-06 04:33:16 +00:00
using NzbDrone.Common.Instrumentation;
using NzbDrone.Common.Security;
2013-05-21 04:03:05 +00:00
using NzbDrone.Update.UpdateEngine;
namespace NzbDrone.Update
{
public class UpdateApp
{
2013-05-21 04:03:05 +00:00
private readonly IInstallUpdateService _installUpdateService;
2013-05-10 23:53:50 +00:00
private readonly IProcessProvider _processProvider;
private static IContainer _container;
2011-11-13 20:31:02 +00:00
2013-08-31 01:42:30 +00:00
private static readonly Logger logger = NzbDroneLogger.GetLogger();
2011-11-13 20:31:02 +00:00
public UpdateApp(IInstallUpdateService installUpdateService, IProcessProvider processProvider)
2011-11-13 20:31:02 +00:00
{
2013-05-21 04:03:05 +00:00
_installUpdateService = installUpdateService;
2011-11-13 20:31:02 +00:00
_processProvider = processProvider;
}
public static void Main(string[] args)
{
try
{
2013-08-31 01:42:30 +00:00
var startupArgument = new StartupArguments(args);
LogTargets.Register(startupArgument, true, true);
2011-11-13 20:31:02 +00:00
Console.WriteLine("Starting NzbDrone Update Client");
2013-08-16 03:32:23 +00:00
IgnoreCertErrorPolicy.Register();
2013-06-07 19:00:48 +00:00
GlobalExceptionHandlers.Register();
2013-04-20 00:05:48 +00:00
2013-08-31 01:42:30 +00:00
_container = UpdateContainerBuilder.Build(startupArgument);
2013-02-28 06:59:08 +00:00
logger.Info("Updating NzbDrone to version {0}", BuildInfo.Version);
_container.Resolve<UpdateApp>().Start(args);
2011-11-13 20:31:02 +00:00
}
catch (Exception e)
{
logger.FatalException("An error has occurred while applying update package.", e);
2011-11-13 20:31:02 +00:00
}
}
2011-11-13 20:31:02 +00:00
public void Start(string[] args)
{
int processId = ParseProcessId(args);
var exeFileInfo = new FileInfo(_processProvider.GetProcessById(processId).StartPath);
string targetFolder = exeFileInfo.Directory.FullName;
2011-11-13 20:31:02 +00:00
logger.Info("Starting update process. Target Path:{0}", targetFolder);
2013-05-21 04:03:05 +00:00
_installUpdateService.Start(targetFolder);
2011-11-13 20:31:02 +00:00
}
private int ParseProcessId(string[] args)
{
int id;
2013-05-23 05:32:54 +00:00
if (args == null || !Int32.TryParse(args[0], out id) || id <= 0)
2011-11-13 20:31:02 +00:00
{
2013-05-23 05:32:54 +00:00
throw new ArgumentOutOfRangeException("args", "Invalid process ID");
2011-11-13 20:31:02 +00:00
}
logger.Debug("NzbDrone processId:{0}", id);
2011-11-13 20:31:02 +00:00
return id;
}
}
}