using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.Drawing;
using System.Globalization;
using System.IO;
using System.Linq;
using System.Text;
using System.Text.RegularExpressions;
using System.Threading.Tasks;
using System.Windows.Documents;
using OMI;
using OMI.Formats.GameRule;
using OMI.Formats.Languages;
using OMI.Formats.Pck;
using OMI.Workers.GameRule;
using OMI.Workers.Language;
using OMI.Workers.Pck;
using PckStudio.Core.Deserializer;
using PckStudio.Core.Extensions;
using PckStudio.Core.Interfaces;

namespace PckStudio.Core.DLC
{
    public sealed class DLCManager
    {
        public static DLCManager Default { get; } = new DLCManager(default, AppLanguage.SystemDefault);
       
        internal const string DEFAULT_TEXTURE_PACK_FILENAME = "TexturePack.pck";
        internal const string DEFAULT_MINIGAME_PACK_FILENAME = "WorldPack.pck";
        internal const string DATA_DIRECTORY_NAME = "Data";
        internal const string PACKAGE_DISPLAYNAME_ID = "IDS_DISPLAY_NAME";

        public ByteOrder ByteOrder => _byteOrder;

        public ConsolePlatform Platform => _platform;

        /// <summary>
        /// See <see cref="AvailableLanguages"/> for details.
        /// </summary>
        public string PreferredLanguage { get; private set; }

        private readonly DLCPackageRegistry _packageRegistry = new DLCPackageRegistry();
        private readonly Random _rng = new Random();
        private ByteOrder _byteOrder;
        private ConsolePlatform _platform;


        /// <param name="byteOrder"></param>
        /// <param name="platform"></param>
        /// <param name="preferredLanguage">See <see cref="AvailableLanguages"/> for details.</param>
        public DLCManager(ConsolePlatform platform, AppLanguage preferredLanguage)
        {
            _platform = platform;
            _byteOrder = GetByteOrderForPlatform(Platform);
            PreferredLanguage = GetPreferredLanguage(preferredLanguage);
        }

        private static ByteOrder GetByteOrderForPlatform(ConsolePlatform platform)
        {
            return platform switch
            {
                ConsolePlatform.Switch => ByteOrder.LittleEndian,
                ConsolePlatform.PS4 => ByteOrder.LittleEndian,
                _ => ByteOrder.BigEndian
            };
        }

        public IDLCPackage CreateNewPackage(string name, DLCPackageType packageType)
        {
            int identifier = _rng.Next(8000, GameConstants.MAX_PACK_ID);
            IDLCPackage package = packageType switch
            {
                DLCPackageType.SkinPack    => DLCSkinPackage.CreateEmpty(name, identifier),
                DLCPackageType.TexturePack => DLCTexturePackage.CreateDefaultPackage(name, "", identifier),
                DLCPackageType.MashUpPack  => new DLCMashUpPackage(name, "", identifier),
                //! TODO: implemnt minigame dlc packages -null
                DLCPackageType.MG01        => new DLCBattlePackage(name, identifier),
                DLCPackageType.MG02        => new DLCMiniGamePackage(name, identifier, packageType, MiniGameId.Tumble),
                DLCPackageType.MG03        => new DLCMiniGamePackage(name, identifier, packageType, MiniGameId.Glide),
                DLCPackageType.Invalid     => InvalidDLCPackage.Instance,
                _ => throw new ArgumentException("Unable to create DLC Package of 'Unknown' type."),
            };

            LOCFile localisation = new LOCFile();
            localisation.AddLanguage(PreferredLanguage);
            localisation.AddLocKey(PACKAGE_DISPLAYNAME_ID, name);
            _packageRegistry.RegisterPackage(identifier, package, localisation);
            
            return package;
        }

        public IDLCPackage OpenDLCPackage(string filepath) => OpenDLCPackage(new FileInfo(filepath));

        public IDLCPackage OpenDLCPackage(FileInfo fileInfo)
        {
            if (Platform == ConsolePlatform.Unknown)
                throw new Exception($"{nameof(Platform)} is Unknown.");

            if (!IsValidPckFile(fileInfo))
                return InvalidDLCPackage.Instance;

            using Stream stream = fileInfo.OpenRead();

            PckFileReader fileReader = new PckFileReader(ByteOrder);

            PckFile pckFile = fileReader.FromStream(stream);

            if (!pckFile.TryGetAsset("0", PckAssetType.InfoFile, out PckAsset zeroAsset))
            {
                Trace.TraceError("Could not find asset named:'0'.");
                return new UnknownDLCPackage(fileInfo.Name, pckFile);
            }

            int identifier = (zeroAsset?.HasProperty("PACKID") ?? default) ? zeroAsset.GetProperty("PACKID", int.Parse) : -1;
            if (identifier <= 0 || identifier > GameConstants.MAX_PACK_ID)
            {
                Trace.TraceError($"{nameof(identifier)}({identifier}) was out of range!");
                return new UnknownDLCPackage(fileInfo.Name, pckFile);
            }

            if (_packageRegistry.ContainsPackage(identifier))
                return _packageRegistry[identifier];

            LOCFile localisation = pckFile.GetAssetsByType(PckAssetType.LocalisationFile).FirstOrDefault()?.GetData(new LOCFileReader());
            if (localisation is null)
                return new UnknownDLCPackage(fileInfo.Name, pckFile);

            IDLCPackage package = ScanForPackageType(fileInfo, identifier, pckFile, localisation, fileReader);
            if (package.GetDLCPackageType() != DLCPackageType.Invalid)
            {
                _packageRegistry.RegisterPackage(identifier, package, localisation);
            }
            return package;
        }

        internal LOCFile GetLocalisation(int identifier)
        {
            return _packageRegistry.ContainsPackage(identifier) ? _packageRegistry.GetLocalisation(identifier) : default;
        }

        private bool IsValidPckFile(FileInfo fileInfo)
        {
            if (fileInfo is null)
                throw new ArgumentNullException(nameof(fileInfo));

            if (!fileInfo.Exists)
                throw new FileNotFoundException(fileInfo.FullName);

            if (fileInfo.Extension != ".pck")
                throw new FileFormatException("File does not end with '.pck'.");
            return true;
        }

        private IDLCPackage ScanForPackageType(FileInfo fileInfo, int identifier, PckFile pckFile, LOCFile localisation, PckFileReader fileReader)
        {
            bool hasLanguage = localisation?.Languages?.Contains(PreferredLanguage) ?? default;

            string name = hasLanguage && (localisation?.HasLocEntry(PACKAGE_DISPLAYNAME_ID) ?? default)
                ? localisation.GetLocEntry(PACKAGE_DISPLAYNAME_ID, PreferredLanguage) : fileInfo.Name;

            string description = hasLanguage && (localisation?.HasLocEntry(DLCTexturePackage.TEXTUREPACK_DESCRIPTION_ID) ?? default)
                ? localisation.GetLocEntry(DLCTexturePackage.TEXTUREPACK_DESCRIPTION_ID, PreferredLanguage) : string.Empty;

            bool couldBeTexturePack = fileInfo.Name == DEFAULT_TEXTURE_PACK_FILENAME;
            bool couldBeMiniGamePack = fileInfo.Name == DEFAULT_MINIGAME_PACK_FILENAME;

            bool hasSkins = pckFile.Contains(PckAssetType.SkinFile) || pckFile.Contains(PckAssetType.SkinDataFile);

            DLCPackageType dlcPackageType = hasSkins ? DLCPackageType.SkinPack : DLCPackageType.Unknown;

            DirectoryInfo dataDirectoryInfo = fileInfo.Directory.EnumerateDirectories().Where(dirInfo => dirInfo.Name == DATA_DIRECTORY_NAME).FirstOrDefault();

            PckAsset texturePackInfo = pckFile.GetAssetsByType(PckAssetType.TexturePackInfoFile).FirstOrDefault();
            string dataPath = texturePackInfo is not null ? texturePackInfo.GetProperty("DATAPATH") : string.Empty;

            if (dataDirectoryInfo is null)
            {
                return skinPackage;
            }

            if (!string.IsNullOrWhiteSpace(dataPath))
            {
                PckFile infoPck = texturePackInfo.GetData(fileReader);
                FileInfo texturePackFileInfo = dataDirectoryInfo.EnumerateFiles().Where(fileInfo => fileInfo.Name == dataPath).FirstOrDefault();
                if (IsValidPckFile(texturePackFileInfo))
                {
                    using Stream texturePackFileStream = texturePackFileInfo.OpenRead();
                    PckFile texturePackPck = fileReader.FromStream(texturePackFileStream);
                    IDLCPackage texturePackage = GetTexturePackageFromPckFile(infoPck, texturePackPck);
                    dlcPackageType = DLCPackageType.TexturePack;
                }
            }

            IEnumerable<FileInfo> audioFiles = dataDirectoryInfo.EnumerateFiles("*.binka");
            IDictionary<string, byte[]> audios = new Dictionary<string, byte[]>();
            foreach (FileInfo audioFile in audioFiles)
            {
                MemoryStream dataStream = new MemoryStream();
                using (Stream audioStream = audioFile.OpenRead())
                {
                    audioStream.CopyTo(dataStream);
                }
                audios.Add(audioFile.Name, dataStream.ToArray());
            }

            IDictionary<string, IDictionary<string, byte[]>> saves = new Dictionary<string, IDictionary<string, byte[]>>();
            foreach (FileInfo worldFile in dataDirectoryInfo.EnumerateFiles("*.mcs"))
            {
                IDictionary<string, byte[]> save = MapReader.OpenSave(worldFile.OpenRead());
                saves.Add(worldFile.Name, save);
            }

            if (pckFile.Contains("GameRules.grf", PckAssetType.GameRulesFile) && pckFile.TryGetAsset("GameRules.grf", PckAssetType.GameRulesFile, out PckAsset gameRuleAsset))
            {
                GameRuleFile gameRuleFile = gameRuleAsset.GetData(new GameRuleFileReader(GetPlatformCompressionType()));
                dlcPackageType = DLCPackageType.MashUpPack;
            }

            Debug.WriteLine(dlcPackageType);
            return new UnknownDLCPackage(name, pckFile);
        }

        private GameRuleFile.CompressionType GetPlatformCompressionType()
        {
            switch (Platform)
            {
                case ConsolePlatform.Xbox360:
                    return GameRuleFile.CompressionType.XMem;

                case ConsolePlatform.PS3:
                    return GameRuleFile.CompressionType.Deflate;

                case ConsolePlatform.XboxOne:
                case ConsolePlatform.PS4:
                case ConsolePlatform.PSVita:
                case ConsolePlatform.WiiU:
                case ConsolePlatform.Switch:
                    return GameRuleFile.CompressionType.Zlib;

                case ConsolePlatform.Unknown:
                default:
                    return GameRuleFile.CompressionType.Unknown;
            }
        }

        private IDLCPackage GetTexturePackageFromPckFile(PckFile infoPck, PckFile dataPck)
        {
            if (!infoPck.TryGetAsset("comparison.png", PckAssetType.TextureFile, out PckAsset comparisonAsset))
            {
                Trace.TraceError($"Could not find 'comparison.png'.");
                return InvalidDLCPackage.Instance;
            }
            if (!infoPck.TryGetAsset("icon.png", PckAssetType.TextureFile, out PckAsset iconnAsset))
            {
                Trace.TraceError($"Could not find 'icon.png'.");
                return InvalidDLCPackage.Instance;
            }

            Image comparisonImg = comparisonAsset.GetTexture();
            Image iconImg = iconnAsset.GetTexture();
            DLCTexturePackage.MetaData metaData = new DLCTexturePackage.MetaData(comparisonImg, iconImg);

            bool hasTerrainAtlas  = TryGetAtlasFromResourceCategory(dataPck, ResourceCategory.BlockAtlas, out Atlas terrainAtlas);
            bool hasItemAtlas     = TryGetAtlasFromResourceCategory(dataPck, ResourceCategory.ItemAtlas, out Atlas itemAtlas);
            bool hasParticleAtlas = TryGetAtlasFromResourceCategory(dataPck, ResourceCategory.ParticleAtlas, out Atlas particleAtlas);
            bool hasPaintingAtlas = TryGetAtlasFromResourceCategory(dataPck, ResourceCategory.PaintingAtlas, out Atlas paintingAtlas);

            string itemAnimationResourcePath = ResourceLocation.GetPathFromCategory(ResourceCategory.ItemAnimation);
            if (dataPck != null &&
                dataPck.TryGetAsset(itemAnimationResourcePath + "/compass.png", PckAssetType.TextureFile, out PckAsset compassAsset) &&
                dataPck.TryGetAsset(itemAnimationResourcePath + "/clock.png", PckAssetType.TextureFile, out PckAsset clockAsset))
            {
                
            }
            return null;
        }

        private bool TryGetAtlasFromResourceCategory(PckFile pck, ResourceCategory resourceCategory, out Atlas atlas)
        {
            ResourceLocation resourceLocation = ResourceLocation.GetFromCategory(resourceCategory);
            if (!pck.TryGetAsset(resourceLocation.ToString(), PckAssetType.TextureFile, out PckAsset asset))
            {
                Trace.TraceWarning($"Could not find '{resourceLocation}'.");
                atlas = null;
                return false;
            }
            atlas = asset.GetDeserializedData(new AtlasDeserializer(resourceLocation));
            return true;
        }

        private IDLCPackage GetDLCSkinPackage(string name, int identifier, PckFile pck, PckFileReader fileReader, IDLCPackage parentPackage = null)
        {
            Skin.Skin GetSkinWithCape(PckAsset skinAsset)
            {
                Skin.Skin skin = skinAsset.GetSkin();
                if (skinAsset.TryGetProperty("CAPEPATH", out string capeAssetPath) && pck.TryGetAsset(capeAssetPath, PckAssetType.CapeFile, out PckAsset capeAsset))
                    skin.CapeTexture = capeAsset.GetTexture();
                return skin;
            }

            IEnumerable<Skin.Skin> skins = pck.GetAssetsByType(PckAssetType.SkinFile).Select(GetSkinWithCape);

            skins = skins.Concat(pck.GetAssetsByType(PckAssetType.SkinDataFile)
                .Select(asset => asset.GetData(fileReader))
                .SelectMany(pck => pck.GetAssetsByType(PckAssetType.SkinFile))
                .Select(GetSkinWithCape)
                );

            return new DLCSkinPackage(name, identifier, skins, null, parentPackage);
        }

        private static string GetPreferredLanguage(AppLanguage appLanguage)
        {
            return appLanguage switch
            {
                AppLanguage.SystemDefault => LOCFile.ValidLanguages.Contains(CultureInfo.CurrentUICulture.Name) ? CultureInfo.CurrentUICulture.Name : AvailableLanguages.English,
                AppLanguage.CzechCzechia => AvailableLanguages.CzechCzechia,
                AppLanguage.Czechia => AvailableLanguages.Czechia,
                AppLanguage.Danish => AvailableLanguages.Danish,
                AppLanguage.DenmarkDanish => AvailableLanguages.DenmarkDanish,
                AppLanguage.GermanAustria => AvailableLanguages.GermanAustria,
                AppLanguage.German => AvailableLanguages.German,
                AppLanguage.GreekGreece => AvailableLanguages.GreekGreece,
                AppLanguage.Greece => AvailableLanguages.Greece,
                AppLanguage.EnglishAustralia => AvailableLanguages.EnglishAustralia,
                AppLanguage.EnglishCanada => AvailableLanguages.EnglishCanada,
                AppLanguage.English => AvailableLanguages.English,
                AppLanguage.EnglishUnitedKingdom => AvailableLanguages.EnglishUnitedKingdom,
                AppLanguage.EnglishIreland => AvailableLanguages.EnglishIreland,
                AppLanguage.EnglishNewZealand => AvailableLanguages.EnglishNewZealand,
                AppLanguage.EnglishUnitedStatesOfAmerica => AvailableLanguages.EnglishUnitedStatesOfAmerica,
                AppLanguage.SpanishSpain => AvailableLanguages.SpanishSpain,
                AppLanguage.SpanishMexico => AvailableLanguages.SpanishMexico,
                AppLanguage.FinnishFinland => AvailableLanguages.FinnishFinland,
                AppLanguage.FrenchFrance => AvailableLanguages.FrenchFrance,
                AppLanguage.FrenchCanada => AvailableLanguages.FrenchCanada,
                AppLanguage.ItalianItaly => AvailableLanguages.ItalianItaly,
                AppLanguage.JapaneseJapan => AvailableLanguages.JapaneseJapan,
                AppLanguage.KoreanSouthKorea => AvailableLanguages.KoreanSouthKorea,
                AppLanguage.Latin => AvailableLanguages.Latin,
                AppLanguage.NorwegianNorway => AvailableLanguages.NorwegianNorway,
                AppLanguage.NorwegianBokmålNorway => AvailableLanguages.NorwegianBokmålNorway,
                AppLanguage.DutchNetherlands => AvailableLanguages.DutchNetherlands,
                AppLanguage.DutchBelgium => AvailableLanguages.DutchBelgium,
                AppLanguage.PolishPoland => AvailableLanguages.PolishPoland,
                AppLanguage.PortugueseBrazil => AvailableLanguages.PortugueseBrazil,
                AppLanguage.PortuguesePortugal => AvailableLanguages.PortuguesePortugal,
                AppLanguage.RussianRussia => AvailableLanguages.RussianRussia,
                AppLanguage.SlovakSlovakia => AvailableLanguages.SlovakSlovakia,
                AppLanguage.SwedishSweden => AvailableLanguages.SwedishSweden,
                AppLanguage.TurkishTurkey => AvailableLanguages.TurkishTurkey,
                AppLanguage.ChineseChina => AvailableLanguages.ChineseChina,
                AppLanguage.ChineseHongKong => AvailableLanguages.ChineseHongKong,
                AppLanguage.ChineseSingapore => AvailableLanguages.ChineseSingapore,
                AppLanguage.ChineseTaiwan => AvailableLanguages.ChineseTaiwan,
                _ => AvailableLanguages.English,
            };
        }

        public void SetPreferredLanguage(AppLanguage lang) => PreferredLanguage = GetPreferredLanguage(lang);

        public void SetPlatform(ConsolePlatform platform)
        {
            _platform = platform;
            _byteOrder = GetByteOrderForPlatform(platform);
        }
    }
}