feat: Implement WsRpcClient and support all required rpc methods.

src/main/java/com/limechain/Main.java

@@ -4,6 +4,8 @@
 import com.limechain.client.LightClient;
 import com.limechain.rpc.RPCFunction;
 import com.limechain.rpc.RpcClient;
+import com.limechain.rpc.WsRpcClient;
+import com.limechain.rpc.WsRpcClientImpl;
 import com.limechain.rpc.server.RpcApp;
 import com.limechain.utils.DivLogger;
 import org.teavm.jso.JSBody;
@@ -13,13 +15,16 @@
 
 public class Main {
 
-    private static final String RPC_VARIABLE_NAME = "rpc";
+    private static final String HTTP_RPC = "rpc";
+    private static final String WS_RPC = "wsRpc";
 
     private static final DivLogger log = new DivLogger();
 
     public static void main(String[] args) {
+        exportHttpRpc(RpcClient::sendRpcRequest, JSString.valueOf(HTTP_RPC));
+        exportWsRpc(new WsRpcClientImpl(), JSString.valueOf(WS_RPC));
+
         log.log("Starting LimeChain node...");
-        exportAPI(RpcClient::sendRpcRequest, JSString.valueOf(RPC_VARIABLE_NAME));
 
         RpcApp rpcApp = new RpcApp();
         rpcApp.start();
@@ -33,6 +38,9 @@ public static void main(String[] args) {
     }
 
     @JSBody(params = {"f", "apiName"}, script = "window[apiName] = f;" +
-        "window.fruzhin.HTTP.changeRpcExported(true);")
-    private static native void exportAPI(RPCFunction f, JSString apiName);
+            "window.fruzhin.HTTP.changeRpcExported(true);")
+    private static native void exportHttpRpc(RPCFunction f, JSString apiName);
+
+    @JSBody(params = {"c", "apiName"}, script = "window[apiName] = c;")
+    private static native void exportWsRpc(WsRpcClient c, JSString apiName);
 }

src/main/java/com/limechain/rpc/RpcClient.java

@@ -34,7 +34,7 @@ public sealed class RpcClient permits ChainRpcClient, GrandpaRpcClient {
      */
     public static String sendRpcRequest(String method, Object[] params) {
         return HttpRequest.createHttpRequest(POST, LOAD_BALANCER.getNextEndpoint(),
-            createRpcRequestJson(method, List.of(params)));
+                createRpcRequestJson(method, List.of(params)));
     }
 
     /**
@@ -46,7 +46,7 @@ public static String sendRpcRequest(String method, Object[] params) {
      */
     protected static RpcResponse sendRpcRequest(RpcMethod method, List<Object> params) {
         String jsonResult = HttpRequest.createHttpRequest(POST, LOAD_BALANCER.getNextEndpoint(),
-            createRpcRequestJson(method.getMethod(), params));
+                createRpcRequestJson(method.getMethod(), params));
         return OBJECT_MAPPER.mapToClass(jsonResult, RpcResponse.class);
     }
 
@@ -67,7 +67,7 @@ private static String createRpcRequestJson(String method, List<Object> params) {
     protected static <T> T getResult(RpcResponse response, Class<T> klazz) {
         if (response.getError() != null) {
             throw new IllegalStateException("RPC request resulted in an error with code:" + response.getError().getCode()
-                + " and message:" + response.getError().getMessage());
+                    + " and message:" + response.getError().getMessage());
         }
 
         return OBJECT_MAPPER.mapToClass(JsonUtil.stringify(response.getResult()), klazz);

src/main/java/com/limechain/rpc/WsRpcClient.java

@@ -0,0 +1,14 @@
+package com.limechain.rpc;
+
+import org.teavm.jso.JSObject;
+import org.teavm.jso.core.JSString;
+
+/**
+ * TeaVM overlay interface for a client used to communicate with a full node's RPC server.
+ */
+public interface WsRpcClient extends JSObject {
+
+    void send(JSString rpcString);
+
+    String nextResponse();
+}

src/main/java/com/limechain/rpc/WsRpcClientImpl.java

@@ -0,0 +1,84 @@
+package com.limechain.rpc;
+
+import com.limechain.constants.RpcConstants;
+import lombok.SneakyThrows;
+import org.teavm.jso.browser.Window;
+import org.teavm.jso.core.JSString;
+import org.teavm.jso.websocket.WebSocket;
+
+import java.util.ArrayDeque;
+
+/**
+ * The implementation of {@link WsRpcClient}. Uses a native JS Websocket implementation.
+ */
+public class WsRpcClientImpl implements WsRpcClient {
+
+    private WebSocket ws;
+    private final ArrayDeque<String> rpcResponses;
+
+    public WsRpcClientImpl() {
+        rpcResponses = new ArrayDeque<>();
+        openWebsocketConnection();
+    }
+
+    private void openWebsocketConnection() {
+        System.out.println("Initializing RPC websocket connection...");
+        //TODO change when configuring chain.
+        ws = new WebSocket(RpcConstants.POLKADOT_WS_RPC);
+        initHandlers();
+    }
+
+    private void initHandlers() {
+        ws.onClose(e -> {
+            System.out.println("RPC websocket connection was closed.");
+            System.out.println("Retrying connection...");
+            Window.setTimeout(this::openWebsocketConnection, 1000);
+        });
+
+        ws.onError(e -> {
+            System.out.println("There was an error in the RPC websocket connection. Closing connection...");
+            ws.close();
+        });
+
+        ws.onOpen(e -> System.out.println("Websocket connection is open."));
+        ws.onMessage(e -> rpcResponses.offerLast(e.getDataAsString()));
+    }
+
+    /**
+     * Waits for the current ws connection to be in an opened state then sends an RPC request to the full node.
+     */
+    @Override
+    public void send(JSString rpcString) {
+        new Thread(() -> {
+            handleSocketState();
+            ws.send(rpcString.stringValue());
+        }).start();
+    }
+
+    /**
+     * Handles the state of the websocket when sending a message. If the connection is in a closing (2) or a closed (3)
+     * state the client throws an error.
+     */
+    @SneakyThrows
+    private void handleSocketState() {
+        var startState = ws.getReadyState();
+
+        while (startState != 1) {
+            var currentState = ws.getReadyState();
+            if (currentState > 1) {
+                throw new Exception("Calling function of a closed websocket is prohibited.");
+            }
+
+            startState = currentState;
+            Thread.sleep(50);
+        }
+    }
+
+    /**
+     * Polls the first item in the queue and returns it as a string.
+     */
+    @Override
+    public String nextResponse() {
+        return rpcResponses.poll();
+    }
+}

src/main/java/com/limechain/utils/json/JsonParser.java

@@ -159,10 +159,10 @@ private Object parseNumber() {
 
     private boolean isValidNumberChar() {
         return (Character.isDigit(json.charAt(index))
-            || json.charAt(index) == '-'
-            || json.charAt(index) == '.'
-            || json.charAt(index) == 'e'
-            || json.charAt(index) == 'E');
+                || json.charAt(index) == '-'
+                || json.charAt(index) == '.'
+                || json.charAt(index) == 'e'
+                || json.charAt(index) == 'E');
     }
 
     private void skipWhitespace() {