92 lines
1.5 KiB
Go
92 lines
1.5 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"log"
|
|
"net/http"
|
|
|
|
"github.com/gorilla/websocket"
|
|
)
|
|
|
|
var upgrader = websocket.Upgrader{
|
|
CheckOrigin: func(r *http.Request) bool { return true },
|
|
}
|
|
|
|
func ServeWS(room *Room, w http.ResponseWriter, r *http.Request) {
|
|
conn, err := upgrader.Upgrade(w, r, nil)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
var auth AuthMessage
|
|
if err := conn.ReadJSON(&auth); err != nil || auth.Type != "auth" {
|
|
conn.Close()
|
|
return
|
|
}
|
|
|
|
data, err := VerifyTelegramInitData(auth.InitData, "7697757472:AAESD9HfkWwbIZe-HXR7IazUShr69hZTLmE")
|
|
if err != nil {
|
|
conn.Close()
|
|
return
|
|
}
|
|
|
|
userID := data.User.ID
|
|
username := data.User.Username
|
|
|
|
if username == "" {
|
|
username = data.User.FirstName
|
|
}
|
|
|
|
log.Printf("Player %d connected!", userID)
|
|
|
|
player := &Player{
|
|
ID: userID,
|
|
Username: username,
|
|
Conn: conn,
|
|
X: 180,
|
|
Y: 320,
|
|
}
|
|
|
|
room.Players[player.ID] = player
|
|
|
|
player.Conn.WriteJSON(map[string]any{
|
|
"type": "init",
|
|
"payload": map[string]string{
|
|
"id": fmt.Sprintf("%d", player.ID),
|
|
},
|
|
})
|
|
|
|
go readLoop(room, player)
|
|
}
|
|
|
|
func readLoop(room *Room, player *Player) {
|
|
defer func() {
|
|
delete(room.Players, player.ID)
|
|
player.Conn.Close()
|
|
}()
|
|
|
|
for {
|
|
var msg InputMessage
|
|
err := player.Conn.ReadJSON(&msg)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
msg.PlayerID = player.ID
|
|
room.Input <- msg
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
room := NewRoom()
|
|
|
|
go room.Run()
|
|
|
|
http.HandleFunc("/ws", func(w http.ResponseWriter, r *http.Request) {
|
|
ServeWS(room, w, r)
|
|
})
|
|
|
|
log.Println("Server started on :8080")
|
|
log.Fatal(http.ListenAndServe(":8080", nil))
|
|
}
|