mirror of
https://github.com/Radarr/Radarr.git
synced 2024-09-11 12:02:35 +02:00
138 lines
4.2 KiB
C#
138 lines
4.2 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Diagnostics;
|
|
using System.Linq;
|
|
using NLog;
|
|
using NzbDrone.Common.Model;
|
|
|
|
namespace NzbDrone.Common
|
|
{
|
|
public interface IProcessProvider
|
|
{
|
|
ProcessInfo GetCurrentProcess();
|
|
ProcessInfo GetProcessById(int id);
|
|
IEnumerable<ProcessInfo> GetProcessByName(string name);
|
|
void Start(string path);
|
|
Process Start(ProcessStartInfo startInfo);
|
|
void WaitForExit(Process process);
|
|
void Kill(int processId);
|
|
void SetPriority(int processId, ProcessPriorityClass priority);
|
|
void KillAll(string processName);
|
|
}
|
|
|
|
public class ProcessProvider : IProcessProvider
|
|
{
|
|
private static readonly Logger Logger = LogManager.GetCurrentClassLogger();
|
|
|
|
public const string NzbDroneProcessName = "NzbDrone";
|
|
public const string NzbDroneConsoleProcessName = "NzbDrone.Console";
|
|
|
|
public ProcessInfo GetCurrentProcess()
|
|
{
|
|
return ConvertToProcessInfo(Process.GetCurrentProcess());
|
|
}
|
|
|
|
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 IEnumerable<ProcessInfo> GetProcessByName(string name)
|
|
{
|
|
return Process.GetProcessesByName(name).Select(ConvertToProcessInfo).Where(p => p != null);
|
|
}
|
|
|
|
public void Start(string path)
|
|
{
|
|
Process.Start(path);
|
|
}
|
|
|
|
public Process Start(ProcessStartInfo startInfo)
|
|
{
|
|
Logger.Info("Starting process. [{0}]", startInfo.FileName);
|
|
|
|
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 Kill(int processId)
|
|
{
|
|
if (processId == 0 || Process.GetProcesses().All(p => p.Id != processId))
|
|
{
|
|
Logger.Warn("Cannot find process with id: {0}", processId);
|
|
return;
|
|
}
|
|
|
|
var process = Process.GetProcessById(processId);
|
|
|
|
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);
|
|
}
|
|
|
|
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;
|
|
}
|
|
|
|
private static ProcessInfo ConvertToProcessInfo(Process process)
|
|
{
|
|
if (process == null || process.Id <= 0 || process.HasExited) return null;
|
|
|
|
return new ProcessInfo
|
|
{
|
|
Id = process.Id,
|
|
Priority = process.PriorityClass,
|
|
StartPath = process.MainModule.FileName,
|
|
Name = process.ProcessName
|
|
};
|
|
}
|
|
|
|
public void KillAll(string processName)
|
|
{
|
|
var processToKill = GetProcessByName(processName);
|
|
|
|
foreach (var processInfo in processToKill)
|
|
{
|
|
Kill(processInfo.Id);
|
|
}
|
|
}
|
|
}
|
|
} |