Add shared memory support to the Swift runtime

Author: kateinoigakukun

Plugins/PackageToJS/Templates/instantiate.d.ts

@@ -1,3 +1,7 @@
+/* #if USE_SHARED_MEMORY */
+import type { SwiftRuntimeThreadChannel } from "./runtime.js";
+/* #endif */
+
 export type Import = {
     // TODO: Generate type from imported .d.ts files
 }
@@ -65,6 +69,7 @@ export declare function instantiate(
         /* #endif */
         /* #if USE_SHARED_MEMORY */
         memory: WebAssembly.Memory
+        threadChannel: SwiftRuntimeThreadChannel
         /* #endif */
     }
 ): Promise<{

Plugins/PackageToJS/Templates/instantiate.js

@@ -27,18 +27,24 @@ export async function instantiate(
     const { wasi } = options;
     /* #endif */
     const instantiator = await createInstantiator(imports, options);
-    const swift = new SwiftRuntime();
+    const swift = new SwiftRuntime({
+        /* #if USE_SHARED_MEMORY */
+        sharedMemory: true,
+        threadChannel: options.threadChannel,
+        /* #endif */
+    });
 
     /** @type {WebAssembly.Imports} */
     const importObject = {
         javascript_kit: swift.wasmImports,
-        /* #if IS_WASI */
+/* #if IS_WASI */
         wasi_snapshot_preview1: wasi.wasiImport,
-        /* #if USE_SHARED_MEMORY */ env: {
+/* #if USE_SHARED_MEMORY */
+        env: {
             memory: options.memory,
         },
-        /* #endif */
-        /* #endif */
+/* #endif */
+/* #endif */
     };
     instantiator.addImports(importObject);
 

Plugins/PackageToJS/Templates/test.js

@@ -37,13 +37,23 @@ Please ensure you are using Node.js v18.x or newer.
             returnOnExit: false,
         })
         const dirname = path.dirname(fileURLToPath(import.meta.url))
+        const module = await WebAssembly.compile(await readFile(path.join(dirname, MODULE_PATH)))
+        const options = { wasi }
+        /* #if USE_SHARED_MEMORY */
+        // @ts-ignore
+        const memoryType = WebAssembly.Module.imports(module).find(i => i.module === "env" && i.name === "memory")?.type
+        const memory = new WebAssembly.Memory({
+            initial: memoryType.minimum,
+            maximum: memoryType.maximum,
+            shared: memoryType.shared,
+        })
+        options.memory = memory
+        /* #endif */
         const { swift } = await instantiate(
-            await readFile(path.join(dirname, MODULE_PATH)),
-            {}, {
-                wasi,
-                /* #if USE_SHARED_MEMORY */
-                /* #endif */
-            }
+            module,
+            {},
+            // @ts-ignore
+            options
         )
         swift.main()
     }