From 96507dfee37f5f4daaa03f74498dfb7d7d340462 Mon Sep 17 00:00:00 2001
From: Gravit <gravit.min@ya.ru>
Date: Mon, 17 Sep 2018 16:30:19 +0700
Subject: [PATCH] =?UTF-8?q?WebSocket=20=D0=BD=D0=B0=D1=87=D0=B0=D0=BB?=
 =?UTF-8?q?=D0=BE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 libLauncher/build.gradle                      |  1 +
 .../utils/websocket/WebSocketClient.java      | 87 +++++++++++++++++++
 2 files changed, 88 insertions(+)
 create mode 100644 libLauncher/src/main/java/ru/gravit/utils/websocket/WebSocketClient.java

diff --git a/libLauncher/build.gradle b/libLauncher/build.gradle
index 6a5c5d66..e8006282 100644
--- a/libLauncher/build.gradle
+++ b/libLauncher/build.gradle
@@ -6,4 +6,5 @@
 	compileOnly 'org.javassist:javassist:3.23.1-GA'
 	compile files('../compat/authlib/authlib-clean.jar')
 	compile 'com.eclipsesource.minimal-json:minimal-json:0.9.4'
+	compile group: 'org.eclipse.jetty.websocket', name: 'websocket-client', version: '9.4.12.v20180830'
 }
diff --git a/libLauncher/src/main/java/ru/gravit/utils/websocket/WebSocketClient.java b/libLauncher/src/main/java/ru/gravit/utils/websocket/WebSocketClient.java
new file mode 100644
index 00000000..0ebcbf14
--- /dev/null
+++ b/libLauncher/src/main/java/ru/gravit/utils/websocket/WebSocketClient.java
@@ -0,0 +1,87 @@
+package ru.gravit.utils.websocket;
+
+import java.io.ByteArrayInputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.nio.ByteBuffer;
+import java.util.concurrent.*;
+
+import jdk.nashorn.internal.ir.RuntimeNode;
+import org.eclipse.jetty.websocket.api.Session;
+import org.eclipse.jetty.websocket.api.StatusCode;
+import org.eclipse.jetty.websocket.api.annotations.OnWebSocketClose;
+import org.eclipse.jetty.websocket.api.annotations.OnWebSocketConnect;
+import org.eclipse.jetty.websocket.api.annotations.OnWebSocketMessage;
+import org.eclipse.jetty.websocket.api.annotations.WebSocket;
+import ru.gravit.launcher.Launcher;
+import ru.gravit.launcher.serialize.HInput;
+import ru.gravit.utils.helper.IOHelper;
+
+/**
+ * Basic Echo Client Socket
+ */
+@WebSocket(maxTextMessageSize = 64 * 1024)
+public class WebSocketClient
+{
+    private final CountDownLatch closeLatch;
+    @SuppressWarnings("unused")
+    private Session session;
+
+    public  WebSocketClient()
+    {
+        this.closeLatch = new CountDownLatch(1);
+    }
+
+    public boolean awaitClose(int duration, TimeUnit unit) throws InterruptedException
+    {
+        return this.closeLatch.await(duration,unit);
+    }
+
+    @OnWebSocketClose
+    public void onClose(int statusCode, String reason)
+    {
+        System.out.printf("Connection closed: %d - %s%n",statusCode,reason);
+        this.session = null;
+        this.closeLatch.countDown(); // trigger latch
+    }
+
+    @OnWebSocketConnect
+    public void onConnect(Session session)
+    {
+        System.out.printf("Got connect: %s%n",session);
+        this.session = session;
+        try
+        {
+            Future<Void> fut;
+            fut = session.getRemote().sendStringByFuture("Hello");
+            fut.get(2,TimeUnit.SECONDS); // wait for send to complete.
+
+            fut = session.getRemote().sendStringByFuture("Thanks for the conversation.");
+            fut.get(2,TimeUnit.SECONDS); // wait for send to complete.
+
+            session.close(StatusCode.NORMAL,"I'm done");
+        }
+        catch (Throwable t)
+        {
+            t.printStackTrace();
+        }
+    }
+    public void request(ByteBuffer buffer) throws InterruptedException, ExecutionException, TimeoutException {
+        Future<Void> fut;
+        fut = session.getRemote().sendBytesByFuture(buffer);
+        fut.get(2,TimeUnit.SECONDS);
+    }
+    @OnWebSocketMessage
+    public void onMessage(String msg) throws IOException {
+        byte[] bytes = msg.getBytes();
+        InputStream stream = new ByteArrayInputStream(bytes);
+        HInput input = new HInput(stream);
+        long handshake = input.readLong();
+        if(handshake != Launcher.PROTOCOL_MAGIC)
+        {
+            throw new IOException("INVALID HANDSHAKE");
+        }
+        int type = input.readInt();
+        System.out.printf("Got msg: %l %d",handshake,type);
+    }
+}