hypha-rpc 0.1.9 → 0.1.11

package/dist/hypha-rpc-websocket.js

@@ -243,9 +243,7 @@ class RPC extends _utils_js__WEBPACK_IMPORTED_MODULE_0__.MessageEmitter {
             });
           }
         } else {
-          console.log(
-            "Connection established", connectionInfo
-          );
+          console.log("Connection established", connectionInfo);
         }
         this._fire("connected", connectionInfo);
       };
@@ -388,11 +386,9 @@ class RPC extends _utils_js__WEBPACK_IMPORTED_MODULE_0__.MessageEmitter {
         Object.assign(main, extra.value);
       }
       this._fire(main["type"], main);
-    }
-    else if (typeof message === "object") {
+    } else if (typeof message === "object") {
       this._fire(message["type"], message);
-    }
-    else {
+    } else {
       throw new Error("Invalid message format");
     }
   }
@@ -797,9 +793,9 @@ class RPC extends _utils_js__WEBPACK_IMPORTED_MODULE_0__.MessageEmitter {
       typeof main_message === "object" && main_message.type,
       "Invalid message, must be an object with a `type` fields.",
     );
-    if(!main_message.to) {
+    if (!main_message.to) {
       this._fire(main_message.type, main_message);
-      return
+      return;
     }
     let message_package = (0,_msgpack_msgpack__WEBPACK_IMPORTED_MODULE_2__.encode)(main_message);
     if (extra_data) {
@@ -1118,7 +1114,18 @@ class RPC extends _utils_js__WEBPACK_IMPORTED_MODULE_0__.MessageEmitter {
         this._method_annotations.has(method) &&
         this._method_annotations.get(method).require_context
       ) {
+        // if args.length + 1 is less than the required number of arguments we will pad with undefined
+        // so we make sure the last argument is the context
+        if (args.length + 1 < method.length) {
+          for (let i = args.length; i < method.length - 1; i++) {
+            args.push(undefined);
+          }
+        }
         args.push(data.ctx);
+        (0,_utils_js__WEBPACK_IMPORTED_MODULE_0__.assert)(
+          args.length === method.length,
+          `Runtime Error: Invalid number of arguments for method ${method_name}, expected ${method.length} but got ${args.length}`,
+        );
       }
       // console.log("Executing method: " + method_name);
       if (data.promise) {
@@ -1989,7 +1996,8 @@ class WebRTCConnection {
     this._handle_connected = () => {};
     this.manager_id = null;
     this._data_channel.onopen = async () => {
-      this._handle_connected && this._handle_connected({channel: this._data_channel});
+      this._handle_connected &&
+        this._handle_connected({ channel: this._data_channel });
     };
     this._data_channel.onmessage = async (event) => {
       let data = event.data;
@@ -4301,7 +4309,7 @@ class WebsocketRPCConnection {
           console.log(
             `Successfully connected to the server, workspace: ${this.connection_info.workspace}, manager_id: ${this.manager_id}`,
           );
-          if(this.connection_info.announcement){
+          if (this.connection_info.announcement) {
             console.log(`${this.connection_info.announcement}`);
           }
           resolve(this.connection_info);
@@ -4311,8 +4319,15 @@ class WebsocketRPCConnection {
           reject(new Error(error));
           return;
         } else {
-          console.error("ConnectionAbortedError: Unexpected message received from the server:", data);
-          reject(new Error("ConnectionAbortedError: Unexpected message received from the server"));
+          console.error(
+            "ConnectionAbortedError: Unexpected message received from the server:",
+            data,
+          );
+          reject(
+            new Error(
+              "ConnectionAbortedError: Unexpected message received from the server",
+            ),
+          );
           return;
         }
       };
@@ -4376,7 +4391,7 @@ class WebsocketRPCConnection {
     ) {
       if ([1000, 1001].includes(event.code)) {
         console.info(
-          `Websocket connection closed (code: ${event.code}): ${event.reason}`  
+          `Websocket connection closed (code: ${event.code}): ${event.reason}`,
         );
         if (this._handle_disconnected) {
           this._handle_disconnected(event.reason);
@@ -4395,7 +4410,9 @@ class WebsocketRPCConnection {
               `Reconnecting to ${this._server_url.split("?")[0]} (attempt #${retry})`,
             );
             await this.open();
-            console.warn(`Successfully reconnected to server ${this._server_url}`); 
+            console.warn(
+              `Successfully reconnected to server ${this._server_url}`,
+            );
           } catch (e) {
             if (`${e}`.includes("ConnectionAbortedError")) {
               console.warn("Failed to reconnect, connection aborted:", e);