using System; using System.Collections; using System.Collections.Generic; using System.Linq; using System.Threading; namespace Newtonsoft.Json.Utilities { internal class DictionaryWrapper<TKey, TValue> : IDictionary<TKey, TValue>, ICollection<KeyValuePair<TKey, TValue>>, IEnumerable<KeyValuePair<TKey, TValue>>, IWrappedDictionary, IDictionary, ICollection, IEnumerable { private struct DictionaryEnumerator<TEnumeratorKey, TEnumeratorValue> : IDictionaryEnumerator, IEnumerator { private readonly IEnumerator<KeyValuePair<TEnumeratorKey, TEnumeratorValue>> _e; public DictionaryEntry Entry => (DictionaryEntry)Current; public object Key => Entry.Key; public object Value => Entry.Value; public object Current => new DictionaryEntry(_e.Current.Key, _e.Current.Value); public DictionaryEnumerator(IEnumerator<KeyValuePair<TEnumeratorKey, TEnumeratorValue>> e) { ValidationUtils.ArgumentNotNull(e, "e"); _e = e; } public bool MoveNext() { return _e.MoveNext(); } public void Reset() { _e.Reset(); } } private readonly IDictionary _dictionary; private readonly IDictionary<TKey, TValue> _genericDictionary; private readonly IReadOnlyDictionary<TKey, TValue> _readOnlyDictionary; private object _syncRoot; public ICollection<TKey> Keys { get { if (_dictionary != null) return Enumerable.ToList<TKey>(Enumerable.Cast<TKey>((IEnumerable)_dictionary.Keys)); if (_readOnlyDictionary != null) return Enumerable.ToList<TKey>(_readOnlyDictionary.Keys); return _genericDictionary.Keys; } } public ICollection<TValue> Values { get { if (_dictionary != null) return Enumerable.ToList<TValue>(Enumerable.Cast<TValue>((IEnumerable)_dictionary.Values)); if (_readOnlyDictionary != null) return Enumerable.ToList<TValue>(_readOnlyDictionary.Values); return _genericDictionary.Values; } } public TValue this[TKey key] { get { if (_dictionary != null) return (TValue)_dictionary[key]; if (_readOnlyDictionary != null) return _readOnlyDictionary[key]; return _genericDictionary[key]; } set { if (_dictionary != null) _dictionary[key] = value; else { if (_readOnlyDictionary != null) throw new NotSupportedException(); _genericDictionary[key] = value; } } } public int Count { get { if (_dictionary != null) return _dictionary.Count; if (_readOnlyDictionary != null) return _readOnlyDictionary.Count; return _genericDictionary.Count; } } public bool IsReadOnly { get { if (_dictionary != null) return _dictionary.IsReadOnly; if (_readOnlyDictionary != null) return true; return _genericDictionary.IsReadOnly; } } object IDictionary.this[object key] { get { if (_dictionary != null) return _dictionary[key]; if (_readOnlyDictionary != null) return _readOnlyDictionary[(TKey)key]; return _genericDictionary[(TKey)key]; } set { if (_dictionary != null) _dictionary[key] = value; else { if (_readOnlyDictionary != null) throw new NotSupportedException(); _genericDictionary[(TKey)key] = (TValue)value; } } } bool IDictionary.IsFixedSize { get { if (_genericDictionary != null) return false; if (_readOnlyDictionary != null) return true; return _dictionary.IsFixedSize; } } ICollection IDictionary.Keys { get { if (_genericDictionary != null) return Enumerable.ToList<TKey>((IEnumerable<TKey>)_genericDictionary.Keys); if (_readOnlyDictionary != null) return Enumerable.ToList<TKey>(_readOnlyDictionary.Keys); return _dictionary.Keys; } } ICollection IDictionary.Values { get { if (_genericDictionary != null) return Enumerable.ToList<TValue>((IEnumerable<TValue>)_genericDictionary.Values); if (_readOnlyDictionary != null) return Enumerable.ToList<TValue>(_readOnlyDictionary.Values); return _dictionary.Values; } } bool ICollection.IsSynchronized { get { if (_dictionary != null) return _dictionary.IsSynchronized; return false; } } object ICollection.SyncRoot { get { if (_syncRoot == null) Interlocked.CompareExchange(ref _syncRoot, new object(), null); return _syncRoot; } } public object UnderlyingDictionary { get { if (_dictionary != null) return _dictionary; if (_readOnlyDictionary != null) return _readOnlyDictionary; return _genericDictionary; } } public DictionaryWrapper(IDictionary dictionary) { ValidationUtils.ArgumentNotNull(dictionary, "dictionary"); _dictionary = dictionary; } public DictionaryWrapper(IDictionary<TKey, TValue> dictionary) { ValidationUtils.ArgumentNotNull(dictionary, "dictionary"); _genericDictionary = dictionary; } public DictionaryWrapper(IReadOnlyDictionary<TKey, TValue> dictionary) { ValidationUtils.ArgumentNotNull(dictionary, "dictionary"); _readOnlyDictionary = dictionary; } public void Add(TKey key, TValue value) { if (_dictionary != null) _dictionary.Add(key, value); else { if (_genericDictionary == null) throw new NotSupportedException(); _genericDictionary.Add(key, value); } } public bool ContainsKey(TKey key) { if (_dictionary != null) return _dictionary.Contains(key); if (_readOnlyDictionary != null) return _readOnlyDictionary.ContainsKey(key); return _genericDictionary.ContainsKey(key); } public bool Remove(TKey key) { if (_dictionary != null) { if (_dictionary.Contains(key)) { _dictionary.Remove(key); return true; } return false; } if (_readOnlyDictionary != null) throw new NotSupportedException(); return _genericDictionary.Remove(key); } public bool TryGetValue(TKey key, out TValue value) { if (_dictionary != null) { if (!_dictionary.Contains(key)) { value = default(TValue); return false; } value = (TValue)_dictionary[key]; return true; } if (_readOnlyDictionary != null) throw new NotSupportedException(); return _genericDictionary.TryGetValue(key, out value); } public void Add(KeyValuePair<TKey, TValue> item) { if (_dictionary != null) ((IList)_dictionary).Add(item); else { if (_readOnlyDictionary != null) throw new NotSupportedException(); if (_genericDictionary != null) _genericDictionary.Add(item); } } public void Clear() { if (_dictionary != null) _dictionary.Clear(); else { if (_readOnlyDictionary != null) throw new NotSupportedException(); _genericDictionary.Clear(); } } public bool Contains(KeyValuePair<TKey, TValue> item) { if (_dictionary != null) return ((IList)_dictionary).Contains(item); if (_readOnlyDictionary != null) return Enumerable.Contains<KeyValuePair<TKey, TValue>>((IEnumerable<KeyValuePair<TKey, TValue>>)_readOnlyDictionary, item); return _genericDictionary.Contains(item); } public void CopyTo(KeyValuePair<TKey, TValue>[] array, int arrayIndex) { if (_dictionary != null) { IDictionaryEnumerator enumerator = _dictionary.GetEnumerator(); try { while (enumerator.MoveNext()) { DictionaryEntry dictionaryEntry = (DictionaryEntry)enumerator.Current; array[arrayIndex++] = new KeyValuePair<TKey, TValue>((TKey)dictionaryEntry.Key, (TValue)dictionaryEntry.Value); } } finally { (enumerator as IDisposable)?.Dispose(); } } else { if (_readOnlyDictionary != null) throw new NotSupportedException(); _genericDictionary.CopyTo(array, arrayIndex); } } public bool Remove(KeyValuePair<TKey, TValue> item) { if (_dictionary != null) { if (_dictionary.Contains(item.Key)) { object objA = _dictionary[item.Key]; if (object.Equals(objA, item.Value)) { _dictionary.Remove(item.Key); return true; } return false; } return true; } if (_readOnlyDictionary != null) throw new NotSupportedException(); return _genericDictionary.Remove(item); } public IEnumerator<KeyValuePair<TKey, TValue>> GetEnumerator() { if (_dictionary != null) return Enumerable.Select<DictionaryEntry, KeyValuePair<TKey, TValue>>(Enumerable.Cast<DictionaryEntry>((IEnumerable)_dictionary), (Func<DictionaryEntry, KeyValuePair<TKey, TValue>>)((DictionaryEntry de) => new KeyValuePair<TKey, TValue>((TKey)de.Key, (TValue)de.Value))).GetEnumerator(); if (_readOnlyDictionary != null) return _readOnlyDictionary.GetEnumerator(); return _genericDictionary.GetEnumerator(); } IEnumerator IEnumerable.GetEnumerator() { return GetEnumerator(); } void IDictionary.Add(object key, object value) { if (_dictionary != null) _dictionary.Add(key, value); else { if (_readOnlyDictionary != null) throw new NotSupportedException(); _genericDictionary.Add((TKey)key, (TValue)value); } } IDictionaryEnumerator IDictionary.GetEnumerator() { if (_dictionary != null) return _dictionary.GetEnumerator(); if (_readOnlyDictionary != null) return new DictionaryEnumerator<TKey, TValue>(_readOnlyDictionary.GetEnumerator()); return new DictionaryEnumerator<TKey, TValue>(_genericDictionary.GetEnumerator()); } bool IDictionary.Contains(object key) { if (_genericDictionary != null) return _genericDictionary.ContainsKey((TKey)key); if (_readOnlyDictionary != null) return _readOnlyDictionary.ContainsKey((TKey)key); return _dictionary.Contains(key); } public void Remove(object key) { if (_dictionary != null) _dictionary.Remove(key); else { if (_readOnlyDictionary != null) throw new NotSupportedException(); _genericDictionary.Remove((TKey)key); } } void ICollection.CopyTo(Array array, int index) { if (_dictionary != null) _dictionary.CopyTo(array, index); else { if (_readOnlyDictionary != null) throw new NotSupportedException(); _genericDictionary.CopyTo((KeyValuePair<TKey, TValue>[])array, index); } } } }