Fixed: Preserve startup arguments during restart

Closes #325
This commit is contained in:
Mark McDowall 2015-09-02 00:27:46 -07:00
parent 741dc8f50b
commit 5acaf9d60b
3 changed files with 22 additions and 6 deletions

View File

@ -1,5 +1,4 @@
using System.Collections.Generic;
using System.Runtime.InteropServices.ComTypes;
namespace NzbDrone.Common.EnvironmentInfo
{

View File

@ -1,7 +1,6 @@
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using NLog;
using NzbDrone.Common;
using NzbDrone.Common.Disk;

View File

@ -14,12 +14,14 @@ namespace NzbDrone.Host
{
private readonly IRuntimeInfo _runtimeInfo;
private readonly IProcessProvider _processProvider;
private readonly IStartupContext _startupContext;
private readonly Logger _logger;
public SpinService(IRuntimeInfo runtimeInfo, IProcessProvider processProvider, Logger logger)
public SpinService(IRuntimeInfo runtimeInfo, IProcessProvider processProvider, IStartupContext startupContext, Logger logger)
{
_runtimeInfo = runtimeInfo;
_processProvider = processProvider;
_startupContext = startupContext;
_logger = logger;
}
@ -30,13 +32,29 @@ namespace NzbDrone.Host
Thread.Sleep(1000);
}
_logger.Debug("wait loop was terminated.");
_logger.Debug("Wait loop was terminated.");
if (_runtimeInfo.RestartPending)
{
_logger.Info("attemptig restart.");
_processProvider.SpawnNewProcess(_runtimeInfo.ExecutingApplication, "--restart --nobrowser");
var restartArgs = GetRestartArgs();
_logger.Info("Attempting restart with arguments: {0}", restartArgs);
_processProvider.SpawnNewProcess(_runtimeInfo.ExecutingApplication, restartArgs);
}
}
private string GetRestartArgs()
{
var args = _startupContext.PreservedArguments;
args += " /restart";
if (!args.Contains("/nobrowser"))
{
args += " /nobrowser";
}
return args;
}
}
}