You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
95 lines
2.8 KiB
95 lines
2.8 KiB
package disgoman
|
|
|
|
/* helpers.go:
|
|
* Utility functions to make my life easier.
|
|
*
|
|
* Disgoman (c) 2020 Dusty.P/dustinpianalto
|
|
*/
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/bwmarrin/discordgo"
|
|
"sort"
|
|
)
|
|
|
|
// GetDefaultStatusManager returns a default Status Manager
|
|
func GetDefaultStatusManager() StatusManager {
|
|
return StatusManager{
|
|
[]string{
|
|
"Golang!",
|
|
"DiscordGo!",
|
|
"Disgoman!",
|
|
}, "10s"}
|
|
}
|
|
|
|
// HasHigherRole checks if the caller has a higher role than the target in the given guild
|
|
func HasHigherRole(session *discordgo.Session, guildID string, callerID string, targetID string) bool {
|
|
caller, _ := session.GuildMember(guildID, callerID)
|
|
target, _ := session.GuildMember(guildID, targetID)
|
|
var callerRoles []*discordgo.Role
|
|
for _, roleID := range caller.Roles {
|
|
role, _ := session.State.Role(guildID, roleID)
|
|
callerRoles = append(callerRoles, role)
|
|
}
|
|
sort.Slice(callerRoles, func(i, j int) bool { return callerRoles[i].Position > callerRoles[j].Position })
|
|
var targetRoles []*discordgo.Role
|
|
for _, roleID := range target.Roles {
|
|
role, _ := session.State.Role(guildID, roleID)
|
|
targetRoles = append(targetRoles, role)
|
|
}
|
|
sort.Slice(targetRoles, func(i, j int) bool { return targetRoles[i].Position > targetRoles[j].Position })
|
|
if callerRoles[0].Position > targetRoles[0].Position {
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
// CheckPermissions checks the channel and guild permissions to see if the member has the needed permissions
|
|
func CheckPermissions(session *discordgo.Session, memberID string, channel discordgo.Channel, perms Permission) bool {
|
|
if perms == 0 {
|
|
return true // If no permissions are required then just return true
|
|
}
|
|
|
|
for _, overwrite := range channel.PermissionOverwrites {
|
|
if overwrite.ID == memberID {
|
|
if overwrite.Allow&int(perms) != 0 {
|
|
return true // If the channel has an overwrite for the user then true
|
|
} else if overwrite.Deny&int(perms) != 0 {
|
|
return false // If there is an explicit deny then false
|
|
}
|
|
}
|
|
}
|
|
|
|
member, err := session.State.Member(channel.GuildID, memberID)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
return false
|
|
}
|
|
for _, roleID := range member.Roles {
|
|
role, err := session.State.Role(channel.GuildID, roleID)
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
|
|
for _, overwrite := range channel.PermissionOverwrites {
|
|
if overwrite.ID == roleID {
|
|
if overwrite.Allow&int(perms) != 0 {
|
|
return true // If the channel has an overwrite for the role then true
|
|
} else if overwrite.Deny&int(perms) != 0 {
|
|
return false // If there is an explicit deny then false
|
|
}
|
|
}
|
|
}
|
|
|
|
if role.Permissions&int(PermissionAdministrator) != 0 {
|
|
return true // If they are an administrator then they automatically have all permissions
|
|
}
|
|
|
|
if role.Permissions&int(perms) != 0 {
|
|
return true // The role has the required permissions
|
|
}
|
|
}
|
|
|
|
return false // Default to false
|
|
}
|