fix(website): fix WASM loading race condition and format parsing

Two bugs prevented the playground from working:
1. Race condition: go.run() is async but code immediately tried to
   use registered functions. Now polls for gosqlxParse availability.
2. Format function returns raw SQL string, not JSON. callAndParse
   now falls back to returning raw string if JSON.parse fails.

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

Author: 2 people

website/src/components/WasmLoader.tsx

@@ -32,11 +32,18 @@ export interface UseWasmResult {
 
 function callAndParse(fn: (sql: string, dialect?: string) => string, sql: string, dialect?: string): unknown {
   const raw = dialect ? fn(sql, dialect) : fn(sql);
-  const result = JSON.parse(raw);
-  if (result && typeof result === "object" && "error" in result && result.error) {
-    throw new Error(result.error);
+  // Try to parse as JSON first; if it fails, return as raw string
+  // (e.g., format() returns plain SQL text, not JSON)
+  try {
+    const result = JSON.parse(raw);
+    if (result && typeof result === "object" && "error" in result && result.error) {
+      throw new Error(result.error);
+    }
+    return result;
+  } catch {
+    // Not JSON — return as-is (raw string result)
+    return raw;
   }
-  return result;
 }
 
 function loadScript(src: string): Promise<void> {
@@ -70,9 +77,24 @@ export async function initWasm(): Promise<GoSQLXApi> {
     const wasmPath = base + "wasm/gosqlx.wasm";
     const result = await WebAssembly.instantiateStreaming(fetch(wasmPath), go.importObject);
 
-    // Run the Go program (registers global functions)
+    // Run the Go program (registers global functions).
+    // go.run() never resolves (Go blocks with select{}), so don't await it.
+    // The global functions are registered synchronously in main(), but we
+    // need to yield to let the Go runtime initialize.
     go.run(result.instance);
 
+    // Wait for the Go runtime to register the global functions
+    await new Promise<void>((resolve) => {
+      const check = () => {
+        if (typeof window.gosqlxParse === "function") {
+          resolve();
+        } else {
+          setTimeout(check, 10);
+        }
+      };
+      check();
+    });
+
     const api: GoSQLXApi = {
       parse(sql: string, dialect?: string) {
         return callAndParse(window.gosqlxParse, sql, dialect);