From 8ed59e901d1fbc077912b46e6690ce8ebf38a6c4 Mon Sep 17 00:00:00 2001
From: Toshit Chawda <r58playz@gmail.com>
Date: Sun, 7 Jul 2024 14:21:04 -0700
Subject: [PATCH] make set transport async for the user

---
 README.md     | 4 ++--
 src/worker.ts | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index 9f8ea42..6beb5f7 100644
--- a/README.md
+++ b/README.md
@@ -43,10 +43,10 @@ Here is an example of using bare-mux:
 import { BareMuxConnection } from "@mercuryworkshop/bare-mux";
 const conn = new BareMuxConnection("/bare-mux/worker.js");
 // Set Bare-Client transport
-await conn.setTransport(`return (async ()=>{
+await conn.setTransport(`
     const exports = await import("/bare-mux/index.js");
     return new exports.BareClient("https://tomp.app");
-})()`);
+`);
 
 /// As a proxy developer
 import { BareClient } from "@mercuryworkshop/bare-mux";
diff --git a/src/worker.ts b/src/worker.ts
index 8508d71..0a71f99 100644
--- a/src/worker.ts
+++ b/src/worker.ts
@@ -8,7 +8,7 @@ function handleConnection(port: MessagePort) {
 		const port = event.data.port;
 		const message: WorkerMessage = event.data.message;
 		if (message.type === "set") {
-			const func = new Function(message.client);
+			const func = new Function("return (async ()=>{" + message.client + "})()");
 			currentTransport = await func();
 			console.log("set transport to ", currentTransport);
 			port.postMessage(<WorkerResponse>{ type: "set" });