75 lines
2.3 KiB
C#
75 lines
2.3 KiB
C#
using System;
|
|
using System.IO;
|
|
using NLog;
|
|
using NzbDrone.Common;
|
|
using NzbDrone.Common.Composition;
|
|
using NzbDrone.Common.EnvironmentInfo;
|
|
using NzbDrone.Common.Instrumentation;
|
|
using NzbDrone.Common.Security;
|
|
using NzbDrone.Update.UpdateEngine;
|
|
|
|
namespace NzbDrone.Update
|
|
{
|
|
public class UpdateApp
|
|
{
|
|
private readonly IInstallUpdateService _installUpdateService;
|
|
private readonly IProcessProvider _processProvider;
|
|
private static IContainer _container;
|
|
|
|
private static readonly Logger logger = NzbDroneLogger.GetLogger();
|
|
|
|
public UpdateApp(IInstallUpdateService installUpdateService, IProcessProvider processProvider)
|
|
{
|
|
_installUpdateService = installUpdateService;
|
|
_processProvider = processProvider;
|
|
}
|
|
|
|
public static void Main(string[] args)
|
|
{
|
|
try
|
|
{
|
|
var startupArgument = new StartupArguments(args);
|
|
LogTargets.Register(startupArgument, true, true);
|
|
|
|
Console.WriteLine("Starting NzbDrone Update Client");
|
|
|
|
IgnoreCertErrorPolicy.Register();
|
|
|
|
GlobalExceptionHandlers.Register();
|
|
|
|
_container = UpdateContainerBuilder.Build(startupArgument);
|
|
|
|
logger.Info("Updating NzbDrone to version {0}", BuildInfo.Version);
|
|
_container.Resolve<UpdateApp>().Start(args);
|
|
}
|
|
catch (Exception e)
|
|
{
|
|
logger.FatalException("An error has occurred while applying update package.", e);
|
|
}
|
|
}
|
|
|
|
public void Start(string[] args)
|
|
{
|
|
int processId = ParseProcessId(args);
|
|
|
|
var exeFileInfo = new FileInfo(_processProvider.GetProcessById(processId).StartPath);
|
|
string targetFolder = exeFileInfo.Directory.FullName;
|
|
|
|
logger.Info("Starting update process. Target Path:{0}", targetFolder);
|
|
_installUpdateService.Start(targetFolder);
|
|
}
|
|
|
|
private int ParseProcessId(string[] args)
|
|
{
|
|
int id;
|
|
if (args == null || !Int32.TryParse(args[0], out id) || id <= 0)
|
|
{
|
|
throw new ArgumentOutOfRangeException("args", "Invalid process ID");
|
|
}
|
|
|
|
logger.Debug("NzbDrone processId:{0}", id);
|
|
return id;
|
|
}
|
|
}
|
|
}
|