From 692cc6b4be4a13ec6172327eeab49a6f03d73cb8 Mon Sep 17 00:00:00 2001
From: Stephen Zhou <38493346+hyoban@users.noreply.github.com>
Date: Thu, 31 Jul 2025 17:22:49 +0800
Subject: [PATCH] chore: load sqlite online faster

---
 .../src/modules/app/EnvironmentIndicator.tsx  | 46 ++++++++++++-------
 1 file changed, 30 insertions(+), 16 deletions(-)

diff --git a/apps/desktop/layer/renderer/src/modules/app/EnvironmentIndicator.tsx b/apps/desktop/layer/renderer/src/modules/app/EnvironmentIndicator.tsx
index 4ee78e75d..947de8293 100644
--- a/apps/desktop/layer/renderer/src/modules/app/EnvironmentIndicator.tsx
+++ b/apps/desktop/layer/renderer/src/modules/app/EnvironmentIndicator.tsx
@@ -69,6 +69,8 @@ export const EnvironmentIndicator = () => {
 }
 
 if (DEV) {
+  const sqliteOnlineWebsite = "https://sqlite-online.vercel.app"
+
   DebugRegistry.add("SQLite Online", () => {
     window.presentModal({
       title: "SQLite Online",
@@ -76,29 +78,41 @@ if (DEV) {
         <div className="h-full p-16" onClick={dismiss}>
           <iframe
             id="sql-viewer"
-            src="https://sqlite-online.vercel.app/"
+            src={sqliteOnlineWebsite}
             className="size-full"
             onLoad={() => {
               const iframe = document.querySelector("#sql-viewer") as HTMLIFrameElement
               if (!iframe) return
               const win = iframe.contentWindow
               if (!win) return
-              getDBFile()
-                .then(async (blob) => {
-                  const arrayBuffer = await blob.arrayBuffer()
-                  await new Promise((resolve) => setTimeout(resolve, 1000))
 
-                  win.postMessage(
-                    {
-                      type: "invokeLoadDatabaseBuffer",
-                      buffer: arrayBuffer,
-                    },
-                    "https://sqlite-online.vercel.app/",
-                  )
-                })
-                .catch((error) => {
-                  console.error("Failed to load database file into SQLite Online", error)
-                })
+              let isDatabaseLoaded = false
+
+              window.addEventListener("message", (event) => {
+                if (event.origin !== sqliteOnlineWebsite) {
+                  console.warn("Blocked message from unauthorized origin:", event.origin)
+                  return
+                }
+
+                if (event.data.type === "loadDatabaseBufferReady" && !isDatabaseLoaded) {
+                  getDBFile()
+                    .then(async (blob) => {
+                      const arrayBuffer = await blob.arrayBuffer()
+
+                      win.postMessage(
+                        {
+                          type: "invokeLoadDatabaseBuffer",
+                          buffer: arrayBuffer,
+                        },
+                        sqliteOnlineWebsite,
+                      )
+                      isDatabaseLoaded = true
+                    })
+                    .catch((error) => {
+                      console.error("Failed to load database file into SQLite Online", error)
+                    })
+                }
+              })
             }}
           />
         </div>