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

using UUIDs for tag command

Browse Source
This commit is contained in:
Yannik
2024-02-19 12:40:19 +01:00
parent 8fb88677c3
commit f98b55480c
5 changed files with 58 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

70
manage_data.go
View File

@@ -2,68 +2,92 @@ package main
import (
"log"
"github.com/google/uuid"
)
func initTable() {
createTableQuery := `CREATE TABLE IF NOT EXISTS tags (
tag_id TEXT NOT NULL,
tag_name TEXT NOT NULL,
tag_content TEXT,
guild_id TEXT NOT NULL,
PRIMARY KEY (tag_id,guild_id)
);`
_, err := db.Exec(createTableQuery)
if err != nil {
log.Fatal(err)
}
}
func addTag(guildID, tagName, tagContent string) bool {
var exists bool
err := db.QueryRow("SELECT EXISTS (SELECT 1 FROM tags WHERE guild_id = $1 AND tag_name = $2)", guildID, tagName).Scan(&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 {
log.Println(err)
}
// If the tag exists it updates it but TODO: needs to return a discord message to use the modify command with autocomplete
if exists {
_, err = db.Exec("UPDATE tags SET tag_content = $1 WHERE guild_id = $2 AND tag_name = $3", tagContent, guildID, tagName)
if err != nil {
log.Println(err)
}
} else {
_, err = db.Exec("INSERT INTO tags (guild_id, tag_name, tag_content) VALUES ($1, $2, $3)", guildID, tagName, tagContent)
// TODO: add modify command
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 {
log.Println(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 {
log.Println(err)
}
return exists
}
func removeTag(guildID, tagContent string) {
func removeTag(guildID string, tagID string) {
var exists bool
err := db.QueryRow("SELECT EXISTS (SELECT 1 FROM tags WHERE guild_id = $1 AND tag_content = $2)", guildID, tagContent).Scan(&exists) // that is so dumb next commit i will make the tag IDs
err := db.QueryRow("SELECT EXISTS (SELECT 1 FROM tags WHERE guild_id = $1 AND tag_id = $2)", guildID, tagID).Scan(&exists)
if err != nil {
log.Println(err)
}
if exists {
_, err = db.Exec("DELETE FROM tags WHERE guild_id = $1 AND tag_content = $2", guildID, tagContent)
_, err = db.Exec("DELETE FROM tags WHERE guild_id = $1 AND tag_id = $2", guildID, tagID)
if err != nil {
log.Println(err)
}
}
}
func getTagKeys(guildID string) ([]string, error) {
var keys []string
rows, err := db.Query("SELECT tag_name FROM tags WHERE guild_id = $1", guildID)
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 key string
if err := rows.Scan(&key); err != nil {
var id string
if err := rows.Scan(&id); err != nil {
return nil, err
}
keys = append(keys, key)
IDs = append(IDs, id)
}
if err := rows.Err(); err != nil {
return nil, err
}
return keys, nil
return IDs, nil
}
func getTag(guildID, tagName string) (string, error) {
var tagContent string
err := db.QueryRow("SELECT tag_content FROM tags WHERE guild_id = $1 AND tag_name = $2", guildID, tagName).Scan(&tagContent)
return tagContent, err
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
}