1 files changed, 104 insertions, 0 deletions
diff --git a/backend/message.go b/backend/message.go
new file mode 100644
index 0000000..f466a8f
--- /dev/null
+++ b/backend/message.go
@@ -0,0 +1,104 @@
+package main
+
+import (
+	"encoding/json"
+	"fmt"
+)
+
+type MessageWithClient struct {
+	Client  *GameClient
+	Message *Message
+}
+
+type Message struct {
+	Type string      `json:"type"`
+	Data MessageData `json:"data"`
+}
+
+type MessageData interface{}
+
+type MessageDataConnect struct {
+}
+
+type MessageDataPrepare struct {
+	Problem string `json:"problem"`
+}
+
+type MessageDataReady struct {
+}
+
+type MessageDataStart struct {
+	StartTime string `json:"startTime"`
+}
+
+type MessageDataCode struct {
+	Code string `json:"code"`
+}
+
+type MessageDataScore struct {
+	Score int `json:"score"`
+}
+
+type MessageDataFinish struct {
+	YourScore     *int `json:"yourScore"`
+	OpponentScore *int `json:"opponentScore"`
+}
+
+type MessageDataWatch struct {
+	Problem string `json:"problem"`
+	ScoreA  *int   `json:"scoreA"`
+	CodeA   string `json:"codeA"`
+	ScoreB  *int   `json:"scoreB"`
+	CodeB   string `json:"codeB"`
+}
+
+func (m *Message) UnmarshalJSON(data []byte) error {
+	var raw map[string]json.RawMessage
+	if err := json.Unmarshal(data, &raw); err != nil {
+		return err
+	}
+
+	if err := json.Unmarshal(raw["type"], &m.Type); err != nil {
+		return err
+	}
+
+	var err error
+	switch m.Type {
+	case "connect":
+		var data MessageDataConnect
+		err = json.Unmarshal(raw["data"], &data)
+		m.Data = data
+	case "prepare":
+		var data MessageDataPrepare
+		err = json.Unmarshal(raw["data"], &data)
+		m.Data = data
+	case "ready":
+		var data MessageDataReady
+		err = json.Unmarshal(raw["data"], &data)
+		m.Data = data
+	case "start":
+		var data MessageDataStart
+		err = json.Unmarshal(raw["data"], &data)
+		m.Data = data
+	case "code":
+		var data MessageDataCode
+		err = json.Unmarshal(raw["data"], &data)
+		m.Data = data
+	case "score":
+		var data MessageDataScore
+		err = json.Unmarshal(raw["data"], &data)
+		m.Data = data
+	case "finish":
+		var data MessageDataFinish
+		err = json.Unmarshal(raw["data"], &data)
+		m.Data = data
+	case "watch":
+		var data MessageDataWatch
+		err = json.Unmarshal(raw["data"], &data)
+		m.Data = data
+	default:
+		err = fmt.Errorf("unknown message type: %s", m.Type)
+	}
+
+	return err
+}