From 1c9c8bdb005a4d6289011a9624d8e6c5ee04f00e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jan=20Pokorn=C3=BD?= Date: Tue, 7 Jan 2025 13:21:00 +0100 Subject: [PATCH] feat: improve ready state handling MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Jan Pokorný --- src/entry.ts | 5 +++++ src/python/run.py | 1 + 2 files changed, 6 insertions(+) diff --git a/src/entry.ts b/src/entry.ts index 419775f..6643eb1 100644 --- 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 index be3e5a2..fafaa0a 100644 --- 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: