2019-03-13 06:09:24 +01:00
|
|
|
package common
|
|
|
|
|
|
|
|
import (
|
|
|
|
"encoding/json"
|
2019-03-14 20:21:53 +01:00
|
|
|
"errors"
|
2019-03-13 06:09:24 +01:00
|
|
|
"fmt"
|
|
|
|
"strconv"
|
|
|
|
)
|
|
|
|
|
2019-03-14 20:21:53 +01:00
|
|
|
type DataInterface interface {
|
|
|
|
HTML() string
|
2019-03-14 02:24:14 +01:00
|
|
|
}
|
|
|
|
|
2019-03-14 20:21:53 +01:00
|
|
|
type ChatData struct {
|
|
|
|
Hidden bool
|
|
|
|
Type DataType
|
|
|
|
Data json.RawMessage
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c ChatData) GetData() (DataInterface, error) {
|
|
|
|
var data DataInterface
|
|
|
|
var err error
|
|
|
|
|
|
|
|
switch c.Type {
|
|
|
|
case DTInvalid:
|
|
|
|
return nil, errors.New("data type is invalid")
|
|
|
|
case DTChat:
|
|
|
|
d := DataMessage{}
|
|
|
|
err = json.Unmarshal(c.Data, &d)
|
|
|
|
data = d
|
|
|
|
case DTError:
|
|
|
|
d := DataError{}
|
|
|
|
err = json.Unmarshal(c.Data, &d)
|
|
|
|
data = d
|
|
|
|
case DTCommand:
|
|
|
|
d := DataCommand{}
|
|
|
|
err = json.Unmarshal(c.Data, &d)
|
|
|
|
data = d
|
|
|
|
case DTEvent:
|
|
|
|
d := DataEvent{}
|
|
|
|
err = json.Unmarshal(c.Data, &d)
|
|
|
|
data = d
|
|
|
|
case DTClient:
|
|
|
|
d := ClientData{}
|
|
|
|
err = json.Unmarshal(c.Data, &d)
|
|
|
|
data = d
|
|
|
|
}
|
2019-03-14 02:24:14 +01:00
|
|
|
|
2019-03-14 20:21:53 +01:00
|
|
|
return data, err
|
2019-03-14 05:21:41 +01:00
|
|
|
}
|
|
|
|
|
2019-03-14 20:21:53 +01:00
|
|
|
func newChatData(hidden bool, dtype DataType, d DataInterface) (ChatData, error) {
|
|
|
|
rawData, err := json.Marshal(d)
|
|
|
|
return ChatData{
|
|
|
|
Hidden: hidden,
|
|
|
|
Type: dtype,
|
|
|
|
Data: rawData,
|
|
|
|
}, err
|
2019-03-14 05:21:41 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
type ClientData struct {
|
|
|
|
Type ClientDataType
|
|
|
|
Message string
|
|
|
|
}
|
|
|
|
|
2019-03-13 06:09:24 +01:00
|
|
|
type DataError struct {
|
|
|
|
Message string
|
|
|
|
}
|
|
|
|
|
2019-03-13 18:42:38 +01:00
|
|
|
type DataMessage struct {
|
|
|
|
From string
|
|
|
|
Color string
|
|
|
|
Message string
|
|
|
|
Type MessageType
|
2019-03-13 06:09:24 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
type DataCommand struct {
|
|
|
|
Command CommandType
|
|
|
|
Arguments []string
|
|
|
|
}
|
|
|
|
|
|
|
|
type DataEvent struct {
|
|
|
|
Event EventType
|
|
|
|
User string
|
|
|
|
Color string
|
|
|
|
}
|
|
|
|
|
2019-03-14 05:21:41 +01:00
|
|
|
type ClientDataType int
|
|
|
|
|
|
|
|
const (
|
|
|
|
CdMessage ClientDataType = iota // a normal message from the client meant to be broadcast
|
|
|
|
CdUsers // get a list of users
|
|
|
|
)
|
|
|
|
|
2019-03-13 06:09:24 +01:00
|
|
|
type DataType int
|
|
|
|
|
2019-03-14 02:24:14 +01:00
|
|
|
// Data Types
|
2019-03-13 06:09:24 +01:00
|
|
|
const (
|
2019-03-14 02:24:14 +01:00
|
|
|
DTInvalid DataType = iota
|
|
|
|
DTChat // chat message
|
|
|
|
DTError // something went wrong with the previous request
|
|
|
|
DTCommand // non-chat function
|
|
|
|
DTEvent // join/leave/kick/ban events
|
2019-03-14 05:21:41 +01:00
|
|
|
DTClient // a message coming from the client
|
2019-03-13 06:09:24 +01:00
|
|
|
)
|
|
|
|
|
|
|
|
func ParseDataType(token json.Token) (DataType, error) {
|
|
|
|
d := fmt.Sprintf("%.0f", token)
|
|
|
|
val, err := strconv.ParseInt(d, 10, 32)
|
|
|
|
if err != nil {
|
2019-03-14 20:21:53 +01:00
|
|
|
return DTInvalid, fmt.Errorf("invalid data type value: %q\n", d)
|
2019-03-13 06:09:24 +01:00
|
|
|
}
|
|
|
|
return DataType(val), nil
|
|
|
|
}
|
|
|
|
|
|
|
|
type CommandType int
|
|
|
|
|
2019-03-14 02:24:14 +01:00
|
|
|
// Command Types
|
2019-03-13 06:09:24 +01:00
|
|
|
const (
|
2019-03-14 02:24:14 +01:00
|
|
|
CmdPlaying CommandType = iota
|
|
|
|
CmdRefreshPlayer
|
|
|
|
CmdPurgeChat
|
|
|
|
CmdHelp
|
2019-03-13 06:09:24 +01:00
|
|
|
)
|
|
|
|
|
|
|
|
type EventType int
|
|
|
|
|
2019-03-14 02:24:14 +01:00
|
|
|
// Event Types
|
2019-03-13 06:09:24 +01:00
|
|
|
const (
|
2019-03-14 02:24:14 +01:00
|
|
|
EvJoin EventType = iota
|
|
|
|
EvLeave
|
|
|
|
EvKick
|
|
|
|
EvBan
|
|
|
|
EvServerMessage
|
2019-03-13 06:09:24 +01:00
|
|
|
)
|
|
|
|
|
2019-03-13 18:42:38 +01:00
|
|
|
type MessageType int
|
|
|
|
|
2019-03-14 02:24:14 +01:00
|
|
|
// Message Types
|
2019-03-13 18:42:38 +01:00
|
|
|
const (
|
2019-03-14 02:24:14 +01:00
|
|
|
MsgChat MessageType = iota // standard chat
|
|
|
|
MsgAction // /me command
|
|
|
|
MsgServer // server message
|
|
|
|
MsgError
|
2019-03-13 18:42:38 +01:00
|
|
|
)
|
|
|
|
|
2019-03-14 05:21:41 +01:00
|
|
|
func (c ClientData) HTML() string {
|
|
|
|
// Client data is for client to server communication only, so clients should not see this
|
|
|
|
return `<div style="color: red;"><span>The developer messed up. You should not be seeing this.</span></div>`
|
|
|
|
}
|
|
|
|
|
2019-03-13 06:09:24 +01:00
|
|
|
// TODO: Read this HTML from a template somewhere
|
2019-03-13 18:42:38 +01:00
|
|
|
func (dc DataMessage) HTML() string {
|
|
|
|
switch dc.Type {
|
2019-03-14 02:24:14 +01:00
|
|
|
case MsgAction:
|
2019-03-13 19:10:45 +01:00
|
|
|
return `<div style="color:` + dc.Color + `"><span class="name">` + dc.From +
|
|
|
|
`</span> <span class="cmdme">` + dc.Message + `</span></div>`
|
2019-03-13 06:09:24 +01:00
|
|
|
|
2019-03-14 02:24:14 +01:00
|
|
|
case MsgServer:
|
2019-03-13 06:09:24 +01:00
|
|
|
return `<div class="announcement">` + dc.Message + `</div>`
|
|
|
|
|
2019-03-14 02:24:14 +01:00
|
|
|
case MsgError:
|
2019-03-13 18:42:38 +01:00
|
|
|
return `<div class="error">` + dc.Message + `</div>`
|
|
|
|
|
|
|
|
default:
|
2019-03-13 19:10:45 +01:00
|
|
|
return `<div><span class="name" style="color:` + dc.Color + `">` + dc.From +
|
|
|
|
`</span><b>:</b> <span class="msg">` + dc.Message + `</span></div>`
|
2019-03-13 18:42:38 +01:00
|
|
|
}
|
2019-03-13 06:09:24 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
func (de DataEvent) HTML() string {
|
|
|
|
switch de.Event {
|
2019-03-14 02:24:14 +01:00
|
|
|
case EvKick:
|
2019-03-13 19:10:45 +01:00
|
|
|
return `<div class="event"><span class="name" style="color:` + de.Color + `">` +
|
|
|
|
de.User + `</span> has been kicked.</div>`
|
2019-03-14 02:24:14 +01:00
|
|
|
case EvLeave:
|
2019-03-13 19:10:45 +01:00
|
|
|
return `<div class="event"><span class="name" style="color:` + de.Color + `">` +
|
|
|
|
de.User + `</span> has left the chat.</div>`
|
2019-03-14 02:24:14 +01:00
|
|
|
case EvBan:
|
2019-03-13 19:10:45 +01:00
|
|
|
return `<div class="event"><span class="name" style="color:` + de.Color + `">` +
|
|
|
|
de.User + `</span> has been banned.</div>`
|
2019-03-14 02:24:14 +01:00
|
|
|
case EvJoin:
|
2019-03-13 19:10:45 +01:00
|
|
|
return `<div class="event"><span class="name" style="color:` + de.Color + `">` +
|
|
|
|
de.User + `</span> has joined the chat.</div>`
|
2019-03-13 06:09:24 +01:00
|
|
|
}
|
|
|
|
return ""
|
|
|
|
}
|
|
|
|
|
|
|
|
func (de DataError) HTML() string {
|
2019-03-13 19:10:45 +01:00
|
|
|
return `<div class="svmsg"><b>Error</b>: ` + de.Message + `</div>`
|
2019-03-13 06:09:24 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
func (de DataCommand) HTML() string {
|
|
|
|
return ""
|
|
|
|
}
|
|
|
|
|
2019-03-13 18:42:38 +01:00
|
|
|
func EncodeMessage(name, color, msg string, msgtype MessageType) (string, error) {
|
2019-03-14 20:21:53 +01:00
|
|
|
d, err := newChatData(false, DTChat, DataMessage{
|
|
|
|
From: name,
|
|
|
|
Color: color,
|
|
|
|
Message: msg,
|
|
|
|
Type: msgtype,
|
|
|
|
})
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
2019-03-13 06:09:24 +01:00
|
|
|
}
|
2019-03-14 20:21:53 +01:00
|
|
|
return jsonifyChatData(d)
|
2019-03-13 06:09:24 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
func EncodeError(message string) (string, error) {
|
2019-03-14 20:21:53 +01:00
|
|
|
d, err := newChatData(false, DTError, DataError{Message: message})
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
2019-03-13 06:09:24 +01:00
|
|
|
}
|
|
|
|
return jsonifyChatData(d)
|
|
|
|
}
|
|
|
|
|
|
|
|
func EncodeCommand(command CommandType, args []string) (string, error) {
|
2019-03-14 20:21:53 +01:00
|
|
|
d, err := newChatData(false, DTCommand, DataCommand{
|
|
|
|
Command: command,
|
|
|
|
Arguments: args,
|
|
|
|
})
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
2019-03-13 06:09:24 +01:00
|
|
|
}
|
|
|
|
return jsonifyChatData(d)
|
|
|
|
}
|
|
|
|
|
|
|
|
func EncodeEvent(event EventType, name, color string) (string, error) {
|
2019-03-14 20:21:53 +01:00
|
|
|
d, err := newChatData(false, DTEvent, DataEvent{
|
|
|
|
Event: event,
|
|
|
|
User: name,
|
|
|
|
Color: color,
|
|
|
|
})
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
2019-03-13 06:09:24 +01:00
|
|
|
}
|
|
|
|
return jsonifyChatData(d)
|
|
|
|
}
|
|
|
|
|
|
|
|
func jsonifyChatData(data ChatData) (string, error) {
|
|
|
|
raw, err := json.Marshal(data)
|
|
|
|
if err != nil {
|
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
return string(raw), nil
|
|
|
|
}
|
|
|
|
|
2019-03-14 20:21:53 +01:00
|
|
|
func DecodeData(rawjson string) (ChatData, error) {
|
|
|
|
var data ChatData
|
|
|
|
err := json.Unmarshal([]byte(rawjson), &data)
|
|
|
|
return data, err
|
2019-03-13 06:09:24 +01:00
|
|
|
}
|