From 09f4db6b570136dab3d3581d8b78afdf9d41cd80 Mon Sep 17 00:00:00 2001
From: Peter Clement <peter@budibase.com>
Date: Mon, 27 Jan 2025 09:01:39 +0000
Subject: [PATCH] finding of screen onload

---
 .../[screenId]/_components/StatePanel.svelte  | 28 ++++++++++++++++---
 1 file changed, 24 insertions(+), 4 deletions(-)

diff --git a/packages/builder/src/pages/builder/app/[application]/design/[screenId]/_components/StatePanel.svelte b/packages/builder/src/pages/builder/app/[application]/design/[screenId]/_components/StatePanel.svelte
index 847b078887..1c0473d8db 100644
--- a/packages/builder/src/pages/builder/app/[application]/design/[screenId]/_components/StatePanel.svelte
+++ b/packages/builder/src/pages/builder/app/[application]/design/[screenId]/_components/StatePanel.svelte
@@ -47,6 +47,10 @@
       previousScreenId = $selectedScreen._id
     }
 
+    if (keyOptions.length > 0 && !keyOptions.includes(selectedKey)) {
+      selectedKey = keyOptions[0]
+    }
+
     if (selectedKey) {
       searchComponents(selectedKey)
       editorValue = previewContext.state?.[selectedKey] ?? ""
@@ -180,14 +184,30 @@
       return
     }
 
+    const componentStateUpdates = findComponentsUpdatingState(
+      $selectedScreen.props,
+      stateKey
+    )
+
     componentsUsingState = findComponentsUsingState(
       $selectedScreen.props,
       stateKey
     )
-    componentsUpdatingState = findComponentsUpdatingState(
-      $selectedScreen.props,
-      stateKey
-    )
+
+    const screenStateUpdates =
+      $selectedScreen?.onLoad
+        ?.filter(
+          (handler: any) =>
+            handler["##eventHandlerType"] === "Update State" &&
+            handler.parameters?.key === stateKey
+        )
+        .map(() => ({
+          id: $selectedScreen._id,
+          name: "Screen onLoad",
+          settings: ["onLoad"],
+        })) || []
+
+    componentsUpdatingState = [...componentStateUpdates, ...screenStateUpdates]
   }
 
   const handleStateKeySelect = (key: CustomEvent) => {