using System.Collections.Generic;
using UnityEngine;
using UnityEngine.U2D;
using YIUIFramework;
using YooAsset;

namespace ET.Client
{
    [FriendOf(typeof(YIUIYooAssetsSpriteComponent))]
    [EntitySystemOf(typeof(YIUIYooAssetsSpriteComponent))]
    public static partial class YIUIYooAssetsSpriteComponentSystem
    {
        [EntitySystem]
        private static void Awake(this YIUIYooAssetsSpriteComponent self)
        {
            self.m_YIUILoadRef = self.GetParent<YIUILoadComponent>();
        }

        public static async ETTask LoadAtlasAsync(this YIUIYooAssetsSpriteComponent self)
        {
            EntityRef<YIUIYooAssetsSpriteComponent> selfRef = self;
            var handle = YooAssets.LoadAssetAsync<YIUIAtlasData>(YIUIConstAsset.AtlasDataName);

            await handle.Task;

            var atlasData = handle.AssetObject as YIUIAtlasData;
            if (atlasData == null)
            {
                return;
            }

            self = selfRef;
            self.m_SpritePathMap.Clear();
            foreach (var info in atlasData.Infos)
            {
                foreach (var spriteName in info.SpriteNames)
                {
                    self.m_SpritePathMap.Add(spriteName, info.AtlasName);
                }
            }

            handle.Release();
        }

        public static string GetSpriteAtlasPath(this YIUIYooAssetsSpriteComponent self, string spriteName)
        {
            return self.m_SpritePathMap.GetValueOrDefault(spriteName);
        }

        #if !YIUIMACRO_SYNCLOAD_CLOSE
        public static Sprite GetSprite(this YIUIYooAssetsSpriteComponent self, string spriteName)
        {
            Sprite sprite = null;

            if (!self.m_SpritePathMap.TryGetValue(spriteName, out string atlasPath))
            {
                #if UNITY_EDITOR
                if (!self.YIUILoad.VerifyAssetValidity(spriteName))
                {
                    Log.Error($"验证资产有效性 没有这个资源 图片无法加载 请检查 {spriteName}");
                    return null;
                }
                #endif

                sprite = self.YIUILoad.LoadAsset<Sprite>(spriteName);

                if (sprite == null)
                {
                    Log.Error($"加载失败 没有这个资源 图片无法加载 请检查 {spriteName}");
                }
            }
            else
            {
                var spriteAtlas = self.YIUILoad.LoadAsset<SpriteAtlas>(atlasPath);
                sprite = spriteAtlas?.GetSprite(spriteName);
                if (sprite == null)
                {
                    Log.Error($"找到图集{atlasPath},但是没有找不到Sprite：{spriteName}");
                    return null;
                }

                if (!self.m_LoadedSprites.TryAdd(sprite, spriteAtlas))
                {
                    Log.Error($"重复添加Sprite：{spriteName}");
                }
            }

            return sprite;
        }
        #endif

        public static async ETTask<Sprite> GetSpriteAsync(this YIUIYooAssetsSpriteComponent self, string spriteName)
        {
            Sprite sprite = null;

            if (!self.m_SpritePathMap.TryGetValue(spriteName, out string atlasPath))
            {
                #if UNITY_EDITOR
                if (!self.YIUILoad.VerifyAssetValidity(spriteName))
                {
                    Log.Error($"验证资产有效性 没有这个资源 图片无法加载 请检查 {spriteName}");
                    return null;
                }
                #endif

                sprite = await self.YIUILoad.LoadAssetAsync<Sprite>(spriteName);

                if (sprite == null)
                {
                    Log.Error($"加载失败 没有这个资源 图片无法加载 请检查 {spriteName}");
                }
            }
            else
            {
                EntityRef<YIUIYooAssetsSpriteComponent> selfRef = self;
                var spriteAtlas = await self.YIUILoad.LoadAssetAsync<SpriteAtlas>(atlasPath);
                sprite = spriteAtlas?.GetSprite(spriteName);
                if (sprite == null)
                {
                    Log.Error($"找到图集{atlasPath},但是没有找不到Sprite：{spriteName}");
                    return null;
                }

                self = selfRef;
                if (!self.m_LoadedSprites.TryAdd(sprite, spriteAtlas))
                {
                    Log.Error($"重复添加Sprite：{spriteName}");
                }
            }

            return sprite;
        }

        public static void ReleaseSprite(this YIUIYooAssetsSpriteComponent self, Sprite sprite)
        {
            if (self.m_LoadedSprites.ContainsKey(sprite))
            {
                var spriteAtlas = self.m_LoadedSprites[sprite];
                self.m_LoadedSprites.Remove(sprite);
                self.YIUILoad.Release(spriteAtlas);
            }
            else
            {
                self.YIUILoad.Release(sprite);
            }
        }
    }
}