80 lines
2.4 KiB
Go
80 lines
2.4 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/jackc/pgx/v5/pgxpool"
|
|
)
|
|
|
|
var dbConn *pgxpool.Pool
|
|
|
|
func DbInit(ctx context.Context) {
|
|
var err error
|
|
dbConn, err = pgxpool.New(ctx, "postgres://master:secret@localhost:5432") // TODO change to env in production
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
_, err = dbConn.Exec(ctx, `
|
|
CREATE TABLE IF NOT EXISTS client (
|
|
id SERIAL PRIMARY KEY,
|
|
name VARCHAR(20) UNIQUE NOT NULL,
|
|
pass_hash VARCHAR(60) NOT NULL,
|
|
pronouns VARCHAR(15) DEFAULT NULL,
|
|
color_red SMALLINT DEFAULT NULL,
|
|
color_green SMALLINT DEFAULT NULL,
|
|
color_blue SMALLINT DEFAULT NULL,
|
|
created_at TIMESTAMP NOT NULL DEFAULT NOW()
|
|
)
|
|
`)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
_, err = dbConn.Exec(ctx, `
|
|
CREATE TABLE IF NOT EXISTS chat_groups (
|
|
id SERIAL PRIMARY KEY,
|
|
name VARCHAR(48) NOT NULL,
|
|
creator_id INTEGER NOT NULL REFERENCES client(id) ON DELETE CASCADE,
|
|
owner_id INTEGER NOT NULL REFERENCES client(id) ON DELETE CASCADE,
|
|
enable_user_colors BOOLEAN NOT NULL DEFAULT true,
|
|
color_red SMALLINT DEFAULT NULL,
|
|
color_green SMALLINT DEFAULT NULL,
|
|
color_blue SMALLINT DEFAULT NULL,
|
|
created_at TIMESTAMP NOT NULL DEFAULT NOW()
|
|
)
|
|
`)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
_, err = dbConn.Exec(ctx, `
|
|
CREATE TABLE IF NOT EXISTS chat_group_members (
|
|
group_id INTEGER NOT NULL REFERENCES chat_groups(id) ON DELETE CASCADE,
|
|
user_id INTEGER NOT NULL REFERENCES client(id) ON DELETE CASCADE,
|
|
joined_at TIMESTAMP NOT NULL DEFAULT NOW(),
|
|
PRIMARY KEY (group_id, user_id)
|
|
)
|
|
`)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func DbSaveClientWithoutGroups(ctx context.Context, client *Client) error {
|
|
err := dbConn.QueryRow(ctx, `
|
|
INSERT INTO clients (name, pass_hash, pronouns, color_red, color_green, color_blue, created_at)
|
|
VALUES ($1, $2, $3, $4, $5, $6, $7)
|
|
RETURNING id
|
|
`, client.Name, client.PasswordHash, client.Pronouns, client.Color[0], client.Color[1], client.Color[2], client.CreatedAt).
|
|
Scan(&client.Id)
|
|
return err
|
|
}
|
|
|
|
func DbSetClientByName(ctx context.Context, client *Client) error {
|
|
err := dbConn.QueryRow(ctx, `
|
|
SELECT name, pass_hash, color_red, color_green, color_blue, created_at FROM clients WHERE name = $1
|
|
`, client.Name).Scan(&client.Name, &client.PasswordHash, client.Pronouns, client.Color[0], client.Color[1], client.Color[2], client.CreatedAt)
|
|
return err
|
|
}
|