MainForm.cs

using Microsoft.Win32;
using System.Diagnostics;
using System.Threading;

namespace WinTombstone
{
    public partial class MainForm : Form
    {
        private readonly CoreManager _core;
        private const int HOTKEY_ID = 9000;
        private DateTime _ignoreTimerUntil = DateTime.MinValue;

        private Rectangle _dragBox;
        private int _dragRowIndex = -1;
        private int _timerBusy;

        public MainForm()
        {
            InitializeComponent();
            InitializeCustomIcon();

            _core = new CoreManager();

            InitGrid();
            InitEvents();
            ApplyLanguage(); // 应用语言

            RegisterGlobalHotkey();
        }

        private void InitializeCustomIcon()
        {
            UpdateThemedIcon();
            SystemEvents.UserPreferenceChanged += (s, e) =>
            {
                if (e.Category == UserPreferenceCategory.General)
                {
                    BeginInvoke(new Action(UpdateThemedIcon));
                }
            };
        }

        private void UpdateThemedIcon()
        {
            bool isDark = IsSystemDarkMode();
            Color iconColor = isDark ? Color.White : Color.Black;

            using (Bitmap bmp = new Bitmap(32, 32))
            using (Graphics g = Graphics.FromImage(bmp))
            {
                g.Clear(Color.Transparent);
                g.TextRenderingHint = System.Drawing.Text.TextRenderingHint.AntiAlias;
                g.SmoothingMode = System.Drawing.Drawing2D.SmoothingMode.HighQuality;

                using (Font f = new Font("Segoe UI Emoji", 20, FontStyle.Bold))
                using (Brush b = new SolidBrush(iconColor))
                {
                    var size = g.MeasureString("❄", f);
                    g.DrawString("❄", f, b, (32 - size.Width) / 2, (32 - size.Height) / 2);
                }

                var oldIcon = this.Icon;
                Icon newIcon = Icon.FromHandle(bmp.GetHicon());
                this.Icon = newIcon;
                notifyIcon1.Icon = newIcon;

                if (oldIcon != null && oldIcon != newIcon) oldIcon.Dispose();
            }
        }

        private bool IsSystemDarkMode()
        {
            try
            {
                const string keyName = @"HKEY_CURRENT_USER\Software\Microsoft\Windows\CurrentVersion\Themes\Personalize";
                object value = Registry.GetValue(keyName, "SystemUsesLightTheme", null);
                if (value == null) value = Registry.GetValue(keyName, "AppsUseLightTheme", 1);
                if (value is int intVal) return intVal == 0;
            }
            catch { }
            return false;
        }

        private void InitGrid()
        {
            dgvList.AutoGenerateColumns = false;
            dgvList.AllowUserToAddRows = false;
            dgvList.AllowDrop = true;

            // 列定义使用Name查找，Text在ApplyLanguage中设置
            dgvList.Columns.Add(new DataGridViewTextBoxColumn { Name = "colName", DataPropertyName = "AppName", ReadOnly = true });
            dgvList.Columns.Add(new DataGridViewCheckBoxColumn { Name = "colFrozen", DataPropertyName = "IsFrozen" });
            dgvList.Columns.Add(new DataGridViewCheckBoxColumn { Name = "colAuto", DataPropertyName = "AutoFreeze" });
            dgvList.Columns.Add(new DataGridViewTextBoxColumn { Name = "colTimeout", DataPropertyName = "TimeoutSeconds" });

            var btnDel = new DataGridViewButtonColumn
            {
                Name = "colAction",
                UseColumnTextForButtonValue = true
            };
            dgvList.Columns.Add(btnDel);

            dgvList.DataSource = new BindingSource { DataSource = _core.Settings.Whitelist };
        }

        private void ApplyLanguage()
        {
            Text = Localization.Get("Title");
            notifyIcon1.Text = Localization.Get("Title");

            dgvList.Columns["colName"].HeaderText = Localization.Get("ColName");
            dgvList.Columns["colFrozen"].HeaderText = Localization.Get("ColFrozen");
            dgvList.Columns["colAuto"].HeaderText = Localization.Get("ColAuto");
            dgvList.Columns["colTimeout"].HeaderText = Localization.Get("ColTimeout");
            dgvList.Columns["colAction"].HeaderText = Localization.Get("ColAction");
            (dgvList.Columns["colAction"] as DataGridViewButtonColumn).Text = Localization.Get("ActionDelete");

            btnUnfreezeAll.Text = Localization.Get("UnfreezeAll");
            btnHelp.Text = Localization.Get("Help");
            btnSettings.Text = Localization.Get("Settings"); // 新增：设置按钮文本

            settingsMenuItemMain.Text = Localization.Get("Settings");
            settingsMenuItemTray.Text = Localization.Get("Settings");
            exitMenuItem.Text = Localization.Get("Exit");

            RefreshGrid(); // 刷新表格以更新按钮文本等
        }

        private void RegisterGlobalHotkey()
        {
            NativeHelper.UnregisterHotKey(this.Handle, HOTKEY_ID);
            NativeHelper.RegisterHotKey(this.Handle, HOTKEY_ID, (uint)_core.Settings.HotkeyModifier, (uint)_core.Settings.HotkeyKey);
        }

        private void InitEvents()
        {
            _core.AppStateChanged += (app) =>
            {
                if (IsDisposed) return;
                BeginInvoke(new Action(RefreshGrid));
            };

            dgvList.CellContentClick += DgvList_CellContentClick;
            dgvList.CellValueChanged += (s, e) =>
            {
                if (e.RowIndex < 0) return;

                // 获取当前变更的列名
                string colName = dgvList.Columns[e.ColumnIndex].Name;

                // 只有在修改“自动冻结(colAuto)”或“超时时间(colTimeout)”时才保存
                if (colName == "colAuto" || colName == "colTimeout")
                {
                    _core.SaveSettings();
                }
            };
            dgvList.CurrentCellDirtyStateChanged += (s, e) =>
            {
                if (dgvList.IsCurrentCellDirty && dgvList.CurrentCell is DataGridViewCheckBoxCell)
                    dgvList.CommitEdit(DataGridViewDataErrorContexts.Commit);
            };

            dgvList.MouseMove += DgvList_MouseMove;
            dgvList.MouseDown += DgvList_MouseDown;
            dgvList.DragOver += (s, e) => e.Effect = DragDropEffects.Move;
            dgvList.DragDrop += DgvList_DragDrop;

            this.FormClosing += (s, e) => {
                if (e.CloseReason == CloseReason.UserClosing)
                {
                    e.Cancel = true;
                    Hide();
                    // 隐藏时可以顺便触发一次保存，以防万一
                    _core.SaveSettings();
                }
                else
                {
                    // 系统关机或任务管理器结束任务时的保存尝试
                    _core.SaveSettings(forceImmediate: true);
                }
            };

            exitMenuItem.Click += (s, e) => {
                _core.RestoreAll();
                _core.SaveSettings(forceImmediate: true); // 强制立即保存
                Application.Exit();
            };

            // 绑定设置菜单和按钮
            settingsMenuItemMain.Click += OpenSettings;
            settingsMenuItemTray.Click += OpenSettings;
            btnSettings.Click += OpenSettings; // 新增：绑定主界面设置按钮

            // 绑定说明按钮
            btnHelp.Click += (s, e) => MessageBox.Show(Localization.Get("HelpContent"), Localization.Get("HelpTitle"), MessageBoxButtons.OK, MessageBoxIcon.Information);

            notifyIcon1.MouseClick += (s, e) => {
                if (e.Button == MouseButtons.Left) { Show(); WindowState = FormWindowState.Normal; Activate(); }
            };

            btnUnfreezeAll.Click += (s, e) => {
                SetFreezeCooldown(2);
                _core.RestoreAll();
                RefreshGrid();
                _core.SaveSettings();
                notifyIcon1.ShowBalloonTip(2000, Localization.Get("MsgOpDone"), Localization.Get("MsgAllRestored"), ToolTipIcon.Info);
            };
        }

        private void OpenSettings(object sender, EventArgs e)
        {
            using (var settingsForm = new SettingsForm(_core))
            {
                if (settingsForm.ShowDialog() == DialogResult.OK)
                {
                    // 重新注册热键
                    RegisterGlobalHotkey();
                    // 如果语言改变，刷新UI
                    if (settingsForm.LanguageChanged)
                    {
                        ApplyLanguage();
                    }
                }
            }
        }

        private void DgvList_MouseDown(object sender, MouseEventArgs e)
        {
            _dragRowIndex = dgvList.HitTest(e.X, e.Y).RowIndex;
            if (_dragRowIndex != -1)
            {
                Size dragSize = SystemInformation.DragSize;
                _dragBox = new Rectangle(new Point(e.X - (dragSize.Width / 2), e.Y - (dragSize.Height / 2)), dragSize);
            }
            else _dragBox = Rectangle.Empty;
        }

        private void DgvList_MouseMove(object sender, MouseEventArgs e)
        {
            if ((e.Button & MouseButtons.Left) == MouseButtons.Left)
            {
                if (_dragBox != Rectangle.Empty && !_dragBox.Contains(e.X, e.Y))
                    dgvList.DoDragDrop(dgvList.Rows[_dragRowIndex], DragDropEffects.Move);
            }
        }

        private void DgvList_DragDrop(object sender, DragEventArgs e)
        {
            Point clientPoint = dgvList.PointToClient(new Point(e.X, e.Y));
            int targetRowIndex = dgvList.HitTest(clientPoint.X, clientPoint.Y).RowIndex;

            if (targetRowIndex != -1 && _dragRowIndex != -1 && _dragRowIndex != targetRowIndex)
            {
                var list = _core.Settings.Whitelist;
                var item = list[_dragRowIndex];
                list.RemoveAt(_dragRowIndex);
                list.Insert(targetRowIndex, item);
                RefreshGrid();
                _core.SaveSettings();
            }
        }

        private void DgvList_CellContentClick(object sender, DataGridViewCellEventArgs e)
        {
            if (e.RowIndex < 0) return;
            var app = _core.Settings.Whitelist[e.RowIndex];

            if (dgvList.Columns[e.ColumnIndex].Name == "colAction")
            {
                if (app.IsFrozen) _core.UnfreezeApp(app);
                _core.Settings.Whitelist.RemoveAt(e.RowIndex);
                RefreshGrid();
                _core.SaveSettings();
            }
            else if (dgvList.Columns[e.ColumnIndex].Name == "colFrozen")
            {
                dgvList.CommitEdit(DataGridViewDataErrorContexts.Commit);
                SetFreezeCooldown(2);
                if (app.IsFrozen) _core.FreezeApp(app);
                else _core.UnfreezeApp(app);
            }
        }

        private void RefreshGrid()
        {
            if (dgvList.IsCurrentCellInEditMode) return;
            (dgvList.DataSource as BindingSource)?.ResetBindings(false);
        }

        protected override void WndProc(ref Message m)
        {
            if (m.Msg == 0x0312 && m.WParam.ToInt32() == HOTKEY_ID)
            {
                AddNewAppAndFreezeFromForeground();
            }
            base.WndProc(ref m);
        }

        private void SetFreezeCooldown(double seconds) => _ignoreTimerUntil = DateTime.Now.AddSeconds(seconds);

        private void AddNewAppAndFreezeFromForeground()
        {
            IntPtr hWnd = NativeHelper.GetForegroundWindow();
            if (hWnd == IntPtr.Zero) return;

            NativeHelper.GetWindowThreadProcessId(hWnd, out uint pid);

            // 修复：使用 NativeHelper 获取路径，避免因权限问题导致 MainModule 抛出异常
            string path = NativeHelper.GetRealProcessPath((int)pid);

            if (string.IsNullOrEmpty(path))
            {
                // 如果实在获取不到路径，可以在这里记录日志或者播放错误提示音
                System.Diagnostics.Debug.WriteLine($"[Error] 无法获取进程路径, PID: {pid}");
                return;
            }

            // 检查黑名单
            if (_core.IsBlacklisted(path))
            {
                Console.WriteLine($"[Core] 跳过黑名单应用: {path}");
                return;
            }

            SetFreezeCooldown(2.0);

            var targetApp = _core.Settings.Whitelist.FirstOrDefault(a => a.ExePath.Equals(path, StringComparison.OrdinalIgnoreCase));

            if (targetApp == null)
            {
                targetApp = new TargetApp
                {
                    AppName = Path.GetFileNameWithoutExtension(path),
                    ExePath = path,
                    LastActiveTime = DateTime.Now,
                    TimeoutSeconds = 30
                };
                _core.Settings.Whitelist.Add(targetApp);
                _core.SaveSettings();
                RefreshGrid();

                _core.FreezeApp(targetApp);
                notifyIcon1.ShowBalloonTip(1000, Localization.Get("MsgAddSuccess"), $"{targetApp.AppName} " + Localization.Get("MsgFrozen"), ToolTipIcon.Info);
            }
            else if (!targetApp.IsFrozen)
            {
                _core.FreezeApp(targetApp);
            }
        }

        private async void timerCheck_Tick(object sender, EventArgs e)
        {
            if (DateTime.Now < _ignoreTimerUntil) return;
            if (Interlocked.CompareExchange(ref _timerBusy, 1, 0) != 0) return;

            try
            {
                IntPtr fgWin = NativeHelper.GetForegroundWindow();
                string fgPath = null;

                if (fgWin != IntPtr.Zero)
                {
                    NativeHelper.GetWindowThreadProcessId(fgWin, out uint fgPid);
                    fgPath = NativeHelper.GetRealProcessPath((int)fgPid);
                }

                TargetApp toUnfreeze = null;
                TargetApp toFreeze = null;

                foreach (var app in _core.Settings.Whitelist)
                {
                    bool isForeground = fgPath != null &&
                        string.Equals(app.ExePath, fgPath, StringComparison.OrdinalIgnoreCase);

                    if (isForeground)
                    {
                        app.LastActiveTime = DateTime.Now;
                        if (app.IsFrozen && NativeHelper.IsWindowVisible(fgWin))
                            toUnfreeze = app;
                    }
                    else
                    {
                        if (app.AutoFreeze && !app.IsFrozen)
                        {
                            if ((DateTime.Now - app.LastActiveTime).TotalSeconds > app.TimeoutSeconds)
                            {
                                SetFreezeCooldown(1.0);
                                toFreeze = app;
                            }
                        }
                    }
                }

                if (toUnfreeze != null)
                {
                    await Task.Run(() => _core.UnfreezeApp(toUnfreeze));
                }
                else if (toFreeze != null)
                {
                    await Task.Run(() => _core.FreezeApp(toFreeze));
                }
            }
            finally
            {
                Interlocked.Exchange(ref _timerBusy, 0);
            }
        }
    }
}