MainForm.cs

using Microsoft.Win32;
using System;
using System.Drawing;
using System.Windows.Forms;
using System.Net.NetworkInformation;
using System.Net;
using System.Linq;

namespace GateSwitchWay
{
    public partial class MainForm : Form
    {
        private static bool isAutoStartOn = false;
        private static bool startHidden = false;
        private static bool isLoadingSettings = false;
        private static bool isAutoAlterOn = false;

        private NetworkHelper.NetworkInfo currentNetworkInfo;
        private NetworkHelper.NetworkInfo alternativeNetworkInfo;

        // Auto-refresh timer for network status
        private System.Windows.Forms.Timer autoRefreshTimer = new System.Windows.Forms.Timer();

        public MainForm()
        {
            InitializeComponent();
            isAutoStartOn = AppAutoStart.GetAutoStart();
            autoStartMenu.Checked = isAutoStartOn;

            startHidden = Settings.Default.StartHidden;
            startHiddenMenu.Checked = startHidden;

            // Load AutoAlter setting
            isAutoAlterOn = Settings.Default.AutoAlter;
            autoAlterMenu.Checked = isAutoAlterOn;
            
            isSwitchedOn = false;
            notifyIcon1.Icon = isSwitchedOn ? Res.gw64_yg_TEA_icon : Res.gw64_g_vzI_icon;
            this.Icon = isSwitchedOn ? Res.gw64_yg_TEA_icon : Res.gw64_g_vzI_icon;

            currentNetworkInfo = NetworkHelper.GetCurrentNetworkInfo(); // Call the method to update the network info on startup

            isLoadingSettings = true;
            NetworkHelper.LoadAlterNativeSettings(textBoxGw4, textBoxGw6, textBoxDns4, textBoxDns6, checkBoxGw4, checkBoxGw6, checkBoxDns4, checkBoxDns6);

            // Load native settings to UI
            NetworkHelper.LoadNativeSettingsToUI(textBoxNativeGw4, textBoxNativeGw6, textBoxNativeDns4, textBoxNativeDns6, checkBoxNativeGw4, checkBoxNativeGw6, checkBoxNativeDns4, checkBoxNativeDns6);

            // Initialize auto-refresh settings
            checkBoxAutoRefresh.Checked = Settings.Default.AutoRefreshEnabled;
            numericUpDownRefreshInterval.Value = Settings.Default.AutoRefreshInterval / 1000; // Convert to seconds
            numericUpDownRefreshInterval.Minimum = 1;
            numericUpDownRefreshInterval.Maximum = 300; // 5 minutes max

            // Check if there are any saved AlterNative settings
            alternativeNetworkInfo = NetworkHelper.GetAlterNativeSettings();
            if ((alternativeNetworkInfo.Gateway4 == "N/A") && (alternativeNetworkInfo.Gateway6 == "N/A") && (alternativeNetworkInfo.Dns4 == "N/A") && (alternativeNetworkInfo.Dns6 == "N/A"))
            {
                // No saved alternative settings, save current settings as native settings
                NetworkHelper.SaveNativeSettings(currentNetworkInfo);
            }
            else
            {
                // Compare current settings with saved alternative settings
                if (NetworkHelper.AreSettingsEqual(currentNetworkInfo, alternativeNetworkInfo))
                {
                    // Current settings are equal to saved alternative settings, restore native settings from storage
                    currentNetworkInfo = NetworkHelper.LoadNativeSettings();
                    //check if restored native settings are valid
                    if (currentNetworkInfo.Gateway4 == "N/A" && currentNetworkInfo.Gateway6 == "N/A" && currentNetworkInfo.Dns4 == "N/A" && currentNetworkInfo.Dns6 == "N/A")
                    {
                        // restored settings are invalid, check if need to do anything else here
                    }
                    else
                    {
                        // restore native settings from saved settings from previous run
                        // Update the network info to reflect the restored native settings
                        NetworkHelper.TemporaryModifyNetworkInfo(currentNetworkInfo);
                    }
                }
                else
                {
                    // Current settings are different from saved alternative settings, save current settings as new native settings
                    NetworkHelper.SaveNativeSettings(currentNetworkInfo);
                }
            }
            // Update the native network info box to be consistent with native settings
            NetworkHelper.PopulateNetworkInfoTextBoxes(currentNetworkInfo, textBoxNativeGw4, textBoxNativeGw6, textBoxNativeDns4, textBoxNativeDns6);

            // Switch to AlterNative mode if AutoAlter is enabled
            if (isAutoAlterOn)
            {
                ToggleSwitch();
            }

            NetworkHelper.UpdateTaskbarNetworkInfo(currentNetworkInfo, notifyIcon1, isSwitchedOn); // Display the current network info
            NetworkHelper.PopulateNetworkInfoTextBoxes(currentNetworkInfo, textBoxCurrentGw4, textBoxCurrentGw6, textBoxCurrentDns4, textBoxCurrentDns6);

            isLoadingSettings = false;
            TrackBar_Set(isSwitchedOn);

            // Setup auto-refresh timer
            SetupAutoRefreshTimer();

            clickTimer.Interval = SystemInformation.DoubleClickTime - 1; // Just under the double-click speed
            clickTimer.Tick += (s, e) =>
            {
                clickTimer.Stop(); // Stop the timer

                // Perform the single-click action if not a double-click
                if (!isDoubleClick)
                {
                    ToggleSwitch(); // Switch mode with single-click
                }
                isDoubleClick = false; // Reset the flag
            };

            if (notifyIcon1.ContextMenuStrip is not null)
            {
                notifyIcon1.ContextMenuStrip.Opening += (sender, e) =>
                {
                    UpdateContextMenuTheme(notifyIcon1.ContextMenuStrip);
                    autoStartMenu.Checked = AppAutoStart.GetAutoStart();
                    mainToolStripMenuItem.Checked = this.Visible;
                    refreshIntervalMenu.Checked = Settings.Default.AutoRefreshEnabled;
                };
            }

            // Register the MouseClick event
            this.notifyIcon1.MouseClick += new MouseEventHandler(notifyIcon1_MouseClick);

            // Hide main window if startHidden is true
            if (startHidden)
            {
                this.Hide();
            }
        }

        private void SetupAutoRefreshTimer()
        {
            autoRefreshTimer.Interval = Settings.Default.AutoRefreshInterval; // Default 5 seconds from settings
            autoRefreshTimer.Tick += AutoRefreshTimer_Tick;
            
            // Start timer only if auto-refresh is enabled
            if (Settings.Default.AutoRefreshEnabled)
            {
                autoRefreshTimer.Start();
            }
        }

        private void AutoRefreshTimer_Tick(object sender, EventArgs e)
        {
            // Only update if auto-refresh is enabled
            if (!Settings.Default.AutoRefreshEnabled)
            {
                autoRefreshTimer.Stop();
                return;
            }

            // Get current network information
            var latestNetworkInfo = NetworkHelper.GetCurrentNetworkInfo();
            
            // Update current network info textboxes
            NetworkHelper.PopulateNetworkInfoTextBoxes(latestNetworkInfo, textBoxCurrentGw4, textBoxCurrentGw6, textBoxCurrentDns4, textBoxCurrentDns6);
            
            // Update taskbar network info
            NetworkHelper.UpdateTaskbarNetworkInfo(latestNetworkInfo, notifyIcon1, isSwitchedOn);
            
            // Update currentNetworkInfo for consistency
            if (!isSwitchedOn)
            {
                currentNetworkInfo = latestNetworkInfo;
            }
        }

        private void buttonSaveNative_Click(object sender, EventArgs e)
        {
            try
            {
                NetworkHelper.SaveNativeSettingsFromUI(textBoxNativeGw4, textBoxNativeGw6, textBoxNativeDns4, textBoxNativeDns6, checkBoxNativeGw4, checkBoxNativeGw6, checkBoxNativeDns4, checkBoxNativeDns6);
                MessageBox.Show("Native network settings saved successfully!", "Settings Saved", MessageBoxButtons.OK, MessageBoxIcon.Information);
            }
            catch (Exception ex)
            {
                MessageBox.Show($"Failed to save native settings: {ex.Message}", "Error", MessageBoxButtons.OK, MessageBoxIcon.Error);
            }
        }

        private void buttonLoadNative_Click(object sender, EventArgs e)
        {
            try
            {
                var currentInfo = NetworkHelper.GetCurrentNetworkInfo();
                NetworkHelper.PopulateNetworkInfoTextBoxes(currentInfo, textBoxNativeGw4, textBoxNativeGw6, textBoxNativeDns4, textBoxNativeDns6);
                
                // Enable checkboxes for valid network settings
                checkBoxNativeGw4.Checked = currentInfo.Gateway4_enable;
                checkBoxNativeGw6.Checked = currentInfo.Gateway6_enable;
                checkBoxNativeDns4.Checked = currentInfo.Dns4_enable;
                checkBoxNativeDns6.Checked = currentInfo.Dns6_enable;
                
                MessageBox.Show("Current network settings loaded into Native settings!", "Settings Loaded", MessageBoxButtons.OK, MessageBoxIcon.Information);
            }
            catch (Exception ex)
            {
                MessageBox.Show($"Failed to load current settings: {ex.Message}", "Error", MessageBoxButtons.OK, MessageBoxIcon.Error);
            }
        }

        protected override void OnShown(EventArgs e)
        {
            base.OnShown(e);
            // Update AlterNative settings display
            AlterCheckBoxes_ReEnable();
            // Update Native settings display
            NativeCheckBoxes_ReEnable();
            DisplayCurrentMode(isSwitchedOn);
            TrackBar_Set(isSwitchedOn);
            
            // Initialize auto-refresh UI state
            isLoadingSettings = true;
            checkBoxAutoRefresh.Checked = Settings.Default.AutoRefreshEnabled;
            numericUpDownRefreshInterval.Value = Settings.Default.AutoRefreshInterval / 1000;
            isLoadingSettings = false;
            
            // Hide main window if startHidden is true
            if (startHidden)
            {
                this.Hide();
            }
        }

        private void exitToolStripMenuItem_Click(object sender, EventArgs e)
        {
            Application.Exit();
        }

        private static bool firstTimeCloseClicked = true;
        protected override void OnFormClosing(FormClosingEventArgs e)
        {
            // Check if the close reason is a user action (e.g., clicking the 'X' button)
            if (e.CloseReason == CloseReason.UserClosing)
            {
                // Cancel the form's closure
                e.Cancel = true;

                // Hide the form
                this.Hide();

                // Optionally, show a notification or message indicating the application is still running
                if (firstTimeCloseClicked)
                {
                    notifyIcon1.ShowBalloonTip(600, "Attempt to close", $"{AppAutoStart.taskName} running in the background.\nDouble click system tray icon to show main window. Once click to toggle active mode.", ToolTipIcon.Info);
                    firstTimeCloseClicked = false;
                }
            }
            else
            {
                // Allow the form to close for other close reasons (e.g., Application.Exit())
                base.OnFormClosing(e);
            }
        }

        // add theme support to context menu
        public class DarkModeToolStripRenderer : ToolStripProfessionalRenderer
        {
            public DarkModeToolStripRenderer() : base(new DarkModeColorTable())
            {
            }

            protected override void OnRenderToolStripBackground(ToolStripRenderEventArgs e)
            {
                // Background of the entire menu
                e.Graphics.FillRectangle(new SolidBrush(Color.FromArgb(40, 40, 40)), e.AffectedBounds);
            }

            protected override void OnRenderMenuItemBackground(ToolStripItemRenderEventArgs e)
            {
                if (e.Item.Selected)
                {
                    // Highlight color for selected items
                    e.Graphics.FillRectangle(new SolidBrush(Color.FromArgb(70, 70, 70)), new Rectangle(Point.Empty, e.Item.Size));
                }
                else
                {
                    // Regular item background color
                    e.Graphics.FillRectangle(new SolidBrush(Color.FromArgb(40, 40, 40)), new Rectangle(Point.Empty, e.Item.Size));
                }
            }

            protected override void OnRenderImageMargin(ToolStripRenderEventArgs e)
            {
                // Override the space on the left of items (where check marks or images go)
                e.Graphics.FillRectangle(new SolidBrush(Color.FromArgb(40, 40, 40)), e.AffectedBounds);
            }

            // Add overrides for other elements as needed, like separators, arrows, etc.
        }
        public class DarkModeColorTable : ProfessionalColorTable
        {
            public override Color MenuItemSelected => Color.FromArgb(70, 70, 70);
            public override Color MenuItemBorder => Color.FromArgb(70, 70, 70);
            public override Color MenuBorder => Color.FromArgb(40, 40, 40);
            public override Color MenuStripGradientBegin => Color.FromArgb(40, 40, 40);
            public override Color MenuStripGradientEnd => Color.FromArgb(40, 40, 40);
            public override Color SeparatorDark => Color.FromArgb(32, 100, 32); // Dark background
        }

        // Timer to handle single-click action delay
        private System.Windows.Forms.Timer clickTimer = new System.Windows.Forms.Timer();

        // Flag to distinguish between single and double clicks
        private bool isDoubleClick = false;

        private void notifyIcon1_MouseDoubleClick(object sender, MouseEventArgs e)
        {
            // Check if the double click was with the left mouse button
            if (e.Button == MouseButtons.Left)
            {
                isDoubleClick = true; // It's a double-click
                clickTimer.Stop(); // Stop the timer to prevent the single-click action

                if (this.Visible)
                {
                    this.Hide();
                }
                else
                {
                    activateMainWindow();
                }
            }
        }

        private void notifyIcon1_MouseClick(object sender, MouseEventArgs e)
        {
            // Check if the click was with the left mouse button
            if (e.Button == MouseButtons.Left)
            {
                isDoubleClick = false; // Assume it's not a double-click
                clickTimer.Start(); // Start the delay timer
            }
        }

        private void activateMainWindow()
        {
            this.Show();
            // Ensure the window is not minimized, but keep maximized or other state
            if (this.WindowState == FormWindowState.Minimized)
            {
                this.WindowState = FormWindowState.Normal;
            }
            this.Activate(); // Bring the form to the forefront
        }

        private static bool isSwitchedOn = false;
        private void ToggleSwitch()
        {
            // Implement the switch on/off logic here
            isSwitchedOn = !isSwitchedOn;

            // set icon to `gw64_1_Jnv_icon` as not defined yet mode
            notifyIcon1.Icon = Res.gw64_1_Jnv_icon;
            this.Icon = Res.gw64_1_Jnv_icon;

            if (isSwitchedOn)
            {
                // Switched on
                UpdateAlternativeNetworkInfo();
                NetworkHelper.TemporaryModifyNetworkInfo(alternativeNetworkInfo);
            }
            else
            {
                // Switched off
                currentNetworkInfo = NetworkHelper.LoadNativeSettings();
                NetworkHelper.TemporaryModifyNetworkInfo(currentNetworkInfo);
            }
            // Update the icon based on the new state
            notifyIcon1.Icon = isSwitchedOn ? Res.gw64_yg_TEA_icon : Res.gw64_g_vzI_icon;
            this.Icon = isSwitchedOn ? Res.gw64_yg_TEA_icon : Res.gw64_g_vzI_icon;

            this.Text = isSwitchedOn ? "AlterNative GateWay" : "Native GateWay";
            // Update the network info to be consistent with current mode
            currentNetworkInfo = NetworkHelper.GetCurrentNetworkInfo();
            // update current network info textboxes
            NetworkHelper.PopulateNetworkInfoTextBoxes(currentNetworkInfo, textBoxCurrentGw4, textBoxCurrentGw6, textBoxCurrentDns4, textBoxCurrentDns6);
            // Update taskbar popup network info to be consistent with current mode
            NetworkHelper.UpdateTaskbarNetworkInfo(currentNetworkInfo, notifyIcon1, isSwitchedOn);
            // Update the display of the current mode
            DisplayCurrentMode(isSwitchedOn);
            TrackBar_Set(isSwitchedOn);
        }

        private static bool currentLightThemeEnabled = false;
        private bool IsLightThemeEnabled()
        {
            var key = @"Software\Microsoft\Windows\CurrentVersion\Themes\Personalize";
            var valueName = "AppsUseLightTheme";

            currentLightThemeEnabled = false;
            using (var regKey = Registry.CurrentUser.OpenSubKey(key))
            {
                if (regKey != null)
                {
                    var value = regKey.GetValue(valueName);
                    if (value != null)
                    {
                        currentLightThemeEnabled = Convert.ToInt32(value) != 0;
                    }
                }
            }
            return currentLightThemeEnabled; // Default to dark theme if unable to read the registry
        }
        private void UpdateContextMenuTheme(ContextMenuStrip contextMenu)
        {
            if (IsLightThemeEnabled())
            {
                contextMenu.BackColor = SystemColors.ControlLightLight; // Light background
                contextMenu.ForeColor = Color.Black; // Dark text
            }
            else
            {
                contextMenu.BackColor = Color.FromArgb(255, 32, 32, 32); // Dark background
                contextMenu.ForeColor = Color.LightGray; // Light text
            }

            // add dark mode custom renderer to display menu items properly
            contextMenu.Renderer = IsLightThemeEnabled() ? new ToolStripProfessionalRenderer() : new DarkModeToolStripRenderer();
        }

        private void autoStartMenu_Click(object sender, EventArgs e)
        {
            // Toggle the auto-start setting based on its current state
            bool currentState = AppAutoStart.GetAutoStart();
            bool success = AppAutoStart.SetAutoStart(!currentState);

            if (success)
            {
                // Update the check state of the menu item to reflect the new state
                autoStartMenu.Checked = !currentState;
            }
            else
            {
                MessageBox.Show("Failed to toggle auto-start setting.", "Error", MessageBoxButtons.OK, MessageBoxIcon.Error);
            }
        }

        private void mainToolStripMenuItem_Click(object sender, EventArgs e)
        {
            if (this.Visible)
            {
                if (this.WindowState == FormWindowState.Minimized)
                {
                    activateMainWindow();
                }
                else
                {
                    this.Hide();
                }
            }
            else
            {
                activateMainWindow();
            }
        }
        private void startHiddenMenu_Click(object sender, EventArgs e)
        {
            startHidden = !startHidden;
            startHiddenMenu.Checked = startHidden;
            Settings.Default.StartHidden = startHidden;
            Settings.Default.Save();
        }
        private void autoAlterMenu_Click(object sender, EventArgs e)
        {
            isAutoAlterOn = !isAutoAlterOn;
            autoAlterMenu.Checked = isAutoAlterOn;
            Settings.Default.AutoAlter = isAutoAlterOn;
            Settings.Default.Save();
        }
        private void CheckBox_CheckedChanged(object sender, EventArgs e)
        {
            CheckBox? checkBox = sender as CheckBox;
            if (checkBox != null)
            {
                bool needSave = false;
                bool needNativeSave = false;
                switch (checkBox.Name)
                {
                    case "checkBoxGw4":
                    case "checkBoxGw6":
                    case "checkBoxDns4":
                    case "checkBoxDns6":
                        {
                            AlterCheckBoxes_ReEnable();
                            needSave = true;
                            break;
                        }
                    case "checkBoxNativeGw4":
                    case "checkBoxNativeGw6":
                    case "checkBoxNativeDns4":
                    case "checkBoxNativeDns6":
                        {
                            NativeCheckBoxes_ReEnable();
                            needNativeSave = true;
                            break;
                        }
                    case "checkBoxAutoRefresh":
                        {
                            if (!isLoadingSettings)
                            {
                                Settings.Default.AutoRefreshEnabled = checkBoxAutoRefresh.Checked;
                                Settings.Default.Save();
                                
                                if (checkBoxAutoRefresh.Checked)
                                {
                                    autoRefreshTimer.Start();
                                }
                                else
                                {
                                    autoRefreshTimer.Stop();
                                }
                            }
                            break;
                        }
                }
                if (!isLoadingSettings)
                {
                    if (needSave)
                    {
                        NetworkHelper.SaveAlterNativeSettings(textBoxGw4, textBoxGw6, textBoxDns4, textBoxDns6, checkBoxGw4, checkBoxGw6, checkBoxDns4, checkBoxDns6);
                    }
                    if (needNativeSave)
                    {
                        NetworkHelper.SaveNativeSettingsFromUI(textBoxNativeGw4, textBoxNativeGw6, textBoxNativeDns4, textBoxNativeDns6, checkBoxNativeGw4, checkBoxNativeGw6, checkBoxNativeDns4, checkBoxNativeDns6);
                    }
                }
            }
        }

        private void refreshIntervalMenu_Click(object sender, EventArgs e)
        {
            // Toggle auto-refresh setting
            checkBoxAutoRefresh.Checked = !checkBoxAutoRefresh.Checked;
        }

        private void NumericUpDownRefreshInterval_ValueChanged(object sender, EventArgs e)
        {
            if (!isLoadingSettings)
            {
                int newInterval = (int)(numericUpDownRefreshInterval.Value * 1000); // Convert to milliseconds
                Settings.Default.AutoRefreshInterval = newInterval;
                Settings.Default.Save();
                
                // Update timer interval
                autoRefreshTimer.Interval = newInterval;
                
                // Restart timer if it's running
                if (autoRefreshTimer.Enabled)
                {
                    autoRefreshTimer.Stop();
                    autoRefreshTimer.Start();
                }
            }
        }
        private void AlterCheckBoxes_ReEnable()
        {
            textBoxGw4.Enabled = checkBoxGw4.Checked;
            textBoxGw6.Enabled = checkBoxGw6.Checked;
            textBoxDns4.Enabled = checkBoxDns4.Checked;
            textBoxDns6.Enabled = checkBoxDns6.Checked;
        }
        private void NativeCheckBoxes_ReEnable()
        {
            textBoxNativeGw4.Enabled = checkBoxNativeGw4.Checked;
            textBoxNativeGw6.Enabled = checkBoxNativeGw6.Checked;
            textBoxNativeDns4.Enabled = checkBoxNativeDns4.Checked;
            textBoxNativeDns6.Enabled = checkBoxNativeDns6.Checked;
        }
        private void TrackBarToggle_ValueChanged(object sender, EventArgs e)
        {
            if (!isLoadingSettings)
            {
                if (trackBarToggle.Value == 0)
                {
                    if (isSwitchedOn)
                    {
                        ToggleSwitch();
                    }
                }
                else
                {
                    if (!isSwitchedOn)
                    {
                        ToggleSwitch();
                    }
                }
            }
        }
        private void TrackBar_Set(bool isSwitchedOn)
        {
            isLoadingSettings = true;
            trackBarToggle.Value = isSwitchedOn ? 1 : 0; // Set the trackbar value to match the current mode
            isLoadingSettings = false;
        }
        private void DisplayCurrentMode(bool isSwitchedOn)
        {
            if (isSwitchedOn)
            {
                groupBoxAlterNative.Font = new Font(groupBoxAlterNative.Font, FontStyle.Bold);
                groupBoxNative.Font = new Font(groupBoxNative.Font, FontStyle.Regular);
            }
            else
            {
                groupBoxAlterNative.Font = new Font(groupBoxAlterNative.Font, FontStyle.Regular);
                groupBoxNative.Font = new Font(groupBoxNative.Font, FontStyle.Bold);
            }
        }
        private void UpdateAlternativeNetworkInfo()
        {
            alternativeNetworkInfo.Gateway4_enable = checkBoxGw4.Checked;
            alternativeNetworkInfo.Gateway4 = textBoxGw4.Text;
            alternativeNetworkInfo.Gateway6_enable = checkBoxGw6.Checked;
            alternativeNetworkInfo.Gateway6 = textBoxGw6.Text;
            alternativeNetworkInfo.Dns4_enable = checkBoxDns4.Checked;
            alternativeNetworkInfo.Dns4 = textBoxDns4.Text;
            alternativeNetworkInfo.Dns6_enable = checkBoxDns6.Checked;
            alternativeNetworkInfo.Dns6 = textBoxDns6.Text;
        }
    }
}