92 lines
2.1 KiB
Go
92 lines
2.1 KiB
Go
package db
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
)
|
|
|
|
type User struct {
|
|
Id int
|
|
Name string
|
|
Role int
|
|
Character string
|
|
Color string
|
|
Pronouns string
|
|
CreatedAt time.Time
|
|
Hotbar map[string]string
|
|
Stats UserStats
|
|
}
|
|
|
|
type UserStats struct {
|
|
Id int
|
|
CoreVersion string
|
|
SystemId string
|
|
SystemVersion string
|
|
CreatedTime int64
|
|
ModifiedTime int64
|
|
LastModifiedBy string
|
|
}
|
|
|
|
type Users []User
|
|
|
|
func (users Users) GetById(id int) *User {
|
|
return &users[id]
|
|
}
|
|
|
|
func (m FoundryStateModel) InsertUser(user *User, stateId int) error {
|
|
query := `
|
|
INSERT INTO users (state_id, name, role, character, color, pronouns)
|
|
VALUES ($1, $2, $3, $4, $5, $6)
|
|
RETURNING id, created_at`
|
|
|
|
args := []any{stateId, user.Name, user.Role, user.Character, user.Color, user.Pronouns}
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
|
|
defer cancel()
|
|
|
|
err := m.DB.QueryRowContext(ctx, query, args...).Scan(&user.Id, &user.CreatedAt)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for k, v := range user.Hotbar {
|
|
err = m.InsertUserHotbar(k, v, user.Id)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return m.InsertUserStats(&user.Stats, user.Id)
|
|
}
|
|
|
|
func (m FoundryStateModel) InsertUserHotbar(key string, value string, userId int) error {
|
|
query := `
|
|
INSERT INTO users_hotbar (user_id, key, value)
|
|
VALUES ($1, $2, $3)`
|
|
|
|
args := []any{userId, key, value}
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
|
|
defer cancel()
|
|
|
|
_, err := m.DB.ExecContext(ctx, query, args...)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (m FoundryStateModel) InsertUserStats(stats *UserStats, userId int) error {
|
|
query := `
|
|
INSERT INTO users_stats (user_id, core_version, system_id, system_version, created_time, modified_time, last_modified_by)
|
|
VALUES ($1, $2, $3, $4, $5, $6, $7)
|
|
RETURNING id`
|
|
|
|
args := []any{userId, stats.CoreVersion, stats.SystemId, stats.SystemVersion, stats.CreatedTime, stats.ModifiedTime, stats.LastModifiedBy}
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
|
|
defer cancel()
|
|
|
|
return m.DB.QueryRowContext(ctx, query, args...).Scan(&stats.Id)
|
|
}
|