vaporvee/acecore
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

set permission requirements for more commands

Browse Source
This commit is contained in:
Yannik
2024-04-14 02:46:34 +02:00
parent a1fd95cefe
commit 357355b18e
5 changed files with 151 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

85
manage_data.go
View File

@@ -57,8 +57,10 @@ func initTables() {
);
CREATE TABLE IF NOT EXISTS blockpolls (
guild_id TEXT NOT NULL,
channel_id TEXT NOT NULL,
PRIMARY KEY (guild_id, channel_id)
channel_id TEXT,
global BOOLEAN,
allowed_role TEXT,
PRIMARY KEY (guild_id)
)
`
_, err := db.Exec(createTableQuery)
@@ -66,15 +68,11 @@ func initTables() {
log.Fatal(err)
}
if slices.Contains(os.Args, "--form-db-update") {
_, err := db.Exec("ALTER TABLE form_manage ADD moderator_id TEXT;")
_, err = db.Exec("ALTER TABLE blockpolls ADD global BOOLEAN;")
if err != nil {
log.Fatal(err)
}
_, err = db.Exec("ALTER TABLE form_manage ADD comment_category TEXT;")
if err != nil {
log.Fatal(err)
}
_, err = db.Exec("ALTER TABLE form_manage DROP COLUMN mods_can_comment;")
_, err = db.Exec("ALTER TABLE blockpolls ADD allowed_role TEXT;")
if err != nil {
log.Fatal(err)
}
@@ -89,6 +87,12 @@ type FormResult struct {
ModeratorID string
}
type BlockPoll struct {
ChannelID string
Global bool
AllowedRole string
}
func addTag(guildID, tagName, tagContent string) bool {
var exists bool = true
//TODO: add modify command
@@ -400,33 +404,84 @@ func isAutopublishEnabled(guildID string, newsChannelID string) bool {
return enabled
}
func toggleBlockPolls(guildID string, channelID string) bool {
func isGlobalBlockPolls(guildID string) bool {
var globalexists bool
err := db.QueryRow("SELECT EXISTS (SELECT 1 FROM blockpolls WHERE guild_id = $1 AND global = true)", guildID).Scan(&globalexists)
if err != nil {
logrus.Error(err)
}
return globalexists
}
func toggleBlockPolls(guildID string, channelID string, global bool, allowedRole string) (e bool, isGlobal bool) {
globalexists := isGlobalBlockPolls(guildID)
var exists bool
err := db.QueryRow("SELECT EXISTS (SELECT 1 FROM blockpolls WHERE guild_id = $1 AND channel_id = $2)", guildID, channelID).Scan(&exists)
if err != nil {
logrus.Error(err)
}
if exists {
if globalexists {
_, err := db.Exec("DELETE FROM blockpolls WHERE guild_id = $1 AND global = true", guildID)
if err != nil {
logrus.Error(err)
}
return true, true
} else if global {
_, err = db.Exec("DELETE FROM blockpolls WHERE guild_id = $1", guildID)
if err != nil {
logrus.Error(err)
}
_, err := db.Exec("INSERT INTO blockpolls (guild_id, global, channel_id, allowed_role) VALUES ($1, true, $2, $3)", guildID, channelID, allowedRole)
if err != nil {
logrus.Error(err)
}
return false, true
} else if exists && !globalexists {
_, err := db.Exec("DELETE FROM blockpolls WHERE guild_id = $1 AND channel_id = $2", guildID, channelID)
if err != nil {
logrus.Error(err)
}
} else {
_, err := db.Exec("INSERT INTO blockpolls (guild_id, channel_id) VALUES ($1, $2)", guildID, channelID)
return true, false
} else if !globalexists {
_, err := db.Exec("INSERT INTO blockpolls (guild_id, channel_id, allowed_role) VALUES ($1, $2, $3)", guildID, channelID, allowedRole)
if err != nil {
logrus.Error(err)
}
return false, false
} else {
return false, false
}
return exists
}
func isBlockPollsEnabled(guildID string, channelID string) bool {
func listBlockPolls(guildID string) []BlockPoll {
var list []BlockPoll
rows, err := db.Query("SELECT channel_id, global, allowed_role FROM blockpolls WHERE guild_id = $1", guildID)
if err != nil {
log.Fatal(err)
}
for rows.Next() {
var bp BlockPoll
err := rows.Scan(&bp.ChannelID, &bp.Global, &bp.AllowedRole)
if err != nil {
log.Fatal(err)
}
list = append(list, bp)
}
return list
}
func getBlockPollsEnabled(guildID string, channelID string) (isEnabled bool, allowedRole string) {
var enabled bool
var v_allowedRole string
err := db.QueryRow("SELECT EXISTS (SELECT 1 FROM blockpolls WHERE guild_id = $1 AND channel_id = $2)", guildID, channelID).Scan(&enabled)
if err != nil {
logrus.Error(err)
}
return enabled
err = db.QueryRow("SELECT allowed_role FROM blockpolls WHERE guild_id = $1 AND channel_id = $2", guildID, channelID).Scan(&v_allowedRole)
if err != nil && err.Error() != "sql: no rows in result set" {
logrus.Error(err)
}
return enabled, v_allowedRole
}
func tryDeleteUnusedMessage(messageID string) {