working Gateway for incoming Mod/Bot ws-conn's; simulated Mod client via sim.go
This commit is contained in:
314
sim.go
314
sim.go
@@ -2,15 +2,23 @@ package main
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"log"
|
||||
"net/url"
|
||||
"os"
|
||||
"os/signal"
|
||||
"syscall"
|
||||
"time"
|
||||
|
||||
"github.com/gorilla/websocket"
|
||||
)
|
||||
|
||||
type MessageEnvelope struct {
|
||||
const (
|
||||
gatewayURL = "ws://localhost:3333/push"
|
||||
apiKey = "gateway"
|
||||
serverID = "test-server-001"
|
||||
)
|
||||
|
||||
type Handshake struct {
|
||||
Type string `json:"type"`
|
||||
Data json.RawMessage `json:"data"`
|
||||
}
|
||||
@@ -19,173 +27,163 @@ type ModHandshake struct {
|
||||
ServerID string `json:"server_id"`
|
||||
}
|
||||
|
||||
type BotHandshake struct {
|
||||
BotID string `json:"bot_id"`
|
||||
type GatewayAck struct {
|
||||
Status string `json:"status"`
|
||||
Type string `json:"type"`
|
||||
}
|
||||
|
||||
type User struct {
|
||||
type MinecraftUser struct {
|
||||
ID string `json:"id"`
|
||||
Name string `json:"name"`
|
||||
}
|
||||
|
||||
type ModMessage struct {
|
||||
MsgID string `json:"msg_id"`
|
||||
Server string `json:"server"`
|
||||
User User `json:"user"`
|
||||
Content string `json:"content"`
|
||||
Meta map[string]interface{} `json:"meta,omitempty"`
|
||||
Ts string `json:"ts,omitempty"`
|
||||
type Destination struct {
|
||||
ChannelID string `json:"channel_id"`
|
||||
}
|
||||
|
||||
type BotMessage struct {
|
||||
MsgID string `json:"msg_id"`
|
||||
ChannelID string `json:"channel_id"`
|
||||
Author string `json:"author"`
|
||||
Content string `json:"content"`
|
||||
Meta map[string]interface{} `json:"meta,omitempty"`
|
||||
Ts string `json:"ts,omitempty"`
|
||||
}
|
||||
|
||||
func simulateMod() {
|
||||
u := url.URL{Scheme: "ws", Host: "localhost:3333", Path: "/ws"}
|
||||
u.RawQuery = "api_key=gateway"
|
||||
|
||||
conn, _, err := websocket.DefaultDialer.Dial(u.String(), nil)
|
||||
if err != nil {
|
||||
log.Fatalf("mod dial error: %v", err)
|
||||
}
|
||||
defer conn.Close()
|
||||
|
||||
fmt.Println("[MOD] Connected")
|
||||
|
||||
// Send handshake
|
||||
handshake := ModHandshake{ServerID: "survival_server"}
|
||||
hsData, _ := json.Marshal(handshake)
|
||||
envelope := MessageEnvelope{
|
||||
Type: "mod",
|
||||
Data: hsData,
|
||||
}
|
||||
envData, _ := json.Marshal(envelope)
|
||||
|
||||
if err := conn.WriteMessage(websocket.TextMessage, envData); err != nil {
|
||||
log.Fatalf("mod handshake write error: %v", err)
|
||||
}
|
||||
fmt.Println("[MOD] Sent handshake")
|
||||
|
||||
// Read handshake response
|
||||
_, resp, err := conn.ReadMessage()
|
||||
if err != nil {
|
||||
log.Fatalf("mod handshake response error: %v", err)
|
||||
}
|
||||
fmt.Printf("[MOD] Handshake response: %s\n", string(resp))
|
||||
|
||||
// Send a few messages
|
||||
for i := 1; i <= 3; i++ {
|
||||
msg := ModMessage{
|
||||
MsgID: fmt.Sprintf("mod_msg_%d", i),
|
||||
Server: "survival_server",
|
||||
User: User{ID: "player_123", Name: "Steve"},
|
||||
Content: fmt.Sprintf("Message %d from Minecraft!", i),
|
||||
Meta: map[string]interface{}{
|
||||
"coordinates": map[string]int{"x": 100 + i*10, "y": 64, "z": 200},
|
||||
},
|
||||
Ts: time.Now().UTC().Format(time.RFC3339),
|
||||
}
|
||||
|
||||
msgData, _ := json.Marshal(msg)
|
||||
if err := conn.WriteMessage(websocket.TextMessage, msgData); err != nil {
|
||||
log.Fatalf("mod message write error: %v", err)
|
||||
}
|
||||
fmt.Printf("[MOD] Sent message: %s\n", msg.Content)
|
||||
|
||||
// Read response
|
||||
_, resp, err := conn.ReadMessage()
|
||||
if err != nil {
|
||||
log.Fatalf("mod response error: %v", err)
|
||||
}
|
||||
fmt.Printf("[MOD] Response: %s\n", string(resp))
|
||||
|
||||
time.Sleep(1 * time.Second)
|
||||
}
|
||||
|
||||
fmt.Println("[MOD] Closing connection")
|
||||
}
|
||||
|
||||
func simulateBot() {
|
||||
time.Sleep(2 * time.Second) // Let mod connect first
|
||||
|
||||
u := url.URL{Scheme: "ws", Host: "localhost:3333", Path: "/ws"}
|
||||
u.RawQuery = "api_key=gateway"
|
||||
|
||||
conn, _, err := websocket.DefaultDialer.Dial(u.String(), nil)
|
||||
if err != nil {
|
||||
log.Fatalf("bot dial error: %v", err)
|
||||
}
|
||||
defer conn.Close()
|
||||
|
||||
fmt.Println("[BOT] Connected")
|
||||
|
||||
// Send handshake
|
||||
handshake := BotHandshake{BotID: "discord_bot_1"}
|
||||
hsData, _ := json.Marshal(handshake)
|
||||
envelope := MessageEnvelope{
|
||||
Type: "bot",
|
||||
Data: hsData,
|
||||
}
|
||||
envData, _ := json.Marshal(envelope)
|
||||
|
||||
if err := conn.WriteMessage(websocket.TextMessage, envData); err != nil {
|
||||
log.Fatalf("bot handshake write error: %v", err)
|
||||
}
|
||||
fmt.Println("[BOT] Sent handshake")
|
||||
|
||||
// Read handshake response
|
||||
_, resp, err := conn.ReadMessage()
|
||||
if err != nil {
|
||||
log.Fatalf("bot handshake response error: %v", err)
|
||||
}
|
||||
fmt.Printf("[BOT] Handshake response: %s\n", string(resp))
|
||||
|
||||
// Send a few messages
|
||||
for i := 1; i <= 3; i++ {
|
||||
msg := BotMessage{
|
||||
MsgID: fmt.Sprintf("bot_msg_%d", i),
|
||||
ChannelID: "987654321",
|
||||
Author: "DiscordUser#1234",
|
||||
Content: fmt.Sprintf("Message %d from Discord!", i),
|
||||
Meta: map[string]interface{}{
|
||||
"reactions": []string{"👍", "❤️"},
|
||||
},
|
||||
Ts: time.Now().UTC().Format(time.RFC3339),
|
||||
}
|
||||
|
||||
msgData, _ := json.Marshal(msg)
|
||||
if err := conn.WriteMessage(websocket.TextMessage, msgData); err != nil {
|
||||
log.Fatalf("bot message write error: %v", err)
|
||||
}
|
||||
fmt.Printf("[BOT] Sent message: %s\n", msg.Content)
|
||||
|
||||
// Read response
|
||||
_, resp, err := conn.ReadMessage()
|
||||
if err != nil {
|
||||
log.Fatalf("bot response error: %v", err)
|
||||
}
|
||||
fmt.Printf("[BOT] Response: %s\n", string(resp))
|
||||
|
||||
time.Sleep(1 * time.Second)
|
||||
}
|
||||
|
||||
fmt.Println("[BOT] Closing connection")
|
||||
type GatewayModMessageIn struct {
|
||||
MsgID string `json:"msg_id"`
|
||||
Server string `json:"server"`
|
||||
Destination Destination `json:"destination"`
|
||||
Author MinecraftUser `json:"author"`
|
||||
Content string `json:"content"`
|
||||
Meta map[string]interface{} `json:"meta,omitempty"`
|
||||
Ts string `json:"ts,omitempty"`
|
||||
}
|
||||
|
||||
func main() {
|
||||
fmt.Println("Starting WebSocket client simulator...")
|
||||
fmt.Println("Connecting to ws://localhost:3333/ws with api_key=gateway")
|
||||
interrupt := make(chan os.Signal, 1)
|
||||
signal.Notify(interrupt, os.Interrupt, syscall.SIGTERM)
|
||||
|
||||
go simulateMod()
|
||||
go simulateBot()
|
||||
// Build WebSocket URL with API key
|
||||
u, err := url.Parse(gatewayURL)
|
||||
if err != nil {
|
||||
log.Fatalf("Failed to parse URL: %v", err)
|
||||
}
|
||||
q := u.Query()
|
||||
q.Set("api_key", apiKey)
|
||||
u.RawQuery = q.Encode()
|
||||
|
||||
// Let them run
|
||||
time.Sleep(15 * time.Second)
|
||||
fmt.Println("Simulator finished")
|
||||
log.Printf("Connecting to %s", u.String())
|
||||
|
||||
// Connect to WebSocket
|
||||
conn, _, err := websocket.DefaultDialer.Dial(u.String(), nil)
|
||||
if err != nil {
|
||||
log.Fatalf("Failed to connect: %v", err)
|
||||
}
|
||||
defer conn.Close()
|
||||
|
||||
log.Println("Connected to gateway")
|
||||
|
||||
// Set up ping handler - respond to pings from server
|
||||
conn.SetPingHandler(func(appData string) error {
|
||||
log.Println("Received ping from server, sending pong")
|
||||
err := conn.WriteControl(websocket.PongMessage, []byte(appData), time.Now().Add(5*time.Second))
|
||||
if err != nil {
|
||||
log.Printf("Failed to send pong: %v", err)
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
})
|
||||
|
||||
// Send handshake
|
||||
modHS := ModHandshake{ServerID: serverID}
|
||||
modHSData, err := json.Marshal(modHS)
|
||||
if err != nil {
|
||||
log.Fatalf("Failed to marshal mod handshake: %v", err)
|
||||
}
|
||||
|
||||
handshake := Handshake{
|
||||
Type: "mod",
|
||||
Data: modHSData,
|
||||
}
|
||||
|
||||
if err := conn.WriteJSON(handshake); err != nil {
|
||||
log.Fatalf("Failed to send handshake: %v", err)
|
||||
}
|
||||
|
||||
log.Println("Handshake sent")
|
||||
|
||||
// Read acknowledgment
|
||||
var ack GatewayAck
|
||||
if err := conn.ReadJSON(&ack); err != nil {
|
||||
log.Fatalf("Failed to read acknowledgment: %v", err)
|
||||
}
|
||||
|
||||
log.Printf("Received acknowledgment: status=%s, type=%s", ack.Status, ack.Type)
|
||||
|
||||
// Channel for incoming messages
|
||||
done := make(chan struct{})
|
||||
|
||||
// Read loop - handles incoming messages and processes control frames
|
||||
go func() {
|
||||
defer close(done)
|
||||
for {
|
||||
messageType, message, err := conn.ReadMessage()
|
||||
if err != nil {
|
||||
if websocket.IsUnexpectedCloseError(err, websocket.CloseGoingAway, websocket.CloseNormalClosure) {
|
||||
log.Printf("WebSocket error: %v", err)
|
||||
} else {
|
||||
log.Printf("Connection closed: %v", err)
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// Only log text/binary messages (ping/pong handled by handlers)
|
||||
if messageType == websocket.TextMessage || messageType == websocket.BinaryMessage {
|
||||
log.Printf("Received from server: %s", string(message))
|
||||
}
|
||||
}
|
||||
}()
|
||||
|
||||
// Optional: Send a test message after connecting
|
||||
time.Sleep(2 * time.Second)
|
||||
testMsg := GatewayModMessageIn{
|
||||
MsgID: "test-msg-001",
|
||||
Server: serverID,
|
||||
Destination: Destination{
|
||||
ChannelID: "123456789",
|
||||
},
|
||||
Author: MinecraftUser{
|
||||
ID: "player-uuid-123",
|
||||
Name: "TestPlayer",
|
||||
},
|
||||
Content: "Hello from simulated mod!",
|
||||
Ts: time.Now().UTC().Format(time.RFC3339),
|
||||
}
|
||||
|
||||
if err := conn.WriteJSON(testMsg); err != nil {
|
||||
log.Printf("Failed to send test message: %v", err)
|
||||
} else {
|
||||
log.Println("Sent test message to gateway")
|
||||
}
|
||||
|
||||
log.Println("Connection established. Responding to pings. Press Ctrl+C to disconnect.")
|
||||
|
||||
// Wait for interrupt or connection close
|
||||
for {
|
||||
select {
|
||||
case <-done:
|
||||
log.Println("Connection closed")
|
||||
return
|
||||
case <-interrupt:
|
||||
log.Println("Interrupt received, closing connection...")
|
||||
|
||||
// Send close message
|
||||
err := conn.WriteMessage(
|
||||
websocket.CloseMessage,
|
||||
websocket.FormatCloseMessage(websocket.CloseNormalClosure, ""),
|
||||
)
|
||||
if err != nil {
|
||||
log.Printf("Write close error: %v", err)
|
||||
return
|
||||
}
|
||||
|
||||
select {
|
||||
case <-done:
|
||||
case <-time.After(time.Second):
|
||||
}
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user