71 lines
1.5 KiB
Go
71 lines
1.5 KiB
Go
package utils
|
|
|
|
import (
|
|
"github.com/TicketsBot/GoPanel/config"
|
|
"github.com/TicketsBot/GoPanel/database/table"
|
|
"github.com/TicketsBot/GoPanel/utils/discord/endpoints/guild"
|
|
"github.com/TicketsBot/GoPanel/utils/discord/objects"
|
|
"github.com/apex/log"
|
|
"github.com/gin-gonic/contrib/sessions"
|
|
"github.com/robfig/go-cache"
|
|
"strconv"
|
|
"time"
|
|
)
|
|
|
|
var roleCache = cache.New(time.Minute, time.Minute)
|
|
|
|
func IsAdmin(store sessions.Session, guild objects.Guild, guildId, userId int64, res chan bool) {
|
|
if Contains(config.Conf.Admins, strconv.Itoa(int(userId))) {
|
|
res <- true
|
|
}
|
|
|
|
if guild.Owner {
|
|
res <- true
|
|
}
|
|
|
|
if table.IsAdmin(guildId, userId) {
|
|
res <- true
|
|
}
|
|
|
|
if guild.Permissions & 0x8 != 0 {
|
|
res <- true
|
|
}
|
|
|
|
userRolesChan := make(chan []int64)
|
|
go table.GetCachedRoles(guildId, userId, userRolesChan)
|
|
userRoles := <-userRolesChan
|
|
|
|
adminRolesChan := make(chan []int64)
|
|
go table.GetAdminRoles(strconv.Itoa(int(guildId)), adminRolesChan)
|
|
adminRoles := <- adminRolesChan
|
|
|
|
hasAdminRole := false
|
|
for _, userRole := range userRoles {
|
|
for _, adminRole := range adminRoles {
|
|
if userRole == adminRole {
|
|
hasAdminRole = true
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
if hasAdminRole {
|
|
res <- true
|
|
}
|
|
|
|
res <- false
|
|
}
|
|
|
|
func GetRolesRest(store sessions.Session, guildId, userId int64) *[]int64 {
|
|
var member objects.Member
|
|
endpoint := guild.GetGuildMember(int(guildId), int(userId))
|
|
|
|
if err, _ := endpoint.Request(store, nil, nil, &member); err != nil {
|
|
log.Error(err.Error())
|
|
return nil
|
|
}
|
|
|
|
roles := []int64(member.Roles)
|
|
return &roles
|
|
}
|