using System.Collections.Generic;
using System.IO;
using System.Linq;
using UnityEngine;
using System;
using System.Collections;
using UnityEngine.Rendering;
using Newtonsoft.Json.Linq;

public class UmaViewerMain : MonoBehaviour
{
    private const string EnglishNamesCacheFileName = "english_names.cache";
    private const string EnglishNamesCacheMagic = "UmaViewerEnglishNames";
    private const int EnglishNamesCacheVersion = 1;

    public static UmaViewerMain Instance;
    public static bool WasEscapeConsumedThisFrame { get; private set; }
    private UmaViewerUI UI => UmaViewerUI.Instance;
    private UmaViewerBuilder Builder => UmaViewerBuilder.Instance;

    public RenderPipelineAsset DefaultRenderPipeline;
    public List<CharaEntry> Characters = new List<CharaEntry>();
    public List<CharaEntry> MobCharacters = new List<CharaEntry>();
    public List<LiveEntry> Lives = new List<LiveEntry>();
    public List<CostumeEntry> Costumes = new List<CostumeEntry>();
    public Dictionary<string,UmaDatabaseEntry> AbList = new Dictionary<string, UmaDatabaseEntry>();
    public List<UmaDatabaseEntry> AbMotions = new List<UmaDatabaseEntry>();
    public List<UmaDatabaseEntry> AbSounds = new List<UmaDatabaseEntry>();
    public List<UmaDatabaseEntry> AbChara = new List<UmaDatabaseEntry>();
    public List<UmaDatabaseEntry> AbEffect = new List<UmaDatabaseEntry>();
    public List<UmaDatabaseEntry> CostumeList = new List<UmaDatabaseEntry>();

    private void Awake()
    {
        Instance = this;
        new Config();
        ApplyFrameRateLimit();

        AbList = UmaDatabaseController.Instance.MetaEntries;
        if (AbList == null) return;
        var chara_3d = AbList.Where(ab => ab.Value.Type == UmaFileType._3d_cutt).Select(ab => ab.Value).ToList();
        AbChara = chara_3d.FindAll(ab => ab.Name.StartsWith(UmaDatabaseController.CharaPath));
        AbMotions = chara_3d.FindAll(ab => ab.Name.StartsWith(UmaDatabaseController.MotionPath));
        AbEffect = chara_3d.FindAll(ab => ab.Name.StartsWith(UmaDatabaseController.EffectPath));
        AbSounds = AbList.Where(ab => ab.Value.Type == UmaFileType.sound).Select(ab => ab.Value).ToList();
        var outgame = AbList.Where(ab => ab.Value.Type == UmaFileType.outgame).Select(ab => ab.Value).ToList();
        CostumeList = outgame.FindAll(e => e.Name.StartsWith(UmaDatabaseController.CostumePath));
    }

    public static void ApplyFrameRateLimit()
    {
        QualitySettings.vSyncCount = 0;
        Application.targetFrameRate = Config.Instance.GetTargetFrameRate();
    }

    private void Update()
    {
        WasEscapeConsumedThisFrame = false;
        TryConsumeEscapeForFullScreen();
    }

    public static bool TryConsumeEscapeForFullScreen()
    {
        if (!Input.GetKeyDown(KeyCode.Escape) || !Screen.fullScreen)
            return false;

        Screen.fullScreenMode = FullScreenMode.Windowed;
        Screen.fullScreen = false;
        WasEscapeConsumedThisFrame = true;
        return true;
    }

    private IEnumerator Start()
    {
        if (AbList == null) yield break;
        int loadingStep = 0;
        int loadingStepsTotal = 10;
        var UmaCharaData = UmaDatabaseController.Instance.CharaData;
        var MobCharaData = UmaDatabaseController.Instance.MobCharaData;
        var loadingUI = UmaSceneController.instance;


        //修改(载入通用服装ColorSet相关)
        var charaDressColor = UmaDatabaseController.Instance.CharaDressColor;



        // Reuse downloaded English names; only fetch when no usable cache exists.
        bool downloadTranslations = false;
        string cachePath = Path.Combine(Application.persistentDataPath, EnglishNamesCacheFileName);
        if (Config.Instance.Language == Language.En)
            downloadTranslations = !TryLoadEnglishNamesCache(cachePath);

        loadingUI.LoadingProgressChange(
            loadingStep++,
            loadingStepsTotal,
            downloadTranslations ? "Downloading Translations" : "Loading Translations");
        if (downloadTranslations)
        {

            yield return UmaViewerDownload.DownloadText(LocalizeEn.TranslationUrl, json =>
            {
                if (string.IsNullOrEmpty(json)) return;

                try
                {
                    if (!TryParseEnglishNames(json, out Dictionary<int, string> charaNames, out Dictionary<int, string> mobNames))
                    {
                        Debug.LogWarning("Online English translations did not contain the required name tables. Using LocalizeEn fallback.");
                        return;
                    }

                    ApplyEnglishNames(charaNames, mobNames);
                    TrySaveEnglishNamesCache(cachePath, charaNames, mobNames);
                    Debug.Log($"Loaded online English names: {charaNames.Count} characters, {mobNames.Count} mobs.");
                }
                catch (Exception ex)
                {
                    Debug.LogWarning($"Could not parse online English translations. Using LocalizeEn fallback: {ex.Message}");
                }
            });
        }

        if (Config.Instance.WorkMode == WorkMode.Standalone)
        {
            var umaIcons = UmaCharaData.Select(item => AbList.TryGetValue($"chara/chr{item["id"]}/chr_icon_{item["id"]}", out UmaDatabaseEntry entry) ? entry : null).Where(entry => entry != null);
            var mobIcons = MobCharaData.Select(item => AbList.TryGetValue($"mob/mob_chr_icon_{item["mob_id"]}_000001_01", out UmaDatabaseEntry entry) ? entry : null).Where(entry => entry != null);
            var costumeIcons = CostumeList;
            List<UmaDatabaseEntry> filesToDownload = umaIcons.Concat(mobIcons).Concat(costumeIcons).ToList();
            yield return UmaViewerDownload.DownloadAssets(filesToDownload, UmaSceneController.instance.LoadingProgressChange);
            filesToDownload.Clear();
        }

        loadingUI.LoadingProgressChange(loadingStep++, loadingStepsTotal, "Loading Character Data");
        yield return null;
        foreach (var item in UmaCharaData)
        {
            var id = Convert.ToInt32(item["id"]);
            if (!Characters.Where(c => c.Id == id).Any())
            {
                Characters.Add(new CharaEntry()
                {
                    Name = item["charaname"].ToString(),
                    EnName = LocalizeEn.GetCharaName(id),
                    Icon = UmaViewerBuilder.Instance.LoadCharaIcon(id.ToString()),
                    Id = id,
                    ThemeColor = "#" + item["ui_nameplate_color_1"].ToString()
                });
            }
        }

        loadingUI.LoadingProgressChange(loadingStep++, loadingStepsTotal, "Loading Mob Data");
        yield return null;
        foreach (var item in MobCharaData)
        {
            if (Convert.ToInt32(item["use_live"]) == 0) 
            {
                continue;
            }

            var id = Convert.ToInt32(item["mob_id"]);
            MobCharacters.Add(new CharaEntry()
            {
                Name = item["charaname"].ToString(),
                EnName = LocalizeEn.GetMobName(id),
                Icon = UmaViewerBuilder.Instance.LoadMobCharaIcon(id.ToString()),
                Id = id,
                IsMob = true
            });
        }

        loadingUI.LoadingProgressChange(loadingStep++, loadingStepsTotal, "Loading Costumes Data");
        yield return null;
        foreach (var item in CostumeList)
        {
            var costume = new CostumeEntry();
            var name = Path.GetFileName(item.Name);
            costume.Id = name.Replace("dress_","");
            costume.Icon = Builder.LoadSprite(item);
            Costumes.Add(costume);
        }

        var DressData = UmaDatabaseController.Instance.DressData;
        foreach (var data in DressData)
        {
            var costume = Costumes.FirstOrDefault(a => a.Id.Split('_')[0].Contains(data["id"].ToString()) );
            if (costume != null)
            {
                costume.CharaId = Convert.ToInt32(data["chara_id"]);
                costume.DressName = data["dressname"].ToString();
                costume.BodyType = Convert.ToInt32(data["body_type"]);
                costume.BodyTypeSub = Convert.ToInt32(data["body_type_sub"]);
            }
        }

        loadingUI.LoadingProgressChange(loadingStep++, loadingStepsTotal, "Loading Live Data");
        yield return null;
        var asset = AbList["livesettings"];
        if (asset != null)
        {
            string filePath = asset.FilePath;
            if (File.Exists(filePath))
            {
                AssetBundle bundle = UmaAssetManager.LoadAssetBundle(asset, true);
                foreach (var item in UmaDatabaseController.Instance.LiveData)
                {
                    var musicId = Convert.ToInt32(item["music_id"]);
                    var songName = item["songname"].ToString();
                    var membercount = Convert.ToInt32(item["live_member_number"]);
                    var defaultdress = Convert.ToInt32(item["default_main_dress"]);

                    if (!Lives.Where(c => c.MusicId == musicId).Any())
                    {
                        if (bundle.Contains(musicId.ToString()))
                        {
                            TextAsset liveData = bundle.LoadAsset<TextAsset>(musicId.ToString());

                            Lives.Add(new LiveEntry(liveData.text)
                            {
                                MusicId = musicId,
                                SongName = songName,
                                MemberCount = membercount,
                                DefaultDress = defaultdress,
                                Icon = UmaViewerBuilder.Instance.LoadLiveIcon(musicId)
                            }); 
                        }
                    }
                }
            }
        }

        loadingUI.LoadingProgressChange(loadingStep++, loadingStepsTotal, "Loading UI");
        yield return null;
        UI.LoadModelPanels();
        loadingUI.LoadingProgressChange(loadingStep++, loadingStepsTotal, "Loading UI");
        yield return null;
        UI.LoadMiniModelPanels();
        loadingUI.LoadingProgressChange(loadingStep++, loadingStepsTotal, "Loading UI");
        yield return null;
        UI.LoadPropPanel();
        loadingUI.LoadingProgressChange(loadingStep++, loadingStepsTotal, "Loading UI");
        yield return null;
        UI.LoadMapPanel();
        loadingUI.LoadingProgressChange(loadingStep++, loadingStepsTotal, "Loading UI");
        yield return null;
        UI.LoadLivePanels();
        loadingUI.LoadingProgressChange(-1, -1);

        //Load Shader First
        var shaders = UmaAssetManager.LoadAssetBundle(AbList["shader"], true);
        Builder.ShaderList = new List<Shader>(shaders.LoadAllAssets<Shader>()); 
        Gallop.ShaderManager.InitManager();
        Gallop.ShaderManager.WarmupDofBloomShader();
    }

    private static bool TryParseEnglishNames(
        string json,
        out Dictionary<int, string> charaNames,
        out Dictionary<int, string> mobNames)
    {
        JObject translations = JObject.Parse(json);
        charaNames = translations[LocalizeEn.CharaTranslationTableKey]
            ?.ToObject<Dictionary<int, string>>();
        mobNames = translations[LocalizeEn.MobTranslationTableKey]
            ?.ToObject<Dictionary<int, string>>();

        return charaNames != null && charaNames.Count > 0 &&
               mobNames != null && mobNames.Count > 0;
    }

    private static void ApplyEnglishNames(
        Dictionary<int, string> charaNames,
        Dictionary<int, string> mobNames)
    {
        foreach (KeyValuePair<int, string> pair in charaNames)
        {
            if (!string.IsNullOrEmpty(pair.Value))
                LocalizeEn.SetCharaName(pair.Key, pair.Value);
        }

        foreach (KeyValuePair<int, string> pair in mobNames)
        {
            if (!string.IsNullOrEmpty(pair.Value))
                LocalizeEn.SetMobName(pair.Key, pair.Value);
        }
    }

    private static bool TryLoadEnglishNamesCache(string cachePath)
    {
        if (!File.Exists(cachePath)) return false;

        try
        {
            using (var stream = File.OpenRead(cachePath))
            using (var reader = new BinaryReader(stream))
            {
                if (reader.ReadString() != EnglishNamesCacheMagic ||
                    reader.ReadInt32() != EnglishNamesCacheVersion)
                {
                    throw new InvalidDataException("Unsupported English name cache format.");
                }

                Dictionary<int, string> charaNames = ReadEnglishNameTable(reader);
                Dictionary<int, string> mobNames = ReadEnglishNameTable(reader);
                if (charaNames.Count == 0 || mobNames.Count == 0)
                    throw new InvalidDataException("English name cache is missing required name tables.");
                ApplyEnglishNames(charaNames, mobNames);
                Debug.Log($"Loaded cached English names: {charaNames.Count} characters, {mobNames.Count} mobs.");
            }
            return true;
        }
        catch (Exception ex)
        {
            Debug.LogWarning($"Could not load the cached English names. Using LocalizeEn fallback: {ex.Message}");
            return false;
        }
    }

    private static void TrySaveEnglishNamesCache(
        string cachePath,
        Dictionary<int, string> charaNames,
        Dictionary<int, string> mobNames)
    {
        try
        {
            using (var stream = File.Create(cachePath))
            using (var writer = new BinaryWriter(stream))
            {
                writer.Write(EnglishNamesCacheMagic);
                writer.Write(EnglishNamesCacheVersion);
                WriteEnglishNameTable(writer, charaNames);
                WriteEnglishNameTable(writer, mobNames);
            }
        }
        catch (Exception ex)
        {
            Debug.LogWarning($"Could not save the updated English name cache: {ex.Message}");
        }
    }

    private static Dictionary<int, string> ReadEnglishNameTable(BinaryReader reader)
    {
        int count = reader.ReadInt32();
        if (count < 0 || count > 100000)
            throw new InvalidDataException("Invalid English name cache entry count.");

        var names = new Dictionary<int, string>(count);
        for (int i = 0; i < count; i++)
        {
            int id = reader.ReadInt32();
            string value = reader.ReadString();
            if (!string.IsNullOrEmpty(value))
                names[id] = value;
        }

        return names;
    }

    private static void WriteEnglishNameTable(
        BinaryWriter writer,
        Dictionary<int, string> names)
    {
        List<KeyValuePair<int, string>> validNames = names
            .Where(pair => !string.IsNullOrEmpty(pair.Value))
            .ToList();
        writer.Write(validNames.Count);
        foreach (KeyValuePair<int, string> pair in validNames)
        {
            writer.Write(pair.Key);
            writer.Write(pair.Value);
        }
    }

    public void OpenUrl(string url)
    {
        Application.OpenURL(url);
    }

}