jazz-tools 0.19.21 → 0.20.0

package/src/worker/JazzMessageChannel.ts

@@ -0,0 +1,73 @@
+import { CojsonMessageChannel, type Peer } from "cojson";
+import type {
+  WaitForConnectionOptions,
+  ExposeOptions,
+  PostMessageTarget,
+} from "cojson/src/CojsonMessageChannel/types.js";
+import { Account, AnonymousJazzAgent } from "jazz-tools";
+import { activeAccountContext } from "../tools/implementation/activeAccountContext.js";
+
+/**
+ * Options for JazzMessageChannel.expose()
+ */
+export interface JazzMessageChannelExposeOptions extends ExposeOptions {
+  /**
+   * The account or anonymous agent to use for the connection.
+   * If not provided, falls back to the active account context.
+   */
+  loadAs?: Account | AnonymousJazzAgent;
+}
+
+/**
+ * JazzMessageChannel provides a high-level API for creating Jazz connections
+ * via the MessageChannel API. It wraps cojson's CojsonMessageChannel and
+ * automatically manages the node connection.
+ */
+export class JazzMessageChannel {
+  /**
+   * Expose a Jazz connection to a target.
+   * This is the host-side API, typically called from the main thread.
+   *
+   * @param target - Any object with a postMessage method (Worker, Window, etc.)
+   * @param opts - Configuration options including the account to use
+   * @returns A promise that resolves once the connection is established
+   */
+  static async expose(
+    target: PostMessageTarget,
+    opts: JazzMessageChannelExposeOptions = {},
+  ): Promise<void> {
+    const { loadAs, ...cojsonOpts } = opts;
+
+    // Get account from loadAs or fall back to active account context
+    const accountOrAgent = loadAs ?? activeAccountContext.maybeGet();
+
+    if (!accountOrAgent) {
+      throw new Error(
+        "No account provided and no active account context available",
+      );
+    }
+
+    const node =
+      accountOrAgent instanceof AnonymousJazzAgent
+        ? accountOrAgent.node
+        : accountOrAgent.$jazz.localNode;
+
+    const peer = await CojsonMessageChannel.expose(target, cojsonOpts);
+
+    node.syncManager.addPeer(peer);
+  }
+
+  /**
+   * Accept an incoming Jazz connection.
+   * Same as cojson CojsonMessageChannel.waitForConnection().
+   */
+  static waitForConnection(opts?: WaitForConnectionOptions): Promise<Peer> {
+    return CojsonMessageChannel.waitForConnection(opts);
+  }
+}
+
+// Re-export types for convenience
+export type {
+  WaitForConnectionOptions,
+  AcceptFromPortOptions,
+} from "cojson/src/CojsonMessageChannel/types.js";

package/src/worker/edge-wasm.ts

@@ -1,5 +1,6 @@
 // InitWasm allow to load the wasm code in edge runtimes (ex. cloudflare worker and vercel edge functions)
-import { init as InitWasm } from "cojson/crypto/WasmCrypto/edge";
+import { init as InitWasm, initSync } from "cojson/crypto/WasmCrypto/edge";
 import { WasmCrypto } from "cojson/crypto/WasmCrypto";
 
 WasmCrypto.setInit(InitWasm);
+WasmCrypto.setInitSync(initSync);

package/src/worker/index.ts

@@ -29,6 +29,11 @@ type WorkerOptions<
 > = {
   accountID?: string;
   accountSecret?: string;
+  /**
+   * A peer to connect to for synchronization.
+   * If provided, syncServer is ignored.
+   */
+  peer?: Peer;
   syncServer?: string;
   WebSocket?: AnyWebSocketConstructor;
   AccountSchema?: S;
@@ -63,21 +68,28 @@ export async function startWorker<
 
   const peers: Peer[] = [];
 
-  const wsPeer = new WebSocketPeerWithReconnection({
-    peer: syncServer,
-    reconnectionTimeout: 100,
-    addPeer: (peer) => {
-      if (node) {
-        node.syncManager.addPeer(peer);
-      } else {
-        peers.push(peer);
-      }
-    },
-    removePeer: () => {},
-    WebSocketConstructor: options.WebSocket,
-  });
-
-  wsPeer.enable();
+  // If a peer is provided directly, use it instead of WebSocket
+  let wsPeer: WebSocketPeerWithReconnection | undefined;
+
+  if (options.peer) {
+    peers.push(options.peer);
+  } else {
+    wsPeer = new WebSocketPeerWithReconnection({
+      peer: syncServer,
+      reconnectionTimeout: 100,
+      addPeer: (peer) => {
+        if (node) {
+          node.syncManager.addPeer(peer);
+        } else {
+          peers.push(peer);
+        }
+      },
+      removePeer: () => {},
+      WebSocketConstructor: options.WebSocket,
+    });
+
+    wsPeer.enable();
+  }
 
   if (!accountID) {
     throw new Error("No accountID provided");
@@ -117,7 +129,7 @@ export async function startWorker<
   async function done() {
     await context.account.$jazz.waitForAllCoValuesSync();
 
-    wsPeer.disable();
+    wsPeer?.disable();
     context.done();
   }
 
@@ -146,11 +158,18 @@ export async function startWorker<
      * Wait for the connection to the sync server to be established.
      *
      * If already connected, it will resolve immediately.
+     * Returns immediately if using a custom peer.
      */
     waitForConnection() {
-      return wsPeer.waitUntilConnected();
+      return wsPeer?.waitUntilConnected() ?? Promise.resolve();
     },
     subscribeToConnectionChange(listener: (connected: boolean) => void) {
+      if (!wsPeer) {
+        // For custom peers, immediately notify as connected
+        listener(true);
+        return () => {};
+      }
+
       wsPeer.subscribe(listener);
 
       return () => {

package/src/worker/wasm.ts

@@ -0,0 +1 @@
+export { initWasmCrypto as initWasm } from "cojson/crypto/WasmCrypto";

package/tsup.config.ts

@@ -42,7 +42,6 @@ export default defineConfig([
     entry: {
       index: "src/expo/index.ts",
       testing: "src/expo/testing.ts",
-      crypto: "src/expo/crypto.ts",
       polyfills: "src/react-native-core/polyfills/index.js",
     },
     outDir: "dist/expo",
@@ -115,7 +114,6 @@ export default defineConfig([
     entry: {
       index: "src/react-native/index.ts",
       testing: "src/react-native/testing.ts",
-      crypto: "src/react-native/crypto.ts",
       polyfills: "src/react-native-core/polyfills/index.js",
     },
     outDir: "dist/react-native",
@@ -132,8 +130,6 @@ export default defineConfig([
     entry: {
       index: "src/react-native-core/index.ts",
       testing: "src/react-native-core/testing.tsx",
-      crypto: "src/react-native-core/crypto/index.ts",
-      "crypto/RNCrypto": "src/react-native-core/crypto/RNCrypto.ts",
     },
     outDir: "dist/react-native-core",
   },

package/dist/browser/storageOptions.d.ts

@@ -1,8 +0,0 @@
-type StorageOption = "indexedDB";
-type CombinedStorageOption = ["indexedDB"];
-export type StorageConfig = StorageOption | CombinedStorageOption | [StorageOption];
-export declare function getStorageOptions(storage?: StorageConfig): {
-    useIndexedDB: boolean;
-};
-export {};
-//# sourceMappingURL=storageOptions.d.ts.map

package/dist/browser/storageOptions.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"storageOptions.d.ts","sourceRoot":"","sources":["../../src/browser/storageOptions.ts"],"names":[],"mappings":"AAAA,KAAK,aAAa,GAAG,WAAW,CAAC;AACjC,KAAK,qBAAqB,GAAG,CAAC,WAAW,CAAC,CAAC;AAC3C,MAAM,MAAM,aAAa,GACrB,aAAa,GACb,qBAAqB,GACrB,CAAC,aAAa,CAAC,CAAC;AAEpB,wBAAgB,iBAAiB,CAAC,OAAO,CAAC,EAAE,aAAa,GAAG;IAC1D,YAAY,EAAE,OAAO,CAAC;CACvB,CAOA"}

package/dist/browser/tests/storageOptions.test.d.ts

@@ -1,2 +0,0 @@
-export {};
-//# sourceMappingURL=storageOptions.test.d.ts.map

package/dist/browser/tests/storageOptions.test.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"storageOptions.test.d.ts","sourceRoot":"","sources":["../../../src/browser/tests/storageOptions.test.ts"],"names":[],"mappings":""}

package/dist/chunk-M2HGBOXS.js.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../src/tools/ssr/ssr.ts"],"sourcesContent":["import { WebSocketPeerWithReconnection } from \"cojson-transport-ws\";\nimport { PureJSCrypto } from \"cojson/dist/crypto/PureJSCrypto\";\nimport { createAnonymousJazzContext } from \"jazz-tools\";\n\nexport function createSSRJazzAgent(opts: { peer: string }) {\n  const ssrNode = createAnonymousJazzContext({\n    crypto: new PureJSCrypto(),\n    peers: [],\n  });\n\n  const wsPeer = new WebSocketPeerWithReconnection({\n    peer: opts.peer,\n    reconnectionTimeout: 100,\n    addPeer: (peer) => {\n      ssrNode.agent.node.syncManager.addPeer(peer);\n    },\n    removePeer: () => {},\n  });\n\n  wsPeer.enable();\n\n  return ssrNode.agent;\n}\n"],"mappings":";AAAA,SAAS,qCAAqC;AAC9C,SAAS,oBAAoB;AAC7B,SAAS,kCAAkC;AAEpC,SAAS,mBAAmB,MAAwB;AACzD,QAAM,UAAU,2BAA2B;AAAA,IACzC,QAAQ,IAAI,aAAa;AAAA,IACzB,OAAO,CAAC;AAAA,EACV,CAAC;AAED,QAAM,SAAS,IAAI,8BAA8B;AAAA,IAC/C,MAAM,KAAK;AAAA,IACX,qBAAqB;AAAA,IACrB,SAAS,CAAC,SAAS;AACjB,cAAQ,MAAM,KAAK,YAAY,QAAQ,IAAI;AAAA,IAC7C;AAAA,IACA,YAAY,MAAM;AAAA,IAAC;AAAA,EACrB,CAAC;AAED,SAAO,OAAO;AAEd,SAAO,QAAQ;AACjB;","names":[]}