Skip to content

Commit

Permalink
ensure hydrating user and author when getting message [#5]
Browse files Browse the repository at this point in the history
  • Loading branch information
zekroTJA committed Mar 17, 2022
1 parent 90d5704 commit b7d3183
Showing 1 changed file with 11 additions and 0 deletions.
11 changes: 11 additions & 0 deletions messages.go
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,7 @@ func (s *State) SetMessage(message *discordgo.Message) (err error) {
// is disabled, nil is returned.
func (s *State) Message(channelID, messageID string) (v *discordgo.Message, err error) {
v = &discordgo.Message{}

ok, err := s.get(s.joinKeys(KeyMessage, channelID, messageID), v)
if !ok {
if s.options.FetchAndStore {
Expand All @@ -47,6 +48,16 @@ func (s *State) Message(channelID, messageID string) (v *discordgo.Message, err
v = nil
}
}
if err != nil || v == nil {
return
}

if v.Author == nil && v.Member != nil && v.Member.User != nil {
v.Author = v.Member.User
} else if v.Member != nil && v.Member.User == nil && v.Author != nil {
v.Member.User = v.Author
}

return
}

Expand Down

0 comments on commit b7d3183

Please sign in to comment.