Relaunch-Process/MainWindow.cs

using System;
using System.ComponentModel;
using System.Diagnostics;
using System.Runtime.InteropServices;
using System.Security.Principal;
using System.Threading;
using System.Windows.Forms;
using RelaunchProcess.Properties;
using static System.Windows.Forms.VisualStyles.VisualStyleElement;
using Discord;
using Discord.Webhook;
using System.Data;
using static System.Net.Mime.MediaTypeNames;
using RelaunchProcess;
using System.Timers;
using System.Threading.Tasks;
using System.Runtime.CompilerServices;
using System.Configuration;

namespace Process_Auto_Relaunch
{
    public partial class MainWindow : Form
    {
        [Flags]
        public enum NotifyLevel
        {
            logNone = 0,
            logAlways = 1,          // писать везде
            logUpdateStatus = 2,    // писать в строке состояния
            logHistory = 4,         // писать в окне истории перезапусков
            logDiscord = 8          // писать в Дискорд
        }
        private delegate void UpdateLogDelegate(string text, NotifyLevel level = NotifyLevel.logUpdateStatus);
        private readonly UpdateLogDelegate updateLogDelegate;
        public DiscordWebhook dwhHook;
        private DiscordMessage dwhMessage;
        private Process WatchedProcess;
        private double cpuLastTime = 0;
        private Stopwatch cpuMeasureTimer;
        private System.Timers.Timer waitResponceTimer;

        /// <summary>
        /// Процесс для наблюдения
        /// </summary>
        public string ProcessName { get { return textBoxProcessName.Text; } set { textBoxProcessName.Text = value; } }

        public MainWindow()
        {
            InitializeComponent();
            if (Settings.Default.upgradeSettings)
            {
                Debug.WriteLine("Обновление настроек.");
                try
                {
                    Settings.Default.Upgrade();
                }
                catch (ConfigurationErrorsException ex)
                {
                    Debug.WriteLine($"Ошибка обновления настроек: {ex.Message}");
                    Settings.Default.Reset();
                }
                finally
                {
                    Settings.Default.upgradeSettings = false;
                    Settings.Default.Save();
                }
            }
            this.updateLogDelegate = this.UpdateStatus;
            this.updateLogDelegate += this.SendDiscordMessage;
            this.updateLogDelegate += this.HistoryLog;
            myBackgroundWorker.WorkerSupportsCancellation = true;
            tipProgramStartPath.SetToolTip(this.labelProgramStartPath, Settings.Default.startProgramPath);
            dwhHook = new DiscordWebhook();
            cpuMeasureTimer = new Stopwatch();
        }

        /// <summary>
        /// Событие запуска формы
        /// </summary>
        /// <param name="sender"></param>
        /// <param name="e"></param>
        private void Form1_Load(object sender, EventArgs e)
        {
            LoadOldState();
            CheckProgramState();
        }

        /// <summary>
        /// Восстановление настроек
        /// </summary>
        private void LoadOldState()
        {
            if (Settings.Default.saveOldState)
            {
                radioButtonEnableWathing.Checked = Settings.Default.enableWatching;
            }

        }

        /// <summary>
        /// Метод для события отключения
        /// </summary>
        /// <param name="sender"></param>
        /// <param name="e"></param>
        private void radioButtonDisableWathing_CheckedChanged(object sender, EventArgs e)
        {
            CheckProgramState();

            if (!radioButtonDisableWathing.Checked)
            {
                return;
            }

            if (myBackgroundWorker.WorkerSupportsCancellation && myBackgroundWorker.IsBusy)
            {
                myBackgroundWorker.CancelAsync();
                UpdateStatus("Отменяем...", NotifyLevel.logUpdateStatus);
            }
        }

        /// <summary>
        /// Метод для события включения
        /// </summary>
        /// <param name="sender"></param>
        /// <param name="e"></param>
        private void radioButtonEnableWathing_CheckedChanged(object sender, EventArgs e)
        {
            if (!radioButtonEnableWathing.Checked)
            {
                return;
            }
            bool error = false;

            if (String.IsNullOrEmpty(textBoxProcessName.Text))
            {
                error = true;
                MessageBox.Show("Имя процесса не может быть пустым!" +
                    "\nУкажите имя процесса", "Ошибка", MessageBoxButtons.OK, MessageBoxIcon.Error);
            }

            if (String.IsNullOrEmpty(Settings.Default.startProgramPath))
            {
                error = true;
                MessageBox.Show("Программа для запуска не указана." +
                    "\nУкажите программу для запуска", "Ошибка", MessageBoxButtons.OK, MessageBoxIcon.Error);
            }

            if (error)
            {
                radioButtonEnableWathing.Checked = false;
                radioButtonDisableWathing.Checked = true;
                return;
            }

            if (!myBackgroundWorker.IsBusy)
            {
                myBackgroundWorker.RunWorkerAsync();
                Status($"Запуск наблюдения за процессом {ProcessName}", NotifyLevel.logDiscord);
            }
        }

        /// <summary>
        /// Обновление статуса в программе
        /// </summary>
        /// <param name="text">Текст для отображения/отправки </param>
        /// <param name="level">Флаги для назначения отправки</param>
        public void UpdateStatus(string text, NotifyLevel level)
        {
            if (!level.HasFlag(NotifyLevel.logAlways) && !level.HasFlag(NotifyLevel.logUpdateStatus)) return;
            labelStatus.Text = text;
        }

        /// <summary>
        /// Добавление строки в Истории Запусков
        /// </summary>
        /// <param name="text">Текст для отображения/отправки </param>
        /// <param name="level">Флаги для назначения отправки</param>
        private void HistoryLog(string text, NotifyLevel level)
        {
            if (!level.HasFlag(NotifyLevel.logAlways) && !level.HasFlag(NotifyLevel.logHistory)) return;
            richTextBoxHistory.Text += DateTime.Now.ToString() + ": " + text + "\n";
        }

        /// <summary>
        /// Отправка статуса в Discord
        /// </summary>
        /// <param name="text">Текст для отображения/отправки </param>
        /// <param name="level">Флаги для назначения отправки</param>
        public void SendDiscordMessage(string text, NotifyLevel level)
        {
            if (!level.HasFlag(NotifyLevel.logAlways) && !level.HasFlag(NotifyLevel.logDiscord)) return;
            if (Settings.Default.dwhEnabled)
            {
                dwhHook.Url = Settings.Default.dwhURL;
                dwhMessage.Username = Settings.Default.dwhBotname;
                dwhMessage.AvatarUrl = Settings.Default.dwhAvatarURL;
                dwhMessage.Content = ":arrows_counterclockwise: " + text;
                try
                {
                    dwhHook.Send(dwhMessage);
                }
                catch (Exception ex)
                {
                    Status($"Ошибка отправки в дискорд.", NotifyLevel.logHistory);
                    Debug.WriteLine($"Discord messaging error: {ex.Message}");
                    //Settings.Default.dwhEnabled = false;
                    //Settings.Default.Save();
                }
            }
        }

        /// <summary>
        /// Обновление статуса в программе
        /// </summary>
        /// <param name="text">Текст для отображения/отправки </param>
        /// <param name="level">Флаги для назначения отправки</param>
        public void Status(string text, NotifyLevel level = NotifyLevel.logUpdateStatus)
        {
            Invoke(updateLogDelegate, text, level);
        }

        private void CheckProgramState()
        {
            bool watching = radioButtonEnableWathing.Checked;
            Debug.WriteLine($"Наблюдение: {watching}");

            groupBoxProcessName.Enabled = !watching;
            groupBoxProgramStart.Enabled = !watching;
            groupBoxActions.Enabled = !watching;
            // btnShowDiscordSettings.Enabled = !watching; //отключаем кнопку настроек дискорда
            // webhookDiscordToolStripMenuItem.Enabled = !watching;
            // отключаем меню настроек
            settingsToolStripMenuItem.Enabled = !watching;

            Settings.Default.enableWatching = watching;
        }

        /// <summary>
        /// Выбор файла для запуска
        /// </summary>
        /// <param name="sender"></param>
        /// <param name="e"></param>
        private void ButtonSetProgramStart_Click(object sender, EventArgs e)
        {
            OpenFileDialog openFile = new OpenFileDialog();
            openFile.Filter = "Исполняемые файлы (*.exe)|*.exe";
            openFile.Title = "Укажите программу запуска";

            if (openFile.ShowDialog() == DialogResult.Cancel)
            {
                return;
            }

            int lastSlash = openFile.FileName.LastIndexOf("\\");
            textBoxProcessName.Text = openFile.FileName.Substring(lastSlash + 1);
            textBoxProcessName.Text = textBoxProcessName.Text.Remove(textBoxProcessName.Text.Length - 4);
            Settings.Default.startProgramPath = openFile.FileName;
            Settings.Default.Save();
            tipProgramStartPath.SetToolTip(this.labelProgramStartPath, Settings.Default.startProgramPath);
            openFile.Dispose();
        }

        /// <summary>
        /// Событие перед закрытием формы
        /// </summary>
        /// <param name="sender"></param>
        /// <param name="e"></param>
        private void Form1_FormClosing(object sender, FormClosingEventArgs e)
        {
            dwhHook.CancelProcessing();
            Status("Наблюдение отменено - приложение закрыто.", NotifyLevel.logAlways);
            Settings.Default.Save();
        }

        private bool ProcessByNameIsRuning(string name)
        {
            var sessionid = Process.GetCurrentProcess().SessionId;
            var processes = Process.GetProcessesByName(name);
            foreach (var process in processes)
            {
                Debug.WriteLine($"Found proces: {process.ProcessName}. Session Id: {process.SessionId}. Current Session Id: {sessionid}");
                if (process.SessionId == sessionid)
                {
                    WatchedProcess = process;
                    return true;
                }
            }
            Debug.WriteLine($"Process {name} for current session id {sessionid} not found");
            return false;
        }

        /// <summary>
        /// Обнаружение зависшего процесса по нагрузке
        /// </summary>
        /// <param name="cpuResponding"></param>
        /// <param name="cpuPrecent"></param>
        private void ProcessCheckResponding(bool cpuResponding, double cpuPrecent)
        {
            if (cpuResponding || cpuPrecent > 0.01)
            {
                // Процесс отвечает
                // Если таймер ожидания запущен - отключаем.
                if (TimerResponceRuning())
                {
                    StopTimerWaitingResponce();
                }
                return;
            }

            // Процесс не отвечает
            // Если таймер ожидания не запущен - запускаем
            if (!TimerResponceRuning())
            {
                StartTimerWaitingResponce(5000, ProcessNotResponding);
            }
        }

        /// <summary>
        /// Проверяет запущен ли таймер ожидания процесса
        /// </summary>
        /// <returns>True, если запущен</returns>
        private bool TimerResponceRuning()
        {
            return waitResponceTimer != null /*&& waitResponceTimer.Enabled*/;
        }

        /// <summary>
        /// Метод запуска таймера для ожидания процесса
        /// </summary>
        /// <param name="interval">Интервал таймера</param>
        /// <param name="timerElapsed">Метод, вызываемый после окончания таймера</param>
        private void StartTimerWaitingResponce(double interval, ElapsedEventHandler timerElapsed)
        {
            Debug.WriteLine($"Запуск таймера {interval} ms.");
            waitResponceTimer = new System.Timers.Timer(interval);
            waitResponceTimer.Elapsed += timerElapsed;
            waitResponceTimer.AutoReset = false;
            waitResponceTimer.Enabled = true;
        }

        /// <summary>
        /// Остановка таймера ожидания процесса
        /// </summary>
        private void StopTimerWaitingResponce()
        {
            Debug.WriteLine("Остановка таймера ожидания процесса.");
            waitResponceTimer.Dispose();
            waitResponceTimer = null;
        }

        /// <summary>
        /// Метод события для таймера если процесс не отвечает
        /// </summary>
        private void ProcessNotResponding(Object source, ElapsedEventArgs e)
        {
            Debug.WriteLine("Таймер ожидания ответа процесса вышел.");
            Status($"Процесс {ProcessName} не отвечает уже какое-то время, возможно завис!", NotifyLevel.logHistory | NotifyLevel.logDiscord);

            // Если процесс не завершаем, тогда только информируем
            if (!checkBoxCloseFreezeProcess.Checked)
            {
                Status("@everyone Завершение процесса отключено. Сделайте что-нибудь!", NotifyLevel.logHistory | NotifyLevel.logDiscord);
                return;
            }

            // Пытаемся самостоятельно завершить процесс
            // ! Обнаружение зависшего процесса может быть некорректным !
            if (ProcessByNameIsRuning(textBoxProcessName.Text))
            {
                ProcessKill(WatchedProcess);
            }
        }

        private void ProcessStart(string path, string args)
        {
            if (ProcessByNameIsRuning(path))
            {
                return;
            }

            // Процесс не запущен
            WatchedProcess = Process.Start(path, args);
            cpuLastTime = 0;
            cpuMeasureTimer.Start();
            Status($"Процесс {ProcessName} запущен.", NotifyLevel.logAlways);
        }

        /// <summary>
        /// Метод принудительного завершения процесса
        /// </summary>
        /// <param name="process"></param>
        private void ProcessKill(Process process)
        {
            if (process == null || process.HasExited)
            {
                return;
            }

            try
            {
                Status($"Попытка завершения процесса {process.ProcessName}", NotifyLevel.logHistory | NotifyLevel.logDiscord);
                process.Kill();
            }
            catch (Exception ex)
            {
                Status(ex.Message, NotifyLevel.logHistory | NotifyLevel.logDiscord);
                return;
            }

            Status($"Процесс был успешно завершён.", NotifyLevel.logHistory | NotifyLevel.logDiscord);
        }

        private void BackgroundWorkerDoWork(object sender, System.ComponentModel.DoWorkEventArgs e)
        {
            BackgroundWorker worker = sender as BackgroundWorker;
            int i = (int)numericUpDown1.Value;

            while (!worker.CancellationPending)
            {
                if (ProcessByNameIsRuning(textBoxProcessName.Text))
                {
                    double cpuTotalTime, cpuPercent;
                    string ProcessAnswer;

                    cpuMeasureTimer.Stop();
                    try
                    {
                        cpuTotalTime = WatchedProcess.TotalProcessorTime.TotalMilliseconds - cpuLastTime;
                        cpuLastTime = WatchedProcess.TotalProcessorTime.TotalMilliseconds;
                        cpuPercent = cpuTotalTime * 100 / (Environment.ProcessorCount * cpuMeasureTimer.ElapsedMilliseconds);
                        ProcessAnswer = (WatchedProcess.Responding) ? "Активен" : "Неактивен";
                        ProcessCheckResponding(WatchedProcess.Responding, cpuPercent);
                    }
                    catch
                    {
                        cpuTotalTime = 0;
                        cpuPercent = 0;
                        ProcessAnswer = "Неактивен";
                    }
                    cpuMeasureTimer.Reset();
                    cpuMeasureTimer.Start();

                    Status($"Процесс {ProcessName} уже запущен.", NotifyLevel.logUpdateStatus);
                    processInformationLabel.Text = $"Интерфейс: {ProcessAnswer}. ЦПУ: {cpuPercent:f2}% {cpuTotalTime:f2}мсек";
                    if (i < (int)numericUpDown1.Value) SendDiscordMessage($"Процесс {ProcessName} запущен.", NotifyLevel.logDiscord);
                    i = (int)numericUpDown1.Value;
                }
                else
                {
                    processInformationLabel.Text = "";
                    if (radioButtonRestartTimer.Checked)
                    {
                        if (i == (int)numericUpDown1.Value) Status($"Процесс {ProcessName} не найден. Запуск через {i} сек", NotifyLevel.logDiscord);
                        i--;
                        Status($"Процесс {ProcessName} не найден. Запуск через {i}", NotifyLevel.logUpdateStatus);
                    }

                    if (i <= 0 || radioButtonRestartNow.Checked)
                    {
                        i = (int)numericUpDown1.Value;
                        Status($"Запускаем {ProcessName}", NotifyLevel.logUpdateStatus | NotifyLevel.logDiscord);
                        ProcessStart(Settings.Default.startProgramPath, textBoxArguments.Text);
                    }
                }

                Thread.Sleep(1000);
            }

            if (worker.CancellationPending)
            {
                e.Cancel = true;
                processInformationLabel.Text = "";
            }
        }

        private void backgroundWorker1_RunWorkerCompleted(object sender, RunWorkerCompletedEventArgs e)
        {
            if (e.Cancelled)
            {
                Status("Наблюдение отменено.", NotifyLevel.logUpdateStatus | NotifyLevel.logDiscord);
            }
            else if (e.Error != null)
            {
                string error_message = "Error: " + e.Error.Message + "\n" + e.Error.StackTrace;
                Status("Произошла ошибка! Наблюдение остановлено.\n" + error_message, NotifyLevel.logUpdateStatus | NotifyLevel.logDiscord);
                MessageBox.Show(error_message, "Ошибка наблюдения", MessageBoxButtons.OK, MessageBoxIcon.Stop);
                radioButtonDisableWathing.Checked = true;
            }
            else
            {
                Status("Наблюдение остановлено.", NotifyLevel.logUpdateStatus | NotifyLevel.logDiscord);
            }
        }

        private void webhookDiscordToolStripMenuItem_Click(object sender, EventArgs e)
        {
            WebhookSettings discordSettings;
            discordSettings = new WebhookSettings(this);
            discordSettings.ShowDialog(this);
            discordSettings.Dispose();
        }

        private void AboutProgram_Click(object sender, EventArgs e)
        {
            AboutBox aboutBox = new AboutBox();
            aboutBox.ShowDialog(this);
            aboutBox.Dispose();
        }
    }
}