feat: improve ready state handling

Signed-off-by: Jan Pokorný <[email protected]>
i-am-bee · Jan 7, 2025 · 1c9c8bd · 1c9c8bd
1 parent ec2d57a
commit 1c9c8bd
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/entry.ts b/src/entry.ts
@@ -85,9 +85,14 @@ const ALLOWED_ORIGINS = (import.meta.env.VITE_ALLOWED_FRAME_ANCESTORS ?? '').spl
     app.kernel._worker.addEventListener('message', (event: MessageEvent) => {
       const data = typeof event.data === 'string' ? JSON.parse(event.data) : event.data;
       switch (data.type) {
+        case 'bee:ready':
+          ALLOWED_ORIGINS.forEach((origin: string) => parent.postMessage({ type: 'bee:ready' }, origin))
+          return;
+
         case 'bee:request':
           ALLOWED_ORIGINS.forEach((origin: string) => parent.postMessage(data, origin));
           return;
+
         default:
           return;
       }

diff --git a/src/python/run.py b/src/python/run.py
@@ -232,6 +232,7 @@ def fix_markdown(body):
 
 
 async def run():
+    js.postMessage(json.dumps({ "type": "bee:ready" }))
     try:
         code = pathlib.Path("app.py").read_text()
         if st.session_state.get("_bee_last_code") != code: