2017-10-25 02:31:37 +00:00
|
|
|
using NLog;
|
2011-10-23 05:26:43 +00:00
|
|
|
using NzbDrone.Common;
|
2017-10-25 02:31:37 +00:00
|
|
|
using NzbDrone.Common.EnvironmentInfo;
|
|
|
|
using NzbDrone.Common.Processes;
|
2018-04-27 00:56:19 +00:00
|
|
|
using NzbDrone.Host.AccessControl;
|
2017-10-25 02:31:37 +00:00
|
|
|
using IServiceProvider = NzbDrone.Common.IServiceProvider;
|
|
|
|
|
2013-08-07 05:32:22 +00:00
|
|
|
namespace NzbDrone.Host
|
2011-10-09 02:16:11 +00:00
|
|
|
{
|
2021-08-04 20:42:41 +00:00
|
|
|
public interface IUtilityModeRouter
|
|
|
|
{
|
|
|
|
void Route(ApplicationModes applicationModes);
|
|
|
|
}
|
|
|
|
|
|
|
|
public class UtilityModeRouter : IUtilityModeRouter
|
2011-10-09 02:16:11 +00:00
|
|
|
{
|
2013-05-10 23:53:50 +00:00
|
|
|
private readonly IServiceProvider _serviceProvider;
|
2013-04-16 04:52:41 +00:00
|
|
|
private readonly IConsoleService _consoleService;
|
2017-10-25 02:31:37 +00:00
|
|
|
private readonly IProcessProvider _processProvider;
|
2018-04-27 00:56:19 +00:00
|
|
|
private readonly IRemoteAccessAdapter _remoteAccessAdapter;
|
2023-03-27 03:53:10 +00:00
|
|
|
private readonly IAppFolderFactory _appFolderFactory;
|
2013-03-01 00:50:50 +00:00
|
|
|
private readonly Logger _logger;
|
2011-10-09 02:16:11 +00:00
|
|
|
|
2021-08-04 20:42:41 +00:00
|
|
|
public UtilityModeRouter(IServiceProvider serviceProvider,
|
2017-10-25 02:31:37 +00:00
|
|
|
IConsoleService consoleService,
|
|
|
|
IProcessProvider processProvider,
|
2018-04-27 00:56:19 +00:00
|
|
|
IRemoteAccessAdapter remoteAccessAdapter,
|
2023-03-27 03:53:10 +00:00
|
|
|
IAppFolderFactory appFolderFactory,
|
2017-10-25 02:31:37 +00:00
|
|
|
Logger logger)
|
2011-10-09 02:16:11 +00:00
|
|
|
{
|
|
|
|
_serviceProvider = serviceProvider;
|
2013-04-16 04:52:41 +00:00
|
|
|
_consoleService = consoleService;
|
2017-10-25 02:31:37 +00:00
|
|
|
_processProvider = processProvider;
|
2018-04-27 00:56:19 +00:00
|
|
|
_remoteAccessAdapter = remoteAccessAdapter;
|
2023-03-27 03:53:10 +00:00
|
|
|
_appFolderFactory = appFolderFactory;
|
2013-03-01 00:50:50 +00:00
|
|
|
_logger = logger;
|
2011-10-09 02:16:11 +00:00
|
|
|
}
|
|
|
|
|
2013-03-01 00:50:50 +00:00
|
|
|
public void Route(ApplicationModes applicationModes)
|
2011-10-14 01:22:51 +00:00
|
|
|
{
|
2013-03-01 00:50:50 +00:00
|
|
|
_logger.Info("Application mode: {0}", applicationModes);
|
2013-01-15 23:17:12 +00:00
|
|
|
|
2013-03-01 00:50:50 +00:00
|
|
|
switch (applicationModes)
|
2013-01-06 08:11:14 +00:00
|
|
|
{
|
2013-03-01 00:50:50 +00:00
|
|
|
case ApplicationModes.InstallService:
|
2013-01-06 08:11:14 +00:00
|
|
|
{
|
2014-03-13 20:12:42 +00:00
|
|
|
_logger.Debug("Install Service selected");
|
2017-09-27 02:06:05 +00:00
|
|
|
if (_serviceProvider.ServiceExist(ServiceProvider.SERVICE_NAME))
|
2011-10-14 01:22:51 +00:00
|
|
|
{
|
2013-04-16 04:52:41 +00:00
|
|
|
_consoleService.PrintServiceAlreadyExist();
|
2011-10-14 01:22:51 +00:00
|
|
|
}
|
2013-01-06 08:11:14 +00:00
|
|
|
else
|
2011-10-14 01:22:51 +00:00
|
|
|
{
|
2018-04-27 00:56:19 +00:00
|
|
|
_remoteAccessAdapter.MakeAccessible(true);
|
2017-09-27 02:06:05 +00:00
|
|
|
_serviceProvider.Install(ServiceProvider.SERVICE_NAME);
|
2017-10-25 02:31:37 +00:00
|
|
|
_serviceProvider.SetPermissions(ServiceProvider.SERVICE_NAME);
|
|
|
|
|
|
|
|
// Start the service and exit.
|
2017-10-31 02:25:19 +00:00
|
|
|
// Ensures that there isn't an instance of Lidarr already running that the service account cannot stop.
|
2017-10-25 02:31:37 +00:00
|
|
|
_processProvider.SpawnNewProcess("sc.exe", $"start {ServiceProvider.SERVICE_NAME}", null, true);
|
2011-10-14 01:22:51 +00:00
|
|
|
}
|
2020-01-03 12:49:24 +00:00
|
|
|
|
2013-01-06 08:11:14 +00:00
|
|
|
break;
|
|
|
|
}
|
2020-01-03 12:49:24 +00:00
|
|
|
|
2013-03-01 00:50:50 +00:00
|
|
|
case ApplicationModes.UninstallService:
|
2013-01-06 08:11:14 +00:00
|
|
|
{
|
2014-03-13 20:12:42 +00:00
|
|
|
_logger.Debug("Uninstall Service selected");
|
2017-09-27 02:06:05 +00:00
|
|
|
if (!_serviceProvider.ServiceExist(ServiceProvider.SERVICE_NAME))
|
2011-10-14 01:22:51 +00:00
|
|
|
{
|
2013-07-09 00:47:09 +00:00
|
|
|
_consoleService.PrintServiceDoesNotExist();
|
2011-10-14 01:22:51 +00:00
|
|
|
}
|
2013-01-06 08:11:14 +00:00
|
|
|
else
|
2011-10-14 01:22:51 +00:00
|
|
|
{
|
2017-10-25 02:31:37 +00:00
|
|
|
_serviceProvider.Uninstall(ServiceProvider.SERVICE_NAME);
|
2011-10-14 01:22:51 +00:00
|
|
|
}
|
2013-01-06 08:11:14 +00:00
|
|
|
|
2018-04-27 00:56:19 +00:00
|
|
|
break;
|
|
|
|
}
|
2020-01-03 12:49:24 +00:00
|
|
|
|
2018-04-27 00:56:19 +00:00
|
|
|
case ApplicationModes.RegisterUrl:
|
|
|
|
{
|
2022-10-30 21:53:02 +00:00
|
|
|
_logger.Debug("Register URL selected");
|
2018-04-27 00:56:19 +00:00
|
|
|
_remoteAccessAdapter.MakeAccessible(false);
|
2023-03-27 03:53:10 +00:00
|
|
|
_appFolderFactory.SetPermissions();
|
2018-04-27 00:56:19 +00:00
|
|
|
|
2013-01-06 08:11:14 +00:00
|
|
|
break;
|
|
|
|
}
|
2020-01-03 12:49:24 +00:00
|
|
|
|
2013-01-06 08:11:14 +00:00
|
|
|
default:
|
|
|
|
{
|
2013-04-16 04:52:41 +00:00
|
|
|
_consoleService.PrintHelp();
|
2013-01-06 08:11:14 +00:00
|
|
|
break;
|
|
|
|
}
|
2011-10-14 01:22:51 +00:00
|
|
|
}
|
|
|
|
}
|
2011-10-09 02:16:11 +00:00
|
|
|
}
|
|
|
|
}
|