#277: Enforce lua_run casing via postcompiler

CHANGELOG.md

@@ -18,7 +18,7 @@
 * Fix various incorrect usages of the FGD `frustum()` helper.
 * Normals are now correctly calculated when generating propcombined models. This requires all previous models to be rebuilt, unfortunately.
 * Fix overlays not functioning with `material_modify_control` parent searching.
-* #274: Force `light_environment` to use the required `SunSpreadAngle` casing.
+* #274, #277: Force required keyvalue casing for `light_environment`'s `SunSpreadAngle` and `lua_run`'s `Code` keys.
 
 
 --------------------

transforms/fgd_tweaks/case_sensitive.py

@@ -0,0 +1,27 @@
+"""Some keyvalues are case-sensitive, force the required casing."""
+from srctools import VMF, logger
+
+from hammeraddons.bsp_transform import Context, trans
+from hammeraddons.bsp_transform.common import ent_description
+
+LOGGER = logger.get_logger(__name__)
+
+
+@trans('FGD - Fix key casing')
+def force_case_sensitivity(ctx: Context) -> None:
+    """Force case-sensitivity on some keyvalues that require it."""
+    fix_casing(ctx.vmf, 'light_environment', 'SunSpreadAngle')
+    fix_casing(ctx.vmf, 'light_directional', 'SunSpreadAngle')
+    fix_casing(ctx.vmf, 'lua_run', 'Code')
+
+
+def fix_casing(vmf: VMF, classname: str, *keys: str) -> None:
+    """Fix the casing for one entity."""
+    for ent in vmf.by_class[classname]:
+        for key in keys:
+            if value := ent.pop(key):
+                LOGGER.warning(
+                    'Correcting case of "{}" for {}',
+                    key, ent_description(ent),
+                )
+                ent[key] = value