using System;
using System.Collections.Generic;

namespace Admin.Core.Common.Extensions
{
    public static class DictionaryExtensions
    {
        public static TValue GetOrAdd<TKey, TValue>(
           this IDictionary<TKey, TValue> dictionary,
           TKey key,
           Func<TKey, TValue> factory)
        {
            TValue obj;
            if (dictionary.TryGetValue(key, out obj))
            {
                return obj;
            }

            return dictionary[key] = factory(key);
        }

        public static TValue GetOrAdd<TKey, TValue>(
           this IDictionary<TKey, TValue> dictionary,
           TKey key,
           Func<TValue> factory)
        {
            return dictionary.GetOrAdd(key, k => factory());
        }

        public static TValue AddOrUpdate<TKey, TValue>(
           this IDictionary<TKey, TValue> dictionary,
           TKey key,
           Func<TKey, TValue> addFactory,
           Func<TKey, TValue, TValue> updateFactory)
        {
            TValue obj;
            if (dictionary.TryGetValue(key, out obj))
            {
                obj = updateFactory(key, obj);
            }
            else
            {
                obj = addFactory(key);
            }
            dictionary[key] = obj;
            return obj;
        }

        public static TValue AddOrUpdate<TKey, TValue>(
           this IDictionary<TKey, TValue> dictionary,
           TKey key,
           Func<TValue> addFactory,
           Func<TValue, TValue> updateFactory)
        {
            return dictionary.AddOrUpdate(key, k => addFactory(), (k, v) => updateFactory(v));
        }
    }
}