add simple ws logger
This commit is contained in:
GitProtogen
-216
@@ -1,216 +0,0 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
|
||||
"golang.org/x/crypto/bcrypt"
|
||||
|
||||
"github.com/jackc/pgx/v5"
|
||||
)
|
||||
|
||||
var dbConnection *pgx.Conn
|
||||
|
||||
func InitDatabase(ctx context.Context) {
|
||||
conn, err := pgx.Connect(ctx, "postgres://master:secret@localhost:5432")
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
_, err = conn.Exec(ctx, `
|
||||
CREATE TABLE IF NOT EXISTS users (
|
||||
id SERIAL PRIMARY KEY,
|
||||
name VARCHAR(20) UNIQUE NOT NULL,
|
||||
pass_hash VARCHAR(60) NOT NULL,
|
||||
color VARCHAR(3) DEFAULT NULL
|
||||
);
|
||||
CREATE TABLE IF NOT EXISTS chat_groups (
|
||||
id SERIAL PRIMARY KEY,
|
||||
name VARCHAR(48) NOT NULL,
|
||||
creator_id INTEGER NOT NULL REFERENCES users(id) ON DELETE CASCADE,
|
||||
owner_id INTEGER NOT NULL REFERENCES users(id) ON DELETE CASCADE,
|
||||
enable_user_colors BOOLEAN NOT NULL DEFAULT true,
|
||||
group_color VARCHAR(3),
|
||||
created_at TIMESTAMP NOT NULL DEFAULT NOW()
|
||||
);
|
||||
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 users(id) ON DELETE CASCADE,
|
||||
joined_at TIMESTAMP NOT NULL DEFAULT NOW(),
|
||||
PRIMARY KEY (group_id, user_id)
|
||||
);
|
||||
`)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
dbConnection = conn
|
||||
InitCache()
|
||||
}
|
||||
|
||||
func AddNewUser(ctx context.Context, user *User) (uint64, error) {
|
||||
var id uint64
|
||||
var err error
|
||||
|
||||
if len(user.Name) == 0 || len(user.Name) > 20 {
|
||||
return 0, errors.New("username bad length")
|
||||
}
|
||||
if user.IsPasswordHashed {
|
||||
if len(user.Password) != 60 {
|
||||
return 0, errors.New("password bad length")
|
||||
}
|
||||
} else {
|
||||
var hashed []byte
|
||||
hashed, err = bcrypt.GenerateFromPassword([]byte(user.Password), bcrypt.DefaultCost)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
user.Password = string(hashed)
|
||||
}
|
||||
if user.Color == ([3]byte{}) {
|
||||
user.Color = [3]byte{'x', 'x', 'x'}
|
||||
}
|
||||
c := string(user.Color[:])
|
||||
err = dbConnection.QueryRow(ctx, `
|
||||
INSERT INTO users (name, pass_hash, color)
|
||||
VALUES ($1, $2, $3)
|
||||
RETURNING id
|
||||
`, user.Name, user.Password, c).Scan(&id)
|
||||
return id, err
|
||||
}
|
||||
|
||||
func isPassValid(ctx context.Context, id uint64, plainPassword string) bool {
|
||||
var controlHash string
|
||||
err := dbConnection.QueryRow(ctx, "SELECT pass_hash FROM users WHERE id = $1", id).Scan(&controlHash)
|
||||
if err != nil {
|
||||
return false
|
||||
}
|
||||
|
||||
return bcrypt.CompareHashAndPassword([]byte(controlHash), []byte(plainPassword)) == nil
|
||||
}
|
||||
|
||||
func GetAllUsers(ctx context.Context) ([]User, error) {
|
||||
rows, err := dbConnection.Query(ctx, "SELECT id, name, color FROM users")
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
var users []User
|
||||
for rows.Next() {
|
||||
var user User
|
||||
if err := rows.Scan(&user.Id, &user.Name, &user.Color); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
users = append(users, user)
|
||||
}
|
||||
return users, rows.Err()
|
||||
}
|
||||
|
||||
func GetUserDataById(ctx context.Context, id uint64) (User, error) {
|
||||
var user User
|
||||
err := dbConnection.QueryRow(ctx, "SELECT id, name, pass_hash, color FROM users WHERE id = $1", id).
|
||||
Scan(&user.Id, &user.Name, &user.Password, &user.Color)
|
||||
if err != nil {
|
||||
return User{}, err
|
||||
}
|
||||
user.IsPasswordHashed = true
|
||||
return user, nil
|
||||
}
|
||||
func GetUserDataByName(ctx context.Context, name string) (User, error) {
|
||||
var user User
|
||||
err := dbConnection.QueryRow(ctx, "SELECT id, name, pass_hash, color FROM users WHERE name = $1", name).
|
||||
Scan(&user.Id, &user.Name, &user.Password, &user.Color)
|
||||
if err != nil {
|
||||
return User{}, err
|
||||
}
|
||||
user.IsPasswordHashed = true
|
||||
return user, nil
|
||||
}
|
||||
|
||||
func CreateChatGroupWithoutMembers(ctx context.Context, group *ChatGroup) (uint64, error) {
|
||||
if len(group.Name) < 1 {
|
||||
return 0, errors.New("group name too short")
|
||||
}
|
||||
if len(group.Name) > 48 {
|
||||
return 0, errors.New("group name too long")
|
||||
}
|
||||
|
||||
var id uint64
|
||||
err := dbConnection.QueryRow(ctx, `INSERT INTO chat_groups (name, creator_id, owner_id, created_at )
|
||||
VALUES ($1, $2, $3, $4)
|
||||
RETURNING id
|
||||
`, group.Name, group.CreatorId, group.OwnerId, group.CreatedAt).Scan(&id)
|
||||
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
|
||||
AddOrUpdateGroupToCache(&mu, *group)
|
||||
return id, err
|
||||
}
|
||||
|
||||
func GetAllChatGroups(ctx context.Context) ([]ChatGroup, error) {
|
||||
rows, err := dbConnection.Query(ctx, "SELECT id, name, creator_id, owner_id, enable_user_colors, group_color, created_at FROM chat_groups")
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
var groups []ChatGroup
|
||||
for rows.Next() {
|
||||
var group ChatGroup
|
||||
if err := rows.Scan(&group.Id, &group.Name, &group.CreatorId, &group.OwnerId, &group.EnableUserColors, &group.Color, &group.CreatedAt); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
groups = append(groups, group)
|
||||
}
|
||||
return groups, rows.Err()
|
||||
}
|
||||
|
||||
func GetChatGroupWithoutMembers(ctx context.Context, id uint64) (ChatGroup, error) {
|
||||
var group ChatGroup
|
||||
err := dbConnection.QueryRow(ctx, `SELECT name, creator_id, owner_id, enable_user_colors, group_color, created_at FROM chat_groups WHERE id = $1`,
|
||||
id).Scan(&group.Name, &group.CreatorId, &group.OwnerId, &group.EnableUserColors, &group.Color, &group.CreatedAt)
|
||||
return group, err
|
||||
}
|
||||
|
||||
func GetUserMemberGroupIds(ctx context.Context, userId uint64) ([]uint64, error) {
|
||||
rows, err := dbConnection.Query(ctx, "SELECT group_id FROM chat_group_members WHERE user_id = $1", userId)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
var groupIds []uint64
|
||||
for rows.Next() {
|
||||
var groupId uint64
|
||||
if err := rows.Scan(&groupId); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
groupIds = append(groupIds, groupId)
|
||||
}
|
||||
return groupIds, rows.Err()
|
||||
}
|
||||
|
||||
func GetChatGroupMembers(ctx context.Context, groupId uint64) ([]User, error) {
|
||||
rows, err := dbConnection.Query(ctx, `
|
||||
SELECT usr.id, usr.name, usr.color FROM users usr
|
||||
JOIN chat_group_members members ON usr.id = members.user_id
|
||||
WHERE members.group_id = $1
|
||||
`, groupId)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
var members []User
|
||||
for rows.Next() {
|
||||
var user User
|
||||
if err := rows.Scan(&user.Id, &user.Name, &user.Color); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
members = append(members, user)
|
||||
}
|
||||
return members, rows.Err()
|
||||
}
|
||||
Reference in New Issue
Block a user