feat: client struct

1 files changed, 24 insertions, 9 deletions

diff --git a/internal/client.go b/internal/client.go
index 164a29d..f1939f9 100644
--- a/internal/client.go
+++ b/internal/client.go
@@ -4,28 +4,43 @@ import (
 	"fmt"
 	"log"
 	"net"
+	"strings"
 )
 
-func HandleClientConnection(conn net.Conn) {
-	defer conn.Close()
-	log.Printf("New client connected from %s\n", conn.RemoteAddr())
+type Client struct {
+	conn net.Conn
+}
+
+func NewIRCClient(conn net.Conn) *Client {
+	c := Client{conn: conn}
+
+	return &c
+}
+
+func (c *Client) HandleConnection() {
+	defer c.conn.Close()
+	log.Printf("New client connected from %s\n", c.conn.RemoteAddr())
 
 	buffer := make([]byte, 512)
 
 	for {
-		n, err := conn.Read(buffer)
+		n, err := c.conn.Read(buffer)
 		if err != nil {
-			log.Printf("Error reading from client %s: %v\n", conn.RemoteAddr(), err)
+			log.Printf("Error reading from client %s: %v\n", c.conn.RemoteAddr(), err)
 			return
 		}
 
-		data := string(buffer[:n])
-		fmt.Printf("%s sent %s\n", conn.RemoteAddr(), data)
+		data := strings.TrimSpace(string(buffer[:n]))
+		fmt.Printf("%s sent %s\n", c.conn.RemoteAddr(), data)
 
-		_, err = conn.Write([]byte("you said " + data))
+		_, err = c.conn.Write([]byte("you said " + data))
 		if err != nil {
-			log.Printf("Error sending to client %s: %v", conn.RemoteAddr(), err)
+			log.Printf("Error sending to client %s: %v", c.conn.RemoteAddr(), err)
 			return
 		}
+
+		if data == "QUIT" {
+			break
+		}
 	}
 }