Use notreallyafactory instead

ZeroK-RTS · Jun 14, 2024 · c176132 · c176132
1 parent 279d7fc
commit c176132
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/LuaUI/Widgets/unit_start_state.lua b/LuaUI/Widgets/unit_start_state.lua
@@ -146,9 +146,13 @@ for i = 1, #UnitDefs do
 	end
 end
 
-local notAFactory = {
-	[UnitDefNames["athena"]] = true, -- use stupid hax to avoid not treating striderhubs like a factory.
-}
+local notAFactory = {}
+for i = 1, #UnitDefs do
+	local ud = UnitDefs[i]
+	if ud.customParams.notreallyafactory then
+		notAFactory[ud.name] = true
+	end
+end
 
 options_path = 'Settings/Unit Behaviour/Default States'
 options_order = {