Add feature to grid protection; players can't add blocks to protected grids

Author: rexxar-tc

EssentialsPlugin/Settings/PluginSettings.cs

@@ -81,11 +81,13 @@ public class PluginSettings
         private bool _dynamicConcealEnabled;
         private float _dynamicConcealDistance;
         private bool _dynamicConcealIncludeLargeGrids;
+        private bool _dynamicConcealIncludeStations;
         private string[] _dynamicConcealIgnoreSubTypeList = { };
         private bool _dynamicConcealIncludeMedBays;
         private bool _dynamicShowMessages;
         private bool _dynamicConcealPirates;
-        private int _dynamicConcealUpdateSpeed;
+        private bool _dynamicConcealPhysics;
+        private bool _dynamicConcealProduction;
         private bool _dynamicTurretManagementEnabled;
         private int _dynamicTurretTargetDistance;
         private bool _dynamicTurretAllowExemption;
@@ -124,7 +126,9 @@ public class PluginSettings
 	    private bool _atmosphericCargoShipsEnabled;
 	    private float _atmosphericCargoShipSpawnTime;
 
-        private List<TicketPlayerItem> _ticketPlayers; 
+        private List<TicketPlayerItem> _ticketPlayers;
+
+        private MTObservableCollection<BlacklistItem> _blacklistItems;
 
         #endregion
 
@@ -621,7 +625,27 @@ public bool DynamicConcealPirates
                 Save( );
             }
         }
-        
+
+        public bool DynamicConcealPhysics
+        {
+            get {return _dynamicConcealPhysics;}
+            set
+            {
+                _dynamicConcealPhysics = value;
+                Save();
+            }
+        }
+
+        public bool DynamicConcealProduction 
+        {
+            get { return _dynamicConcealProduction; }
+            set
+            {
+                _dynamicConcealProduction = value;
+                Save();
+            }
+        }
+
         public float DynamicConcealDistance
 		{
 			get { return _dynamicConcealDistance; }
@@ -642,12 +666,12 @@ public bool ConcealIncludeLargeGrids
 			}
 		}
 
-        public int DynamicConcealUpdateSpeed
+        public bool ConcealIncludeStations
         {
-            get {return _dynamicConcealUpdateSpeed;}
+            get { return _dynamicConcealIncludeStations; }
             set
             {
-                _dynamicConcealUpdateSpeed = value;
+                _dynamicConcealIncludeStations = value;
                 Save();
             }
         }
@@ -1001,6 +1025,17 @@ public List<TicketPlayerItem> TicketPlayers
                 Save(  );
             }
         }
+
+        public MTObservableCollection<BlacklistItem> BlacklistItems
+        {
+            get { return _blacklistItems; }
+            set
+            {
+                _blacklistItems = value;
+                BlacklistManager.Instance.UpdateBlacklist();
+                Save();
+            }
+        }
         #endregion
 
         #region Constructor
@@ -1046,7 +1081,8 @@ public PluginSettings()
 
 			_dynamicConcealDistance = 8000;
             _dynamicConcealPirates = false;
-            _dynamicConcealUpdateSpeed = 500;
+            _dynamicConcealPhysics = false;
+            _dynamicConcealProduction = true;
             _dynamicShowMessages = false;
 			_dynamicTurretTargetDistance = 2000;
 			_dynamicTurretManagementEnabled = false;
@@ -1077,21 +1113,24 @@ public PluginSettings()
             _timedCommandsItem = new MTObservableCollection<TimedCommandItem>(  );
             _timedCommandsItem.CollectionChanged += ItemsCollectionChanged;
 
+            _blacklistItems = new MTObservableCollection<BlacklistItem>();
+            _blacklistItems.CollectionChanged += ItemsCollectionChanged;
+            _blacklistItems.CollectionChanged += BlacklistManager.Instance._blacklistItems_CollectionChanged;
+
             _atmosphericCargoShipsEnabled = false;
             _atmosphericCargoShipSpawnTime = 10.0f;
 
             _ticketPlayers = new List<TicketPlayerItem>();
 		}
+        
+        #endregion
 
+        #region Loading and Saving
 
-		#endregion
-
-		#region Loading and Saving
-
-		/// <summary>
-		/// Loads our settings
-		/// </summary>
-		public void Load()
+        /// <summary>
+        /// Loads our settings
+        /// </summary>
+        public void Load()
 		{
 			_loading = true;
 

EssentialsPlugin/Settings/SettingsProtectedItem.cs

@@ -18,5 +18,19 @@ public long EntityId
 			get { return entityId; }
 			set { entityId = value; }
 		}
+
+	    private bool protectBlockAdd;
+	    public bool ProtectBockAdd
+	    {
+	        get { return protectBlockAdd; }
+            set { protectBlockAdd = value; }
+	    }
+
+	    private string protectBlockWarning;
+	    public string ProtectBlockWarning
+	    {
+	        get {return protectBlockWarning;}
+            set { protectBlockWarning = value; }
+	    }
 	}
 }

EssentialsPlugin/Utility/Protection.cs

@@ -3,49 +3,86 @@
     using System;
     using System.Collections.Generic;
     using System.Linq;
-    using System.Threading;
     using EssentialsPlugin.Settings;
-    using Sandbox.Common.ObjectBuilders.Definitions;
     using Sandbox.Game.Entities;
     using Sandbox.Game.Entities.Cube;
+    using Sandbox.Game.World;
     using Sandbox.ModAPI;
-    using Sandbox.Game.Weapons;
-    using Sandbox.ModAPI.Interfaces;
     using SEModAPIInternal.API.Common;
     using VRage.Game;
+    using VRage.Game.Entity.EntityComponents;
     using VRage.Game.ModAPI;
-    using VRage.Game.ModAPI.Interfaces;
-    using VRage.ModAPI;
-    using VRage.ObjectBuilders;
-    using VRage.Utils;
 
-    public static class Protection
+    public class Protection
     {
-        private static bool _init = false;
-        private static DateTime _lastLog;
-        private static SortedList<IMySlimBlock, float> toDamage = new SortedList<IMySlimBlock, float>( );
+        private static Protection _instance;
+        private bool _init = false;
+        private DateTime _lastLog;
 
-        public static void Init( )
+        public static Protection Instance
+        {
+            get
+            {
+                if(_instance == null)
+                    _instance = new Protection();
+                return _instance;
+            }
+        }
+
+        public void Init( )
         {
             if ( _init )
                 return;
 
             _init = true;
             MyAPIGateway.Session.DamageSystem.RegisterBeforeDamageHandler( 0, DamageHandler );
+            RegisterGridHandlers();
+        }
+
+        public void RegisterGridHandlers()
+        {
+            foreach ( var item in PluginSettings.Instance.ProtectedItems )
+            {
+                var grid = MyEntities.GetEntityById( item.EntityId ) as MyCubeGrid;
+                if ( grid == null )
+                {
+                    Essentials.Log.Error( $"Error getting entity in Protection.RegisterGridHandlers. ID: {item.EntityId}" );
+                    continue;
+                }
+                
+                grid.OnBlockAdded -= OnBlockAdded;
+                grid.OnBlockAdded += OnBlockAdded;
+            }
         }
 
-        private static void DamageHandler( object target, ref MyDamageInformation info )
+        private void OnBlockAdded( MySlimBlock block )
+        {
+            var protectionItem = PluginSettings.Instance.ProtectedItems.FirstOrDefault( x => x.EntityId == block.CubeGrid.EntityId );
+            if ( protectionItem == null || !protectionItem.Enabled || !protectionItem.ProtectBockAdd )
+                return;
+
+            if ( block.OwnerId != 0 )
+            {
+                var steamId = PlayerMap.Instance.GetSteamIdFromPlayerId( block.OwnerId );
+                //if ( PlayerManager.Instance.IsUserAdmin( steamId ) )
+                //    return;
+
+                Communication.Notification( steamId, MyFontEnum.Red, 5, protectionItem.ProtectBlockWarning ?? "You cannot add blocks to this grid!" );
+            }
+            MyAPIGateway.Utilities.InvokeOnGameThread(()=>block.CubeGrid.RazeBlock( block.Position ));
+            Essentials.Log.Info( $"Removed block from protected grid {protectionItem.EntityId}. Block owner: {PlayerMap.Instance.GetPlayerNameFromPlayerId( block.OwnerId )}" );
+        }
+
+        private void DamageHandler( object target, ref MyDamageInformation info )
         {
             if ( !PluginSettings.Instance.ProtectedEnabled )
                 return;
 
-            IMySlimBlock block = target as IMySlimBlock;
+            MySlimBlock block = target as MySlimBlock;
             if ( block == null )
                 return;
 
-            IMyCubeGrid grid = block.CubeGrid;
-            
-            ulong steamId = PlayerMap.Instance.GetSteamId( info.AttackerId );
+            MyCubeGrid grid = block.CubeGrid;
 
             foreach ( ProtectedItem item in PluginSettings.Instance.ProtectedItems )
             {
@@ -59,7 +96,6 @@ private static void DamageHandler( object target, ref MyDamageInformation info )
                     }
                 }
             }
-
         }
     }
 }