using FlybyScript;
using System;
using System.Diagnostics;
using System.Drawing;
using System.IO;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.Windows.Forms;

namespace Views
{
    public partial class MediaView : UserControl
    {
        private USBManagement usbManagement;
        private Logger logger;

        private string selectedIsoPath;

        public MediaView()
        {
            InitializeComponent();
            SetStyle();

            InitializeUSBDriveList();
            logger = new Logger(rtbLog);
            usbManagement = new USBManagement(logger, updateLabelStatus, progressBar);
        }

        private void SetStyle()
        {
            // Segoe MDL2 Assets
            btnBack.Text = "\uE72B";
        }

        private void updateLabelStatus(string statusText)
        {
            if (lblStatus.InvokeRequired)
            {
                lblStatus.Invoke(new Action(() => lblStatus.Text = statusText));
            }
            else
            {
                lblStatus.Text = statusText;
            }
        }

        private void InitializeUSBDriveList()
        {
            // Initialize disk numbers
            comboBoxDrives.Items.Clear();
            comboBoxDrives.Items.AddRange(new object[] { "Select (e.g., 1, 2, 3...)", "1", "2", "3" });
            comboBoxDrives.SelectedIndex = 0;

            // Initialize removable drives
            comboBoxDriveLetters.Items.Clear();
            var removableDrives = DriveInfo.GetDrives()
                                           .Where(d => d.DriveType == DriveType.Removable)
                                           .Select(d => d.Name)
                                           .ToArray();

            if (removableDrives.Any())
            {
                comboBoxDriveLetters.Items.AddRange(removableDrives);
                comboBoxDriveLetters.SelectedIndex = 0;
            }
            else
            {
                // MessageBox.Show("No removable drives found.", "Warning", MessageBoxButtons.OK, MessageBoxIcon.Warning);
                comboBoxDriveLetters.Enabled = comboBoxDrives.Enabled = false;
            }

            // Initialize partition schemes
            comboBoxPartitionScheme.Items.Clear();
            comboBoxPartitionScheme.Items.Add("GPT");
            comboBoxPartitionScheme.Items.Add("MBR");
            comboBoxPartitionScheme.SelectedIndex = 0; // default to gpt
        }

        private async Task ShowDiskList()
        {
            try
            {
                // Use chcp command to get the current code page of the user's system
                ProcessStartInfo chcpInfo = new ProcessStartInfo
                {
                    FileName = "cmd.exe",
                    Arguments = "/C chcp",
                    RedirectStandardOutput = true,
                    UseShellExecute = false,
                    CreateNoWindow = true
                };

                string codePageOutput;
                using (Process chcpProcess = Process.Start(chcpInfo))
                {
                    codePageOutput = await chcpProcess.StandardOutput.ReadToEndAsync();
                    chcpProcess.WaitForExit();
                }

                // Extract the code page number from chcp output
                int codePage = 850; // Default to 850 if detection fails
                var match = System.Text.RegularExpressions.Regex.Match(codePageOutput, @"\d+");
                if (match.Success)
                {
                    codePage = int.Parse(match.Value); // Parse our detected code page
                }

                ProcessStartInfo processInfo = new ProcessStartInfo
                {
                    FileName = "diskpart.exe",
                    RedirectStandardInput = true,
                    RedirectStandardOutput = true,
                    UseShellExecute = false,
                    CreateNoWindow = true,
                    StandardOutputEncoding = Encoding.GetEncoding(codePage)
                };

                // Start the process
                using (Process process = Process.Start(processInfo))
                {
                    using (StreamWriter writer = process.StandardInput)
                    {
                        if (writer.BaseStream.CanWrite)
                        {
                            writer.WriteLine("list disk");
                        }
                    }

                    string output = await process.StandardOutput.ReadToEndAsync();
                    process.WaitForExit();
                    MessageBox.Show(output, "Available Disks", MessageBoxButtons.OK, MessageBoxIcon.Information);
                }
            }
            catch (Exception ex)
            {
                MessageBox.Show($"An error occurred while listing disks: {ex.Message}", "Error", MessageBoxButtons.OK, MessageBoxIcon.Error);
            }
        }

        // Event handler for starting the format and copy process
        private async void btnStart_Click(object sender, EventArgs e)
        {
            btnStart.Enabled = false; // Disable the Start button

            // Validate Disk Number
            if (string.IsNullOrEmpty(comboBoxDrives.Text) || !int.TryParse(comboBoxDrives.Text, out int diskNumber))
            {
                logger.Log("Please enter a valid Disk Number.", Color.Red);
                btnStart.Enabled = true;
                return;
            }

            // Validate Drive Letter
            if (string.IsNullOrEmpty(comboBoxDriveLetters.SelectedItem?.ToString()))
            {
                logger.Log("Please select a valid Drive Letter.", Color.Red);
                btnStart.Enabled = true;
                return;
            }

            // Get the selected drive letter and partition scheme
            string driveLetter = comboBoxDriveLetters.SelectedItem.ToString().Trim();
            string partitionScheme = comboBoxPartitionScheme.SelectedItem.ToString().Trim(); // Get the selected partition scheme

            try
            {
                // Check if the selected drive is a USB stick
                DriveInfo selectedDrive = DriveInfo.GetDrives()
                                                   .FirstOrDefault(d => d.Name.StartsWith(driveLetter));

                if (selectedDrive == null || selectedDrive.DriveType != DriveType.Removable)
                {
                    logger.Log("Selected drive is not a compatible USB stick. Only USB sticks are supported.", Color.Red);
                    btnStart.Enabled = true;
                    return;
                }

                logger.Log($"Starting to format Disk {diskNumber} ({driveLetter}) as {partitionScheme}...", Color.Black);

                // Format the drive with the selected partition scheme
                await usbManagement.FormatDriveAsync(diskNumber.ToString(), partitionScheme); // Pass the disk number and selected partition scheme
                logger.Log($"Disk {diskNumber} formatted successfully.", Color.Black);
                logger.Log($"Disk {diskNumber} is now bootable.", Color.Black);

                // Copy ISO and create unattend.xml if checked
                bool createUnattend = checkCreateUnattend.Checked;
                await usbManagement.CopyISOWithUnattendAsync(driveLetter, selectedIsoPath, createUnattend);

                logger.Log("Process completed!", Color.Green);
            }
            catch (Exception ex)
            {
                logger.Log($"An error occurred during the process: {ex.Message}", Color.Red);
            }
            finally
            {
                btnStart.Enabled = true;
            }
        }

        private void btnSelect_Click(object sender, EventArgs e)
        {
            using (OpenFileDialog openFileDialog = new OpenFileDialog())
            {
                openFileDialog.Filter = "ISO Files (*.iso)|*.iso|All Files (*.*)|*.*";
                openFileDialog.Title = "Select an ISO File";

                if (openFileDialog.ShowDialog() == DialogResult.OK)
                {
                    selectedIsoPath = openFileDialog.FileName;
                    logger.Log($"Selected ISO file: {selectedIsoPath}", Color.Black);

                    ToggleIsoControls(true);
                }
                else
                {
                    logger.Log("ISO file selection was cancelled.", Color.Red);
                    ToggleIsoControls(false); // Disable ISO-related controls if selection is canceled
                }

                // MCT and Install Assistant buttons are always enabled
                btnMCT.Enabled = true;
                btnInstallAssistant.Enabled = true;
            }
        }

        // Helper method to toggle ISO-related controls
        private void ToggleIsoControls(bool isEnabled)
        {
            checkCreateUnattend.Visible = isEnabled;
            comboBoxDrives.Enabled = isEnabled;
            comboBoxDriveLetters.Enabled = isEnabled;
            comboBoxPartitionScheme.Enabled = isEnabled;
            lblDeviceIDInfo.Enabled = isEnabled;
            lblDriveInfo.Enabled = isEnabled;
            rtbLog.Enabled = isEnabled;
            lblDriveLetterInfo.Enabled = isEnabled;
            lblPartitionSchemeInfo.Enabled = isEnabled;
            lblStatusInfo.Enabled = isEnabled;
            linkShowDiskList.Enabled = isEnabled;
            btnStart.Enabled = isEnabled;
        }


        private void checkCreateUnattend_CheckedChanged(object sender, EventArgs e)
        {
            if (checkCreateUnattend.Checked)
            {
                checkCreateUnattend.BackColor = Color.LightGreen;

                MessageBox.Show(
                    "You have enabled the bypass for Windows 11 hardware requirements.\n\n" +
                    "This will: \n" +
                    "- Skip TPM 2.0 checks\n" +
                    "- Bypass the 4GB+ RAM requirement\n" +
                    "- Disable Secure Boot requirement\n" +
                    "- Allow creating a local account (no online setup required)\n\n" +
                    "During installation, Windows will offer the 'I don't have internet' option for offline account creation.",
                    "Bypass Enabled",
                    MessageBoxButtons.OK,
                    MessageBoxIcon.Information
                );
            }
            else
            {
                checkCreateUnattend.BackColor = Color.White;
            }
        }

        private async void linkShowDiskList_LinkClicked(object sender, LinkLabelLinkClickedEventArgs e)
        {
            await ShowDiskList();
        }

        private async void btnMCT_Click(object sender, EventArgs e)
        {
            string mediaCreationToolUrl = "https://go.microsoft.com/fwlink/?linkid=2156295";
            await usbManagement.DownloadMediaTool(mediaCreationToolUrl, "MediaCreationTool.exe");
        }

        private async void btnInstallAssistant_Click(object sender, EventArgs e)
        {
            string installAssistantUrl = "https://go.microsoft.com/fwlink/?linkid=2171764";
            await usbManagement.DownloadMediaTool(installAssistantUrl, "Windows11InstallationAssistant.exe");
        }

        private async void linkPCHealthCheckApp_LinkClicked(object sender, LinkLabelLinkClickedEventArgs e)
        {
            string installAssistantUrl = "https://github.com/rcmaehl/WhyNotWin11/releases/download/2.6.1.1/WhyNotWin11.exe";
            await usbManagement.DownloadMediaTool(installAssistantUrl, "WhyNotWin11.exe");
        }

        private void btnBack_Click(object sender, EventArgs e)
        {
            SwitchView.GoBack(this.Parent as Panel);
        }

   
    }
}