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

namespace Oxide.Plugins
{
    [Info("Damage Mod: Deployables", "ColonBlow", "1.1.13")]
    [Description("Prevents/allows damage to numerous deployables")]
    class DMDeployables : RustPlugin
    {
        // Added config option to only block damage if under TC Privledge if set to block damage for deployable

        private Dictionary<string, bool> deployables = new Dictionary<string, bool>();
        private Dictionary<string, string> prefabs = new Dictionary<string, string>();
        private bool init = false;

        void OnServerInitialized() => LoadVariables();

        void Unload()
        {
            deployables.Clear();
            prefabs.Clear();
        }

        object OnEntityTakeDamage(BaseCombatEntity entity, HitInfo hitInfo)
        {
            if (!init || entity == null || hitInfo == null)
                return null;

            var kvp = prefabs.FirstOrDefault(x => x.Key == entity.PrefabName);

            bool underPriv = entity.GetBuildingPrivilege();
            if (!underPriv) return !string.IsNullOrEmpty(kvp.Value) && deployables.ContainsKey(kvp.Value) && deployables[kvp.Value] && !blockOnlyUnderTC ? (object)true : null;
            else return !string.IsNullOrEmpty(kvp.Value) && deployables.ContainsKey(kvp.Value) && deployables[kvp.Value] ? (object)true : null;
        }

        #region Config
        private bool Changed;
        private bool blockOnlyUnderTC = false;

        void LoadVariables()
        {
            CheckCfg("_Global Setting - Only Block Damage if under TC Privledge ? ", ref blockOnlyUnderTC);

            foreach (var itemDef in ItemManager.GetItemDefinitions().ToList())
            {
                var mod = itemDef.GetComponent<ItemModDeployable>();

                if (mod != null)
                {
                    deployables[itemDef.displayName.translated] = Convert.ToBoolean(GetConfig("Deployables", string.Format("Block {0}", itemDef.displayName.translated), false));
                    prefabs[mod.entityPrefab.resourcePath] = itemDef.displayName.translated;
                }
            }

            init = true;

            if (Changed)
            {
                SaveConfig();
                Changed = false;
            }
        }

        protected override void LoadDefaultConfig()
        {
            PrintWarning("Creating a new configuration file");
            Config.Clear();
            LoadVariables();
        }

        private void CheckCfg<T>(string Key, ref T var)
        {
            if (Config[Key] is T)
                var = (T)Config[Key];
            else
                Config[Key] = var;
        }

        object GetConfig(string menu, string datavalue, object defaultValue)
        {
            var data = Config[menu] as Dictionary<string, object>;
            if (data == null)
            {
                data = new Dictionary<string, object>();
                Config[menu] = data;
                Changed = true;
            }
            object value;
            if (!data.TryGetValue(datavalue, out value))
            {
                value = defaultValue;
                data[datavalue] = value;
                Changed = true;
            }
            return value;
        }
        #endregion
    }
}