fix
This commit is contained in:
5
go.mod
5
go.mod
@@ -2,4 +2,7 @@ module server
|
|||||||
|
|
||||||
go 1.25.0
|
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
|
||||||
|
)
|
||||||
|
|||||||
2
go.sum
2
go.sum
@@ -1,2 +1,4 @@
|
|||||||
github.com/gorilla/websocket v1.5.3 h1:saDtZ6Pbx/0u+bgYQ3q96pZgCzfhKXGPqt7kZ72aNNg=
|
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/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=
|
||||||
|
|||||||
27
main.go
27
main.go
@@ -1,6 +1,7 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"fmt"
|
||||||
"log"
|
"log"
|
||||||
"net/http"
|
"net/http"
|
||||||
|
|
||||||
@@ -23,24 +24,21 @@ func ServeWS(room *Room, w http.ResponseWriter, r *http.Request) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
data, ok := VerifyTelegramInitData(auth.InitData, "7697757472:AAESD9HfkWwbIZe-HXR7IazUShr69hZTLmE")
|
data, err := VerifyTelegramInitData(auth.InitData, "7697757472:AAESD9HfkWwbIZe-HXR7IazUShr69hZTLmE")
|
||||||
if !ok {
|
if err != nil {
|
||||||
conn.Close()
|
conn.Close()
|
||||||
return
|
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 == "" {
|
if username == "" {
|
||||||
username = data["user.first_name"]
|
username = data.User.FirstName
|
||||||
}
|
|
||||||
if username == "" {
|
|
||||||
username = "user_" + userID
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
log.Printf("Player %d connected!", userID)
|
||||||
|
|
||||||
player := &Player{
|
player := &Player{
|
||||||
ID: userID,
|
ID: userID,
|
||||||
Username: username,
|
Username: username,
|
||||||
@@ -54,7 +52,7 @@ func ServeWS(room *Room, w http.ResponseWriter, r *http.Request) {
|
|||||||
player.Conn.WriteJSON(map[string]any{
|
player.Conn.WriteJSON(map[string]any{
|
||||||
"type": "init",
|
"type": "init",
|
||||||
"payload": map[string]string{
|
"payload": map[string]string{
|
||||||
"id": player.ID,
|
"id": fmt.Sprintf("%d", player.ID),
|
||||||
},
|
},
|
||||||
})
|
})
|
||||||
|
|
||||||
@@ -62,12 +60,15 @@ func ServeWS(room *Room, w http.ResponseWriter, r *http.Request) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func readLoop(room *Room, player *Player) {
|
func readLoop(room *Room, player *Player) {
|
||||||
|
defer func() {
|
||||||
|
delete(room.Players, player.ID)
|
||||||
|
player.Conn.Close()
|
||||||
|
}()
|
||||||
|
|
||||||
for {
|
for {
|
||||||
var msg InputMessage
|
var msg InputMessage
|
||||||
err := player.Conn.ReadJSON(&msg)
|
err := player.Conn.ReadJSON(&msg)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
delete(room.Players, player.ID)
|
|
||||||
player.Conn.Close()
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -5,7 +5,7 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type Player struct {
|
type Player struct {
|
||||||
ID string
|
ID int64
|
||||||
Username string
|
Username string
|
||||||
Conn *websocket.Conn
|
Conn *websocket.Conn
|
||||||
|
|
||||||
|
|||||||
@@ -1,7 +1,7 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
type InputMessage struct {
|
type InputMessage struct {
|
||||||
PlayerID string
|
PlayerID int64
|
||||||
DX float64 `json:"dx"`
|
DX float64 `json:"dx"`
|
||||||
DY float64 `json:"dy"`
|
DY float64 `json:"dy"`
|
||||||
}
|
}
|
||||||
|
|||||||
6
room.go
6
room.go
@@ -6,14 +6,14 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
type Room struct {
|
type Room struct {
|
||||||
Players map[string]*Player
|
Players map[int64]*Player
|
||||||
Input chan InputMessage
|
Input chan InputMessage
|
||||||
mu sync.Mutex
|
mu sync.Mutex
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewRoom() *Room {
|
func NewRoom() *Room {
|
||||||
return &Room{
|
return &Room{
|
||||||
Players: make(map[string]*Player),
|
Players: make(map[int64]*Player),
|
||||||
Input: make(chan InputMessage, 128),
|
Input: make(chan InputMessage, 128),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -48,7 +48,7 @@ DONE:
|
|||||||
func (r *Room) broadcast() {
|
func (r *Room) broadcast() {
|
||||||
r.mu.Lock()
|
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 {
|
for id, p := range r.Players {
|
||||||
state[id] = map[string]any{
|
state[id] = map[string]any{
|
||||||
"x": p.X,
|
"x": p.X,
|
||||||
|
|||||||
43
telegram.go
43
telegram.go
@@ -1,41 +1,16 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"crypto/hmac"
|
"time"
|
||||||
"crypto/sha256"
|
|
||||||
"encoding/hex"
|
initdata "github.com/telegram-mini-apps/init-data-golang"
|
||||||
"net/url"
|
|
||||||
"sort"
|
|
||||||
"strings"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
func VerifyTelegramInitData(initData, botToken string) (map[string]string, bool) {
|
func VerifyTelegramInitData(initData, botToken string) (initdata.InitData, error) {
|
||||||
values, _ := url.ParseQuery(initData)
|
expIn := 24 * time.Hour
|
||||||
|
err := initdata.Validate(initData, botToken, expIn)
|
||||||
hash := values.Get("hash")
|
if err != nil {
|
||||||
values.Del("hash")
|
return initdata.InitData{}, err
|
||||||
|
|
||||||
var data []string
|
|
||||||
for k, v := range values {
|
|
||||||
data = append(data, k+"="+v[0])
|
|
||||||
}
|
}
|
||||||
sort.Strings(data)
|
return initdata.Parse(initData)
|
||||||
|
|
||||||
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
|
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user