namespace SPTarkov.Server.Core.Extensions
{
    public static class DictionaryExtensions
    {
        /// <summary>
        /// Add a value by key to a dictionary, if the key doesn't exist, create it
        /// </summary>
        /// <typeparam name="T">Dictionary key type</typeparam>
        /// <param name="dict">Dictionary to add/update</param>
        /// <param name="key">Key to update by</param>
        /// <param name="value">Value to add to key</param>
        public static void AddOrUpdate<T>(this IDictionary<T, double> dict, T key, double value)
            where T : notnull
        {
            if (!dict.TryAdd(key, value))
            {
                dict[key] += value;
            }
        }

        /// <summary>
        /// Add a value by key to a dictionary, if the key doesn't exist, create it
        /// </summary>
        /// <typeparam name="T">Dictionary key type</typeparam>
        /// <param name="dict">Dictionary to add/update</param>
        /// <param name="key">Key to update by</param>
        /// <param name="value">Value to add to key</param>
        public static void AddOrUpdate<T>(this IDictionary<T, int> dict, T key, int value)
            where T : notnull
        {
            if (!dict.TryAdd(key, value))
            {
                dict[key] += value;
            }
        }
    }
}