You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
117 lines
3.6 KiB
117 lines
3.6 KiB
package utils
|
|
|
|
import (
|
|
"database/sql"
|
|
"fmt"
|
|
_ "github.com/lib/pq"
|
|
)
|
|
|
|
var (
|
|
Database *sql.DB
|
|
)
|
|
|
|
func ConnectDatabase(dbConnString string) {
|
|
db, err := sql.Open("postgres", dbConnString)
|
|
if err != nil {
|
|
panic(fmt.Sprintf("Can't connect to the database. %v", err))
|
|
} else {
|
|
fmt.Println("Database Connected.")
|
|
}
|
|
Database = db
|
|
}
|
|
|
|
func InitializeDatabase() {
|
|
_, err := Database.Query("CREATE TABLE IF NOT EXISTS users(" +
|
|
"id varchar(30) primary key," +
|
|
"banned bool not null default false," +
|
|
"logging bool not null default true," +
|
|
"steam_id varchar(30) NOT NULL DEFAULT ''," +
|
|
"is_active bool not null default true," +
|
|
"is_staff bool not null default false," +
|
|
"is_admin bool not null default false" +
|
|
")")
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
_, err = Database.Query("CREATE TABLE IF NOT EXISTS guilds(" +
|
|
"id varchar(30) primary key," +
|
|
"welcome_message varchar(1000)," +
|
|
"goodbye_message varchar(1000)," +
|
|
"logging_channel varchar(30) NOT NULL DEFAULT ''," +
|
|
"welcome_channel varchar(30) NOT NULL DEFAULT ''" +
|
|
")")
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
_, err = Database.Query("CREATE TABLE IF NOT EXISTS prefixes(" +
|
|
"id serial primary key," +
|
|
"prefix varchar(10) not null unique default 'Go.'" +
|
|
")")
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
_, err = Database.Query("CREATE TABLE IF NOT EXISTS tags(" +
|
|
"id serial primary key," +
|
|
"tag varchar(100) not null unique," +
|
|
"content varchar(1000) not null," +
|
|
"creator varchar(30) not null references users(id)," +
|
|
"creation_time timestamp not null default NOW()," +
|
|
"guild_id varchar(30) not null references guilds(id)" +
|
|
")")
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
_, err = Database.Query("CREATE TABLE IF NOT EXISTS x_users_guilds(" +
|
|
"guild_id varchar(30) not null references guilds(id)," +
|
|
"user_id varchar(30) not null references users(id)" +
|
|
")")
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
_, err = Database.Query("CREATE TABLE IF NOT EXISTS x_guilds_prefixes(" +
|
|
"guild_id varchar(30) not null references guilds(id)," +
|
|
"prefix_id int not null references prefixes(id)" +
|
|
")")
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
}
|
|
|
|
func LoadTestData() {
|
|
_, err := Database.Query("INSERT INTO users (id, banned, logging, steam_id, is_active, is_staff, is_admin) values " +
|
|
"('351794468870946827', false, true, '76561198024193239', true, true, true)," +
|
|
"('692908139506434065', false, true, '', true, false, false)," +
|
|
"('396588996706304010', false, true, '', true, true, false)")
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
_, err = Database.Query("INSERT INTO guilds (id, welcome_message, goodbye_message) VALUES " +
|
|
"('265828729970753537', 'Hey there is someone new here.', 'Well fine then... Just leave without saying goodbye')")
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
_, err = Database.Query("INSERT INTO prefixes (prefix) VALUES ('Go.'), ('go.'), ('go,')")
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
_, err = Database.Query("INSERT INTO x_users_guilds (guild_id, user_id) VALUES " +
|
|
"('265828729970753537', '351794468870946827')," +
|
|
"('265828729970753537', '692908139506434065')," +
|
|
"('265828729970753537', '396588996706304010')")
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
_, err = Database.Query("INSERT INTO x_guilds_prefixes (guild_id, prefix_id) VALUES " +
|
|
"('265828729970753537', 1)," +
|
|
"('265828729970753537', 2)," +
|
|
"('265828729970753537', 3)")
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
_, err = Database.Query("INSERT INTO tags (tag, content, creator, guild_id) VALUES " +
|
|
"('test', 'This is a test of the tag system', '351794468870946827', '265828729970753537')")
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
}
|