fix

Reviewed-on: #5

main.go

@@ -4,6 +4,8 @@ import (
 	"fmt"
 	"log"
 	"net/http"
+	"server/src/controllers"
+	"server/src/models"
 
 	"github.com/gorilla/websocket"
 )
@@ -12,20 +14,23 @@ var upgrader = websocket.Upgrader{
 	CheckOrigin: func(r *http.Request) bool { return true },
 }
 
-func ServeWS(room *Room, w http.ResponseWriter, r *http.Request) {
+func ServeWS(room *controllers.Room, w http.ResponseWriter, r *http.Request) {
 	conn, err := upgrader.Upgrade(w, r, nil)
 	if err != nil {
+		log.Println("Failed to upgrade connection:", err)
 		return
 	}
 
-	var auth AuthMessage
+	var auth models.AuthMessage
 	if err := conn.ReadJSON(&auth); err != nil || auth.Type != "auth" {
+		log.Println("Invalid auth message. Connection closed!")
 		conn.Close()
 		return
 	}
 
-	data, err := VerifyTelegramInitData(auth.InitData, "7697757472:AAESD9HfkWwbIZe-HXR7IazUShr69hZTLmE")
+	data, err := controllers.VerifyTelegramInitData(auth.InitData, "7697757472:AAESD9HfkWwbIZe-HXR7IazUShr69hZTLmE")
 	if err != nil {
+		log.Println("initData is empty. Connection closed!")
 		conn.Close()
 		return
 	}
@@ -39,7 +44,7 @@ func ServeWS(room *Room, w http.ResponseWriter, r *http.Request) {
 
 	log.Printf("Player %d connected!", userID)
 
-	player := &Player{
+	player := &models.Player{
 		ID:       userID,
 		Username: username,
 		Conn:     conn,
@@ -59,26 +64,32 @@ func ServeWS(room *Room, w http.ResponseWriter, r *http.Request) {
 	go readLoop(room, player)
 }
 
-func readLoop(room *Room, player *Player) {
+func readLoop(room *controllers.Room, player *models.Player) {
 	defer func() {
+		log.Printf("Player %d disconnected!", player.ID)
 		delete(room.Players, player.ID)
 		player.Conn.Close()
 	}()
 
 	for {
-		var msg InputMessage
+		var msg models.StateMessage
 		err := player.Conn.ReadJSON(&msg)
 		if err != nil {
 			return
 		}
 
-		msg.PlayerID = player.ID
+		switch msg.Type {
-		room.Input <- msg
+		case models.InputMsgType:
+			var input models.InputMessage = msg.Payload.(models.InputMessage)
+			player.X = input.DX
+			player.Y = input.DY
+			room.Input <- input
+		}
 	}
 }
 
 func main() {
-	room := NewRoom()
+	room := controllers.NewRoom()
 
 	go room.Run()
 

protocol.go

@@ -1,12 +0,0 @@
-package main
-
-type InputMessage struct {
-	PlayerID int64
-	DX       float64 `json:"dx"`
-	DY       float64 `json:"dy"`
-}
-
-type StateMessage struct {
-	Type    string `json:"type"`
-	Payload any    `json:"payload"`
-}

src/controllers/room.go

@@ -1,25 +1,25 @@
-package main
+package controllers
 
 import (
+	"server/src/models"
 	"sync"
 	"time"
 )
 
 type Room struct {
-	Players map[int64]*Player
+	Players map[int64]*models.Player
-	Input   chan InputMessage
+	Input   chan models.InputMessage
 	mu      sync.Mutex
 }
 
 func NewRoom() *Room {
 	return &Room{
-		Players: make(map[int64]*Player),
+		Players: make(map[int64]*models.Player),
-		Input:   make(chan InputMessage, 128),
+		Input:   make(chan models.InputMessage, 128),
 	}
 }
 
 func (r *Room) update() {
-	// 1️⃣ обрабатываем input
 	for {
 		select {
 		case input := <-r.Input:
@@ -59,8 +59,8 @@ func (r *Room) broadcast() {
 
 	r.mu.Unlock()
 
-	msg := StateMessage{
+	msg := models.StateMessage{
-		Type: "state",
+		Type: "input",
 		Payload: map[string]any{
 			"players": state,
 		},

src/controllers/telegram.go

@@ -1,4 +1,4 @@
-package main
+package controllers
 
 import (
 	"time"

src/models/auth.go

@@ -1,4 +1,4 @@
-package main
+package models
 
 type AuthMessage struct {
 	Type     string `json:"type"`

src/models/player.go

@@ -1,4 +1,4 @@
-package main
+package models
 
 import (
 	"github.com/gorilla/websocket"

src/models/protocol.go

@@ -0,0 +1,19 @@
+package models
+
+type MsgType string
+
+const (
+	InputMsgType MsgType = "input"
+	ChatMsgType  MsgType = "chat"
+)
+
+type InputMessage struct {
+	PlayerID int64
+	DX       float64 `json:"dx"`
+	DY       float64 `json:"dy"`
+}
+
+type StateMessage struct {
+	Type    MsgType `json:"type"`
+	Payload any     `json:"payload"`
+}