From 55fdc2ec4afee737aa312bd25b2a9a3926a5c044 Mon Sep 17 00:00:00 2001
From: KoalaTrapLord <265133236+KoalaTrapLord@users.noreply.github.com>
Date: Mon, 30 Mar 2026 20:31:09 -0400
Subject: [PATCH] feat: add CreateBattlegroundScene editor script to build full
 scene programmatically

---
 .../Assets/Editor/CreateBattlegroundScene.cs  | 390 ++++++++++++++++++
 1 file changed, 390 insertions(+)
 create mode 100644 unity-client/Assets/Editor/CreateBattlegroundScene.cs

diff --git a/unity-client/Assets/Editor/CreateBattlegroundScene.cs b/unity-client/Assets/Editor/CreateBattlegroundScene.cs
new file mode 100644
index 0000000..1965673
--- /dev/null
+++ b/unity-client/Assets/Editor/CreateBattlegroundScene.cs
@@ -0,0 +1,390 @@
+// Assets/Editor/CreateBattlegroundScene.cs
+// Unity menu: Commander AI Lab ▶ Create Battleground Scene
+// Builds the entire Battleground scene hierarchy and wires every SerializeField.
+// Run once on an empty scene, then hit Play.
+
+using System.Collections.Generic;
+using UnityEditor;
+using UnityEditor.SceneManagement;
+using UnityEngine;
+using UnityEngine.UI;
+using UnityEngine.EventSystems;
+using TMPro;
+using CommanderAILab.UI;
+using CommanderAILab.Services;
+
+namespace CommanderAILab.Editor
+{
+    public static class CreateBattlegroundScene
+    {
+        // ── Phase names in enum order (matches PhaseTrackerWidget.Phase) ──────
+        private static readonly string[] PhaseNames =
+        {
+            "Untap", "Upkeep", "Draw", "Main1",
+            "Begin Combat", "Attackers", "Blockers",
+            "1st Strike", "Damage", "End Combat",
+            "Main2", "End Step", "Cleanup"
+        };
+
+        // ── Seat labels ───────────────────────────────────────────────────────
+        private static readonly string[] SeatNames = { "Bottom", "Right", "Top", "Left" };
+
+        // ─────────────────────────────────────────────────────────────────────
+        [MenuItem("Commander AI Lab/Create Battleground Scene")]
+        public static void Build()
+        {
+            // 1. Open a fresh scene
+            if (!EditorSceneManager.SaveCurrentModifiedScenesIfUserWantsTo()) return;
+            var scene = EditorSceneManager.NewScene(NewSceneSetup.EmptyScene, NewSceneMode.Single);
+
+            // 2. Services (DontDestroyOnLoad singletons — just need to exist in scene)
+            var services    = NewGO("Services");
+            var apiClient   = services.AddComponent<ApiClient>();
+            // GameWebSocketClient added below so BattlegroundController can find it
+            var wsClientGO  = NewGO("GameWebSocketClient", services.transform);
+
+            // 3. GameStateManager singleton
+            var gsmGO = NewGO("GameStateManager");
+            gsmGO.AddComponent<GameStateManager>();
+
+            // 4. EventSystem (required for all UI input)
+            var esGO = NewGO("EventSystem");
+            esGO.AddComponent<EventSystem>();
+            esGO.AddComponent<StandaloneInputModule>();
+
+            // 5. Main Camera
+            var camGO = NewGO("Main Camera");
+            var cam   = camGO.AddComponent<Camera>();
+            cam.clearFlags       = CameraClearFlags.SolidColor;
+            cam.backgroundColor  = new Color(0.08f, 0.12f, 0.08f); // dark green
+            cam.orthographic     = true;
+            cam.orthographicSize = 5f;
+            cam.tag              = "MainCamera";
+            camGO.AddComponent<AudioListener>();
+
+            // 6. Root Canvas
+            var canvasGO = NewGO("Canvas");
+            var canvas   = canvasGO.AddComponent<Canvas>();
+            canvas.renderMode = RenderMode.ScreenSpaceOverlay;
+            canvasGO.AddComponent<CanvasScaler>();
+            canvasGO.AddComponent<GraphicRaycaster>();
+
+            // ── 7. Seat Panels (4 RectTransforms around the board) ────────────
+            var seatsRoot  = NewGO("SeatPanels", canvasGO.transform);
+            var seatPanels = new RectTransform[4];
+            for (int i = 0; i < 4; i++)
+            {
+                var sp   = NewGO($"SeatPanel_{SeatNames[i]}", seatsRoot.transform);
+                var rt   = sp.AddComponent<RectTransform>();
+                var img  = sp.AddComponent<Image>();
+                img.color = new Color(0f, 0f, 0f, 0.3f);
+                StretchFill(rt);
+                seatPanels[i] = rt;
+            }
+
+            // ── 8. Center panel ───────────────────────────────────────────────
+            var centerGO = NewGO("BattlefieldCenter", canvasGO.transform);
+            StretchFill(centerGO.GetComponent<RectTransform>() ?? centerGO.AddComponent<RectTransform>());
+
+            // ── 9. PhaseTrackerWidget ─────────────────────────────────────────
+            var ptwGO   = NewGO("PhaseTracker", centerGO.transform);
+            var ptWidget = ptwGO.AddComponent<PhaseTrackerWidget>();
+            SetAnchored(ptwGO, new Vector2(0f, 0f), new Vector2(1f, 0.08f)); // bottom strip
+
+            // -- Active player + turn number labels
+            var aplGO = MakeTMPText("ActivePlayerLabel", ptwGO.transform, "Player's Turn", 18);
+            var tnlGO = MakeTMPText("TurnNumberLabel",   ptwGO.transform, "Turn 1",        16);
+
+            // -- 13 PhaseStepButtons
+            var stepButtons = new PhaseStepButton[13];
+            for (int i = 0; i < 13; i++)
+            {
+                var btnGO = NewGO($"PhaseBtn_{PhaseNames[i]}", ptwGO.transform);
+                var img   = btnGO.AddComponent<Image>();
+                img.color = new Color(0.2f, 0.2f, 0.2f);
+
+                var psb   = btnGO.AddComponent<PhaseStepButton>();
+
+                // PhaseStepButton.background + label — set via reflection (private SerializeFields)
+                var labelGO = MakeTMPText("Label", btnGO.transform, PhaseNames[i], 11);
+                SetPrivateField(psb, "background", img);
+                SetPrivateField(psb, "label", labelGO.GetComponent<TMP_Text>());
+
+                stepButtons[i] = psb;
+            }
+
+            // Wire PhaseTrackerWidget SerializeFields
+            SetPrivateField(ptWidget, "stepButtons",       stepButtons);
+            SetPrivateField(ptWidget, "activePlayerLabel", aplGO.GetComponent<TMP_Text>());
+            SetPrivateField(ptWidget, "turnNumberLabel",   tnlGO.GetComponent<TMP_Text>());
+
+            // ── 10. TurnOrderBar ──────────────────────────────────────────────
+            var tobGO  = NewGO("TurnOrderBar", centerGO.transform);
+            var tobBar = tobGO.AddComponent<TurnOrderBar>();
+            SetAnchored(tobGO, new Vector2(0f, 0.92f), new Vector2(1f, 1f)); // top strip
+
+            // ActiveBorder (sliding highlight)
+            var borderGO = NewGO("ActiveBorder", tobGO.transform);
+            var borderRT = borderGO.AddComponent<RectTransform>();
+            var borderImg = borderGO.AddComponent<Image>();
+            borderImg.color = new Color(1f, 0.85f, 0f, 0.35f);
+
+            // 4 TurnOrderPills
+            var pills = new TurnOrderPill[4];
+            for (int i = 0; i < 4; i++)
+            {
+                var pillGO  = NewGO($"Pill_{i}", tobGO.transform);
+                var pillImg = pillGO.AddComponent<Image>();
+                pillImg.color = new Color(0.25f, 0.25f, 0.25f);
+                var pill    = pillGO.AddComponent<TurnOrderPill>();
+
+                var nameLabel = MakeTMPText("PlayerName", pillGO.transform, $"Seat {i}", 13);
+                var lifeLabel = MakeTMPText("Life",       pillGO.transform, "40",        13);
+                var youBadge  = NewGO("YouBadge", pillGO.transform);
+                MakeTMPText("YouText", youBadge.transform, "YOU", 11);
+                youBadge.SetActive(i == 0);
+
+                SetPrivateField(pill, "playerNameLabel", nameLabel.GetComponent<TMP_Text>());
+                SetPrivateField(pill, "lifeLabel",       lifeLabel.GetComponent<TMP_Text>());
+                SetPrivateField(pill, "pillBackground",  pillImg);
+                SetPrivateField(pill, "youBadge",        youBadge);
+
+                pills[i] = pill;
+            }
+
+            SetPrivateField(tobBar, "pills",        pills);
+            SetPrivateField(tobBar, "activeBorder", borderRT);
+
+            // ── 11. LobbySetupModal ───────────────────────────────────────────
+            var modalGO  = NewGO("LobbySetupModal", canvasGO.transform);
+            var modal    = modalGO.AddComponent<LobbySetupModal>();
+            StretchFill(modalGO.GetComponent<RectTransform>() ?? modalGO.AddComponent<RectTransform>());
+
+            // Dark overlay panel
+            var panelGO  = NewGO("Panel", modalGO.transform);
+            var panelImg = panelGO.AddComponent<Image>();
+            panelImg.color = new Color(0f, 0f, 0f, 0.85f);
+            SetAnchored(panelGO, new Vector2(0.2f, 0.1f), new Vector2(0.8f, 0.9f));
+
+            MakeTMPText("Title", panelGO.transform, "Configure Seats", 24);
+
+            // 4 seat rows
+            var seatRows = new LobbyRowUI[4];
+            for (int i = 0; i < 4; i++)
+            {
+                var rowGO  = NewGO($"SeatRow_{i}", panelGO.transform);
+                rowGO.AddComponent<RectTransform>();
+                var row    = rowGO.AddComponent<LobbyRowUI>();
+
+                var seatLbl  = MakeTMPText("SeatLabel",  rowGO.transform, $"Seat {i}", 14);
+                var togGO    = MakeToggle("HumanToggle", rowGO.transform);
+                var deckDD   = MakeTMPDropdown("DeckDropdown",    rowGO.transform);
+                var styleDD  = MakeTMPDropdown("AiStyleDropdown", rowGO.transform);
+
+                SetPrivateField(row, "seatLabel",       seatLbl.GetComponent<TMP_Text>());
+                SetPrivateField(row, "humanToggle",     togGO.GetComponent<Toggle>());
+                SetPrivateField(row, "deckDropdown",    deckDD.GetComponent<TMP_Dropdown>());
+                SetPrivateField(row, "aiStyleDropdown", styleDD.GetComponent<TMP_Dropdown>());
+
+                seatRows[i] = row;
+            }
+
+            // Error label (starts inactive)
+            var errGO  = MakeTMPText("ErrorLabel", panelGO.transform, "", 13);
+            errGO.GetComponent<TMP_Text>().color = Color.red;
+            errGO.SetActive(false);
+
+            // Confirm + Cancel buttons
+            var confirmBtn = MakeButton("ConfirmButton", panelGO.transform, "Confirm");
+            var cancelBtn  = MakeButton("CancelButton",  panelGO.transform, "Cancel");
+
+            // Wire LobbySetupModal SerializeFields
+            SetPrivateField(modal, "seatRows",      seatRows);
+            SetPrivateField(modal, "confirmButton", confirmBtn.GetComponent<Button>());
+            SetPrivateField(modal, "cancelButton",  cancelBtn.GetComponent<Button>());
+            SetPrivateField(modal, "errorLabel",    errGO.GetComponent<TMP_Text>());
+            SetPrivateField(modal, "apiClient",     apiClient);
+
+            // ── 12. Main Menu Button (top-left HUD) ───────────────────────────
+            var mmBtnGO = MakeButton("MainMenuButton", canvasGO.transform, "⬅ Menu");
+            var mmRT    = mmBtnGO.GetComponent<RectTransform>();
+            mmRT.anchorMin        = new Vector2(0f, 1f);
+            mmRT.anchorMax        = new Vector2(0f, 1f);
+            mmRT.pivot            = new Vector2(0f, 1f);
+            mmRT.anchoredPosition = new Vector2(10f, -10f);
+            mmRT.sizeDelta        = new Vector2(120f, 40f);
+
+            // ── 13. BattlegroundController ────────────────────────────────────
+            var bcGO = NewGO("BattlegroundController", canvasGO.transform);
+            var bc   = bcGO.AddComponent<BattlegroundController>();
+
+            SetPrivateField(bc, "seatPanels",    seatPanels);
+            SetPrivateField(bc, "phaseTracker",  ptWidget);
+            SetPrivateField(bc, "turnOrderBar",  tobBar);
+            SetPrivateField(bc, "lobbyModal",    modal);
+            SetPrivateField(bc, "mainMenuButton", mmBtnGO.GetComponent<Button>());
+
+            // ── 14. Save scene ────────────────────────────────────────────────
+            const string scenePath = "Assets/Scripts/UI/Battleground/Battleground.unity";
+            EditorSceneManager.SaveScene(scene, scenePath);
+
+            // Add to build settings if not already present
+            AddSceneToBuild(scenePath);
+
+            Debug.Log("[CreateBattlegroundScene] ✅ Scene built and saved to " + scenePath);
+            EditorUtility.DisplayDialog("Done",
+                "Battleground scene created at:\n" + scenePath +
+                "\n\nHit Play to run it.", "OK");
+        }
+
+        // ═════════════════════════════════════════════════════════════════════
+        // Helpers
+        // ═════════════════════════════════════════════════════════════════════
+
+        static GameObject NewGO(string name, Transform parent = null)
+        {
+            var go = new GameObject(name);
+            if (parent != null) go.transform.SetParent(parent, false);
+            return go;
+        }
+
+        static void StretchFill(RectTransform rt)
+        {
+            rt.anchorMin        = Vector2.zero;
+            rt.anchorMax        = Vector2.one;
+            rt.offsetMin        = Vector2.zero;
+            rt.offsetMax        = Vector2.zero;
+        }
+
+        static void SetAnchored(GameObject go, Vector2 anchorMin, Vector2 anchorMax)
+        {
+            var rt = go.GetComponent<RectTransform>() ?? go.AddComponent<RectTransform>();
+            rt.anchorMin  = anchorMin;
+            rt.anchorMax  = anchorMax;
+            rt.offsetMin  = Vector2.zero;
+            rt.offsetMax  = Vector2.zero;
+        }
+
+        static GameObject MakeTMPText(string name, Transform parent, string text, int fontSize)
+        {
+            var go  = NewGO(name, parent);
+            var rt  = go.AddComponent<RectTransform>();
+            var tmp = go.AddComponent<TextMeshProUGUI>();
+            tmp.text     = text;
+            tmp.fontSize = fontSize;
+            tmp.color    = Color.white;
+            tmp.alignment = TextAlignmentOptions.Center;
+            return go;
+        }
+
+        static GameObject MakeButton(string name, Transform parent, string label)
+        {
+            var go  = NewGO(name, parent);
+            go.AddComponent<RectTransform>();
+            var img = go.AddComponent<Image>();
+            img.color = new Color(0.15f, 0.15f, 0.15f);
+            go.AddComponent<Button>();
+
+            var lblGO = MakeTMPText("Text", go.transform, label, 14);
+            StretchFill(lblGO.GetComponent<RectTransform>());
+            return go;
+        }
+
+        static GameObject MakeToggle(string name, Transform parent)
+        {
+            var go     = NewGO(name, parent);
+            go.AddComponent<RectTransform>();
+            var toggle = go.AddComponent<Toggle>();
+
+            var bgGO   = NewGO("Background", go.transform);
+            var bgImg  = bgGO.AddComponent<Image>();
+            bgImg.color = Color.white;
+
+            var checkGO  = NewGO("Checkmark", bgGO.transform);
+            var checkImg = checkGO.AddComponent<Image>();
+            checkImg.color = new Color(0.2f, 0.8f, 0.2f);
+
+            toggle.targetGraphic = bgImg;
+            toggle.graphic       = checkImg;
+            return go;
+        }
+
+        static GameObject MakeTMPDropdown(string name, Transform parent)
+        {
+            var go  = NewGO(name, parent);
+            go.AddComponent<RectTransform>();
+            var img = go.AddComponent<Image>();
+            img.color = new Color(0.15f, 0.15f, 0.15f);
+            var dd  = go.AddComponent<TMP_Dropdown>();
+
+            // Required child: Label
+            var lblGO = MakeTMPText("Label", go.transform, "Select...", 13);
+            StretchFill(lblGO.GetComponent<RectTransform>());
+            dd.captionText = lblGO.GetComponent<TMP_Text>();
+
+            // Required child: Arrow
+            var arrowGO  = MakeTMPText("Arrow", go.transform, "▼", 12);
+            var arrowRT  = arrowGO.GetComponent<RectTransform>();
+            arrowRT.anchorMin        = new Vector2(1f, 0.5f);
+            arrowRT.anchorMax        = new Vector2(1f, 0.5f);
+            arrowRT.pivot            = new Vector2(1f, 0.5f);
+            arrowRT.anchoredPosition = new Vector2(-5f, 0f);
+            arrowRT.sizeDelta        = new Vector2(20f, 20f);
+
+            // Required child: Template (hidden)
+            var tmplGO  = NewGO("Template", go.transform);
+            var tmplImg = tmplGO.AddComponent<Image>();
+            tmplImg.color = new Color(0.1f, 0.1f, 0.1f);
+            var tmplRT  = tmplGO.GetComponent<RectTransform>() ?? tmplGO.AddComponent<RectTransform>();
+            tmplRT.anchorMin        = new Vector2(0f, 0f);
+            tmplRT.anchorMax        = new Vector2(1f, 0f);
+            tmplRT.pivot            = new Vector2(0.5f, 1f);
+            tmplRT.sizeDelta        = new Vector2(0f, 150f);
+            tmplGO.AddComponent<ScrollRect>();
+            tmplGO.SetActive(false);
+
+            // Viewport > Content > Item > ItemLabel
+            var vpGO    = NewGO("Viewport", tmplGO.transform);
+            vpGO.AddComponent<RectTransform>();
+            vpGO.AddComponent<Mask>();
+            vpGO.AddComponent<Image>();
+
+            var contentGO = NewGO("Content", vpGO.transform);
+            contentGO.AddComponent<RectTransform>();
+
+            var itemGO  = NewGO("Item", contentGO.transform);
+            var itemTog = itemGO.AddComponent<Toggle>();
+            var itemLbl = MakeTMPText("Item Label", itemGO.transform, "Option", 13);
+            itemTog.targetGraphic = itemGO.AddComponent<Image>();
+            itemTog.graphic       = itemLbl.GetComponent<TMP_Text>();
+            dd.itemText           = itemLbl.GetComponent<TMP_Text>();
+
+            return go;
+        }
+
+        /// <summary>Set a private [SerializeField] via reflection.</summary>
+        static void SetPrivateField(object target, string fieldName, object value)
+        {
+            var type  = target.GetType();
+            while (type != null)
+            {
+                var fi = type.GetField(fieldName,
+                    System.Reflection.BindingFlags.NonPublic |
+                    System.Reflection.BindingFlags.Instance);
+                if (fi != null) { fi.SetValue(target, value); return; }
+                type = type.BaseType;
+            }
+            Debug.LogWarning($"[CreateBattlegroundScene] Field not found: {fieldName} on {target.GetType().Name}");
+        }
+
+        static void AddSceneToBuild(string scenePath)
+        {
+            var scenes = new List<EditorBuildSettingsScene>(EditorBuildSettings.scenes);
+            foreach (var s in scenes)
+                if (s.path == scenePath) return; // already present
+
+            scenes.Insert(0, new EditorBuildSettingsScene(scenePath, true));
+            EditorBuildSettings.scenes = scenes.ToArray();
+        }
+    }
+}