add onConnectionSetup event handler

src/guest.ts

@@ -1,19 +1,19 @@
 import { extractMethods, isWorker } from "./helpers";
 import { registerLocalMethods, registerRemoteMethods } from "./rpc";
-import { actions, events, IConnection, ISchema } from "./types";
+import { actions, EventHandlers, events, IConnection, ISchema } from "./types";
 
 const REQUEST_INTERVAL = 10;
 const TIMEOUT_INTERVAL = 3000;
 
 let interval: any = null;
 let connected = false;
 
-function connect(schema: ISchema = {}): Promise<IConnection> {
+function connect(schema: ISchema = {}, eventHandlers?: EventHandlers): Promise<IConnection> {
   return new Promise((resolve, reject) => {
     const localMethods = extractMethods(schema);
 
     // on handshake response
-    function handleHandshakeResponse(event: any) {
+    async function handleHandshakeResponse(event: any) {
       if (event.data.action !== actions.HANDSHAKE_REPLY) return;
 
       // register local methods
@@ -27,6 +27,8 @@ function connect(schema: ISchema = {}): Promise<IConnection> {
         event
       );
 
+      await eventHandlers?.onConnectionSetup?.(remote);
+
       // close the connection and all listeners when called
       const close = () => {
         self.removeEventListener(events.MESSAGE, handleHandshakeResponse);

src/host.ts

@@ -70,6 +70,7 @@ function connect(guest: HTMLIFrameElement | Worker, schema: ISchema = {}): Promi
         listeners.removeEventListener(events.MESSAGE, handleHandshake);
         unregisterRemote();
         unregisterLocal();
+        if (guestIsWorker) (guest as Worker).terminate();
       };
 
       // resolve connection object

src/types.ts

@@ -59,3 +59,7 @@ export interface IRPCResolvePayload {
   callName: string;
   connectionID: string;
 }
+
+export interface EventHandlers {
+  onConnectionSetup: (remote: ISchema) => Promise<void>;
+}