diamondburned/arikawa
1
0
Fork 0
mirror of https://github.com/diamondburned/arikawa.git synced 2025-03-21 09:29:21 +00:00
Code Issues Releases Activity

State: Author* methods now try and use Message.Member

Browse source
This commit is contained in:
diamondburned (Forefront) 2020-02-12 20:19:24 -08:00
parent 76c7760c24
commit 0de1e579f3
2 changed files with 19 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
gateway/op.go
View file

@ -96,6 +96,10 @@ func AssertEvent(driver json.Driver,
}
func HandleEvent(g *Gateway, data []byte) error {
if len(data) == 0 {
return ErrInvalidSession
}
// Parse the raw data into an OP struct
var op *OP
if err := g.Driver.Unmarshal(data, &op); err != nil {

15
state/state.go
View file

@ -81,6 +81,13 @@ func (s *State) AuthorDisplayName(message discord.Message) string {
return message.Author.Username
}
if message.Member != nil {
if message.Member.Nick != "" {
return message.Member.Nick
}
return message.Author.Username
}
n, err := s.MemberDisplayName(message.GuildID, message.Author.ID)
if err != nil {
return message.Author.Username
@ -109,6 +116,14 @@ func (s *State) AuthorColor(message discord.Message) discord.Color {
return discord.DefaultMemberColor
}
if message.Member != nil {
guild, err := s.Guild(message.GuildID)
if err != nil {
return discord.DefaultMemberColor
}
return discord.MemberColor(*guild, *message.Member)
}
return s.MemberColor(message.GuildID, message.Author.ID)
}