using UnityEngine;
using System.Collections.Generic;
using Sirenix.OdinInspector; // 引入 Odin 命名空间

#if UNITY_EDITOR
using UnityEditor.Animations; // 仅编辑器下引用，用于解析Controller
#endif
namespace Cielonos.MainGame.Characters
{
    public class AnimatorStateMapper
    {
        // --- 配置区域 ---
        [Title("Bake Settings")]  
        [SerializeField, LabelText("Target Animator")]
        private Animator _targetAnimatorForBake;
        
        [TableList(ShowIndexLabels = true), Searchable] // Odin: 列表显示为表格，且支持搜索
        [SerializeField]
        private List<StateClipPair> _mappings = new List<StateClipPair>();

        // --- 运行时缓存 ---
        private Dictionary<string, AnimationClip> _clipDict;
        private bool _isInitialized = false;

        // --- 简单的数据结构 ---
        [System.Serializable]
        public struct StateClipPair
        {
            [ReadOnly] // 防止手动误改，建议通过Bake生成
            public string stateName;

            public AnimationClip clip;
        }

        public AnimatorStateMapper()
        {
            _mappings = new List<StateClipPair>();
        }

        // ========================================================================
        // Runtime API (供宿主调用)
        // ========================================================================

        /// <summary>
        /// 必须在宿主的 Awake 中调用此方法构建索引
        /// </summary>
        public void Initialize()
        {
            if (_isInitialized) return;

            _clipDict = new Dictionary<string, AnimationClip>(_mappings.Count);
            foreach (var pair in _mappings)
            {
                if (!string.IsNullOrEmpty(pair.stateName) && pair.clip != null)
                {
                    // 防止重复Key报错，以后面的覆盖前面的（或者你可以选择忽略）
                    _clipDict[pair.stateName] = pair.clip;
                }
            }

            _isInitialized = true;
        }

        public AnimationClip GetClip(string stateName)
        {
            if (!_isInitialized) Initialize();

            if (_clipDict.TryGetValue(stateName, out var clip))
            {
                return clip;
            }

            Debug.LogWarning($"[AnimatorStateMapper] 未找到 State: '{stateName}' 对应的 Clip。请检查是否已 Bake 或 State 名字是否正确。");
            return null;
        }

        public float GetClipLength(string stateName)
        {
            var clip = GetClip(stateName);
            return clip != null ? clip.length : 0f;
        }

        // ========================================================================
        // Editor Baking Logic (Odin Button)
        // ========================================================================

#if UNITY_EDITOR
        public void Bake(Animator animator)
        {
            _targetAnimatorForBake = animator;

            var controller = _targetAnimatorForBake.runtimeAnimatorController as AnimatorController;

            // 处理 Override Controller 的情况
            if (controller == null && _targetAnimatorForBake.runtimeAnimatorController is AnimatorOverrideController overrideCtrl)
            {
                controller = overrideCtrl.runtimeAnimatorController as AnimatorController;
            }

            if (controller == null)
            {
                Debug.LogError("Animator 上未找到有效的 AnimatorController (或 AnimatorOverrideController) 资源！");
                return;
            }

            _mappings.Clear();
            int count = 0;

            // 递归遍历所有的 Layer 和 SubStateMachine
            foreach (var layer in controller.layers)
            {
                count += RecursiveProcessStateMachine(layer.stateMachine);
            }

            Debug.Log($"<color=green>Bake 完成！共提取了 {count} 个 State-Clip 映射。</color>");
        }

        private int RecursiveProcessStateMachine(AnimatorStateMachine stateMachine)
        {
            int count = 0;

            // 1. 遍历当前层级的 State
            foreach (var childState in stateMachine.states)
            {
                var state = childState.state;
                var motion = state.motion;

                // 仅处理直接引用 AnimationClip 的情况
                if (motion is AnimationClip clip)
                {
                    _mappings.Add(new StateClipPair { stateName = state.name, clip = clip });
                    count++;
                }
                // TODO: 如果需要支持 BlendTree，可以在这里扩展逻辑
                // else if (motion is BlendTree tree) { ... }
            }

            // 2. 递归遍历子状态机 (Sub-State Machines)
            foreach (var childMachine in stateMachine.stateMachines)
            {
                count += RecursiveProcessStateMachine(childMachine.stateMachine);
            }

            return count;
        }
#endif
    }
    
#if UNITY_EDITOR
    public partial class AnimationSubcontrollerBase
    {
        [Button]
        private void MapperBake()
        {
            mapper ??= new AnimatorStateMapper();
            mapper.Bake(animator);
        }
    }
#endif
}