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

102 lines
3.4 KiB
C#
Raw Normal View History

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;
2014-01-14 01:35:16 +00:00
using NzbDrone.Common.Processes;
namespace NzbDrone.Common.EnvironmentInfo
{
public interface IRuntimeInfo
{
bool IsUserInteractive { get; }
bool IsAdmin { get; }
2013-11-26 06:53:36 +00:00
bool IsWindowsService { get; }
2014-01-13 01:14:57 +00:00
bool IsConsole { get; }
bool IsRunning { get; set; }
string ExecutingApplication { get; }
}
public class RuntimeInfo : IRuntimeInfo
{
private readonly Logger _logger;
2014-01-13 01:14:57 +00:00
private static readonly string ProcessName = Process.GetCurrentProcess().ProcessName.ToLower();
2013-11-26 06:53:36 +00:00
public RuntimeInfo(Logger logger, IServiceProvider serviceProvider)
{
_logger = logger;
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;
ExecutingApplication = Assembly.GetEntryAssembly().Location;
}
2014-01-13 01:14:57 +00:00
static RuntimeInfo()
{
2014-01-13 01:14:57 +00:00
IsProduction = InternalIsProduction();
}
2014-01-13 01:14:57 +00:00
public bool IsUserInteractive
2013-07-26 06:11:55 +00:00
{
2014-01-13 01:14:57 +00:00
get { return Environment.UserInteractive; }
2013-07-26 06:11:55 +00:00
}
public bool IsAdmin
{
get
{
try
{
var principal = new WindowsPrincipal(WindowsIdentity.GetCurrent());
return principal.IsInRole(WindowsBuiltInRole.Administrator);
}
catch (Exception ex)
{
_logger.WarnException("Error checking if the current user is an administrator.", ex);
return false;
}
}
}
2013-11-26 06:53:36 +00:00
public bool IsWindowsService { get; private set; }
2014-01-13 01:14:57 +00:00
public bool IsConsole
{
get
{
return (OsInfo.IsWindows &&
IsUserInteractive &&
2014-01-14 01:35:16 +00:00
ProcessName.Equals(ProcessProvider.NZB_DRONE_CONSOLE_PROCESS_NAME, StringComparison.InvariantCultureIgnoreCase)) ||
2014-01-13 01:14:57 +00:00
OsInfo.IsLinux;
}
}
public bool IsRunning { get; set; }
public string ExecutingApplication { get; private set; }
2013-07-26 06:11:55 +00:00
public static bool IsProduction { get; private set; }
private static bool InternalIsProduction()
{
2013-07-26 06:11:55 +00:00
if (BuildInfo.IsDebug || Debugger.IsAttached) return false;
if (BuildInfo.Version.Revision > 10000) return false; //Official builds will never have such a high revision
2013-07-26 06:11:55 +00:00
var lowerProcessName = 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;
string lowerCurrentDir = Directory.GetCurrentDirectory().ToLower();
if (lowerCurrentDir.Contains("teamcity")) return false;
if (lowerCurrentDir.StartsWith("/run/")) return false;
2013-07-26 06:11:55 +00:00
return true;
}
}
}