From 246d71048165cbb06dfe9e85bb256c8074de6b49 Mon Sep 17 00:00:00 2001
From: Tom Burrows <tomburrows13@icloud.com>
Date: Thu, 12 Dec 2024 03:18:39 +0000
Subject: [PATCH] Add spidertron requires fuel setting

---
 data-updates.lua     | 18 ++++++++++++++++++
 locale/en/locale.cfg |  2 ++
 settings.lua         | 12 ++++++++++--
 3 files changed, 30 insertions(+), 2 deletions(-)

diff --git a/data-updates.lua b/data-updates.lua
index 17c82f4..e988396 100644
--- a/data-updates.lua
+++ b/data-updates.lua
@@ -1 +1,19 @@
 require "prototypes.dummy-spidertron"
+
+local spidertron_requires_fuel = settings.startup["spidertron-enhancements-spiderton-requires-fuel"].value
+if spidertron_requires_fuel == "Yes" then
+  local spidertron = data.raw["spider-vehicle"]["spidertron"]
+  spidertron.energy_source = {
+    type = "burner",
+    fuel_categories = {"chemical"},
+    effectivity = 1,
+    fuel_inventory_size = 3,
+  }
+  spidertron.movement_energy_consumption = "800kW"
+  spidertron.alert_icon_shift = {0, 0}
+elseif spidertron_requires_fuel == "No" then
+  local spidertron = data.raw["spider-vehicle"]["spidertron"]
+  spidertron.energy_source = {
+    type = "void"
+  }
+end
\ No newline at end of file
diff --git a/locale/en/locale.cfg b/locale/en/locale.cfg
index aa96a16..44b7427 100644
--- a/locale/en/locale.cfg
+++ b/locale/en/locale.cfg
@@ -20,6 +20,7 @@ spidertron-enhancements-recall-shortcut=Call last used __1__
 
 [mod-setting-name]
 spidertron-enhancements-enable-corpse=Save spidertron inventory into corpse upon death
+spidertron-enhancements-spiderton-requires-fuel=Spidertron requires fuel
 spidertron-enhancements-increase-size=Increase spidertron size
 spidertron-enhancements-sound-pause=Spidertron sound average pause
 spidertron-enhancements-volume-scale=Spidertron sound volume scale
@@ -38,6 +39,7 @@ spidertron-enhancements-enter-entity-all-except-spidertrons=All except spidertro
 spidertron-enhancements-enter-entity-all=All
 
 [mod-setting-description]
+spidertron-enhancements-spiderton-requires-fuel=Default means that it will not be modified from whatever vanilla or another mod sets it to.
 spidertron-enhancements-sound-pause=Average number of seconds before a spidertron's sound starts playing again (sound plays for 19s).
 
 spidertron-enhancements-enter-entity=Configure controls in [font=default-bold]Per player[/font] settings tab.
diff --git a/settings.lua b/settings.lua
index 0addbd2..478f440 100644
--- a/settings.lua
+++ b/settings.lua
@@ -49,13 +49,21 @@ data:extend({
         default_value = false,
         order = "b",
     },
+    {
+        type = "string-setting",
+        name = "spidertron-enhancements-spiderton-requires-fuel",
+        setting_type = "startup",
+        default_value = "Default",
+        allowed_values = {"Yes", "Default", "No"},
+        order = "c",
+    },
     {
         type = "double-setting",
         name = "spidertron-enhancements-sound-pause",
         setting_type = "startup",
         default_value = 0,
         minimum_value = 0,
-        order = "c",
+        order = "d",
     },
     {
         type = "double-setting",
@@ -63,7 +71,7 @@ data:extend({
         setting_type = "startup",
         default_value = 1,
         minimum_value = 0,
-        order = "d",
+        order = "e",
     }
 })