fix: ensure wasm loaded (#6)

* fix: ensure wasm loaded

* chore: remove withWasm function

* fix: use correct await when initialization

* fix: catch error for initialization

Author: dadamu

packages/sdk/src/index.ts

@@ -12,7 +12,7 @@ import {
   Transaction,
   payToAddressScript,
 } from "./wasm/kaspa";
-import { rpcClient, watchBalanceChanged } from "./rpc-client";
+import { rpcClient, watchBalanceChanged, wasmReady } from "./rpc-client";
 import { IWalletEventHandler, NetworkId } from "./interfaces";
 import { listeners, ListenerMethod } from "./listener";
 import { sleep } from "./utils";
@@ -583,4 +583,29 @@ export const removeEventListener = (
   listeners[method].delete(handler);
 };
 
+// ------------------------------------------------------------------------
+// --------------------------WASM Utilities--------------------------------
+// ------------------------------------------------------------------------
+
+/**
+ * Promise that resolves when the Kaspa WASM module is fully loaded and initialized
+ * @example
+ * ```typescript
+ * await wasmReady;
+ * const tx = kaspaWasm.createTransaction(...);
+ * ```
+ */
+export { wasmReady } from "./rpc-client";
+
+/**
+ * The Kaspa WASM module
+ * Note: Ensure wasmReady Promise is resolved before using this
+ * @example
+ * ```typescript
+ * import { kaspaWasm, wasmReady } from 'kastle-sdk';
+ *
+ * await wasmReady;
+ * const address = kaspaWasm.addressFromScriptPublicKey(...);
+ * ```
+ */
 export const kaspaWasm = wasm;

packages/sdk/src/rpc-client.ts

@@ -3,9 +3,21 @@ import { config } from "./config";
 import { getNetwork } from "./index";
 import { sleep } from "./utils";
 
+// Initialize WASM module independently
+let wasmInitialized = false;
+export const wasmReady: Promise<void> = (async () => {
+  if (!wasmInitialized) {
+    await init(config.wasm);
+    wasmInitialized = true;
+  }
+})();
+
 export let rpcClient: RpcClient | undefined;
 
 export const connectToRPC = async () => {
+  // Ensure WASM is initialized before connecting to RPC
+  await wasmReady;
+
   if (rpcClient?.isConnected) {
     rpcClient.removeAllEventListeners();
     await rpcClient.disconnect();
@@ -58,5 +70,5 @@ export const watchBalanceChanged = async (address: string | null) => {
 };
 
 (async () => {
-  await init(config.wasm).then(connectToRPC);
+  await wasmReady.then(connectToRPC).catch(console.error);
 })();