/* SPDX-License-Identifier: Apache-2.0
*
* The OpenSearch Contributors require contributions made to
* this file be licensed under the Apache-2.0 license or a
* compatible open source license.
*/
/*
* Modifications Copyright OpenSearch Contributors. See
* GitHub history for details.
*
*  Licensed to Elasticsearch B.V. under one or more contributor
*  license agreements. See the NOTICE file distributed with
*  this work for additional information regarding copyright
*  ownership. Elasticsearch B.V. licenses this file to you under
*  the Apache License, Version 2.0 (the "License"); you may
*  not use this file except in compliance with the License.
*  You may obtain a copy of the License at
*
* 	http://www.apache.org/licenses/LICENSE-2.0
*
*  Unless required by applicable law or agreed to in writing,
*  software distributed under the License is distributed on an
*  "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
*  KIND, either express or implied.  See the License for the
*  specific language governing permissions and limitations
*  under the License.
*/

using System.Collections;
using System.Collections.Generic;
using System.ComponentModel;

namespace OpenSearch.Client
{
	public abstract class IsADictionaryBase<TKey, TValue> : IIsADictionary<TKey, TValue>
	{
		protected IsADictionaryBase() => BackingDictionary = new Dictionary<TKey, TValue>();

		protected IsADictionaryBase(IDictionary<TKey, TValue> backingDictionary)
		{
			// ReSharper disable VirtualMemberCallInConstructor
			if (backingDictionary != null)
				foreach (var key in backingDictionary.Keys)
					ValidateKey(Sanitize(key));
			// ReSharper enable VirtualMemberCallInConstructor

			BackingDictionary = backingDictionary != null
				? new Dictionary<TKey, TValue>(backingDictionary)
				: new Dictionary<TKey, TValue>();
		}

		public TValue this[TKey key]
		{
			get => BackingDictionary[Sanitize(key)];
			set => BackingDictionary[ValidateKey(Sanitize(key))] = value;
		}

		protected Dictionary<TKey, TValue> BackingDictionary { get; }
		int ICollection<KeyValuePair<TKey, TValue>>.Count => BackingDictionary.Count;
		bool ICollection<KeyValuePair<TKey, TValue>>.IsReadOnly => Self.IsReadOnly;

		TValue IDictionary<TKey, TValue>.this[TKey key]
		{
			get => BackingDictionary[Sanitize(key)];
			set => BackingDictionary[ValidateKey(Sanitize(key))] = value;
		}

		ICollection<TKey> IDictionary<TKey, TValue>.Keys => BackingDictionary.Keys;
		private ICollection<KeyValuePair<TKey, TValue>> Self => BackingDictionary;
		ICollection<TValue> IDictionary<TKey, TValue>.Values => BackingDictionary.Values;

		void ICollection<KeyValuePair<TKey, TValue>>.Add(KeyValuePair<TKey, TValue> item)
		{
			ValidateKey(Sanitize(item.Key));
			Self.Add(item);
		}

		void ICollection<KeyValuePair<TKey, TValue>>.Clear() => BackingDictionary.Clear();

		[EditorBrowsable(EditorBrowsableState.Never)]
		bool ICollection<KeyValuePair<TKey, TValue>>.Contains(KeyValuePair<TKey, TValue> item) => Self.Contains(item);

		void ICollection<KeyValuePair<TKey, TValue>>.CopyTo(KeyValuePair<TKey, TValue>[] array, int arrayIndex) => Self.CopyTo(array, arrayIndex);

		bool ICollection<KeyValuePair<TKey, TValue>>.Remove(KeyValuePair<TKey, TValue> item) => Self.Remove(item);

		void IDictionary<TKey, TValue>.Add(TKey key, TValue value) => BackingDictionary.Add(ValidateKey(Sanitize(key)), value);

		[EditorBrowsable(EditorBrowsableState.Never)]
		bool IDictionary<TKey, TValue>.ContainsKey(TKey key) => BackingDictionary.ContainsKey(Sanitize(key));

		bool IDictionary<TKey, TValue>.Remove(TKey key) => BackingDictionary.Remove(Sanitize(key));

		bool IDictionary<TKey, TValue>.TryGetValue(TKey key, out TValue value) => BackingDictionary.TryGetValue(Sanitize(key), out value);

		IEnumerator IEnumerable.GetEnumerator() => BackingDictionary.GetEnumerator();

		IEnumerator<KeyValuePair<TKey, TValue>> IEnumerable<KeyValuePair<TKey, TValue>>.GetEnumerator() => BackingDictionary.GetEnumerator();

		protected virtual TKey ValidateKey(TKey key) => key;

		protected virtual TKey Sanitize(TKey key) => key;
	}
}