using UnityEngine;
using UnityEngine.UI;
using UnityEngine.Events;

using System;
using System.Collections;
using System.Linq.Expressions;
using System.ComponentModel;
using System.Collections.Generic;
using System.Diagnostics;
using Object = UnityEngine.Object;
using Random = UnityEngine.Random;
using Component = UnityEngine.Component;
using Debug = UnityEngine.Debug;

public enum Folder
{
    UI,
    Effect,
    Audio,
    Scene,
    Config,
}

public enum ObjType
{
    Null,

    Firework,
    LightwallUI,

    PlayerPink,
    PlayerBlond,
    PlayerBrown,

    Page,
    Flower,
    Garden,
    Tutorial,

    Canvas,
    GroupA,
    GroupB,
    GroupC,
    GroupD,
    EventSystem,
    MainCamera,

    HudText,
    SkillItem,
    SignItem,
    FlowerItem,
    AchieveItem,

    Music,

    DropGold,
    DropDiamond,

    BeeRed,
    BeeBlue,
    BeeWhite,
    BeePurple,
    BeeYellow,

    ButterflyRed,
    ButterflyBlue,
    ButterflyWhite,
    ButterflyPurple,
    ButterflyYellow,
}

public class AsyncRequest
{
    public UnityAction Callback;
    public AssetBundleRequest Request;
}

public class ManaReso : Regist
{
    #region 变量

    public static bool AsyncLoadLock;
    public static bool AsyncInstantiateLock;
    public static Coroutine CoroAsync;
    public static Coroutine CoroInstantiate;
    public static ManaReso Instance;
    public static UnityAction OnAsyncLoadFinish;

    public static List<UnityAction> AsyncList = new List<UnityAction>();
    public static List<AssetBundleRequest> RequestList = new List<AssetBundleRequest>();

    public static List<KV<AsyncRequest, UnityAction>> InstantiateList = new List<KV<AsyncRequest, UnityAction>>();

    public static Dictionary<string, Object> ObjDic = new Dictionary<string, Object>();
    public static Dictionary<string, Transform> TraDic = new Dictionary<string, Transform>();

    public static Dictionary<ObjType, List<Transform>> ObjectPool = new Dictionary<ObjType, List<Transform>>();

    #endregion

    public override void RegistImmed()
    {
        if (RegistFlag)
        {
            return;
        }
        else
        {
            RegistFlag = true;
        }

        Instance = this;

        Transform objPool = new GameObject("ObjPool").transform;

        objPool.parent = transform;
        objPool.SetActive(false);

        TraDic.Add(objPool.name, objPool);

        CoroAsync = StartCoroutine(IAsyncLoad());
        CoroInstantiate = StartCoroutine(IAsyncInstancitate());
    }

    #region TraDic

    public static T Get<T>(string goName)
    {
        Transform tra;

        if (TraDic.TryGetValue(goName, out tra))
        {
            T t = tra.GetComponent<T>();

            if (t == null)
            {
                throw new Exception(goName);
            }

            return t;
        }
        else
        {
            throw new Exception(goName);
        }
    }

    public static T[] Gets<T>(string goName)
    {
        Transform tra;

        if (TraDic.TryGetValue(goName, out tra))
        {
            T[] ts = tra.GetComponentsInChildren<T>();

            if (ts == null)
            {
                throw new Exception(goName);
            }

            return ts;
        }
        else
        {
            throw new Exception(goName);
        }
    }

    public static Transform Get(string goName)
    {
        Transform tra;

        if (TraDic.TryGetValue(goName, out tra))
        {
            return tra;
        }
        else
        {
            throw new Exception(goName);
        }
    }

    #endregion

    #region ObjPool

    public static void Save<T>(T t, bool aware = false) where T : Component
    {
        Transform tra = t.transform;

        ObjRoot objRoot = tra.GetComponent<ObjRoot>();

        if (objRoot == null)
        {
            throw new Exception();
        }

        List<Transform> traList;

        if (ObjectPool.TryGetValue(objRoot.ObjType, out traList))
        {
            if (traList.Contains(tra))
            {
                if (aware)
                {
                    return;
                }
                else
                {
                    throw new Exception();
                }
            }

            tra.SetActive(false);

            traList.Add(tra);

            tra.transform.SetParent(Get("ObjPool"));
        }
        else
        {
            throw new Exception();
        }
    }

    public static void Save(GameObject go, bool aware = false)
    {
        ObjRoot objRoot = go.GetComponent<ObjRoot>();

        if (objRoot == null)
        {
            throw new Exception();
        }

        List<Transform> traList;

        if (ObjectPool.TryGetValue(objRoot.ObjType, out traList))
        {
            if (traList.Contains(go.transform))
            {
                if (aware)
                {
                    return;
                }
                else
                {
                    throw new Exception();
                }
            }

            go.SetActive(false);

            traList.Add(go.transform);

            go.transform.SetParent(Get("ObjPool"));
        }
        else
        {
            throw new Exception();
        }
    }


    public static Transform Get(ObjType objType)
    {
        List<Transform> traList;
        
        if (ObjectPool.TryGetValue(objType, out traList))
        {
            if (traList.Count > 0)
            {
                Transform tra = traList[0];

                tra.SetActive(true);

                traList.RemoveAt(0);

                return tra.transform;
            }
        }

        return null;
    }

    #endregion

    #region ShortCut

    public static void SetText(string goName)
    {
        Text text = Get<Text>(goName);

        text.text = Language.GetStr("UI", goName);
    }

    public static void SetText(string goName, string str)
    {
        Text text = Get<Text>(goName);

        text.text = str;
    }

    public static void SetSprite(string goName, Sprite sprite)
    {
        Image image = Get<Image>(goName);

        image.sprite = sprite;
    }

    public static void SetActive(string goName, bool active)
    {
        Get(goName).SetActive(active);
    }

    public static void SetButtonEvent(string goName, UnityAction onClick)
    {
        Button button = Get<Button>(goName);

        button.onClick = new Button.ButtonClickedEvent();

        button.onClick.AddListener(onClick);
    }

    public static void AddButtonEvent(string goName, UnityAction onClick)
    {
        Button button = Get<Button>(goName);

        button.onClick.AddListener(onClick);
    }

    public static void PushButtonEvent(string goName, UnityAction onClick)
    {
        Button button = Get<Button>(goName);

        Button.ButtonClickedEvent click = button.onClick;

        button.onClick = new Button.ButtonClickedEvent();

        button.onClick.AddListener(onClick);
        button.onClick.AddListener(click.Invoke);
    }

    public static void AddButtonEventOnetime(string goName, UnityAction onClick)
    {
        Button button = Get<Button>(goName);

        onClick += () =>
        {
            button.onClick.RemoveListener(onClick);
        };

        button.onClick.AddListener(onClick);
    }

    public static void PushButtonEventOnetime(string goName, UnityAction onClick)
    {
        Button button = Get<Button>(goName);

        onClick += () =>
        {
            button.onClick.RemoveListener(onClick);
        };

        Button.ButtonClickedEvent click = button.onClick;

        button.onClick = new Button.ButtonClickedEvent();

        button.onClick.AddListener(onClick);
        button.onClick.AddListener(click.Invoke);
    }

    #endregion


    public static T Load<T>(string goName, Folder folder, ObjType objType = ObjType.Null) where T : Object
    {
        Object obj;
        
        if (ObjDic.TryGetValue(goName, out obj))
        {
            return (T) obj;
        }
        else
        {
            T t;

            if (folder == Folder.UI)
            {
                t = Bundle.UI.LoadAsset<T>(goName);
            }
            else if (folder == Folder.Audio)
            {
                t = Bundle.Audio.LoadAsset<T>(goName);
            }
            else if (folder == Folder.Config)
            {
                t = Bundle.Config.LoadAsset<T>(goName);
            }
            else if (folder == Folder.Effect)
            {
                t = Bundle.Effect.LoadAsset<T>(goName);
            }
            else if (folder == Folder.Scene)
            {
                t = Bundle.Scene.LoadAsset<T>(goName);
            }
            else
            {
                throw new Exception();
            }

            if (t == null)
            {
                throw new Exception(goName);
            }

            ObjDic.Add(goName, t);

            if (objType != ObjType.Null)
            {
                ObjectPool.UniqueAdd(objType, new List<Transform>());
            }

            return t;
        }
    }

    public static Sprite LoadSprite(string goName)
    {
        return Load<Sprite>(goName, Folder.UI);
    }

    public static Transform Get<T>(string goName, Folder folder, bool compile, Transform par, bool worldSpace, ObjType objType = ObjType.Null) where T : ObjRoot
    {
        Transform tra = Get(objType);

        if (tra == null)
        {
            GameObject go = Load<GameObject>(goName, folder, objType);

            go = Instantiate(go, par, worldSpace);
            go.name = go.name.Replace("(Clone)", "");

            if (compile)
            {
                Auxiliary.CompileDic(go.transform, TraDic);
            }

            if (objType != ObjType.Null)
            {
                go.AddComponent<T>().ObjType = objType;
            }

            return go.transform;
        }
        else
        {
            if (compile)
            {
                Auxiliary.CompileDic(tra, TraDic);
            }

            GameObject prefab = Load<GameObject>(goName, folder, objType);

            tra.SetParent(par);

            if (worldSpace)
            {
                tra.position = prefab.transform.position;
            }
            else
            {
                tra.localPosition = prefab.transform.position;
            }

            return tra;
        }
    }

    public static Transform Get<T>(string goName, Folder folder, bool compile, Transform par, Vector3 pos, ObjType objType = ObjType.Null) where T : ObjRoot
    {
        Transform tra = Get(objType);

        if (tra == null)
        {
            GameObject go = Load<GameObject>(goName, folder, objType);
            
            go = Instantiate(go, pos, Quaternion.identity, par);
            go.name = go.name.Replace("(Clone)", "");

            if (compile)
            {
                Auxiliary.CompileDic(go.transform, TraDic);
            }

            if (objType != ObjType.Null)
            {
                go.AddComponent<T>().ObjType = objType;
            }

            return go.transform;
        }
        else
        {
            if (compile)
            {
                Auxiliary.CompileDic(tra, TraDic);
            }

            tra.SetParent(par);

            tra.position = pos;

            return tra;
        }
    }


    public static Drop GetDrop(ObjType objType)
    {
        ManaAudio.PlayClip(Clip.DropClip);

        Vector3 pos = Vector3.Lerp(Get("MiniLeft").position, Get("MiniRight").position, Random.Range(0, 1f));

        Transform tra;

        if (objType == ObjType.DropGold)
        {
            tra = Get<DropGold>(objType.ToString(), Folder.UI, false, null, pos, objType);
        }
        else if(objType == ObjType.DropDiamond)
        {
            tra = Get<DropDiamond>(objType.ToString(), Folder.UI, false, null, pos, objType);
        }
        else
        {
            throw new Exception();
        }

        Drop drop = tra.GetComponent<Drop>();

        drop.RegistImmed();
        drop.ResetStatus();

        return drop;
    }


    public static Flower GetFlower(FlowerInfo flowerInfo, Slot slot, bool collider)
    {
        Transform tra = Get<Flower>("Flower", Folder.Scene, false, slot.transform, false, ObjType.Flower);

        tra.localScale = new Vector3(1, 1, 1);

        Flower flower = tra.GetComponent<Flower>();

        flower.RegistImmed();

        flower.GoldBk.SetActive(false);

        flower.FlowerInfo = flowerInfo;
        flower.Slot = slot;

        flower.SetCollider(collider);

        return flower;
    }

    public static Flower GetFlower(FlowerInfo flowerInfo, Transform par)
    {
        Transform tra = Get<Flower>("Flower", Folder.Scene, false, par, false, ObjType.Flower);

        tra.localScale = new Vector3(1, 1, 1);

        Flower flower = tra.GetComponent<Flower>();

        flower.RegistImmed();

        flower.GoldBk.SetActive(false);

        flower.FlowerInfo = flowerInfo;

        flower.SetCollider(false);

        return flower;
    }


    public static HudText GetHudText(string str, Color color, int size, Transform posTra, Transform parTra, bool scene)
    {
        Vector3 pos;

        if (scene)
        {
            pos = Camera.main.WorldToScreenPoint(posTra.position);
        }
        else
        {
            pos = posTra.position;
        }

        Transform tra = Get<HudText>("HudText", Folder.UI, false, parTra, pos, ObjType.HudText);

        HudText hudText = tra.GetComponent<HudText>();

        hudText.Show(str, color, size);

        return hudText;
    }


    public static Transform GetElf(Flower flower, Vector4 offset, ObjType obj)
    {
        Transform tra;

        Bounds bounds = flower.FlowerIcon.bounds;
        Vector3 pos = flower.FlowerIcon.transform.position;
        Transform par = flower.transform;


        pos.z -= Random.Range(0.001f, 0.1f);

        tra = Get<ObjRoot>(obj.ToString(), Folder.Scene, false, par, pos, obj);

        Elf elf = tra.GetChild(0).GetComponent<Elf>();

        if (elf == null)
        {
            elf = tra.GetChild(0).AddScript<Elf>();
        }


        if (Random.Range(0f, 1f) <= 0.5f)
        {
            tra.SetEY(180);
        }
        else
        {
            tra.SetEY(0);
        }

        float offsetX = Mathf.Lerp(offset.x*bounds.extents.x, offset.y*bounds.extents.x, Random.Range(0f, 1f));
        float offsetY = Mathf.Lerp(offset.z, offset.w*bounds.extents.y, Random.Range(0f, 1f));

        tra.position += new Vector3(offsetX, offsetY, 0);


        elf.Flower = flower;
        elf.Animator.SetTrigger("Play");

        return tra;
    }

    public static Transform GetSkillItem(SkillRoot skillRoot)
    {
        Transform tra;

        if (skillRoot.SkillTab == SkillTab.Elf)
        {
            tra = Get<ObjRoot>("SkillItem", Folder.UI, false, Get("Fd_Grid"), false, ObjType.SkillItem);
        }
        else if (skillRoot.SkillTab == SkillTab.Store)
        {
            tra = Get<ObjRoot>("SkillItem", Folder.UI, false, Get("Fc_Grid"), false, ObjType.SkillItem);
        }
        else if (skillRoot.SkillTab == SkillTab.Magic)
        {
            tra = Get<ObjRoot>("SkillItem", Folder.UI, false, Get("Fb_Grid"), false, ObjType.SkillItem);
        }
        else if (skillRoot.SkillTab == SkillTab.Garden)
        {
            tra = Get<ObjRoot>("SkillItem", Folder.UI, false, Get("Fa_Grid"), false, ObjType.SkillItem);
        }
        else
        {
            throw new Exception();
        }

        skillRoot.SkillItem = tra;

        return tra;
    }

    public static Transform GetAchieveItem()
    {
        Transform tra = Get<ObjRoot>("AchieveItem", Folder.UI, false, ManaReso.Get("M_Grid"), false, ObjType.AchieveItem);

        return tra;
    }


    public static ParticleSystem GetFirework(Vector3 pos)
    {
        Transform tra = Get<Effect>("Firework", Folder.Effect, false, null, pos, ObjType.Firework);

        ParticleSystem particle = tra.GetComponent<ParticleSystem>();

        particle.Play();

        return particle;
    }

    public static ParticleSystem GetLightwall()
    {
        Transform tra = Get<Effect>("LightwallUI", Folder.Effect, false, Get("Canvas"), false, ObjType.LightwallUI);

        ParticleSystem particle = tra.GetComponent<ParticleSystem>();

        particle.Play();

        return particle;
    }


    public static void AsyncLoad(string goName, Folder folder, UnityAction callback = null)
    {
        KV<AsyncRequest, UnityAction> kv = new KV<AsyncRequest, UnityAction>();

        AssetBundleRequest bundleRequest;

        if (folder == Folder.UI)
        {
            bundleRequest = Bundle.UI.LoadAssetAsync(goName);
        }
        else if (folder == Folder.Audio)
        {
            bundleRequest = Bundle.Audio.LoadAssetAsync(goName);
        }
        else if (folder == Folder.Config)
        {
            bundleRequest = Bundle.Config.LoadAssetAsync(goName);
        }
        else if (folder == Folder.Effect)
        {
            bundleRequest = Bundle.Effect.LoadAssetAsync(goName);
        }
        else if (folder == Folder.Scene)
        {
            bundleRequest = Bundle.Scene.LoadAssetAsync(goName);
        }
        else
        {
            throw new Exception();
        }
        
        AsyncRequest asyncRequest = new AsyncRequest();
        asyncRequest.Request = bundleRequest;
        asyncRequest.Callback = callback;

        kv.Key = asyncRequest;
        kv.Value = () =>
        {
            ObjDic.UniqueAdd(goName, bundleRequest.asset);
        };

        RequestList.Add(bundleRequest);

        InstantiateList.Add(kv);
    }

    public static void AsyncLoad<T>(string goName, int amt, Folder folder, ObjType objType, bool ui = false, bool canvas = false, UnityAction callback = null) where T : ObjRoot
    {
        KV<AsyncRequest, UnityAction> kv = new KV<AsyncRequest, UnityAction>();

        AssetBundleRequest bundleRequest;

        if (folder == Folder.UI)
        {
            bundleRequest = Bundle.UI.LoadAssetAsync(goName);
        }
        else if (folder == Folder.Audio)
        {
            bundleRequest = Bundle.Audio.LoadAssetAsync(goName);
        }
        else if (folder == Folder.Config)
        {
            bundleRequest = Bundle.Config.LoadAssetAsync(goName);
        }
        else if (folder == Folder.Effect)
        {
            bundleRequest = Bundle.Effect.LoadAssetAsync(goName);
        }
        else if (folder == Folder.Scene)
        {
            bundleRequest = Bundle.Scene.LoadAssetAsync(goName);
        }
        else
        {
            throw new Exception();
        }

        AsyncRequest asyncRequest = new AsyncRequest();
        asyncRequest.Request = bundleRequest;
        asyncRequest.Callback = callback;

        kv.Key = asyncRequest;
        kv.Value = () =>
        {
            ObjDic.UniqueAdd(goName, bundleRequest.asset);

            if (objType != ObjType.Null)
            {
                ObjectPool.UniqueAdd(objType, new List<Transform>());
            }

            GameObject go;

            if (ui)
            {
                go = (GameObject)Instantiate(bundleRequest.asset, GameObject.Find("ManagerGame").transform.GetChild(0).GetChild(0));
            }
            else
            {
                go = (GameObject)Instantiate(bundleRequest.asset);
            }

            go.AddComponent<T>().ObjType = objType;

            go.name = go.name.Replace("(Clone)", "");

            Save(go);

            if (ui)
            {
                go.SetParent(GameObject.Find("ManagerGame").transform.GetChild(0).GetChild(0));
            }
        };

        RequestList.Add(bundleRequest);

        for (int i = 0; i < amt; i++)
        {
            InstantiateList.Add(kv);
        }
    }

    public static void AddAsyncLoad(string goName, Folder folder, UnityAction callback = null)
    {
        AsyncList.Add
        (
            () =>
            {
                AsyncLoad(goName, folder, callback);
            }
        );
    }

    public static void AddAsyncLoad<T>(string goName, int amt, Folder folder, ObjType objType, bool ui = false, bool canvas = false, UnityAction callback = null) where T : ObjRoot
    {
        AsyncList.Add
        (
            () =>
            {
                AsyncLoad<T>(goName, amt, folder, objType, ui, canvas, callback);
            }
        );
    }

    public static void AddAsyncPlayer(string player)
    {
        AsyncList.Add
        (
            () =>
            {
                KV<AsyncRequest, UnityAction> kv = new KV<AsyncRequest, UnityAction>();

                AssetBundleRequest bundleRequest = Bundle.Scene.LoadAssetAsync("Player");

                AsyncRequest asyncRequest = new AsyncRequest();
                asyncRequest.Request = bundleRequest;
                asyncRequest.Callback = null;

                kv.Key = asyncRequest;
                kv.Value = () =>
                {
                    GameObject go = (GameObject)Instantiate(bundleRequest.asset);

                    go.name = go.name.Replace("(Clone)", "");


                    ObjDic.UniqueAdd("Player", bundleRequest.asset);

                    ObjType objType;

                    if (player == "PlayerPink")
                    {
                        objType = ObjType.PlayerPink;

                        ObjectPool.Add(objType, new List<Transform>());

                        go.AddScript<Player>().BuildPink();

                        go.AddComponent<ObjRoot>().ObjType = objType;
                    }
                    else if (player == "PlayerBlond")
                    {
                        objType = ObjType.PlayerBlond;

                        ObjectPool.Add(objType, new List<Transform>());

                        go.AddScript<Player>().BuildBlond();

                        go.AddComponent<ObjRoot>().ObjType = objType;
                    }
                    else if(player == "PlayerBrown")
                    {
                        objType = ObjType.PlayerBrown;

                        ObjectPool.Add(objType, new List<Transform>());

                        go.AddScript<Player>().BuildBrown();

                        go.AddComponent<ObjRoot>().ObjType = objType;
                    }
                    else
                    {
                        throw new Exception();
                    }

                    Save(go);
                };

                RequestList.Add(bundleRequest);

                InstantiateList.Add(kv);
            }
        );
    }


    public void StopAsync()
    {
        StopCoroutine(CoroAsync);
        StopCoroutine(CoroInstantiate);
    }

    public static IEnumerator IAsyncLoad()
    {
        while (true)
        {
            if (RequestList.Count == 0 && !AsyncLoadLock)
            {
                OnAsyncLoadFinish.SafeInvoke();
                OnAsyncLoadFinish = null;
            }

            if (RequestList.Count > 0)
            {
                yield return null;
            }
            else
            {
                if (AsyncList.Count > 0 && !AsyncLoadLock)
                {
                    AsyncLoadLock = true;

                    Auxiliary.Instance.DelayCall
                    (
                        () =>
                        {
                            AsyncLoadLock = false;

                            AsyncList[0].SafeInvoke();

                            AsyncList.RemoveAt(0);
                        },
                        1
                    );

                    yield return null;
                }
                else
                {
                    yield return null;
                }
            }
        }
    }

    public static IEnumerator IAsyncInstancitate()
    {
        while (true)
        {
            if (AsyncInstantiateLock)
            {
                yield return null;
            }

            if (!InstantiateList.Valid())
            {
                yield return null;
            }
            else
            {
                if (Time.deltaTime > 0.333f)
                {
                    yield return null;
                }

                if (!InstantiateList[0].Key.Request.isDone)
                {
                    yield return null;
                }
                else
                {
                    InstantiateList[0].Key.Callback.SafeInvoke();

                    RequestList.Remove(InstantiateList[0].Key.Request);
                    
                    UnityAction action = InstantiateList[0].Value;

                    InstantiateList.RemoveAt(0);

                    AsyncInstantiateLock = true;

                    Auxiliary.Instance.DelayCall
                    (
                        () =>
                        {
                            action.SafeInvoke();

                            AsyncInstantiateLock = false;
                        },
                        1
                    );

                    yield return null;
                }
            }
        }
    }
}