fix

Reviewed-on: #5

go.mod

@@ -2,4 +2,7 @@ module server
 
 go 1.25.0
 
-require github.com/gorilla/websocket v1.5.3
+require (
+	github.com/gorilla/websocket v1.5.3
+	github.com/telegram-mini-apps/init-data-golang v1.5.0
+)

go.sum

@@ -1,2 +1,4 @@
 github.com/gorilla/websocket v1.5.3 h1:saDtZ6Pbx/0u+bgYQ3q96pZgCzfhKXGPqt7kZ72aNNg=
 github.com/gorilla/websocket v1.5.3/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
+github.com/telegram-mini-apps/init-data-golang v1.5.0 h1:rtpsmQ/nihkicPvnrdRXmHHtTnPvG1FmxMRZJwMKPz0=
+github.com/telegram-mini-apps/init-data-golang v1.5.0/go.mod h1:GG4HnRx9ocjD4MjjzOw7gf9Ptm0NvFbDr5xqnfFOYuY=

main.go

@@ -1,8 +1,11 @@
 package main
 
 import (
+	"fmt"
 	"log"
 	"net/http"
+	"server/src/controllers"
+	"server/src/models"
 
 	"github.com/gorilla/websocket"
 )
@@ -11,37 +14,37 @@ 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, ok := VerifyTelegramInitData(auth.InitData, "7697757472:AAESD9HfkWwbIZe-HXR7IazUShr69hZTLmE")
+	data, err := controllers.VerifyTelegramInitData(auth.InitData, "7697757472:AAESD9HfkWwbIZe-HXR7IazUShr69hZTLmE")
-	if !ok {
+	if err != nil {
+		log.Println("initData is empty. Connection closed!")
 		conn.Close()
 		return
 	}
 
-	log.Println("Client connected! \ninitData: ", data)
+	userID := data.User.ID
-
+	username := data.User.Username
-	userID := data["user.id"]
-	username := data["user.username"]
 
 	if username == "" {
-		username = data["user.first_name"]
+		username = data.User.FirstName
-	}
-	if username == "" {
-		username = "user_" + userID
 	}
 
-	player := &Player{
+	log.Printf("Player %d connected!", userID)
+
+	player := &models.Player{
 		ID:       userID,
 		Username: username,
 		Conn:     conn,
@@ -54,30 +57,39 @@ func ServeWS(room *Room, w http.ResponseWriter, r *http.Request) {
 	player.Conn.WriteJSON(map[string]any{
 		"type": "init",
 		"payload": map[string]string{
-			"id": player.ID,
+			"id": fmt.Sprintf("%d", player.ID),
 		},
 	})
 
 	go readLoop(room, player)
 }
 
-func readLoop(room *Room, player *Player) {
+func readLoop(room *controllers.Room, player *models.Player) {
-	for {
+	defer func() {
-		var msg InputMessage
+		log.Printf("Player %d disconnected!", player.ID)
-		err := player.Conn.ReadJSON(&msg)
-		if err != nil {
 		delete(room.Players, player.ID)
 		player.Conn.Close()
+	}()
+
+	for {
+		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 string
-	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[string]*Player
+	Players map[int64]*models.Player
-	Input   chan InputMessage
+	Input   chan models.InputMessage
 	mu      sync.Mutex
 }
 
 func NewRoom() *Room {
 	return &Room{
-		Players: make(map[string]*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:
@@ -48,7 +48,7 @@ DONE:
 func (r *Room) broadcast() {
 	r.mu.Lock()
 
-	state := make(map[string]map[string]any, len(r.Players))
+	state := make(map[int64]map[string]any, len(r.Players))
 	for id, p := range r.Players {
 		state[id] = map[string]any{
 			"x":    p.X,
@@ -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

@@ -0,0 +1,16 @@
+package controllers
+
+import (
+	"time"
+
+	initdata "github.com/telegram-mini-apps/init-data-golang"
+)
+
+func VerifyTelegramInitData(initData, botToken string) (initdata.InitData, error) {
+	expIn := 24 * time.Hour
+	err := initdata.Validate(initData, botToken, expIn)
+	if err != nil {
+		return initdata.InitData{}, err
+	}
+	return initdata.Parse(initData)
+}

src/models/auth.go

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

src/models/player.go

@@ -1,11 +1,11 @@
-package main
+package models
 
 import (
 	"github.com/gorilla/websocket"
 )
 
 type Player struct {
-	ID       string
+	ID       int64
 	Username string
 	Conn     *websocket.Conn
 

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"`
+}

telegram.go

@@ -1,41 +0,0 @@
-package main
-
-import (
-	"crypto/hmac"
-	"crypto/sha256"
-	"encoding/hex"
-	"net/url"
-	"sort"
-	"strings"
-)
-
-func VerifyTelegramInitData(initData, botToken string) (map[string]string, bool) {
-	values, _ := url.ParseQuery(initData)
-
-	hash := values.Get("hash")
-	values.Del("hash")
-
-	var data []string
-	for k, v := range values {
-		data = append(data, k+"="+v[0])
-	}
-	sort.Strings(data)
-
-	checkString := strings.Join(data, "\n")
-
-	secret := sha256.Sum256([]byte(botToken))
-	h := hmac.New(sha256.New, secret[:])
-	h.Write([]byte(checkString))
-
-	expected := hex.EncodeToString(h.Sum(nil))
-
-	return valuesToMap(values), expected == hash
-}
-
-func valuesToMap(v url.Values) map[string]string {
-	m := make(map[string]string)
-	for k, val := range v {
-		m[k] = val[0]
-	}
-	return m
-}