using pk3DS.Core;
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Text;
using System.Windows.Forms;

namespace pk3DS.WinForms;

public partial class TextEditor : Form
{
    public TextEditor(string[][] infiles, string mode)
    {
        InitializeComponent();
        files = infiles;
        Mode = mode;
        for (int i = 0; i < files.Length; i++)
            CB_Entry.Items.Add(i.ToString());
        CB_Entry.SelectedIndex = 0;
        dgv.EditMode = DataGridViewEditMode.EditOnEnter;
    }

    private readonly string[][] files;
    private readonly string Mode;
    private int entry = -1;

    // IO
    private void B_Export_Click(object sender, EventArgs e)
    {
        if (files.Length == 0)
            return;
        var Dump = new SaveFileDialog { Filter = "Text File|*.txt" };
        var sdr = Dump.ShowDialog();
        if (sdr != DialogResult.OK)
            return;
        bool newline = WinFormsUtil.Prompt(MessageBoxButtons.YesNo, "Remove newline formatting codes? (\\n,\\r,\\c)", "Removing newline formatting will make it more readable but will prevent any importing of that dump.") == DialogResult.Yes;
        string path = Dump.FileName;
        ExportTextFile(path, newline, files);
    }

    private void B_Import_Click(object sender, EventArgs e)
    {
        if (files.Length == 0)
            return;
        var Dump = new OpenFileDialog { Filter = "Text File|*.txt" };
        var odr = Dump.ShowDialog();
        if (odr != DialogResult.OK)
            return;
        string path = Dump.FileName;

        if (!ImportTextFile(path))
            return;

        // Reload the form with the new data.
        ChangeEntry(null, null);
        WinFormsUtil.Alert("Imported Text from Input Path:", path);
    }

    public static void ExportTextFile(string fileName, bool newline, string[][] fileData)
    {
        using var ms = new MemoryStream();
        ms.Write([0xFF, 0xFE], 0, 2); // Write Unicode BOM
        using (var tw = new StreamWriter(ms, new UnicodeEncoding()))
        {
            for (int i = 0; i < fileData.Length; i++)
            {
                // Get Strings for the File
                string[] data = fileData[i];
                // Append the File Header
                tw.WriteLine("~~~~~~~~~~~~~~~");
                tw.WriteLine("Text File : " + i);
                tw.WriteLine("~~~~~~~~~~~~~~~");
                // Write the String to the File
                if (data == null) continue;
                foreach (string line in data)
                {
                    tw.WriteLine(newline
                        ? line.Replace("\\n\\n", " ")
                            .Replace("\\n", " ")
                            .Replace("\\c", "")
                            .Replace("\\r", "")
                            .Replace("\\\\", "\\")
                            .Replace("\\[", "[")
                        : line);
                }
            }
        }
        File.WriteAllBytes(fileName, ms.ToArray());
    }

    private bool ImportTextFile(string fileName)
    {
        string[] fileText = File.ReadAllLines(fileName, Encoding.Unicode);
        string[][] textLines = new string[files.Length][];
        int ctr = 0;
        bool newlineFormatting = false;
        // Loop through all files
        for (int i = 0; i < fileText.Length; i++)
        {
            string line = fileText[i];
            if (line != "~~~~~~~~~~~~~~~")
                continue;
            string[] brokenLine = fileText[i++ + 1].Split([" : "], StringSplitOptions.None);
            if (brokenLine.Length != 2)
            { WinFormsUtil.Error($"Invalid Line @ {i}, expected Text File : {ctr}"); return false; }
            int file = Util.ToInt32(brokenLine[1]);
            if (file != ctr)
            { WinFormsUtil.Error($"Invalid Line @ {i}, expected Text File : {ctr}"); return false; }
            i += 2; // Skip over the other header line
            List<string> Lines = [];
            while (i < fileText.Length && fileText[i] != "~~~~~~~~~~~~~~~")
            {
                Lines.Add(fileText[i]);
                newlineFormatting |= fileText[i].Contains("\\n"); // Check if any line wasn't stripped of ingame formatting codes for human readability.
                i++;
            }
            i--;
            textLines[ctr++] = [.. Lines];
        }

        // Error Check
        if (ctr != files.Length)
        {
            WinFormsUtil.Error("The amount of Text Files in the input file does not match the required for the text file.",
                $"Received: {ctr}, Expected: {files.Length}"); return false;
        }
        if (!newlineFormatting)
        {
            WinFormsUtil.Error("The input Text Files do not have the ingame newline formatting codes (\\n,\\r,\\c).",
                "When exporting text, do not remove newline formatting."); return false;
        }

        // All Text Lines received. Store all back.
        for (int i = 0; i < files.Length; i++)
        {
            try { files[i] = textLines[i]; }
            catch (Exception e) { WinFormsUtil.Error($"The input Text File (# {i}) failed to convert:", e.ToString()); return false; }
        }

        return true;
    }

    private void ChangeEntry(object sender, EventArgs e)
    {
        // Save All the old text
        if (entry > -1 && sender != null)
        {
            try
            {
                files[entry] = GetCurrentDGLines();
            }
            catch (Exception ex) { WinFormsUtil.Error(ex.ToString()); }
        }

        // Reset
        entry = CB_Entry.SelectedIndex;
        SetStringsDataGridView(files[entry]);
    }

    // Main Handling
    private void SetStringsDataGridView(string[] textArray)
    {
        // Clear the datagrid row content to remove all text lines.
        dgv.Rows.Clear();
        // Clear the header columns, these are repopulated every time.
        dgv.Columns.Clear();
        if (textArray == null || textArray.Length == 0)
            return;
        // Reset settings and columns.
        dgv.AllowUserToResizeColumns = false;
        DataGridViewColumn dgvLine = new DataGridViewTextBoxColumn
        {
            HeaderText = "Line",
            DisplayIndex = 0,
            Width = 32,
            ReadOnly = true,
            SortMode = DataGridViewColumnSortMode.NotSortable,
        };
        dgvLine.DefaultCellStyle.Alignment = DataGridViewContentAlignment.MiddleRight;

        var dgvText = new DataGridViewTextBoxColumn
        {
            HeaderText = "Text",
            DisplayIndex = 1,
            SortMode = DataGridViewColumnSortMode.NotSortable,
            AutoSizeMode = DataGridViewAutoSizeColumnMode.Fill,
        };

        dgv.Columns.Add(dgvLine);
        dgv.Columns.Add(dgvText);
        dgv.Rows.Add(textArray.Length);

        // Add the text lines into their cells.
        for (int i = 0; i < textArray.Length; i++)
        {
            dgv.Rows[i].Cells[0].Value = i;
            dgv.Rows[i].Cells[1].Value = textArray[i];
        }
    }

    private string[] GetCurrentDGLines()
    {
        // Get Line Count
        string[] lines = new string[dgv.RowCount];
        for (int i = 0; i < dgv.RowCount; i++)
            lines[i] = (string)dgv.Rows[i].Cells[1].Value;
        return lines;
    }
    // Meta Usage
    private void B_AddLine_Click(object sender, EventArgs e)
    {
        int currentRow = 0;
        try { currentRow = dgv.CurrentRow.Index; }
        catch { dgv.Rows.Add(); }

        if (dgv.Rows.Count != 1 && (currentRow < dgv.Rows.Count - 1 || currentRow == 0))
        {
            if (ModifierKeys != Keys.Control && currentRow != 0)
            {
                if (WinFormsUtil.Prompt(MessageBoxButtons.YesNo, "Inserting in between rows will shift all subsequent lines.", "Continue?") != DialogResult.Yes)
                    return;
            }
            // Insert new Row after current row.
            dgv.Rows.Insert(currentRow + 1);
        }

        for (int i = 0; i < dgv.Rows.Count; i++)
            dgv.Rows[i].Cells[0].Value = i.ToString();
    }

    private void B_RemoveLine_Click(object sender, EventArgs e)
    {
        int currentRow = dgv.CurrentRow.Index;
        if (currentRow < dgv.Rows.Count - 1)
        {
            if (ModifierKeys != Keys.Control && DialogResult.Yes != WinFormsUtil.Prompt(MessageBoxButtons.YesNo, "Deleting a row above other lines will shift all subsequent lines.", "Continue?"))
                return;
        }
        dgv.Rows.RemoveAt(currentRow);

        // Resequence the Index Value column
        for (int i = 0; i < dgv.Rows.Count; i++)
            dgv.Rows[i].Cells[0].Value = i.ToString();
    }

    private void TextEditor_FormClosing(object sender, FormClosingEventArgs e)
    {
        // Save any pending edits
        dgv.EndEdit();
        // Save All the old text
        if (entry > -1) files[entry] = GetCurrentDGLines();
    }

    private void B_Randomize_Click(object sender, EventArgs e)
    {
        // gametext can be horribly broken if randomized
        if (Mode == "gametext" && DialogResult.Yes != WinFormsUtil.Prompt(MessageBoxButtons.YesNo, "Randomizing Game Text is dangerous!", "Continue?"))
            return;

        // get if the user wants to randomize current text file or all files
        var dr = WinFormsUtil.Prompt(MessageBoxButtons.YesNoCancel, $"Yes: Randomize ALL{Environment.NewLine}No: Randomize current textfile{Environment.NewLine}Cancel: Abort");

        if (dr == DialogResult.Cancel)
            return;

        // get if pure shuffle or smart shuffle (no shuffle if variable present)
        var drs = WinFormsUtil.Prompt(MessageBoxButtons.YesNo, $"Smart shuffle:{Environment.NewLine}Yes: Shuffle if no Variable present{Environment.NewLine}No: Pure random!");

        if (drs == DialogResult.Cancel)
            return;

        bool all = dr == DialogResult.Yes;
        bool smart = drs == DialogResult.Yes;

        // save current
        if (entry > -1)
            files[entry] = GetCurrentDGLines();

        // single-entire looping
        int start = all ? 0 : entry;
        int end = all ? files.Length - 1 : entry;

        // Gather strings
        List<string> strings = [];
        for (int i = start; i <= end; i++)
        {
            string[] data = files[i];
            strings.AddRange(smart
                ? data.Where(line => !line.Contains('['))
                : data);
        }

        // Shuffle up
        string[] pool = [.. strings];
        Util.Shuffle(pool);

        // Apply Text
        int ctr = 0;
        for (int i = start; i <= end; i++)
        {
            string[] data = files[i];

            for (int j = 0; j < data.Length; j++) // apply lines
            {
                if (!smart || !data[j].Contains('['))
                    data[j] = pool[ctr++];
            }

            files[i] = data;
        }

        // Load current text file
        SetStringsDataGridView(files[entry]);

        WinFormsUtil.Alert("Strings randomized!");
    }
}