340 lines
9.0 KiB
Go
340 lines
9.0 KiB
Go
package discord
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"time"
|
|
)
|
|
|
|
type Opcode int
|
|
|
|
// https://discord.com/developers/docs/topics/opcodes-and-status-codes#gateway-gateway-opcodes
|
|
// NOTE(ben): I'm not using iota because 5 is missing
|
|
const (
|
|
OpcodeDispatch Opcode = 0
|
|
OpcodeHeartbeat Opcode = 1
|
|
OpcodeIdentify Opcode = 2
|
|
OpcodePresenceUpdate Opcode = 3
|
|
OpcodeVoiceStateUpdate Opcode = 4
|
|
OpcodeResume Opcode = 6
|
|
OpcodeReconnect Opcode = 7
|
|
OpcodeRequestGuildMembers Opcode = 8
|
|
OpcodeInvalidSession Opcode = 9
|
|
OpcodeHello Opcode = 10
|
|
OpcodeHeartbeatACK Opcode = 11
|
|
)
|
|
|
|
type Intent int
|
|
|
|
// https://discord.com/developers/docs/topics/gateway#list-of-intents
|
|
// NOTE(ben): I'm not using iota because the opcode thing made me paranoid
|
|
const (
|
|
IntentGuilds Intent = 1 << 0
|
|
IntentGuildMembers Intent = 1 << 1
|
|
IntentGuildBans Intent = 1 << 2
|
|
IntentGuildEmojisAndStickers Intent = 1 << 3
|
|
IntentGuildIntegrations Intent = 1 << 4
|
|
IntentGuildWebhooks Intent = 1 << 5
|
|
IntentGuildInvites Intent = 1 << 6
|
|
IntentGuildVoiceStates Intent = 1 << 7
|
|
IntentGuildPresences Intent = 1 << 8
|
|
IntentGuildMessages Intent = 1 << 9
|
|
IntentGuildMessageReactions Intent = 1 << 10
|
|
IntentGuildMessageTyping Intent = 1 << 11
|
|
IntentDirectMessages Intent = 1 << 12
|
|
IntentDirectMessageReactions Intent = 1 << 13
|
|
IntentDirectMessageTyping Intent = 1 << 14
|
|
)
|
|
|
|
type GatewayMessage struct {
|
|
Opcode Opcode `json:"op"`
|
|
Data interface{} `json:"d"`
|
|
SequenceNumber *int `json:"s,omitempty"`
|
|
EventName *string `json:"t,omitempty"`
|
|
}
|
|
|
|
func (m *GatewayMessage) ToJSON() []byte {
|
|
mBytes, err := json.Marshal(m)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
// TODO: check if the payload is too big, either here or where we actually send
|
|
// https://discord.com/developers/docs/topics/gateway#sending-payloads
|
|
|
|
return mBytes
|
|
}
|
|
|
|
type Hello struct {
|
|
HeartbeatIntervalMs int `json:"heartbeat_interval"`
|
|
}
|
|
|
|
func HelloFromMap(m interface{}) Hello {
|
|
// TODO: This should probably have some error handling, right?
|
|
return Hello{
|
|
HeartbeatIntervalMs: int(m.(map[string]interface{})["heartbeat_interval"].(float64)),
|
|
}
|
|
}
|
|
|
|
type Identify struct {
|
|
Token string `json:"token"`
|
|
Properties IdentifyConnectionProperties `json:"properties"`
|
|
Intents Intent `json:"intents"`
|
|
}
|
|
|
|
type IdentifyConnectionProperties struct {
|
|
OS string `json:"$os"`
|
|
Browser string `json:"$browser"`
|
|
Device string `json:"$device"`
|
|
}
|
|
|
|
type Ready struct {
|
|
GatewayVersion int `json:"v"`
|
|
User User `json:"user"`
|
|
SessionID string `json:"session_id"`
|
|
}
|
|
|
|
func ReadyFromMap(m interface{}) Ready {
|
|
mmap := m.(map[string]interface{})
|
|
|
|
return Ready{
|
|
GatewayVersion: int(mmap["v"].(float64)),
|
|
User: UserFromMap(mmap["user"]),
|
|
SessionID: mmap["session_id"].(string),
|
|
}
|
|
}
|
|
|
|
type Resume struct {
|
|
Token string `json:"token"`
|
|
SessionID string `json:"session_id"`
|
|
SequenceNumber int `json:"seq"`
|
|
}
|
|
|
|
type ChannelType int
|
|
|
|
const (
|
|
ChannelTypeGuildext ChannelType = 0
|
|
ChannelTypeDM ChannelType = 1
|
|
ChannelTypeGuildVoice ChannelType = 2
|
|
ChannelTypeGroupDM ChannelType = 3
|
|
ChannelTypeGuildCategory ChannelType = 4
|
|
ChannelTypeGuildNews ChannelType = 5
|
|
ChannelTypeGuildStore ChannelType = 6
|
|
ChannelTypeGuildNewsThread ChannelType = 10
|
|
ChannelTypeGuildPublicThread ChannelType = 11
|
|
ChannelTypeGuildPrivateThread ChannelType = 12
|
|
ChannelTypeGuildStageVoice ChannelType = 13
|
|
)
|
|
|
|
type Channel struct {
|
|
ID string `json:"id"`
|
|
Type ChannelType `json:"type"`
|
|
GuildID string `json:"guild_id"`
|
|
Name string `json:"name"`
|
|
Receipients []User `json:"recipients"`
|
|
OwnerID User `json:"owner_id"`
|
|
ParentID *string `json:"parent_id"`
|
|
}
|
|
|
|
type MessageType int
|
|
|
|
const (
|
|
MessageTypeDefault MessageType = 0
|
|
|
|
MessageTypeRecipientAdd MessageType = 1
|
|
MessageTypeRecipientRemove MessageType = 2
|
|
MessageTypeCall MessageType = 3
|
|
|
|
MessageTypeChannelNameChange MessageType = 4
|
|
MessageTypeChannelIconChange MessageType = 5
|
|
MessageTypeChannelPinnedMessage MessageType = 6
|
|
|
|
MessageTypeGuildMemberJoin MessageType = 7
|
|
|
|
MessageTypeUserPremiumGuildSubscription MessageType = 8
|
|
MessageTypeUserPremiumGuildSubscriptionTier1 MessageType = 9
|
|
MessageTypeUserPremiumGuildSubscriptionTier2 MessageType = 10
|
|
MessageTypeUserPremiumGuildSubscriptionTier3 MessageType = 11
|
|
|
|
MessageTypeChannelFollowAdd MessageType = 12
|
|
|
|
MessageTypeGuildDiscoveryDisqualified MessageType = 14
|
|
MessageTypeGuildDiscoveryRequalified MessageType = 15
|
|
MessageTypeGuildDiscoveryGracePeriodInitialWarning MessageType = 16
|
|
MessageTypeGuildDiscoveryGracePeriodFinalWarning MessageType = 17
|
|
|
|
MessageTypeThreadCreated MessageType = 18
|
|
MessageTypeReply MessageType = 19
|
|
MessageTypeApplicationCommand MessageType = 20
|
|
MessageTypeThreadStarterMessage MessageType = 21
|
|
MessageTypeGuildInviteReminder MessageType = 22
|
|
)
|
|
|
|
// https://discord.com/developers/docs/resources/channel#message-object
|
|
type Message struct {
|
|
ID string `json:"id"`
|
|
ChannelID string `json:"channel_id"`
|
|
GuildID *string `json:"guild_id"`
|
|
Content string `json:"content"`
|
|
Author User `json:"author"` // note that this may not be an actual valid user (see the docs)
|
|
Timestamp string `json:"timestamp"`
|
|
Type MessageType `json:"type"`
|
|
|
|
Attachments []Attachment `json:"attachments"`
|
|
// TODO: Embeds
|
|
|
|
originalMap map[string]interface{}
|
|
}
|
|
|
|
func (m *Message) JumpURL() string {
|
|
guildStr := "@me"
|
|
if m.GuildID != nil {
|
|
guildStr = *m.GuildID
|
|
}
|
|
return fmt.Sprintf("https://discord.com/channels/%s/%s/%s", guildStr, m.ChannelID, m.ID)
|
|
}
|
|
|
|
func (m *Message) Time() time.Time {
|
|
t, err := time.Parse(time.RFC3339Nano, m.Timestamp)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return t
|
|
}
|
|
|
|
func (m *Message) OriginalHasFields(fields ...string) bool {
|
|
if m.originalMap == nil {
|
|
return false
|
|
}
|
|
|
|
for _, field := range fields {
|
|
_, ok := m.originalMap[field]
|
|
if !ok {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
|
|
func MessageFromMap(m interface{}) Message {
|
|
/*
|
|
Some gateway events, like MESSAGE_UPDATE, do not contain the
|
|
entire message body. So we need to be defensive on all fields here,
|
|
except the most basic identifying information.
|
|
*/
|
|
|
|
mmap := m.(map[string]interface{})
|
|
msg := Message{
|
|
ID: mmap["id"].(string),
|
|
ChannelID: mmap["channel_id"].(string),
|
|
GuildID: maybeStringP(mmap, "guild_id"),
|
|
Content: maybeString(mmap, "content"),
|
|
Timestamp: maybeString(mmap, "timestamp"),
|
|
Type: MessageType(maybeInt(mmap, "type")),
|
|
|
|
originalMap: mmap,
|
|
}
|
|
|
|
if author, ok := mmap["author"]; ok {
|
|
msg.Author = UserFromMap(author)
|
|
}
|
|
|
|
if iattachments, ok := mmap["attachments"]; ok {
|
|
attachments := iattachments.([]interface{})
|
|
for _, iattachment := range attachments {
|
|
msg.Attachments = append(msg.Attachments, AttachmentFromMap(iattachment))
|
|
}
|
|
}
|
|
|
|
// TODO: Embeds
|
|
|
|
return msg
|
|
}
|
|
|
|
// https://discord.com/developers/docs/resources/user#user-object
|
|
type User struct {
|
|
ID string `json:"id"`
|
|
Username string `json:"username"`
|
|
Discriminator string `json:"discriminator"`
|
|
Avatar *string `json:"avatar"`
|
|
IsBot bool `json:"bot"`
|
|
Locale string `json:"locale"`
|
|
}
|
|
|
|
func UserFromMap(m interface{}) User {
|
|
mmap := m.(map[string]interface{})
|
|
|
|
u := User{
|
|
ID: mmap["id"].(string),
|
|
Username: mmap["username"].(string),
|
|
Discriminator: mmap["discriminator"].(string),
|
|
}
|
|
|
|
if isBot, ok := mmap["bot"]; ok {
|
|
u.IsBot = isBot.(bool)
|
|
}
|
|
|
|
return u
|
|
}
|
|
|
|
type Attachment struct {
|
|
ID string `json:"id"`
|
|
Filename string `json:"filename"`
|
|
ContentType string `json:"content_type"`
|
|
Size int `json:"size"`
|
|
Url string `json:"url"`
|
|
ProxyUrl string `json:"proxy_url"`
|
|
Height *int `json:"height"`
|
|
Width *int `json:"width"`
|
|
}
|
|
|
|
func AttachmentFromMap(m interface{}) Attachment {
|
|
mmap := m.(map[string]interface{})
|
|
a := Attachment{
|
|
ID: mmap["id"].(string),
|
|
Filename: mmap["filename"].(string),
|
|
ContentType: maybeString(mmap, "content_type"),
|
|
Size: int(mmap["size"].(float64)),
|
|
Url: mmap["url"].(string),
|
|
ProxyUrl: mmap["proxy_url"].(string),
|
|
Height: maybeIntP(mmap, "height"),
|
|
Width: maybeIntP(mmap, "width"),
|
|
}
|
|
|
|
return a
|
|
}
|
|
|
|
func maybeString(m map[string]interface{}, k string) string {
|
|
val, ok := m[k]
|
|
if !ok {
|
|
return ""
|
|
}
|
|
return val.(string)
|
|
}
|
|
|
|
func maybeStringP(m map[string]interface{}, k string) *string {
|
|
val, ok := m[k]
|
|
if !ok {
|
|
return nil
|
|
}
|
|
strval := val.(string)
|
|
return &strval
|
|
}
|
|
|
|
func maybeInt(m map[string]interface{}, k string) int {
|
|
val, ok := m[k]
|
|
if !ok {
|
|
return 0
|
|
}
|
|
return int(val.(float64))
|
|
}
|
|
|
|
func maybeIntP(m map[string]interface{}, k string) *int {
|
|
val, ok := m[k]
|
|
if !ok {
|
|
return nil
|
|
}
|
|
intval := int(val.(float64))
|
|
return &intval
|
|
}
|