Update both server and client for websocket support

client-cli/src/main/kotlin/Main.kt

@@ -1,7 +1,9 @@
 package xyz.maxwellj.chookpen.client
 
-import fuel.Fuel
-import fuel.get
+import okhttp3.*
+import java.util.Scanner
+import kotlin.system.exitProcess
+
 import java.io.File
 
 import kotlin.system.exitProcess
@@ -14,7 +16,7 @@ fun md5(input:String): String {
     return BigInteger(1, md.digest(input.toByteArray())).toString(16).padStart(32, '0')
 }
 
-suspend fun main(args: Array<String>) {
+fun main() {
 	// Variables 
     var name = ""
     var server = ""
@@ -23,50 +25,6 @@ suspend fun main(args: Array<String>) {
     var password = ""
     var configFile = File("${System.getProperty("user.home")}/chookpen.profile")
     
-    if (!configFile.exists()) {
-        println("You don't have a Chookpen profile set up yet. If you've got a Chookpen profile, type 'l' to login and press enter. Otherwise, just press enter.")
-        val userInput = readln()
-        if (userInput == "l") {
-            println("Username:")
-            name = readln()
-            println("Password:")
-            password = readln()
-            val request = Fuel.get("http://localhost:7070/api/logintest/username:{$name}token:{${md5(password)}}").body.string()
-            if (request == "Invalid token") {
-                println("Invalid password. Please rerun the program and put in the right password")
-                exitProcess(1)
-            } else {
-                configFile.createNewFile()
-                configFile.writeText("$name:$password:localhost:7070:0")
-                println("Logged in! Run the command again to start talking!")
-                exitProcess(0)
-            }
-        } else {
-            println("Choose a username:")
-            val newName = readln()
-            if (newName == "") {
-                println("Please choose a username! Rerun the program and try again")
-                exitProcess(1)
-            println("Choose a password:")
-            val newPassword = readln()
-            if (newPassword == "") {
-                println("Please choose a password! Rerun the program and try again")
-                exitProcess(1)
-            }
-            val request = Fuel.get("http://localhost:7070/api/createAccount/username:{$newName}token:{${md5(newPassword)}}").body.string()
-            if (request == "That username already exists on the server! Please choose a different username") {
-                println("That username already exists on the server! Rerun the program and choose a different username")
-                exitProcess(1)
-            } else {
-                configFile.createNewFile()
-                configFile.writeText("$newName:$newPassword:localhost:7070:0")
-                println("Account created!")
-                exitProcess(0)
-            }
-        }
-    }
-}
-
     var configStage = 0
     for (char in configFile.readText()) {
         if (char == ':') {configStage ++}
@@ -88,22 +46,67 @@ suspend fun main(args: Array<String>) {
     hasHTTPS = hasHTTPS.replace(":", "")
     password = password.replace(":", "")
 
-    // Actual code
-    println("Chookpen Client initialised!")
-    println("Hello $name@$server")
-    val protocol = "http"
-    if (hasHTTPS == "1") {
-        val protocol = "https"
+    if (password == "x") {
+        println("Enter your password:")
+        password = readln()
     }
-    if (args.count() == 0) {
-        println(Fuel.get("$protocol://$server:$port/api/syncmessages/username:{$name}token:{${md5(password)}}").body.string())
-    } else {
-        val message = args[0]
-        val isSuccessful = Fuel.get("$protocol://$server:$port/api/send/username:{$name}token:{${md5(password)}}message:{$message}").body.string()
-        if (isSuccessful != "Success") {
-            println(isSuccessful)
+
+    val client = OkHttpClient.Builder()
+        //.pingInterval(30, TimeUnit.SECONDS)
+        .build()
+
+    val request = Request.Builder()
+        .url("ws://localhost:7070/api/websocket")
+        .build()
+
+    var webSocket: WebSocket? = null
+
+    val listener = object : WebSocketListener() {
+        override fun onOpen(webSocket: WebSocket, response: Response) {
+            println(password)
+            println(md5(password))
+            println("Connection opened")
+            webSocket.send("username:{$name}token:{${md5(password)}}message:{Joined the room}")
+        }
+
+        override fun onMessage(webSocket: WebSocket, text: String) {
+            println("$text")
+        }
+
+        override fun onClosing(webSocket: WebSocket, code: Int, reason: String) {
+            println("Connection closing: $reason")
+            webSocket.close(1000, null)
+        }
+
+        override fun onFailure(webSocket: WebSocket, t: Throwable, response: Response?) {
+            println("Connection failed: ${t.message}")
+        }
+    }
+
+    // Set up shutdown hook for Ctrl+C handling
+    Runtime.getRuntime().addShutdownHook(Thread {
+        println("\nShutting down gracefully...")
+        webSocket?.close(1000, "Client shutting down")
+        client.dispatcher.executorService.shutdown()
+    })
+
+    // Initialize WebSocket connection
+    webSocket = client.newWebSocket(request, listener)
+
+    // Set up input handling
+    val scanner = Scanner(System.`in`)
+    println("Type your messages (press Enter to send, Ctrl+C to quit):")
+    
+    while (true) {
+        try {
+            val input = scanner.nextLine()
+            if (input.isNotEmpty()) {
+                webSocket?.send("username:{$name}token:{${md5(password)}}message:{$input}")
+            }
+        } catch (e: Exception) {
+            // Handle any input-related exceptions
+            println("Error reading input: ${e.message}")
+            break
         }
-        println(Fuel.get("$protocol://$server:$port/api/syncmessages/username:{$name}token:{${md5(password)}}").body.string())
     }
-    exitProcess(0)
 }