From cc44805337ab713c7aa70a7b87564e592397ae82 Mon Sep 17 00:00:00 2001
From: FuzTheCat <96272894+FuzTheCat@users.noreply.github.com>
Date: Wed, 21 Dec 2022 07:04:08 +1000
Subject: [PATCH] Diggy danger ore - Player warnings limited to once per
 minute. (#1337)

---
 map_gen/maps/diggy/feature/danger_ore.lua  | 27 +++++++++--
 map_gen/maps/diggy/presets/danger_ores.lua | 53 +++++++++++-----------
 2 files changed, 49 insertions(+), 31 deletions(-)

diff --git a/map_gen/maps/diggy/feature/danger_ore.lua b/map_gen/maps/diggy/feature/danger_ore.lua
index 23fbd91c..9d351190 100644
--- a/map_gen/maps/diggy/feature/danger_ore.lua
+++ b/map_gen/maps/diggy/feature/danger_ore.lua
@@ -3,10 +3,18 @@ local RestrictEntities = require 'map_gen.shared.entity_placement_restriction'
 local Event = require 'utils.event'
 local Token = require 'utils.token'
 local ScenarioInfo = require 'features.gui.info'
+local Global = require 'utils.global'
 
 local DangerOre = {}
 
-local function banned_entities(allowed_entities)
+local last_warning_time = {}
+Global.register({
+        last_warning_time = last_warning_time,
+    }, function(tbl)
+        last_warning_time = tbl.last_warning_time
+end)
+
+local function banned_entities(allowed_entities, warning_timeout)
     --- Items explicitly allowed on ores
     RestrictEntities.add_allowed(allowed_entities)
 
@@ -29,10 +37,18 @@ local function banned_entities(allowed_entities)
     )
 
     --- Warning for players when their entities are destroyed
+    --- Note: Edit to limit warning once per minute per player produced with help from ChatGPT & grilledham
     local function on_destroy(event)
         local p = event.player
+        local current_time = game.tick
         if p and p.valid then
-            p.print('You cannot build that on top of ores, only belts, mining drills, and power poles are allowed.')
+            if not last_warning_time[p.index] then
+                last_warning_time[p.index] = -(warning_timeout + 1)
+            end
+            if current_time > last_warning_time[p.index] + warning_timeout then
+                last_warning_time[p.index] = current_time
+                p.print('You cannot build that on top of ores, only belts, mining drills, and power poles are allowed.')
+            end
         end
     end
 
@@ -40,9 +56,10 @@ local function banned_entities(allowed_entities)
 end
 
 function DangerOre.register (config)
-	local allowed_entities = config.allowed_entities
-	banned_entities(allowed_entities)
-	ScenarioInfo.add_map_extra_info([[Danger! Ores are generally unstable to build upon.
+    local allowed_entities = config.allowed_entities
+    local warning_timeout = config.warning_timeout
+    banned_entities(allowed_entities, warning_timeout)
+    ScenarioInfo.add_map_extra_info([[Danger! Ores are generally unstable to build upon.
 Only the following entities have been strengthened for building upon the ores:
  [item=burner-mining-drill] [item=electric-mining-drill] [item=pumpjack] [item=small-electric-pole] [item=medium-electric-pole] [item=big-electric-pole] [item=substation] [item=car] [item=tank] [item=spidertron]
  [item=stone-wall][item=small-lamp][item=transport-belt] [item=fast-transport-belt] [item=express-transport-belt]  [item=underground-belt] [item=fast-underground-belt] [item=express-underground-belt] [item=pipe] [item=pipe-to-ground]
diff --git a/map_gen/maps/diggy/presets/danger_ores.lua b/map_gen/maps/diggy/presets/danger_ores.lua
index a8912b67..5cdeb7cd 100644
--- a/map_gen/maps/diggy/presets/danger_ores.lua
+++ b/map_gen/maps/diggy/presets/danger_ores.lua
@@ -449,32 +449,33 @@ local config = {
             load = function() return require('map_gen.maps.diggy.feature.weapon_balance') end
         },
         danger_ore = {
-			enabled = true,
-			load = function() return require('map_gen.maps.diggy.feature.danger_ore') end,
-			allowed_entities = {
-				'transport-belt',
-				'fast-transport-belt',
-				'express-transport-belt',
-				'underground-belt',
-				'fast-underground-belt',
-				'express-underground-belt',
-				'small-electric-pole',
-				'medium-electric-pole',
-				'big-electric-pole',
-				'substation',
-				'electric-mining-drill',
-				'burner-mining-drill',
-				'pumpjack',
-				'car',
-				'tank',
-				'spidertron',
-				'stone-wall',
-				'small-lamp',
-				'pipe',
-				'pipe-to-ground',
-				'wooden-chest'
-			}
-		}
+            enabled = true,
+            load = function() return require('map_gen.maps.diggy.feature.danger_ore') end,
+            allowed_entities = {
+                'transport-belt',
+                'fast-transport-belt',
+                'express-transport-belt',
+                'underground-belt',
+                'fast-underground-belt',
+                'express-underground-belt',
+                'small-electric-pole',
+                'medium-electric-pole',
+                'big-electric-pole',
+                'substation',
+                'electric-mining-drill',
+                'burner-mining-drill',
+                'pumpjack',
+                'car',
+                'tank',
+                'spidertron',
+                'stone-wall',
+                'small-lamp',
+                'pipe',
+                'pipe-to-ground',
+                'wooden-chest'
+            },
+            warning_timeout = 60 * 60  -- In game ticks -> 1 minute
+        }
     }
 }