using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Text;
using Lumina;
using Lumina.Data;
using Lumina.Excel;
using Lumina.Excel.GeneratedSheets;
using Lumina.Text;
using Pidgin;

namespace SourceGenerator {
    internal class Program {
        private static void Main(string[] args) {
            var data = new Dictionary<Language, GameData>(4);
            foreach (var lang in Languages.Keys) {
                data[lang] = new GameData(args[0], new LuminaOptions {
                    PanicOnSheetChecksumMismatch = false,
                    DefaultExcelLanguage = lang,
                });
            }

            var prog = new Program(data);

            File.WriteAllText(Path.Join(args[1], "duties.rs"), prog.GenerateDuties());
            File.WriteAllText(Path.Join(args[1], "jobs.rs"), prog.GenerateJobs());
            File.WriteAllText(Path.Join(args[1], "roulettes.rs"), prog.GenerateRoulettes());
            File.WriteAllText(Path.Join(args[1], "worlds.rs"), prog.GenerateWorlds());
            File.WriteAllText(Path.Join(args[1], "territory_names.rs"), prog.GenerateTerritoryNames());
            File.WriteAllText(Path.Join(args[1], "auto_translate.rs"), prog.GenerateAutoTranslate());
            File.WriteAllText(Path.Join(args[1], "treasure_maps.rs"), prog.GenerateTreasureMaps());
        }

        private Dictionary<Language, GameData> Data { get; }

        private Program(Dictionary<Language, GameData> data) {
            this.Data = data;
        }

        private static StringBuilder DefaultHeader(bool localisedText = false) {
            var sb = new StringBuilder("use std::collections::HashMap;\n");

            if (localisedText) {
                sb.Append("use super::LocalisedText;\n");
            }

            return sb;
        }

        private static readonly Dictionary<Language, string> Languages = new() {
            [Language.English] = "en",
            [Language.Japanese] = "ja",
            [Language.German] = "de",
            [Language.French] = "fr",
        };

        private string? GetLocalisedStruct<T>(uint rowId, Func<T, SeString?> nameFunc, uint indent = 0, bool capitalise = false) where T : ExcelRow {
            var def = this.Data[Language.English].GetExcelSheet<T>()!.GetRow(rowId)!;
            var defName = nameFunc(def)?.TextValue();
            if (string.IsNullOrEmpty(defName)) {
                return null;
            }

            var sb = new StringBuilder();

            sb.Append("LocalisedText {\n");

            foreach (var (language, key) in Languages) {
                var row = this.Data[language].GetExcelSheet<T>(language)?.GetRow(rowId);
                var name = row == null
                    ? defName
                    : nameFunc(row)?.TextValue().Replace("\"", "\\\"");
                name ??= defName;
                if (capitalise) {
                    name = name[..1].ToUpperInvariant() + name[1..];
                }

                for (var i = 0; i < indent + 4; i++) {
                    sb.Append(' ');
                }

                sb.Append($"{key}: \"{name}\",\n");
            }

            for (var i = 0; i < indent; i++) {
                sb.Append(' ');
            }

            sb.Append('}');

            return sb.ToString();
        }

        private string GenerateDuties() {
            var sb = DefaultHeader(true);
            sb.Append('\n');

            sb.Append("#[derive(Debug)]\n");
            sb.Append("pub struct DutyInfo {\n");
            sb.Append("    pub name: LocalisedText,\n");
            sb.Append("    pub high_end: bool,\n");
            sb.Append("    pub content_kind: ContentKind,\n");
            sb.Append("}\n\n");

            sb.Append("#[derive(Debug, Clone, Copy)]\n");
            sb.Append("#[allow(unused)]\n");
            sb.Append("#[repr(u32)]\n");
            sb.Append("pub enum ContentKind {\n");
            foreach (var kind in this.Data[Language.English].GetExcelSheet<ContentType>()!) {
                var name = kind.Name.TextValue().Replace(" ", "").Replace("&", "");
                if (name.Length > 0) {
                    sb.Append($"    {name} = {kind.RowId},\n");
                }
            }

            sb.Append("    Other(u32),\n");
            sb.Append("}\n\n");

            sb.Append("impl ContentKind {\n");

            sb.Append("    fn from_u32(kind: u32) -> Self {\n");
            sb.Append("        match kind {\n");
            foreach (var kind in this.Data[Language.English].GetExcelSheet<ContentType>()!) {
                var name = kind.Name.TextValue().Replace(" ", "").Replace("&", "");
                if (name.Length > 0) {
                    sb.Append($"            {kind.RowId} => Self::{name},\n");
                }
            }

            sb.Append("            x => Self::Other(x),\n");
            sb.Append("        }\n");
            sb.Append("    }\n\n");

            sb.Append("    pub fn as_u32(self) -> u32 {\n");
            sb.Append("        match self {\n");
            foreach (var kind in this.Data[Language.English].GetExcelSheet<ContentType>()!) {
                var name = kind.Name.TextValue().Replace(" ", "").Replace("&", "");
                if (name.Length > 0) {
                    sb.Append($"            Self::{name} => {kind.RowId},\n");
                }
            }

            sb.Append("            Self::Other(x) => x,\n");
            sb.Append("        }\n");
            sb.Append("    }\n");

            sb.Append("}\n\n");

            sb.Append("lazy_static::lazy_static! {\n");
            sb.Append("    pub static ref DUTIES: HashMap<u32, DutyInfo> = maplit::hashmap! {\n");

            foreach (var cfc in this.Data[Language.English].GetExcelSheet<ContentFinderCondition>()!) {
                if (cfc.RowId == 0) {
                    continue;
                }

                var name = this.GetLocalisedStruct<ContentFinderCondition>(cfc.RowId, row => row.Name, 12, true);
                if (name == null) {
                    continue;
                }

                var highEnd = cfc.HighEndDuty ? "true" : "false";
                var contentType = cfc.ContentType.Value;
                var contentKind = contentType?.Name?.TextValue().Replace(" ", "").Replace("&", "");
                if (string.IsNullOrEmpty(contentKind)) {
                    contentKind = $"Other({contentType?.RowId ?? 0})";
                }

                sb.Append($"        {cfc.RowId} => DutyInfo {{\n");
                sb.Append($"            name: {name},\n");
                sb.Append($"            high_end: {highEnd},\n");
                sb.Append($"            content_kind: ContentKind::{contentKind},\n");
                sb.Append("        },\n");
            }

            sb.Append("    };\n");
            sb.Append("}\n");

            return sb.ToString();
        }

        private string GenerateJobs() {
            var sb = DefaultHeader();
            sb.Append("use ffxiv_types::jobs::{ClassJob, Class, Job, NonCombatJob};\n\n");
            sb.Append("lazy_static::lazy_static! {\n");
            sb.Append("    pub static ref JOBS: HashMap<u32, ClassJob> = maplit::hashmap! {\n");

            foreach (var cj in this.Data[Language.English].GetExcelSheet<ClassJob>()!) {
                if (cj.RowId == 0) {
                    continue;
                }

                var name = cj.NameEnglish.TextValue().Replace(" ", "");
                if (name.Length <= 0) {
                    continue;
                }

                var isCombat = cj.Role != 0;
                var isClass = cj.JobIndex == 0;

                string value;
                if (isCombat) {
                    value = isClass
                        ? $"ClassJob::Class(Class::{name})"
                        : $"ClassJob::Job(Job::{name})";
                } else {
                    value = $"ClassJob::NonCombat(NonCombatJob::{name})";
                }

                sb.Append($"        {cj.RowId} => {value},\n");
            }

            sb.Append("    };\n");
            sb.Append("}\n");

            return sb.ToString();
        }

        private string GenerateRoulettes() {
            var sb = DefaultHeader(true);
            sb.Append('\n');
            sb.Append("#[derive(Debug)]\n");
            sb.Append("pub struct RouletteInfo {\n");
            sb.Append("    pub name: LocalisedText,\n");
            sb.Append("    pub pvp: bool,\n");
            sb.Append("}\n\n");

            sb.Append("lazy_static::lazy_static! {\n");
            sb.Append("    pub static ref ROULETTES: HashMap<u32, RouletteInfo> = maplit::hashmap! {\n");

            foreach (var cr in this.Data[Language.English].GetExcelSheet<ContentRoulette>()!) {
                if (cr.RowId == 0) {
                    continue;
                }

                var name = this.GetLocalisedStruct<ContentRoulette>(cr.RowId, row => row.Name, 12);
                if (name == null) {
                    continue;
                }

                var pvp = cr.IsPvP
                    ? "true"
                    : "false";

                sb.Append($"        {cr.RowId} => RouletteInfo {{\n");
                sb.Append($"            name: {name},\n");
                sb.Append($"            pvp: {pvp},\n");
                sb.Append("        },\n");
            }

            sb.Append("    };\n");
            sb.Append("}\n");

            return sb.ToString();
        }

        private string GenerateWorlds() {
            var sb = DefaultHeader();
            sb.Append("use ffxiv_types::World;\n\n");
            sb.Append("lazy_static::lazy_static! {\n");
            sb.Append("    pub static ref WORLDS: HashMap<u32, World> = maplit::hashmap! {\n");

            foreach (var world in this.Data[Language.English].GetExcelSheet<World>()!) {
                if (world.RowId == 0 || !world.IsPublic || world.UserType == 0 || world.DataCenter.Row == 0) {
                    continue;
                }

                var name = world.Name.TextValue();
                if (name.Length <= 0) {
                    continue;
                }

                sb.Append($"        {world.RowId} => World::{name},\n");
            }

            sb.Append("    };\n");
            sb.Append("}\n");

            return sb.ToString();
        }

        private string GenerateTerritoryNames() {
            var sb = DefaultHeader(true);
            sb.Append("\nlazy_static::lazy_static! {\n");
            sb.Append("    pub static ref TERRITORY_NAMES: HashMap<u32, LocalisedText> = maplit::hashmap! {\n");

            foreach (var tt in this.Data[Language.English].GetExcelSheet<TerritoryType>()!) {
                if (tt.RowId == 0 || tt.PlaceName.Row == 0) {
                    continue;
                }

                var name = this.GetLocalisedStruct<TerritoryType>(
                    tt.RowId,
                    row => row.PlaceName.Value!.Name,
                    8
                );
                if (name == null) {
                    continue;
                }

                sb.Append($"        {tt.RowId} => {name},\n");
            }

            sb.Append("    };\n");
            sb.Append("}\n");

            return sb.ToString();
        }

        private string GenerateAutoTranslate() {
            var sb = DefaultHeader(true);
            sb.Append("\nlazy_static::lazy_static! {\n");
            sb.Append("    pub static ref AUTO_TRANSLATE: HashMap<(u32, u32), LocalisedText> = maplit::hashmap! {\n");

            var parser = AutoTranslate.Parser();
            foreach (var row in this.Data[Language.English].GetExcelSheet<Completion>()!) {
                var lookup = row.LookupTable.TextValue();
                if (lookup is not ("" or "@")) {
                    var (sheetName, selector) = parser.ParseOrThrow(lookup);
                    var sheetType = typeof(Completion)
                        .Assembly
                        .GetType($"Lumina.Excel.GeneratedSheets.{sheetName}")!;
                    var getSheet = this.Data[Language.English]
                        .GetType()
                        .GetMethod("GetExcelSheet", Type.EmptyTypes)!
                        .MakeGenericMethod(sheetType);
                    var sheets = this.Data.ToDictionary(
                        pair => pair.Key,
                        pair => {
                            var sheet = (ExcelSheetImpl) getSheet.Invoke(pair.Value, null)!;
                            return (sheet, sheet.GetRowParsers().ToArray());
                        });

                    var columns = new List<int>();
                    var rows = new List<Range>();
                    if (selector.HasValue) {
                        columns.Clear();
                        rows.Clear();

                        foreach (var part in selector.Value) {
                            switch (part) {
                                case IndexRange range: {
                                    var start = (int) range.Start;
                                    var end = (int) (range.End + 1);
                                    rows.Add(start..end);
                                    break;
                                }
                                case SingleRow single: {
                                    var idx = (int) single.Row;
                                    rows.Add(idx..(idx + 1));
                                    break;
                                }
                                case ColumnSpecifier col:
                                    columns.Add((int) col.Column);
                                    break;
                            }
                        }
                    }

                    if (columns.Count == 0) {
                        columns.Add(0);
                    }

                    if (rows.Count == 0) {
                        rows.Add(..);
                    }

                    var builder = new StringBuilder();
                    foreach (var range in rows) {
                        var validRows = sheets[Language.English]
                            .Item2
                            .Select(parser => parser.RowId)
                            .ToArray();
                        for (var i = range.Start.Value; i < range.End.Value; i++) {
                            if (!validRows.Contains((uint) i)) {
                                continue;
                            }

                            builder.Clear();

                            builder.Append($"        ({row.Group}, {i}) => LocalisedText {{\n");

                            var lines = 0;
                            foreach (var (lang, (_, parsers)) in sheets) {
                                // take the first column that works
                                foreach (var col in columns) {
                                    var rowParser = parsers.FirstOrDefault(parser => parser.RowId == i);
                                    if (rowParser != null) {
                                        var name = rowParser.ReadColumn<SeString>(col)!;
                                        var text = name.TextValue().Replace("\"", "\\\"");
                                        if (text.Length > 0) {
                                            builder.Append($"            {Languages[lang]}: \"{text}\",\n");
                                            lines += 1;
                                            break;
                                        }
                                    }
                                }
                            }

                            builder.Append("        },\n");

                            if (lines != 4) {
                                continue;
                            }

                            sb.Append(builder);
                        }
                    }
                } else {
                    var text = this.GetLocalisedStruct<Completion>(row.RowId, row => row.Text, 8);
                    if (text != null) {
                        sb.Append($"        ({row.Group}, {row.RowId}) => {text},\n");
                    }
                }
            }

            sb.Append("    };\n");
            sb.Append("}\n");

            return sb.ToString();
        }

        private string GenerateTreasureMaps() {
            var sb = DefaultHeader(true);
            sb.Append("\nlazy_static::lazy_static! {\n");
            sb.Append("    pub static ref TREASURE_MAPS: HashMap<u32, LocalisedText> = maplit::hashmap! {\n");
            sb.Append("        0 => LocalisedText {\n");
            sb.Append("            en: \"All Levels\",\n");
            sb.Append("            ja: \"レベルを指定しない\",\n");
            sb.Append("            de: \"Jede Stufe\",\n");
            sb.Append("            fr: \"Tous niveaux\",\n");
            sb.Append("        },\n");

            var i = 1;
            foreach (var row in this.Data[Language.English].GetExcelSheet<TreasureHuntRank>()!) {
                // IS THIS RIGHT?
                if (row.TreasureHuntTexture != 0) {
                    continue;
                }

                SeString? GetMapName(TreasureHuntRank thr) {
                    var name = thr.KeyItemName.Value?.Name;
                    return string.IsNullOrEmpty(name?.TextValue())
                        ? thr.ItemName.Value?.Name
                        : name;
                }

                var name = this.GetLocalisedStruct<TreasureHuntRank>(row.RowId, GetMapName, 8);
                if (!string.IsNullOrEmpty(name)) {
                    sb.Append($"        {i++} => {name},\n");
                }
            }

            sb.Append("    };\n");
            sb.Append("}\n");

            return sb.ToString();
        }
    }
}