mirror of
https://github.com/Radarr/Radarr.git
synced 2024-09-11 03:52:33 +02:00
252 lines
7.4 KiB
C#
252 lines
7.4 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.ComponentModel;
|
|
using System.Diagnostics;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using NLog;
|
|
using NzbDrone.Common.EnvironmentInfo;
|
|
using NzbDrone.Common.Model;
|
|
|
|
namespace NzbDrone.Common
|
|
{
|
|
public interface IProcessProvider
|
|
{
|
|
ProcessInfo GetCurrentProcess();
|
|
ProcessInfo GetProcessById(int id);
|
|
void OpenDefaultBrowser(string url);
|
|
void WaitForExit(Process process);
|
|
void SetPriority(int processId, ProcessPriorityClass priority);
|
|
void KillAll(string processName);
|
|
bool Exists(string processName);
|
|
ProcessPriorityClass GetCurrentProcessPriority();
|
|
Process Start(string path, string args = null, Action<string> onOutputDataReceived = null, Action<string> onErrorDataReceived = null);
|
|
Process SpawnNewProcess(string path, string args = null);
|
|
}
|
|
|
|
public class ProcessProvider : IProcessProvider
|
|
{
|
|
private static readonly Logger Logger = LogManager.GetCurrentClassLogger();
|
|
|
|
public const string NZB_DRONE_PROCESS_NAME = "NzbDrone";
|
|
public const string NZB_DRONE_CONSOLE_PROCESS_NAME = "NzbDrone.Console";
|
|
|
|
private static List<Process> GetProcessesByName(string name)
|
|
{
|
|
var monoProcesses = Process.GetProcessesByName("mono")
|
|
.Where(process => process.Modules.Cast<ProcessModule>().Any(module => module.ModuleName.ToLower() == name.ToLower() + ".exe"));
|
|
return Process.GetProcessesByName(name)
|
|
.Union(monoProcesses).ToList();
|
|
}
|
|
|
|
public ProcessInfo GetCurrentProcess()
|
|
{
|
|
return ConvertToProcessInfo(Process.GetCurrentProcess());
|
|
}
|
|
|
|
public bool Exists(string processName)
|
|
{
|
|
return GetProcessesByName(processName).Any();
|
|
}
|
|
|
|
public ProcessPriorityClass GetCurrentProcessPriority()
|
|
{
|
|
return Process.GetCurrentProcess().PriorityClass;
|
|
}
|
|
|
|
public ProcessInfo GetProcessById(int id)
|
|
{
|
|
Logger.Trace("Finding process with Id:{0}", id);
|
|
|
|
var processInfo = ConvertToProcessInfo(Process.GetProcesses().FirstOrDefault(p => p.Id == id));
|
|
|
|
if (processInfo == null)
|
|
{
|
|
Logger.Warn("Unable to find process with ID {0}", id);
|
|
}
|
|
else
|
|
{
|
|
Logger.Trace("Found process {0}", processInfo.ToString());
|
|
}
|
|
|
|
return processInfo;
|
|
}
|
|
|
|
public void OpenDefaultBrowser(string url)
|
|
{
|
|
Logger.Info("Opening URL [{0}]", url);
|
|
|
|
var process = new Process
|
|
{
|
|
StartInfo = new ProcessStartInfo(url)
|
|
{
|
|
UseShellExecute = true
|
|
}
|
|
};
|
|
|
|
process.Start();
|
|
}
|
|
|
|
|
|
|
|
public Process Start(string path, string args = null, Action<string> onOutputDataReceived = null, Action<string> onErrorDataReceived = null)
|
|
{
|
|
|
|
if (OsInfo.IsMono && path.EndsWith(".exe", StringComparison.InvariantCultureIgnoreCase))
|
|
{
|
|
args = path + " " + args;
|
|
path = "mono";
|
|
}
|
|
|
|
var logger = LogManager.GetLogger(new FileInfo(path).Name);
|
|
|
|
var startInfo = new ProcessStartInfo(path, args)
|
|
{
|
|
CreateNoWindow = true,
|
|
UseShellExecute = false,
|
|
RedirectStandardError = true,
|
|
RedirectStandardOutput = true,
|
|
RedirectStandardInput = true
|
|
};
|
|
|
|
|
|
logger.Info("Starting {0} {1}", path, args);
|
|
|
|
var process = new Process
|
|
{
|
|
StartInfo = startInfo
|
|
};
|
|
|
|
process.OutputDataReceived += (sender, eventArgs) =>
|
|
{
|
|
if (string.IsNullOrWhiteSpace(eventArgs.Data)) return;
|
|
|
|
logger.Debug(eventArgs.Data);
|
|
|
|
if (onOutputDataReceived != null)
|
|
{
|
|
onOutputDataReceived(eventArgs.Data);
|
|
}
|
|
};
|
|
|
|
process.ErrorDataReceived += (sender, eventArgs) =>
|
|
{
|
|
if (string.IsNullOrWhiteSpace(eventArgs.Data)) return;
|
|
|
|
logger.Error(eventArgs.Data);
|
|
|
|
if (onErrorDataReceived != null)
|
|
{
|
|
onErrorDataReceived(eventArgs.Data);
|
|
}
|
|
};
|
|
|
|
process.Start();
|
|
|
|
process.BeginErrorReadLine();
|
|
process.BeginOutputReadLine();
|
|
|
|
|
|
return process;
|
|
}
|
|
|
|
public Process SpawnNewProcess(string path, string args = null)
|
|
{
|
|
if (OsInfo.IsMono && path.EndsWith(".exe", StringComparison.InvariantCultureIgnoreCase))
|
|
{
|
|
args = path + " " + args;
|
|
path = "mono";
|
|
}
|
|
|
|
Logger.Info("Starting {0} {1}", path, args);
|
|
|
|
var startInfo = new ProcessStartInfo(path, args);
|
|
var process = new Process
|
|
{
|
|
StartInfo = startInfo
|
|
};
|
|
|
|
process.Start();
|
|
|
|
return process;
|
|
}
|
|
|
|
public void WaitForExit(Process process)
|
|
{
|
|
Logger.Trace("Waiting for process {0} to exit.", process.ProcessName);
|
|
process.WaitForExit();
|
|
}
|
|
|
|
public void SetPriority(int processId, ProcessPriorityClass priority)
|
|
{
|
|
var process = Process.GetProcessById(processId);
|
|
|
|
Logger.Info("Updating [{0}] process priority from {1} to {2}",
|
|
process.ProcessName,
|
|
process.PriorityClass,
|
|
priority);
|
|
|
|
process.PriorityClass = priority;
|
|
}
|
|
|
|
public void KillAll(string processName)
|
|
{
|
|
var processToKill = GetProcessesByName(processName);
|
|
|
|
foreach (var processInfo in processToKill)
|
|
{
|
|
Kill(processInfo.Id);
|
|
}
|
|
}
|
|
|
|
private static ProcessInfo ConvertToProcessInfo(Process process)
|
|
{
|
|
if (process == null) return null;
|
|
|
|
process.Refresh();
|
|
|
|
try
|
|
{
|
|
if (process.Id <= 0 || process.HasExited) return null;
|
|
|
|
return new ProcessInfo
|
|
{
|
|
Id = process.Id,
|
|
StartPath = process.MainModule.FileName,
|
|
Name = process.ProcessName
|
|
};
|
|
}
|
|
catch (Win32Exception)
|
|
{
|
|
Logger.Warn("Coudn't get process info for " + process.ProcessName);
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
private void Kill(int processId)
|
|
{
|
|
var process = Process.GetProcesses().FirstOrDefault(p => p.Id == processId);
|
|
|
|
if (process == null)
|
|
{
|
|
Logger.Warn("Cannot find process with id: {0}", processId);
|
|
return;
|
|
}
|
|
|
|
process.Refresh();
|
|
|
|
if (process.HasExited)
|
|
{
|
|
return;
|
|
}
|
|
|
|
Logger.Info("[{0}]: Killing process", process.Id);
|
|
process.Kill();
|
|
Logger.Info("[{0}]: Waiting for exit", process.Id);
|
|
process.WaitForExit();
|
|
Logger.Info("[{0}]: Process terminated successfully", process.Id);
|
|
}
|
|
}
|
|
}
|