38 lines
973 B
Go
38 lines
973 B
Go
package table
|
|
|
|
import (
|
|
"github.com/TicketsBot/GoPanel/database"
|
|
)
|
|
|
|
type BlacklistNode struct {
|
|
Assoc int `gorm:"column:ASSOCID;type:int;primary_key;auto_increment"`
|
|
Guild int64 `gorm:"column:GUILDID"`
|
|
User int64 `gorm:"column:USERID"`
|
|
}
|
|
|
|
func (BlacklistNode) TableName() string {
|
|
return "blacklist"
|
|
}
|
|
|
|
func IsBlacklisted(guildId, userId int64) bool {
|
|
var count int
|
|
database.Database.Table("blacklist").Where(&BlacklistNode{Guild: guildId, User: userId}).Count(&count)
|
|
return count > 0
|
|
}
|
|
|
|
func AddBlacklist(guildId, userId int64) {
|
|
database.Database.Create(&BlacklistNode{Guild: guildId, User: userId})
|
|
}
|
|
|
|
func RemoveBlacklist(guildId, userId int64) {
|
|
var node BlacklistNode
|
|
database.Database.Where(BlacklistNode{Guild: guildId, User: userId}).Take(&node)
|
|
database.Database.Delete(&node)
|
|
}
|
|
|
|
func GetBlacklistNodes(guildId int64) []BlacklistNode {
|
|
var nodes []BlacklistNode
|
|
database.Database.Where(&BlacklistNode{Guild: guildId}).Find(&nodes)
|
|
return nodes
|
|
}
|