515 lines
14 KiB
Go
515 lines
14 KiB
Go
package main
|
|
|
|
import (
|
|
"log"
|
|
"os"
|
|
"slices"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
func initTables() {
|
|
createTableQuery := `CREATE TABLE IF NOT EXISTS tags (
|
|
tag_id TEXT NOT NULL,
|
|
tag_name TEXT NOT NULL,
|
|
tag_content TEXT NOT NULL,
|
|
guild_id TEXT NOT NULL,
|
|
PRIMARY KEY (tag_id, guild_id)
|
|
);
|
|
CREATE TABLE IF NOT EXISTS sticky (
|
|
message_id TEXT NOT NULL,
|
|
channel_id TEXT NOT NULL,
|
|
message_content TEXT NOT NULL,
|
|
guild_id TEXT NOT NULL,
|
|
PRIMARY KEY (channel_id, guild_id)
|
|
);
|
|
CREATE TABLE IF NOT EXISTS custom_forms (
|
|
form_type TEXT NOT NULL,
|
|
title TEXT NOT NULL,
|
|
json JSON NOT NULL,
|
|
guild_id TEXT NOT NULL,
|
|
PRIMARY KEY (form_type, guild_id)
|
|
);
|
|
CREATE TABLE IF NOT EXISTS form_manage (
|
|
form_manage_id TEXT NOT NULL,
|
|
form_type TEXT NOT NULL,
|
|
overwrite_title TEXT,
|
|
message_id TEXT NOT NULL,
|
|
channel_id TEXT NOT NULL,
|
|
guild_id TEXT NOT NULL,
|
|
result_channel_id TEXT NOT NULL,
|
|
accept_channel_id TEXT,
|
|
comment_category TEXT,
|
|
moderator_id TEXT,
|
|
PRIMARY KEY (form_manage_id, form_type)
|
|
);
|
|
CREATE TABLE IF NOT EXISTS autojoinroles (
|
|
guild_id TEXT NOT NULL,
|
|
bot_role TEXT,
|
|
user_role TEXT,
|
|
PRIMARY KEY (guild_id)
|
|
);
|
|
CREATE TABLE IF NOT EXISTS autopublish (
|
|
guild_id TEXT NOT NULL,
|
|
news_channel_id TEXT NOT NULL,
|
|
PRIMARY KEY (guild_id, news_channel_id)
|
|
);
|
|
CREATE TABLE IF NOT EXISTS blockpolls (
|
|
guild_id TEXT NOT NULL,
|
|
channel_id TEXT,
|
|
global BOOLEAN,
|
|
allowed_role TEXT,
|
|
PRIMARY KEY (guild_id)
|
|
)
|
|
`
|
|
_, err := db.Exec(createTableQuery)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
if slices.Contains(os.Args, "--form-db-update") {
|
|
_, err = db.Exec("ALTER TABLE blockpolls ADD global BOOLEAN;")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
_, err = db.Exec("ALTER TABLE blockpolls ADD allowed_role TEXT;")
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
}
|
|
|
|
type FormResult struct {
|
|
OverwriteTitle string
|
|
ResultChannelID string
|
|
AcceptChannelID string
|
|
CommentCategoryID string
|
|
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
|
|
id := uuid.New()
|
|
for exists {
|
|
id = uuid.New()
|
|
err := db.QueryRow("SELECT EXISTS (SELECT 1 FROM tags WHERE guild_id = $1 AND tag_id = $2)", guildID, id).Scan(&exists)
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
}
|
|
_, err := db.Exec("INSERT INTO tags (guild_id, tag_name, tag_content, tag_id) VALUES ($1, $2, $3, $4)", guildID, tagName, tagContent, id)
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
|
|
return exists
|
|
}
|
|
func removeTag(guildID string, tagID string) {
|
|
var exists bool
|
|
err := db.QueryRow("SELECT EXISTS (SELECT 1 FROM tags WHERE guild_id = $1 AND tag_id = $2)", guildID, tagID).Scan(&exists)
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
if exists {
|
|
_, err = db.Exec("DELETE FROM tags WHERE guild_id = $1 AND tag_id = $2", guildID, tagID)
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
}
|
|
}
|
|
func getTagIDs(guildID string) ([]string, error) {
|
|
var IDs []string
|
|
rows, err := db.Query("SELECT tag_id FROM tags WHERE guild_id = $1", guildID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer rows.Close()
|
|
|
|
for rows.Next() {
|
|
var id string
|
|
if err := rows.Scan(&id); err != nil {
|
|
return nil, err
|
|
}
|
|
IDs = append(IDs, id)
|
|
}
|
|
|
|
if err := rows.Err(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return IDs, nil
|
|
}
|
|
func getTagName(guildID string, tagID string) string {
|
|
var tagName string
|
|
db.QueryRow("SELECT tag_name FROM tags WHERE guild_id = $1 AND tag_id = $2", guildID, tagID).Scan(&tagName)
|
|
return tagName
|
|
}
|
|
func getTagContent(guildID string, tagID string) string {
|
|
var tagContent string
|
|
db.QueryRow("SELECT tag_content FROM tags WHERE guild_id = $1 AND tag_id = $2", guildID, tagID).Scan(&tagContent)
|
|
return tagContent
|
|
}
|
|
|
|
func addSticky(guildID string, channelID string, messageContent string, messageID string) {
|
|
_, err := db.Exec("INSERT INTO sticky (guild_id, channel_id, message_id, message_content) VALUES ($1, $2, $3, $4)", guildID, channelID, messageID, messageContent)
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
|
|
}
|
|
|
|
func hasSticky(guildID string, channelID string) bool {
|
|
var exists bool
|
|
err := db.QueryRow("SELECT EXISTS (SELECT 1 FROM sticky WHERE guild_id = $1 AND channel_id = $2)", guildID, channelID).Scan(&exists)
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
return exists
|
|
}
|
|
|
|
func getStickyMessageID(guildID string, channelID string) string {
|
|
var messageID string
|
|
exists := hasSticky(guildID, channelID)
|
|
if exists {
|
|
err := db.QueryRow("SELECT message_id FROM sticky WHERE guild_id = $1 AND channel_id = $2", guildID, channelID).Scan(&messageID)
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
}
|
|
return messageID
|
|
}
|
|
func getStickyMessageContent(guildID string, channelID string) string {
|
|
var messageID string
|
|
exists := hasSticky(guildID, channelID)
|
|
if exists {
|
|
err := db.QueryRow("SELECT message_content FROM sticky WHERE guild_id = $1 AND channel_id = $2", guildID, channelID).Scan(&messageID)
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
}
|
|
return messageID
|
|
}
|
|
|
|
func updateStickyMessageID(guildID string, channelID string, messageID string) {
|
|
exists := hasSticky(guildID, channelID)
|
|
if exists {
|
|
_, err := db.Exec("UPDATE sticky SET message_id = $1 WHERE guild_id = $2 AND channel_id = $3", messageID, guildID, channelID)
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func removeSticky(guildID string, channelID string) {
|
|
_, err := db.Exec("DELETE FROM sticky WHERE guild_id = $1 AND channel_id = $2", guildID, channelID)
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
}
|
|
|
|
func getFormManageIdExists(id uuid.UUID) bool {
|
|
var exists bool
|
|
err := db.QueryRow("SELECT EXISTS (SELECT 1 FROM form_manage WHERE form_manage_id = $1)", id).Scan(&exists)
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
return exists
|
|
}
|
|
|
|
func addFormButton(guildID string, channelID string, messageID string, formManageID string, formType string, resultChannelID string, overwriteTitle string, acceptChannelID string, commentCategory string, moderator_id string) {
|
|
_, err := db.Exec(
|
|
`INSERT INTO form_manage (
|
|
guild_id,
|
|
form_manage_id,
|
|
channel_id,
|
|
message_id,
|
|
form_type,
|
|
result_channel_id,
|
|
overwrite_title,
|
|
accept_channel_id,
|
|
comment_category,
|
|
moderator_id) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10)`,
|
|
guildID, formManageID, channelID, messageID, formType, resultChannelID, overwriteTitle, acceptChannelID, commentCategory, moderator_id)
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
}
|
|
|
|
func getFormManageIDs() []string {
|
|
if db == nil {
|
|
return nil
|
|
}
|
|
var IDs []string
|
|
rows, err := db.Query("SELECT form_manage_id FROM form_manage")
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
return nil
|
|
}
|
|
defer rows.Close()
|
|
|
|
for rows.Next() {
|
|
var id string
|
|
if err := rows.Scan(&id); err != nil {
|
|
logrus.Error(err)
|
|
return nil
|
|
}
|
|
IDs = append(IDs, id)
|
|
}
|
|
|
|
if err := rows.Err(); err != nil {
|
|
logrus.Error(err)
|
|
return nil
|
|
}
|
|
return IDs
|
|
}
|
|
|
|
func getFormType(formManageID string) string {
|
|
var formType string
|
|
err := db.QueryRow("SELECT form_type FROM form_manage WHERE form_manage_id = $1", formManageID).Scan(&formType)
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
return formType
|
|
}
|
|
|
|
func getFormResultValues(formManageID string) FormResult {
|
|
var result FormResult
|
|
err := db.QueryRow("SELECT overwrite_title FROM form_manage WHERE form_manage_id = $1", formManageID).Scan(&result.OverwriteTitle)
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
err = db.QueryRow("SELECT result_channel_id FROM form_manage WHERE form_manage_id = $1", formManageID).Scan(&result.ResultChannelID)
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
err = db.QueryRow("SELECT accept_channel_id FROM form_manage WHERE form_manage_id = $1", formManageID).Scan(&result.AcceptChannelID)
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
err = db.QueryRow("SELECT comment_category FROM form_manage WHERE form_manage_id = $1", formManageID).Scan(&result.CommentCategoryID)
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
err = db.QueryRow("SELECT moderator_id FROM form_manage WHERE form_manage_id = $1", formManageID).Scan(&result.ModeratorID)
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
return result
|
|
}
|
|
|
|
func getFormOverwriteTitle(formManageID string) string {
|
|
var overwriteTitle string
|
|
err := db.QueryRow("SELECT overwrite_title FROM form_manage WHERE form_manage_id = $1", formManageID).Scan(&overwriteTitle)
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
return overwriteTitle
|
|
}
|
|
|
|
func updateFormCommentCategory(formManageID string, comment_category string) {
|
|
_, err := db.Exec("UPDATE form_manage SET comment_category = $1 WHERE form_manage_id = $2", comment_category, formManageID)
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
}
|
|
|
|
func setAutoJoinRole(guildID string, option string, roleID string) bool {
|
|
var role_exists bool
|
|
var autojoinroles_exists bool
|
|
err := db.QueryRow("SELECT EXISTS (SELECT 1 FROM autojoinroles WHERE guild_id = $1)", guildID).Scan(&autojoinroles_exists)
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
err = db.QueryRow("SELECT EXISTS (SELECT 1 FROM autojoinroles WHERE guild_id = $1 AND "+option+"_role IS NOT NULL AND "+option+"_role != '')", guildID).Scan(&role_exists)
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
if autojoinroles_exists {
|
|
_, err = db.Exec("UPDATE autojoinroles SET "+option+"_role = $1 WHERE guild_id = $2", roleID, guildID)
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
} else {
|
|
_, err = db.Exec("INSERT INTO autojoinroles (guild_id, "+option+"_role) VALUES ($1, $2)", guildID, roleID)
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
}
|
|
return role_exists
|
|
}
|
|
|
|
func purgeUnusedAutoJoinRoles(guildID string) {
|
|
_, err := db.Exec("DELETE FROM autojoinroles WHERE guild_id = $1 AND user_role = '' OR user_role IS NULL AND bot_role = '' OR bot_role IS NULL", guildID)
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
}
|
|
|
|
func getAutoJoinRole(guildID string, isBot bool) string {
|
|
var isBotString string
|
|
var role string
|
|
if isBot {
|
|
isBotString = "bot"
|
|
} else {
|
|
isBotString = "user"
|
|
}
|
|
var exists bool
|
|
err := db.QueryRow("SELECT EXISTS (SELECT 1 FROM autojoinroles WHERE guild_id = $1)", guildID).Scan(&exists)
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
return role
|
|
}
|
|
if exists {
|
|
err = db.QueryRow("SELECT "+isBotString+"_role FROM autojoinroles WHERE guild_id = $1", guildID).Scan(&role)
|
|
if err != nil {
|
|
logrus.Error(err, guildID)
|
|
}
|
|
}
|
|
return role
|
|
}
|
|
|
|
func toggleAutoPublish(guildID string, newsChannelID string) bool {
|
|
var exists bool
|
|
err := db.QueryRow("SELECT EXISTS (SELECT 1 FROM autopublish WHERE guild_id = $1 AND news_channel_id = $2)", guildID, newsChannelID).Scan(&exists)
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
if exists {
|
|
_, err := db.Exec("DELETE FROM autopublish WHERE guild_id = $1 AND news_channel_id = $2", guildID, newsChannelID)
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
} else {
|
|
_, err := db.Exec("INSERT INTO autopublish (guild_id, news_channel_id) VALUES ($1, $2)", guildID, newsChannelID)
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
}
|
|
return exists
|
|
}
|
|
|
|
func isAutopublishEnabled(guildID string, newsChannelID string) bool {
|
|
var enabled bool
|
|
err := db.QueryRow("SELECT EXISTS (SELECT 1 FROM autopublish WHERE guild_id = $1 AND news_channel_id = $2)", guildID, newsChannelID).Scan(&enabled)
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
return enabled
|
|
}
|
|
|
|
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 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)
|
|
}
|
|
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
|
|
}
|
|
}
|
|
|
|
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)
|
|
}
|
|
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) {
|
|
_, err := db.Exec("DELETE FROM form_manage WHERE message_id = $1", messageID)
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
}
|
|
|
|
func getAllSavedMessages() []MessageIDs {
|
|
var savedMessages []MessageIDs
|
|
rows, err := db.Query("SELECT message_id, channel_id FROM form_manage")
|
|
if err != nil {
|
|
logrus.Error(err)
|
|
return nil
|
|
}
|
|
defer rows.Close()
|
|
for rows.Next() {
|
|
var messageID, channelID string
|
|
if err := rows.Scan(&messageID, &channelID); err != nil {
|
|
logrus.Error(err)
|
|
continue
|
|
}
|
|
savedMessages = append(savedMessages, MessageIDs{ID: messageID, ChannelID: channelID})
|
|
}
|
|
if err := rows.Err(); err != nil {
|
|
logrus.Error(err)
|
|
}
|
|
return savedMessages
|
|
}
|