Add exit command

client/cmd/main.go

@@ -93,13 +93,14 @@ func listenForMessages(conn net.Conn) {
 	reader := bufio.NewReader(conn)
 	for {
 		message, err := reader.ReadString('\n')
-		if err != nil {
-			log.Fatalln("Server disconnected.")
-		}
 
 		// Clear the current input prompt
 		fmt.Print("\r\033[K") // This clears the line
 
+		if err != nil {
+			os.Exit(0)
+		}
+
 		// Print the new message
 		fmt.Print(FgGreen + Bold + message + Reset)
 

server/internals/chatserver/client.go

@@ -54,6 +54,7 @@ func (cs *ChatServer) HandleConnection(conn net.Conn) {
 
 	mode = strings.TrimSpace(mode)
 	log.Println("User", username, "entered mode:", mode)
+
 	switch mode {
 	case "global":
 		cs.handleGlobalChat(conn, reader, username)
@@ -75,7 +76,14 @@ func (cs *ChatServer) handleGlobalChat(conn net.Conn, reader *bufio.Reader, user
 			cs.RemoveClient(conn, username)
 			return
 		}
-		formattedMessage := utils.FormatChatMessage(username, strings.TrimSpace(message))
-		cs.BroadcastMessage(formattedMessage, conn)
+
+		trimmedMessage := strings.TrimSpace(message)
+		if strings.HasPrefix(trimmedMessage, "/") {
+			// Handle command
+			cs.HandleCommand(conn, trimmedMessage[1:])
+		} else {
+			formattedMessage := utils.FormatChatMessage(username, strings.TrimSpace(message))
+			cs.BroadcastMessage(formattedMessage, conn)
+		}
 	}
 }

server/internals/chatserver/commands.go

@@ -0,0 +1,20 @@
+package chatserver
+
+import (
+	"net"
+	"strings"
+)
+
+// HandleCommand processes commands for user
+func(cs *ChatServer) HandleCommand(conn net.Conn, command string) {
+	command = strings.TrimSpace(command)
+	args := strings.Split(command, " ")
+
+	switch args[0] {
+	case "exit":
+		cs.RemoveClient(conn, cs.globalClients[conn])
+		conn.Write([]byte("You have exited the server.\n"))
+		conn.Close()
+
+	}
+}