mirror of
https://github.com/diamondburned/arikawa.git
synced 2024-09-08 03:12:23 +00:00
154 lines
3.1 KiB
Go
154 lines
3.1 KiB
Go
|
package gateway
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
|
||
|
"github.com/diamondburned/arikawa/json"
|
||
|
"github.com/diamondburned/arikawa/wsutil"
|
||
|
"github.com/pkg/errors"
|
||
|
)
|
||
|
|
||
|
type OPCode uint8
|
||
|
|
||
|
const (
|
||
|
DispatchOP OPCode = iota // recv
|
||
|
HeartbeatOP // send/recv
|
||
|
IdentifyOP // send...
|
||
|
StatusUpdateOP //
|
||
|
VoiceStateUpdateOP //
|
||
|
VoiceServerPingOP //
|
||
|
ResumeOP //
|
||
|
ReconnectOP // recv
|
||
|
RequestGuildMembersOP // send
|
||
|
InvalidSessionOP // recv...
|
||
|
HelloOP
|
||
|
HeartbeatAckOP
|
||
|
_
|
||
|
CallConnectOP
|
||
|
GuildSubscriptionsOP
|
||
|
)
|
||
|
|
||
|
type OP struct {
|
||
|
Code OPCode `json:"op"`
|
||
|
Data json.Raw `json:"d,omitempty"`
|
||
|
|
||
|
// Only for Dispatch (op 0)
|
||
|
Sequence int `json:"s,omitempty"`
|
||
|
EventName string `json:"t,omitempty"`
|
||
|
}
|
||
|
|
||
|
var ErrInvalidSession = errors.New("Invalid session")
|
||
|
|
||
|
func DecodeOP(driver json.Driver, ev wsutil.Event) (*OP, error) {
|
||
|
if ev.Error != nil {
|
||
|
return nil, ev.Error
|
||
|
}
|
||
|
|
||
|
var op *OP
|
||
|
if err := driver.Unmarshal(ev.Data, &op); err != nil {
|
||
|
return nil, errors.Wrap(err, "Failed to decode payload")
|
||
|
}
|
||
|
|
||
|
if op.Code == InvalidSessionOP {
|
||
|
return op, ErrInvalidSession
|
||
|
}
|
||
|
|
||
|
return op, nil
|
||
|
}
|
||
|
|
||
|
func DecodeEvent(driver json.Driver,
|
||
|
ev wsutil.Event, v interface{}) (OPCode, error) {
|
||
|
|
||
|
op, err := DecodeOP(driver, ev)
|
||
|
if err != nil {
|
||
|
return 0, err
|
||
|
}
|
||
|
|
||
|
if err := driver.Unmarshal(op.Data, v); err != nil {
|
||
|
return 0, errors.Wrap(err, "Failed to decode data")
|
||
|
}
|
||
|
|
||
|
return op.Code, nil
|
||
|
}
|
||
|
|
||
|
func AssertEvent(driver json.Driver,
|
||
|
ev wsutil.Event, code OPCode, v interface{}) error {
|
||
|
|
||
|
op, err := DecodeOP(driver, ev)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
if op.Code != code {
|
||
|
return fmt.Errorf(
|
||
|
"Unexpected OP Code: %d, expected %d (%s)",
|
||
|
op.Code, code, op.Data,
|
||
|
)
|
||
|
}
|
||
|
|
||
|
if err := driver.Unmarshal(op.Data, v); err != nil {
|
||
|
return errors.Wrap(err, "Failed to decode data")
|
||
|
}
|
||
|
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func HandleEvent(g *Gateway, data []byte) error {
|
||
|
// Parse the raw data into an OP struct
|
||
|
var op *OP
|
||
|
if err := g.Driver.Unmarshal(data, &op); err != nil {
|
||
|
return errors.Wrap(err, "OP error")
|
||
|
}
|
||
|
|
||
|
if g.OP != nil {
|
||
|
g.OP <- op
|
||
|
}
|
||
|
|
||
|
switch op.Code {
|
||
|
case HeartbeatAckOP:
|
||
|
// Heartbeat from the server?
|
||
|
g.Pacemaker.Echo()
|
||
|
|
||
|
case HeartbeatOP:
|
||
|
// Server requesting a heartbeat.
|
||
|
return g.Pacemaker.Pace()
|
||
|
|
||
|
case ReconnectOP:
|
||
|
// Server requests to reconnect, die and retry.
|
||
|
return g.Reconnect()
|
||
|
|
||
|
case InvalidSessionOP:
|
||
|
// Invalid session, respond with Identify.
|
||
|
return g.Identify()
|
||
|
|
||
|
case HelloOP:
|
||
|
// What is this OP doing here???
|
||
|
return nil
|
||
|
|
||
|
case DispatchOP:
|
||
|
// Check if we know the event
|
||
|
fn, ok := EventCreator[op.EventName]
|
||
|
if !ok {
|
||
|
return errors.New("Unknown event: " + op.EventName)
|
||
|
}
|
||
|
|
||
|
// Make a new pointer to the event
|
||
|
var ev = fn()
|
||
|
|
||
|
// Try and parse the event
|
||
|
if err := g.Driver.Unmarshal(op.Data, ev); err != nil {
|
||
|
return errors.Wrap(err, "Failed to parse event "+op.EventName)
|
||
|
}
|
||
|
|
||
|
// Throw the event into a channel, it's valid now.
|
||
|
g.Events <- ev
|
||
|
return nil
|
||
|
|
||
|
default:
|
||
|
return fmt.Errorf(
|
||
|
"Unknown OP code %d (event %s)", op.Code, op.EventName)
|
||
|
}
|
||
|
|
||
|
return nil
|
||
|
}
|