From bdd2c0f91dd652a0f5e11a4c4cf1d8da991f61d3 Mon Sep 17 00:00:00 2001
From: hornet <hornetfighter515@protonmail.com>
Date: Tue, 19 Oct 2021 14:08:05 -0400
Subject: [PATCH] Writing test websocket client

---
 src/client/cli.py      | 34 +++++++++++++++++++++++++++++++++
 src/client/test-cli.py | 43 ++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 77 insertions(+)
 create mode 100644 src/client/cli.py
 create mode 100644 src/client/test-cli.py

diff --git a/src/client/cli.py b/src/client/cli.py
new file mode 100644
index 0000000..02a3d95
--- /dev/null
+++ b/src/client/cli.py
@@ -0,0 +1,34 @@
+import websocket
+import _thread
+import time
+import ssl
+
+def on_message(ws, message):
+	print(f"<< {message}")
+
+def on_error(ws, error):
+	print(error)
+
+def on_close(ws, close_status_code, close_msg):
+	print(f"### closed. reason: {close_msg} ###")
+
+def on_open(ws):
+	def run(*args):
+		running = True
+		while running:
+			outbound = input(">")
+			if outbound == 'q':
+				running = False
+			else:
+				ws.send(outbound)
+		ws.close()
+	_thread.start_new_thread(run, ())
+
+if __name__ == "__main__":
+	websocket.enableTrace(True)
+	ws = websocket.WebSocketApp("wss://hornetfighter.com:9443",
+		on_open=on_open,
+		on_message=on_message,
+		on_error=on_error,
+		on_close=on_close)
+	ws.run_forever(sslopt={"cert_reqs": ssl.CERT_NONE})
diff --git a/src/client/test-cli.py b/src/client/test-cli.py
new file mode 100644
index 0000000..02daae0
--- /dev/null
+++ b/src/client/test-cli.py
@@ -0,0 +1,43 @@
+import websocket
+import _thread
+import ssl
+import json
+
+def on_message(ws, message):
+	print(f"<< {message}")
+
+def on_error(ws, error):
+	print(error)
+
+def on_close(ws, close_status_code, close_msg):
+	print(f"### closed. reason: {close_msg} ###")
+
+def on_open(ws):
+	def run(*args):
+		running = True
+		uname = input('input username:')
+		msg = {
+			"user":uname,
+			"content":None
+		}
+		ws.send(json.dumps(msg))
+		while running:
+			outbound = input(">")
+			if outbound == 'q':
+				running = False
+			else:
+				msg = {
+					"user":uname,
+					"content":outbound
+				}
+				ws.send(json.dumps(msg))
+		ws.close()
+	_thread.start_new_thread(run, ())
+
+if __name__ == "__main__":
+	ws = websocket.WebSocketApp("wss://hornetfighter.com:9443",
+		on_open=on_open,
+		on_message=on_message,
+		on_error=on_error,
+		on_close=on_close)
+	ws.run_forever(sslopt={"cert_reqs": ssl.CERT_NONE})