diff options
Diffstat (limited to 'rolereact/events.go')
| -rw-r--r-- | rolereact/events.go | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/rolereact/events.go b/rolereact/events.go index 090b924..190c0d8 100644 --- a/rolereact/events.go +++ b/rolereact/events.go @@ -3,8 +3,8 @@ package rolereact import ( "context" + "git.anhgelus.world/anhgelus/les-copaings-bot/common" "git.anhgelus.world/anhgelus/les-copaings-bot/config" - oldGokord "github.com/anhgelus/gokord" "github.com/nyttikord/gokord/bot" "github.com/nyttikord/gokord/event" ) @@ -13,28 +13,28 @@ type RoleReact struct { RoleID string } -func HandleReactionAdd(ctx context.Context, s bot.Session, e *event.MessageReactionAdd) { +func HandleReactionAdd(ctx context.Context, dg bot.Session, e *event.MessageReactionAdd) { results := []RoleReact{} - oldGokord.DB.Model(&config.RoleReact{}). + common.GetDB(ctx).Model(&config.RoleReact{}). Joins("JOIN role_react_messages ON role_reacts.role_react_message_id = role_react_messages.id"). Where("role_react_messages.message_id = ? AND role_reacts.reaction = ?", e.MessageID, e.MessageReaction.Emoji.APIName()). Scan(&results) for _, role := range results { - err := s.GuildAPI().MemberRoleAdd(e.GuildID, e.UserID, role.RoleID) + err := dg.GuildAPI().MemberRoleAdd(e.GuildID, e.UserID, role.RoleID).Do(ctx) if err != nil { bot.Logger(ctx).Error("Unable to add role after member added reaction", "error", err) } } } -func HandleReactionRemove(ctx context.Context, s bot.Session, e *event.MessageReactionRemove) { +func HandleReactionRemove(ctx context.Context, dg bot.Session, e *event.MessageReactionRemove) { results := []RoleReact{} - oldGokord.DB.Model(&config.RoleReact{}). + common.GetDB(ctx).Model(&config.RoleReact{}). Joins("JOIN role_react_messages ON role_reacts.role_react_message_id = role_react_messages.id"). Where("role_react_messages.message_id = ? AND role_reacts.reaction = ?", e.MessageID, e.MessageReaction.Emoji.APIName()). Scan(&results) for _, role := range results { - err := s.GuildAPI().MemberRoleRemove(e.GuildID, e.UserID, role.RoleID) + err := dg.GuildAPI().MemberRoleRemove(e.GuildID, e.UserID, role.RoleID).Do(ctx) if err != nil { bot.Logger(ctx).Error("Unable to remove role after member removed reaction", "error", err) } |