mirror of
https://github.com/diamondburned/arikawa.git
synced 2024-09-08 03:12:23 +00:00
https://github.com/golang/go/wiki/CommonMistakes#using-reference-to-loop-iterator-variable
192 lines
5.2 KiB
Go
192 lines
5.2 KiB
Go
package state
|
|
|
|
import (
|
|
"github.com/diamondburned/arikawa/discord"
|
|
"github.com/diamondburned/arikawa/gateway"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
func (s *State) hookSession() error {
|
|
s.unhooker = s.Session.AddHandler(func(iface interface{}) {
|
|
if s.PreHandler != nil {
|
|
s.PreHandler.Call(iface)
|
|
}
|
|
s.onEvent(iface)
|
|
})
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *State) onEvent(iface interface{}) {
|
|
// TODO: voice states
|
|
|
|
switch ev := iface.(type) {
|
|
case *gateway.ReadyEvent:
|
|
// Handle guilds
|
|
for _, g := range ev.Guilds {
|
|
g := g
|
|
|
|
if err := s.Store.GuildSet(&g); err != nil {
|
|
s.stateErr(err, "Failed to set guild in state")
|
|
}
|
|
}
|
|
|
|
// Handle private channels
|
|
for _, ch := range ev.PrivateChannels {
|
|
ch := ch
|
|
|
|
if err := s.Store.ChannelSet(&ch); err != nil {
|
|
s.stateErr(err, "Failed to set channel in state")
|
|
}
|
|
}
|
|
|
|
// Handle user
|
|
if err := s.Store.SelfSet(&ev.User); err != nil {
|
|
s.stateErr(err, "Failed to set self in state")
|
|
}
|
|
|
|
// Set Ready to the state
|
|
s.Ready = *ev
|
|
|
|
case *gateway.GuildCreateEvent:
|
|
if err := s.Store.GuildSet(&ev.Guild); err != nil {
|
|
s.stateErr(err, "Failed to create guild in state")
|
|
}
|
|
|
|
for _, m := range ev.Members {
|
|
m := m
|
|
|
|
if err := s.Store.MemberSet(ev.Guild.ID, &m); err != nil {
|
|
s.stateErr(err, "Failed to add a member from guild in state")
|
|
}
|
|
}
|
|
|
|
for _, ch := range ev.Channels {
|
|
ch := ch
|
|
ch.GuildID = ev.Guild.ID // just to make sure
|
|
|
|
if err := s.Store.ChannelSet(&ch); err != nil {
|
|
s.stateErr(err, "Failed to add a channel from guild in state")
|
|
}
|
|
}
|
|
|
|
for _, p := range ev.Presences {
|
|
p := p
|
|
|
|
if err := s.Store.PresenceSet(ev.Guild.ID, &p); err != nil {
|
|
s.stateErr(err, "Failed to add a presence from guild in state")
|
|
}
|
|
}
|
|
case *gateway.GuildUpdateEvent:
|
|
if err := s.Store.GuildSet((*discord.Guild)(ev)); err != nil {
|
|
s.stateErr(err, "Failed to update guild in state")
|
|
}
|
|
case *gateway.GuildDeleteEvent:
|
|
if err := s.Store.GuildRemove(ev.ID); err != nil {
|
|
s.stateErr(err, "Failed to delete guild in state")
|
|
}
|
|
|
|
case *gateway.GuildMemberAddEvent:
|
|
if err := s.Store.MemberSet(ev.GuildID, &ev.Member); err != nil {
|
|
s.stateErr(err, "Failed to add a member in state")
|
|
}
|
|
case *gateway.GuildMemberUpdateEvent:
|
|
m, err := s.Store.Member(ev.GuildID, ev.User.ID)
|
|
if err != nil {
|
|
// We can't do much here.
|
|
m = &discord.Member{}
|
|
}
|
|
|
|
// Update available fields from ev into m
|
|
ev.Update(m)
|
|
|
|
if err := s.Store.MemberSet(ev.GuildID, m); err != nil {
|
|
s.stateErr(err, "Failed to update a member in state")
|
|
}
|
|
case *gateway.GuildMemberRemoveEvent:
|
|
if err := s.Store.MemberRemove(ev.GuildID, ev.User.ID); err != nil {
|
|
s.stateErr(err, "Failed to remove a member in state")
|
|
}
|
|
|
|
case *gateway.GuildMembersChunkEvent:
|
|
for _, m := range ev.Members {
|
|
m := m
|
|
|
|
if err := s.Store.MemberSet(ev.GuildID, &m); err != nil {
|
|
s.stateErr(err, "Failed to add a member from chunk in state")
|
|
}
|
|
}
|
|
|
|
for _, p := range ev.Presences {
|
|
p := p
|
|
|
|
if err := s.Store.PresenceSet(ev.GuildID, &p); err != nil {
|
|
s.stateErr(err, "Failed to add a presence from chunk in state")
|
|
}
|
|
}
|
|
|
|
case *gateway.GuildRoleCreateEvent:
|
|
if err := s.Store.RoleSet(ev.GuildID, &ev.Role); err != nil {
|
|
s.stateErr(err, "Failed to add a role in state")
|
|
}
|
|
case *gateway.GuildRoleUpdateEvent:
|
|
if err := s.Store.RoleSet(ev.GuildID, &ev.Role); err != nil {
|
|
s.stateErr(err, "Failed to update a role in state")
|
|
}
|
|
case *gateway.GuildRoleDeleteEvent:
|
|
if err := s.Store.RoleRemove(ev.GuildID, ev.RoleID); err != nil {
|
|
s.stateErr(err, "Failed to remove a role in state")
|
|
}
|
|
|
|
case *gateway.GuildEmojisUpdateEvent:
|
|
if err := s.Store.EmojiSet(ev.GuildID, ev.Emojis); err != nil {
|
|
s.stateErr(err, "Failed to update emojis in state")
|
|
}
|
|
|
|
case *gateway.ChannelCreateEvent:
|
|
if err := s.Store.ChannelSet((*discord.Channel)(ev)); err != nil {
|
|
s.stateErr(err, "Failed to create a channel in state")
|
|
}
|
|
case *gateway.ChannelUpdateEvent:
|
|
if err := s.Store.ChannelSet((*discord.Channel)(ev)); err != nil {
|
|
s.stateErr(err, "Failed to update a channel in state")
|
|
}
|
|
case *gateway.ChannelDeleteEvent:
|
|
if err := s.Store.ChannelRemove((*discord.Channel)(ev)); err != nil {
|
|
s.stateErr(err, "Failed to remove a channel in state")
|
|
}
|
|
|
|
// *gateway.ChannelPinsUpdateEvent is not tracked.
|
|
|
|
case *gateway.MessageCreateEvent:
|
|
if err := s.Store.MessageSet((*discord.Message)(ev)); err != nil {
|
|
s.stateErr(err, "Failed to add a message in state")
|
|
}
|
|
case *gateway.MessageUpdateEvent:
|
|
if err := s.Store.MessageSet((*discord.Message)(ev)); err != nil {
|
|
s.stateErr(err, "Failed to update a message in state")
|
|
}
|
|
case *gateway.MessageDeleteEvent:
|
|
if err := s.Store.MessageRemove(ev.ChannelID, ev.ID); err != nil {
|
|
s.stateErr(err, "Failed to delete a message in state")
|
|
}
|
|
case *gateway.MessageDeleteBulkEvent:
|
|
for _, id := range ev.IDs {
|
|
if err := s.Store.MessageRemove(ev.ChannelID, id); err != nil {
|
|
s.stateErr(err, "Failed to delete bulk meessages in state")
|
|
}
|
|
}
|
|
|
|
case *gateway.PresenceUpdateEvent:
|
|
if err := s.Store.PresenceSet(
|
|
ev.GuildID, (*discord.Presence)(ev)); err != nil {
|
|
|
|
s.stateErr(err, "Failed to update presence in state")
|
|
}
|
|
}
|
|
}
|
|
|
|
func (s *State) stateErr(err error, wrap string) {
|
|
s.ErrorLog(errors.Wrap(err, wrap))
|
|
}
|