Update SubUnit.lua to make it possible to remove the button

Basilisk3 · Feb 12, 2024 · 2581b11 · 2581b11
1 parent 7cedaf1
commit 2581b11
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/lua/sim/units/SubUnit.lua b/lua/sim/units/SubUnit.lua
@@ -1,4 +1,3 @@
-
 local EffectTemplate = import("/lua/effecttemplates.lua")
 
 local Entity = import("/lua/sim/entity.lua").Entity
@@ -39,8 +38,10 @@ SubUnit = ClassUnit(MobileUnit) {
         end
 
         if new == 'Top' or new == 'Bottom' then
-            self:AddCommandCap("RULEUCC_Dive")
-            self:RequestRefreshUI()
+			if bp.General.CommandCaps.RULEUCC_Dive then
+				self:AddCommandCap("RULEUCC_Dive")
+				self:RequestRefreshUI()
+			end
         end
     end,
 }