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

220 lines
6.0 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.Security.Principal;
using Microsoft.Extensions.Hosting;
using Microsoft.Extensions.Hosting.WindowsServices;
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(IHostLifetime hostLifetime, Logger logger)
{
_logger = logger;
2013-11-26 06:53:36 +00:00
IsWindowsService = hostLifetime is WindowsServiceLifetime;
2021-12-24 17:40:37 +00:00
// net6.0 will return Lidarr.dll for entry assembly, we need the actual
// executable name (Lidarr on linux). On mono this will return the location of
// the mono executable itself, which is not what we want.
var entry = Process.GetCurrentProcess().MainModule;
2014-02-10 19:40:13 +00:00
if (entry != null)
{
ExecutingApplication = entry.FileName;
IsWindowsTray = OsInfo.IsWindows && entry.ModuleName == $"{ProcessProvider.LIDARR_PROCESS_NAME}.exe";
}
}
static RuntimeInfo()
{
var officialBuild = InternalIsOfficialBuild();
// An build running inside of the testing environment. (Analytics disabled)
IsTesting = InternalIsTesting();
// An official build running outside of the testing environment. (Analytics configurable)
IsProduction = !IsTesting && officialBuild;
// An unofficial build running outside of the testing environment. (Analytics enabled)
IsDevelopment = !IsTesting && !officialBuild && !InternalIsDebug();
}
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
{
if (OsInfo.IsNotWindows)
{
return false;
}
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)
{
2017-09-27 02:06:05 +00:00
return IsUserInteractive && Process.GetCurrentProcess().ProcessName.Equals(ProcessProvider.LIDARR_PROCESS_NAME, StringComparison.InvariantCultureIgnoreCase);
2017-09-04 02:20:56 +00:00
}
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 IsTesting { get; }
public static bool IsProduction { get; }
public static bool IsDevelopment { get; }
2013-07-26 06:11:55 +00:00
private static bool InternalIsTesting()
{
try
{
var lowerProcessName = Process.GetCurrentProcess().ProcessName.ToLower();
if (lowerProcessName.Contains("vshost"))
{
return true;
}
if (lowerProcessName.Contains("nunit"))
{
return true;
}
if (lowerProcessName.Contains("jetbrain"))
{
return true;
}
if (lowerProcessName.Contains("resharper"))
{
return true;
}
}
catch
{
}
try
{
var currentAssemblyLocation = typeof(RuntimeInfo).Assembly.Location;
if (currentAssemblyLocation.ToLower().Contains("_output"))
{
return true;
}
if (currentAssemblyLocation.ToLower().Contains("_tests"))
{
return true;
}
}
catch
{
}
var lowerCurrentDir = Directory.GetCurrentDirectory().ToLower();
if (lowerCurrentDir.Contains("vsts"))
{
return true;
}
if (lowerCurrentDir.Contains("buildagent"))
{
return true;
}
if (lowerCurrentDir.Contains("_output"))
{
return true;
}
if (lowerCurrentDir.Contains("_tests"))
{
return true;
}
return false;
}
private static bool InternalIsDebug()
{
if (BuildInfo.IsDebug || Debugger.IsAttached)
{
return true;
}
return false;
}
private static bool InternalIsOfficialBuild()
{
//Official builds will never have such a high revision
if (BuildInfo.Version.Major >= 10 || BuildInfo.Version.Revision > 10000)
{
return false;
}
2013-07-26 06:11:55 +00:00
return true;
}
public bool IsWindowsTray { get; private set; }
}
}