using DarkControls;
using KsDumper11.Driver;
using Newtonsoft.Json;
using System;
using System.IO;
using System.Collections.Generic;
using System.ComponentModel;
using System.Data;
using System.Diagnostics;
using System.Drawing;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.Windows.Forms;
using static System.Windows.Forms.VisualStyles.VisualStyleElement;

namespace KsDumper11
{
    public partial class ProviderSelector : Form
    {
        protected override CreateParams CreateParams
        {
            get
            {
                // Activate double buffering at the form level.  All child controls will be double buffered as well.
                CreateParams cp = base.CreateParams;
                cp.ExStyle |= 0x02000000;  // Turn on WS_EX_COMPOSITED
                return cp;
            }
        }

        KduWrapper wrapper;

        public ProviderSelector()
        {

            InitializeComponent();

            this.FormBorderStyle = FormBorderStyle.None;
            this.Region = Region.FromHrgn(Utils.CreateRoundRectRgn(0, 0, Width, Height, 10, 10));
            this.closeBtn.Region = Region.FromHrgn(Utils.CreateRoundRectRgn(0, 0, closeBtn.Width, closeBtn.Height, 10, 10));

            this.appIcon1.DragForm = this;

            KduSelfExtract.Extract();

            wrapper = new KduWrapper(KduSelfExtract.KduPath);
            wrapper.DriverLoaded += Wrapper_DriverLoaded;
            wrapper.ProvidersLoaded += Wrapper_ProvidersLoaded;

            wrapper.LoadProviders();
        }

        private void Wrapper_ProvidersLoaded(object sender, EventArgs e)
        {
            foreach (KduProvider p in wrapper.providers)
            {
                ListViewItem item = new ListViewItem(p.ProviderIndex.ToString());
                item.SubItems.Add(p.ProviderName);

                if (p.ProviderName.Contains("[NOT WORKING]"))
                {
                    item.ForeColor = Color.Red;
                }

                if (p.ProviderName.Contains("[WORKING]"))
                {
                    item.ForeColor = Color.Green;
                }

                providerList.Items.Add(item);
            }

            if (wrapper.DefaultProvider != -1)
            { 
                providerList.SelectedIndices.Add(wrapper.DefaultProvider);
            }
            else
            {
                providerList.SelectedIndices.Add(0);
            }
        }

        private void Wrapper_DriverLoaded(object sender, object[] e)
        {
            if (this.InvokeRequired)
            {
                this.Invoke(new EventHandler<object[]>(Wrapper_DriverLoaded), new object[] { sender, e });
            }
            else
            {
                bool res = (bool)e[0];

                int idx = (int)e[1];

                ListViewItem item = providerList.Items[idx];

                if (res)
                {
                    driverLoadedLbl.ForeColor = Color.Green;
                    driverLoadedLbl.Text = "Driver Loaded!";

                    item.SubItems[1].Text = "[WORKING] " + item.SubItems[1].Text;

                    if (providerList.SelectedIndices.Count > 0 && providerList.SelectedIndices[0] == idx)
                    {
                        setDefaultProviderBtn.Enabled = true;
                    }

                    item.ForeColor = Color.Green;
                }
                else
                {
                    driverLoadedLbl.ForeColor = Color.Red;
                    driverLoadedLbl.Text = "Driver failed to load!";

                    item.SubItems[1].Text = "[NOT WORKING] " + item.SubItems[1].Text;

                    if (providerList.SelectedIndices.Count > 0 && providerList.SelectedIndices[0] == idx)
                    {
                        setDefaultProviderBtn.Enabled = false;
                    }

                    item.ForeColor = Color.Red;
                }

                driverLoadedLbl.Visible = true;
                driverLoadedLblTimer.Start();
            }
        }

        protected override void WndProc(ref Message m)
        {
            base.WndProc(ref m);
            if (m.Msg == Utils.WM_NCHITTEST)
                m.Result = (IntPtr)(Utils.HT_CAPTION);
        }

        private void providerList_SelectedIndexChanged(object sender, EventArgs e)
        {
            if (providerList.SelectedIndices.Count > 0)
            {
                int idx = providerList.SelectedIndices[0];

                KduProvider p = wrapper.providers[idx];

                if (p.ProviderName.Contains("[NOT WORKING]") || p.ProviderName.Contains("[WORKING]"))
                {
                    testProviderBtn.Enabled = false;
                    
                }
                else
                {
                    testProviderBtn.Enabled = true;
                    
                }

                if (p.ProviderName.Contains("[NOT WORKING]") )
                {
                    setDefaultProviderBtn.Enabled = false;
                }
                else if (p.ProviderName.Contains("[WORKING]"))
                {
                    setDefaultProviderBtn.Enabled = true;
                }
                else
                {
                    setDefaultProviderBtn.Enabled = false;
                }

                driverNameBox.Text = p.DriverName;
                deviceNameBox.Text = p.DeviceName;
                signerNameBox.Text = p.SignerName;
                minWinBuildBox.Text = p.MinWindowsBuild;
                maxWinBuildBox.Text = p.MaxWindowsBuild;
                driverWhqlSignedBox.Checked = p.IsWHQL_Signed;
                shellcodeMaskBox.Text = p.ShellcodeSupportMask;

                defaultProviderIDBox.Text = wrapper.DefaultProvider.ToString();

                if (p.ExtraInfo.Length > 0)
                {
                    providerExtraInfoBox.Clear();
                    for (int i = 0; i < p.ExtraInfo.Length; i++)
                    {
                        providerExtraInfoBox.AppendText(p.ExtraInfo[i] + Environment.NewLine);
                    }
                }
            }
        }

        private void testProviderBtn_Click(object sender, EventArgs e)
        {
            if (providerList.SelectedIndices.Count > 0)
            {
                testProviderBtn.Enabled = false;

                int idx = providerList.SelectedIndices[0];

                KduProvider p = wrapper.providers[idx];

                wrapper.tryLoad(p.ProviderIndex);
            }
        }

        private void driverLoadedLblTimer_Tick(object sender, EventArgs e)
        {
            testProviderBtn.Enabled = true;
            driverLoadedLbl.Visible = false;
            driverLoadedLblTimer.Stop();
        }

        private void setDefaultProviderBtn_Click(object sender, EventArgs e)
        {
            if (providerList.SelectedIndices.Count > 0)
            {
                testProviderBtn.Enabled = false;

                int idx = providerList.SelectedIndices[0];

                wrapper.SetDefaultProvider(idx);

                defaultProviderIDBox.Text = wrapper.DefaultProvider.ToString();

                wrapper.Start();

                this.Close();
            }
        }
    }
}