Lidarr/src/NzbDrone.Common/EnvironmentInfo/RuntimeInfo.cs

151 lines
4.3 KiB
C#
Raw Normal View History

2017-09-04 02:20:56 +00:00
using System;
using System.Diagnostics;
2013-07-26 06:11:55 +00:00
using System.IO;
using System.Reflection;
using System.Security.Principal;
2013-11-26 06:53:36 +00:00
using System.ServiceProcess;
using NLog;
2017-09-04 02:20:56 +00:00
using NzbDrone.Common.Processes;
namespace NzbDrone.Common.EnvironmentInfo
{
public class RuntimeInfo : IRuntimeInfo
{
private readonly Logger _logger;
2017-09-04 02:20:56 +00:00
private readonly DateTime _startTime = DateTime.UtcNow;
public RuntimeInfo(IServiceProvider serviceProvider, Logger logger)
{
_logger = logger;
2017-09-04 02:20:56 +00:00
2013-11-26 06:53:36 +00:00
IsWindowsService = !IsUserInteractive &&
2014-01-13 01:14:57 +00:00
OsInfo.IsWindows &&
serviceProvider.ServiceExist(ServiceProvider.NZBDRONE_SERVICE_NAME) &&
serviceProvider.GetStatus(ServiceProvider.NZBDRONE_SERVICE_NAME) == ServiceControllerStatus.StartPending;
//Guarded to avoid issues when running in a non-managed process
2014-02-10 19:40:13 +00:00
var entry = Assembly.GetEntryAssembly();
if (entry != null)
{
ExecutingApplication = entry.Location;
}
}
static RuntimeInfo()
{
2014-01-13 01:14:57 +00:00
IsProduction = InternalIsProduction();
}
2017-09-04 02:20:56 +00:00
public DateTime StartTime
{
get
{
return _startTime;
}
}
2016-12-09 06:54:15 +00:00
public static bool IsUserInteractive => Environment.UserInteractive;
2013-07-26 06:11:55 +00:00
2016-12-09 06:54:15 +00:00
bool IRuntimeInfo.IsUserInteractive => IsUserInteractive;
public bool IsAdmin
{
get
{
try
{
var principal = new WindowsPrincipal(WindowsIdentity.GetCurrent());
return principal.IsInRole(WindowsBuiltInRole.Administrator);
}
catch (Exception ex)
{
2016-02-11 21:13:42 +00:00
_logger.Warn(ex, "Error checking if the current user is an administrator.");
return false;
}
}
}
2013-11-26 06:53:36 +00:00
public bool IsWindowsService { get; private set; }
public bool IsExiting { get; set; }
2017-09-04 02:20:56 +00:00
public bool IsTray
{
get
{
if (OsInfo.IsWindows)
{
return IsUserInteractive && Process.GetCurrentProcess().ProcessName.Equals(ProcessProvider.NZB_DRONE_PROCESS_NAME, StringComparison.InvariantCultureIgnoreCase);
}
return false;
}
}
public RuntimeMode Mode
{
get
{
if (IsWindowsService)
{
return RuntimeMode.Service;
}
if (IsTray)
{
return RuntimeMode.Tray;
}
return RuntimeMode.Console;
}
}
public bool RestartPending { get; set; }
public string ExecutingApplication { get; }
public static bool IsProduction { get; }
2013-07-26 06:11:55 +00:00
private static bool InternalIsProduction()
{
2013-07-26 06:11:55 +00:00
if (BuildInfo.IsDebug || Debugger.IsAttached) return false;
//Official builds will never have such a high revision
if (BuildInfo.Version.Revision > 10000) return false;
try
{
var lowerProcessName = Process.GetCurrentProcess().ProcessName.ToLower();
if (lowerProcessName.Contains("vshost")) return false;
if (lowerProcessName.Contains("nunit")) return false;
if (lowerProcessName.Contains("jetbrain")) return false;
if (lowerProcessName.Contains("resharper")) return false;
}
catch
{
}
try
{
var currentAssemblyLocation = typeof(RuntimeInfo).Assembly.Location;
if (currentAssemblyLocation.ToLower().Contains("_output")) return false;
}
catch
{
}
var lowerCurrentDir = Directory.GetCurrentDirectory().ToLower();
if (lowerCurrentDir.Contains("teamcity")) return false;
if (lowerCurrentDir.Contains("buildagent")) return false;
if (lowerCurrentDir.Contains("_output")) return false;
2013-07-26 06:11:55 +00:00
return true;
}
}
}