Net.Like.Xue.Tokyo/Assets/Plugins/SerializedCollections/Runtime/LookupTables/DictionaryLookupTable.cs

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

namespace AYellowpaper.SerializedCollections
{
    internal class DictionaryLookupTable<TKey, TValue> : IKeyable
    {
        private SerializedDictionary<TKey, TValue> _dictionary;
        private Dictionary<TKey, List<int>> _occurences = new Dictionary<TKey, List<int>>();

        private static readonly List<int> EmptyList = new List<int>();

        public IEnumerable Keys => _dictionary.Keys;

        public DictionaryLookupTable(SerializedDictionary<TKey, TValue> dictionary)
        {
            _dictionary = dictionary;
        }

        public IReadOnlyList<int> GetOccurences(object key)
        {
            if (key is TKey castKey && _occurences.TryGetValue(castKey, out var list))
                return list;

            return EmptyList;
        }

        public void RecalculateOccurences()
        {
            _occurences.Clear();

            int count = _dictionary._serializedList.Count;
            for (int i = 0; i < count; i++)
            {
                var kvp = _dictionary._serializedList[i];
                if (!SerializedCollectionsUtility.IsValidKey(kvp.Key))
                    continue;

                if (!_occurences.ContainsKey(kvp.Key))
                    _occurences.Add(kvp.Key, new List<int>() { i });
                else
                    _occurences[kvp.Key].Add(i);
            }
        }

        public void RemoveKey(object key)
        {
            for (int i = _dictionary._serializedList.Count - 1; i >= 0; i--)
            {
                var dictKey = _dictionary._serializedList[i].Key;
                if (SerializedCollectionsUtility.KeysAreEqual(dictKey, key))
                    _dictionary._serializedList.RemoveAt(i);
            }
        }

        public void RemoveAt(int index)
        {
            _dictionary._serializedList.RemoveAt(index);
        }

        public object GetKeyAt(int index)
        {
            return _dictionary._serializedList[index];
        }

        public int GetCount()
        {
            return _dictionary._serializedList.Count;
        }

        public void RemoveDuplicates()
        {
            _dictionary._serializedList = _dictionary._serializedList
                .GroupBy(x => x.Key)
                .Where(x => SerializedCollectionsUtility.IsValidKey(x.Key))
                .Select(x => x.First()).ToList();
        }

        public void AddKey(object key)
        {
            var entry = new SerializedKeyValuePair<TKey, TValue>();
            entry.Key = (TKey) key;
            _dictionary._serializedList.Add(entry);
        }
    }
}