using MyNes.Core;
using System.Collections.Generic;
using System.Linq;
using UnityEngine;

namespace AxibugEmuOnline.Client.Input
{
    public class KeyMapper
    {
        private Dictionary<KeyCode, EnumKeyKind> m_mapper = new Dictionary<KeyCode, EnumKeyKind>();
        private Dictionary<EnumKeyKind, KeyCode> m_mapperOpp = new Dictionary<EnumKeyKind, KeyCode>();
        private Dictionary<EnumKeyKind, int> m_keyIndexTable = new Dictionary<EnumKeyKind, int>();
        private EnumKeyKind[] m_focusKeys;
        private bool[] m_keyStates;

        public void SetKeyMapper(KeyCode inputKeycode, EnumKeyKind joyKey)
        {
            if (m_mapperOpp.TryGetValue(joyKey, out KeyCode keyCode))//如果该映射已设置过,移除之前的映射
            {
                m_mapperOpp.Remove(joyKey);
                m_mapper.Remove(keyCode);
            }
            m_mapper[inputKeycode] = joyKey;
            m_mapperOpp[joyKey] = inputKeycode;
        }

        public void SetComplete()
        {
            m_focusKeys = m_mapperOpp.Keys.ToArray();
            m_keyStates = new bool[m_focusKeys.Length];

            m_keyIndexTable.Clear();
            for (int i = 0; i < m_focusKeys.Length; i++)
            {
                m_keyIndexTable[m_focusKeys[i]] = i;
            }
        }

        public void Update()
        {
            if (m_focusKeys == null) return;

            for (int i = 0; i < m_focusKeys.Length; i++)
            {
                var keyCode = m_mapperOpp[m_focusKeys[i]];
                m_keyStates[i] = UnityEngine.Input.GetKey(keyCode);
            }
        }

        public bool IsPressing(EnumKeyKind keyKind)
        {
            if (!m_keyIndexTable.TryGetValue(keyKind, out int index)) return false;//没有设置映射,直接false

            return m_keyStates[index];
        }
    }
}