using System;
using System.Collections.Generic;
using System.IO;
using WindowsInput;
using Patagames.Ocr;
using System.Timers;
using System.Drawing;
using System.Linq;
using WindowsInput.Native;
using System.Windows.Forms;
using System.Text.RegularExpressions;
using System.Threading;

namespace SoD2_Reroll
{
    public partial class SoD2Reroll : Form
    {
        private System.Timers.Timer rerollTimer;
        private readonly InputSimulator sim = new InputSimulator();
        private bool reroll = false;
        private bool firstIteration = false;
        private short wait = 10000;
        private readonly short interval = 50;
        private Size resolution = new Size(1920, 1080);
        private static readonly int heightTrait = 125, heightSkill = 35;

        //Array that holds the currently selected skills and traits in combo boxes
        private string[] activeSkills = {"", "", ""};
        private string[,] activeTraits = {{"", "", ""}, {"", "", ""}, {"", "", ""}};

        public SoD2Reroll()
        {
            InitializeComponent();
        }

        private void SoD2Reroll_Load(object sender, EventArgs e)
        {
            //Arrays of all skills and traits obtainable by random characters, excludes red talon and heartland exlusives
            string[] skills = Properties.Resources.skills.Split(new[] {'\n'}, StringSplitOptions.None);
            string[] traits = Properties.Resources.traits.Split(new[] {'\n'}, StringSplitOptions.None);

            skills.Select(s => s.Trim())
                .Where(s => s.Length > 0)
                .ToList()
                .ForEach(s =>
                {
                    cbSurvivor1.Items.Add(s);
                    cbSurvivor2.Items.Add(s);
                    cbSurvivor3.Items.Add(s);
                });

            traits.Select(s => s.Trim())
                .Where(s => s.Length > 0)
                .ToList()
                .ForEach(t =>
                {
                    cbS1Trait1.Items.Add(t);
                    cbS1Trait2.Items.Add(t);
                    cbS1Trait3.Items.Add(t);
                    cbS2Trait1.Items.Add(t);
                    cbS2Trait2.Items.Add(t);
                    cbS2Trait3.Items.Add(t);
                    cbS3Trait1.Items.Add(t);
                    cbS3Trait2.Items.Add(t);
                    cbS3Trait3.Items.Add(t);
                });

            cbResolution.Items.AddRange(new string[]
                {"1280x720", "1360x720", "1366x720", "1600x900", "1920x1080", "2560x1440"});
            cbResolution.SelectedIndex = 4;

            rerollTimer = new System.Timers.Timer(wait);
            rerollTimer.Elapsed += RerollTimer_Elapsed;

            Stop();
        }

        private void RerollTimer_Elapsed(object sender, ElapsedEventArgs e)
        {
            rerollTimer.Interval = interval;
            lock (sender)
            {
                if (firstIteration)
                {
                    //Move cursor to the first button
                    sim.Keyboard.KeyPress(VirtualKeyCode.UP);
                    sim.Keyboard.KeyPress(VirtualKeyCode.LEFT);
                    sim.Keyboard.KeyPress(VirtualKeyCode.LEFT);

                    firstIteration = false;
                }

                //int top = (int)Math.Round(resolution.Height / 1.55);
                int topTrait = (int) Math.Round(resolution.Height / 3.65);
                int topSkill = (int) Math.Round(resolution.Height / 1.55);

                string[] traitStrings = new string[3];
                string[] skillStrings = new string[3];
                bool[] matchingSurvivorSkills = {false, false, false};
                bool[] matchingSurvivorTraits = {false, false, false};

                sim.Keyboard.KeyPress(VirtualKeyCode.UP);
                sim.Keyboard.KeyPress(VirtualKeyCode.LEFT);
                sim.Keyboard.KeyPress(VirtualKeyCode.LEFT);

                //Use the screenreader to determine what text is in the screenshot
                using (OcrApi objOcr = OcrApi.Create())
                {
                    objOcr.Init(Patagames.Ocr.Enums.Languages.English);

                    using (Bitmap img = Screenshot())
                    {
                        int leftSkill = 0;
                        int leftTrait = 0;
                        for (int i = 0; i < 3; ++i)
                        {
                            switch (i)
                            {
                                //screenshot locations for different suvivors
                                case 0:
                                    leftTrait = (int) Math.Round(resolution.Width / 5.47);
                                    leftSkill = resolution.Width / 5;
                                    break;
                                case 1:
                                    leftTrait = (int) Math.Round(resolution.Width / 1.995);
                                    leftSkill = (int) Math.Round(resolution.Width / 1.925);
                                    break;
                                case 2:
                                    leftTrait = (int) Math.Round(4.1 * resolution.Width / 5);
                                    leftSkill = (int) Math.Round(4.2 * resolution.Width / 5);
                                    break;
                            }

                            string traitText = objOcr.GetTextFromImage(
                                img,
                                leftTrait,
                                topTrait,
                                375,
                                heightTrait
                            );
                            traitStrings[i] = Regex.Replace(traitText.Trim(), @"\s+", "").ToUpper();

                            string skillText = objOcr.GetTextFromImage(
                                img,
                                leftSkill,
                                topSkill,
                                375,
                                heightSkill
                            );
                            skillStrings[i] = Regex.Replace(skillText.Trim(), @"\s+", "").ToUpper();
                        }


                        for (int a = 0; a < 3; ++a)
                        {
                            if (activeSkills[a].Length == 0)
                            {
                                matchingSurvivorSkills[a] = true;
                            }
                            else
                            {
                                if (activeSkills[a].Trim().ToUpper() == skillStrings[a].Trim().ToUpper())
                                {
                                    matchingSurvivorSkills[a] = true;
                                }
                            }

                            List<string> wantedTraits = new List<string>();
                            for (int at = 0; at < 3; ++at)
                            {
                                if (activeTraits[a, at].Length > 0)
                                {
                                    wantedTraits.Add(activeTraits[a, at].Trim().ToUpper());
                                }
                            }

                            if (wantedTraits.Count == 0)
                            {
                                matchingSurvivorTraits[a] = true;
                            }
                            else
                            {
                                matchingSurvivorTraits[a] = wantedTraits.All(w => traitStrings[a].Contains(w));
                            }
                        }


                        bool modified = false;
                        for (int i = 0; i < 3; ++i)
                        {
                            if (matchingSurvivorSkills[i] == false || matchingSurvivorTraits[i] == false)
                            {
                                sim.Keyboard.KeyPress(VirtualKeyCode.RETURN);
                                sim.Keyboard.KeyPress(VirtualKeyCode.RIGHT);
                                modified = true;
                            }
                        }

                        if (!modified)
                        {
                            rerollTimer.Enabled = false;
                        }

                        sim.Keyboard.KeyPress(VirtualKeyCode.LEFT);
                        sim.Keyboard.KeyPress(VirtualKeyCode.LEFT);
                    }
                }
            }
        }

        private Bitmap Screenshot()
        {
            Bitmap img = new Bitmap(resolution.Width, resolution.Height);
            Graphics g = Graphics.FromImage(img);
            g.CopyFromScreen(0, 0, 0, 0, new Size(resolution.Width, resolution.Height), CopyPixelOperation.SourceCopy);
            return img;
        }

        private void EnableTimer()
        {
            ////Program waits 50ms before iterating to prevent misreading by the OCR
            System.Threading.Thread.Sleep(50);

            //Timer is enabled to continue iterating if needed, program stops when CONTROL is pressed
            if (ModifierKeys.HasFlag(Keys.Control))
            {
                Stop();
            }
            else if (reroll)
            {
                rerollTimer.Enabled = true;
            }
        }

        //Get the distance between two strings of different lengths
        public static int ComputeStringDistance(string input1, string input2)
        {
            var bounds = new {Height = input1.Length + 1, Width = input2.Length + 1};
            int[,] matrix = new int[bounds.Height, bounds.Width];
            for (int height = 0; height < bounds.Height; height++)
            {
                matrix[height, 0] = height;
            }

            for (int width = 0; width < bounds.Width; width++)
            {
                matrix[0, width] = width;
            }

            for (int height = 1; height < bounds.Height; height++)
            {
                for (int width = 1; width < bounds.Width; width++)
                {
                    int cost = (input1[height - 1] == input2[width - 1]) ? 0 : 1;
                    int insertion = matrix[height, width - 1] + 1;
                    int deletion = matrix[height - 1, width] + 1;
                    int substitution = matrix[height - 1, width - 1] + cost;
                    int distance = Math.Min(insertion, Math.Min(deletion, substitution));
                    if (height > 1 && width > 1 && input1[height - 1] == input2[width - 2] &&
                        input1[height - 2] == input2[width - 1])
                    {
                        distance = Math.Min(distance, matrix[height - 2, width - 2] + cost);
                    }

                    matrix[height, width] = distance;
                }
            }

            return matrix[bounds.Height - 1, bounds.Width - 1];
        }

        private void Stop()
        {
            //Reset and disable the timer
            rerollTimer.Interval = wait;
            reroll = false;
            rerollTimer.Enabled = false;
            ToggleButtons(true);
        }

        private void ToggleButtons(bool toggle)
        {
            //Enable and disable buttons
            btnStop.BeginInvoke((Action) delegate() { btnStop.Enabled = !toggle; });
            btnStart.BeginInvoke((Action) delegate() { btnStart.Enabled = toggle; });
        }

        private void SkillChanged(object sender, EventArgs e)
        {
            int index = 0;
            ComboBox cb = (ComboBox) sender;

            switch (cb.Name)
            {
                case "cbSurvivor1":
                    index = 0;
                    break;
                case "cbSurvivor2":
                    index = 1;
                    break;
                case "cbSurvivor3":
                    index = 2;
                    break;
            }

            activeSkills[index] = Regex.Replace(cb.GetItemText(cb.SelectedItem).ToUpper(), @"\s+", "");
        }

        private void TraitChanged(object sender, EventArgs e)
        {
            int i = 0, j = 0;
            ComboBox cb = (ComboBox) sender;

            switch (cb.Name)
            {
                case "cbS1Trait1":
                    i = 0;
                    j = 0;
                    break;
                case "cbS1Trait2":
                    i = 0;
                    j = 1;
                    break;
                case "cbS1Trait3":
                    i = 0;
                    j = 2;
                    break;
                case "cbS2Trait1":
                    i = 1;
                    j = 0;
                    break;
                case "cbS2Trait2":
                    i = 1;
                    j = 1;
                    break;
                case "cbS2Trait3":
                    i = 1;
                    j = 2;
                    break;
                case "cbS3Trait1":
                    i = 2;
                    j = 0;
                    break;
                case "cbS3Trait2":
                    i = 2;
                    j = 1;
                    break;
                case "cbS3Trait3":
                    i = 2;
                    j = 2;
                    break;
            }

            activeTraits[i, j] = Regex.Replace(cb.GetItemText(cb.SelectedItem).ToUpper(), @"\s+", "");
        }

        private void btnStart_Click(object sender, EventArgs e)
        {
            cbResolution.Select(0, 0);
            ToggleButtons(false);
            firstIteration = true;
            reroll = true;
            rerollTimer.Enabled = true;
        }

        private void SoD2Reroll_KeyDown(object sender, KeyEventArgs e)
        {
            e.Handled = true;
        }

        private void btnStop_Click(object sender, EventArgs e)
        {
            Stop();
        }

        private void nudWait_ValueChanged(object sender, EventArgs e)
        {
            wait = Convert.ToInt16(nudWait.Value * 100);
        }

        private void cbResolution_SelectedIndexChanged(object sender, EventArgs e)
        {
            string[] dimensions = cbResolution.GetItemText(cbResolution.SelectedItem).Split('x');
            resolution.Width = Convert.ToInt16(dimensions[0]);
            resolution.Height = Convert.ToInt16(dimensions[1]);
        }
    }
}