forked from chanbakjsd/TerraOceanPlugin
64 lines
1.5 KiB
Go
64 lines
1.5 KiB
Go
package backend
|
|
|
|
import (
|
|
"errors"
|
|
"strings"
|
|
|
|
"github.com/bwmarrin/discordgo"
|
|
|
|
"TerraOceanBot/discord/config"
|
|
)
|
|
|
|
var ErrMemberNotFound = errors.New("discord/backend: the requested member was not found")
|
|
|
|
func GetAllMembers(s *discordgo.Session) ([]*discordgo.Member, error) {
|
|
for _, guild := range s.State.Guilds {
|
|
if guild.ID == config.GuildID {
|
|
return guild.Members, nil
|
|
}
|
|
}
|
|
return nil, ErrMemberNotFound
|
|
}
|
|
|
|
func GetMemberFromUserFriendlyName(s *discordgo.Session, user string) (*discordgo.Member, error) {
|
|
allMembers, err := GetAllMembers(s)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if strings.HasPrefix(user, "@") {
|
|
user = strings.TrimPrefix(user, "@")
|
|
} else if strings.HasSuffix(user, ">") {
|
|
if strings.HasPrefix(user, "<@!") {
|
|
user = strings.TrimSuffix(strings.TrimPrefix(user, "<@!"), ">")
|
|
} else if strings.HasPrefix(user, "<@") {
|
|
user = strings.TrimSuffix(strings.TrimPrefix(user, "<@"), ">")
|
|
}
|
|
}
|
|
if strings.ContainsRune(user, '#') {
|
|
list := strings.Split(user, "#")
|
|
username := strings.Join(list[:len(list)-1], "#")
|
|
discriminator := list[len(list)-1]
|
|
for _, v := range allMembers {
|
|
if v.User.Username == username && v.User.Discriminator == discriminator {
|
|
return v, nil
|
|
}
|
|
}
|
|
}
|
|
for _, v := range allMembers {
|
|
if v.User.ID == user {
|
|
return v, nil
|
|
}
|
|
}
|
|
for _, v := range allMembers {
|
|
if v.Nick == user {
|
|
return v, nil
|
|
}
|
|
}
|
|
for _, v := range allMembers {
|
|
if v.User.Username == user {
|
|
return v, nil
|
|
}
|
|
}
|
|
return nil, ErrMemberNotFound
|
|
}
|