Getting somewhere lol

network/network.go

@@ -3,12 +3,12 @@ package network
 import (
 	"bufio"
 	"encoding/binary"
+	"fmt"
 	"io"
 	"log"
 	"net"
 	"time"
 
-	"gitea.boner.be/bdnugget/goonscape/game"
 	"gitea.boner.be/bdnugget/goonscape/types"
 	pb "gitea.boner.be/bdnugget/goonserver/actions"
 	"google.golang.org/protobuf/proto"
@@ -20,39 +20,62 @@ func SetServerAddr(addr string) {
 	serverAddr = addr
 }
 
-func ConnectToServer() (net.Conn, int32, error) {
+func ConnectToServer(username, password string, isRegistering bool) (net.Conn, int32, error) {
 	conn, err := net.Dial("tcp", serverAddr)
 	if err != nil {
 		log.Printf("Failed to dial server: %v", err)
 		return nil, 0, err
 	}
 
-	log.Println("Connected to server. Waiting for player ID...")
-	reader := bufio.NewReader(conn)
+	log.Println("Connected to server. Authenticating...")
 
-	// Read message length (4 bytes)
+	// Send auth message
+	authAction := &pb.Action{
+		Type:     pb.Action_LOGIN,
+		Username: username,
+		Password: password,
+	}
+	if isRegistering {
+		authAction.Type = pb.Action_REGISTER
+	}
+
+	authBatch := &pb.ActionBatch{
+		Actions: []*pb.Action{authAction},
+	}
+
+	if err := writeMessage(conn, authBatch); err != nil {
+		conn.Close()
+		return nil, 0, fmt.Errorf("failed to send auth: %v", err)
+	}
+
+	// Read server response
+	reader := bufio.NewReader(conn)
 	lengthBuf := make([]byte, 4)
 	if _, err := io.ReadFull(reader, lengthBuf); err != nil {
-		log.Printf("Failed to read message length: %v", err)
-		return nil, 0, err
+		conn.Close()
+		return nil, 0, fmt.Errorf("failed to read auth response: %v", err)
 	}
 	messageLength := binary.BigEndian.Uint32(lengthBuf)
 
-	// Read the full message
 	messageBuf := make([]byte, messageLength)
 	if _, err := io.ReadFull(reader, messageBuf); err != nil {
-		log.Printf("Failed to read message body: %v", err)
-		return nil, 0, err
+		conn.Close()
+		return nil, 0, fmt.Errorf("failed to read auth response body: %v", err)
 	}
 
 	var response pb.ServerMessage
 	if err := proto.Unmarshal(messageBuf, &response); err != nil {
-		log.Printf("Failed to unmarshal server response: %v", err)
-		return nil, 0, err
+		conn.Close()
+		return nil, 0, fmt.Errorf("failed to unmarshal auth response: %v", err)
+	}
+
+	if !response.AuthSuccess {
+		conn.Close()
+		return nil, 0, fmt.Errorf(response.ErrorMessage)
 	}
 
 	playerID := response.GetPlayerId()
-	log.Printf("Successfully connected with player ID: %d", playerID)
+	log.Printf("Successfully authenticated with player ID: %d", playerID)
 	return conn, playerID, nil
 }
 
@@ -172,8 +195,8 @@ func HandleServerCommunication(conn net.Conn, playerID int32, player *types.Play
 				}
 			}
 
-			if g, ok := player.UserData.(*game.Game); ok && len(serverMessage.ChatMessages) > 0 {
-				g.Chat.HandleServerMessages(serverMessage.ChatMessages)
+			if handler, ok := player.UserData.(types.ChatMessageHandler); ok && len(serverMessage.ChatMessages) > 0 {
+				handler.HandleServerMessages(serverMessage.ChatMessages)
 			}
 		}
 	}