hermeticum/server/db/db.go

package db

import (
	"context"
	_ "embed"
	"errors"
	"fmt"
	"log"
	"math/rand"

	"github.com/google/uuid"
	"github.com/jackc/pgx/v4/pgxpool"
)

//go:embed schema.sql
var schema string

// TODO I have a suspicion that I'm going to want to move to an ORM like model where the object struct has a DB member and various methods on it. For now I'm just going to keep adding shit to the DB interface because if it doesn't get too long in the end then it's fine.

type DB interface {
	// Accounts
	CreateAccount(string, string) (*Account, error)
	ValidateCredentials(string, string) (*Account, error)
	GetAccount(string) (*Account, error)
	StartSession(Account) (string, error)
	EndSession(string) error
	ClearSessions() error

	// General
	GetObject(owner, name string) (*Object, error)

	// Defaults
	Ensure() error

	// Presence
	ActiveSessions() ([]Session, error)
	AvatarBySessionID(string) (*Object, error)
	BedroomBySessionID(string) (*Object, error)
	MoveInto(toMove Object, container Object) error
	Earshot(Object) ([]Object, error)
}

type Account struct {
	ID     int
	Name   string
	Pwhash string
	God    bool
}

type Session struct {
	ID        string
	AccountID int
}

type Object struct {
	ID      int
	Avatar  bool
	Bedroom bool
	Data    map[string]string
	OwnerID int
	Script  string
}

type pgDB struct {
	pool *pgxpool.Pool
}

func NewDB(connURL string) (DB, error) {
	pool, err := pgxpool.Connect(context.Background(), connURL)
	if err != nil {
		return nil, err
	}
	pgdb := &pgDB{
		pool: pool,
	}

	return pgdb, nil
}

// Ensure checks for and then creates default resources if they do not exist (like the Foyer)
func (db *pgDB) Ensure() error {
	// TODO this is sloppy but shrug
	_, err := db.pool.Exec(context.Background(), schema)
	//log.Println(err)
	sysAcc, err := db.GetAccount("system")
	if err != nil {
		// TODO actually check error. for now assuming it means does not exist
		sysAcc, err = db.CreateGod("system", "")
		if err != nil {
			return fmt.Errorf("failed to create system account: %w", err)
		}
	}

	log.Printf("%#v", sysAcc)

	if _, err := db.GetObject("system", "foyer"); err != nil {
		data := map[string]string{}
		data["name"] = "foyer"
		data["description"] = "a big room. the ceiling is painted with constellations."
		foyer := &Object{
			Data:   data,
			Script: "",
			// TODO default room script
		}
		if err = db.CreateObject(sysAcc, foyer); err != nil {
			return err
		}
	}

	return nil
}

func (db *pgDB) CreateGod(name, password string) (account *Account, err error) {
	account = &Account{
		Name:   name,
		Pwhash: password,
		God:    true,
	}

	return account, db.createAccount(account)
}

func (db *pgDB) CreateAccount(name, password string) (account *Account, err error) {
	account = &Account{
		Name:   name,
		Pwhash: password,
	}

	return account, db.createAccount(account)
}

func (db *pgDB) createAccount(account *Account) (err error) {
	ctx := context.Background()
	tx, err := db.pool.Begin(ctx)
	if err != nil {
		return
	}

	defer tx.Rollback(ctx)

	stmt := "INSERT INTO accounts (name, pwhash, god) VALUES ( $1, $2, $3 ) RETURNING id"
	err = tx.QueryRow(ctx, stmt, account.Name, account.Pwhash, account.God).Scan(&account.ID)
	// TODO handle and cleanup unqiue violations
	if err != nil {
		return
	}

	data := map[string]string{}
	data["name"] = account.Name
	data["description"] = fmt.Sprintf("a gaseous form. it smells faintly of %s.", randSmell())
	av := &Object{
		Avatar: true,
		Data:   data,
		Script: "",
	}

	stmt = "INSERT INTO objects ( avatar, data, owner, script ) VALUES ( $1, $2, $3, $4 ) RETURNING id"
	err = tx.QueryRow(ctx, stmt, av.Avatar, av.Data, account.ID, av.Script).Scan(&av.ID)
	if err != nil {
		return
	}

	stmt = "INSERT INTO permissions (object) VALUES ( $1 )"
	_, err = tx.Exec(ctx, stmt, av.ID)
	if err != nil {
		return
	}

	data = map[string]string{}
	data["name"] = "your private bedroom"

	bedroom := &Object{
		Bedroom: true,
		Data:    data,
		Script:  "",
	}

	stmt = "INSERT INTO objects ( bedroom, data, owner, script ) VALUES ( $1, $2, $3, $4 ) RETURNING id"
	err = tx.QueryRow(ctx, stmt, bedroom.Bedroom, bedroom.Data, account.ID, bedroom.Script).Scan(&bedroom.ID)
	if err != nil {
		return
	}

	stmt = "INSERT INTO permissions (object) VALUES ( $1 )"
	_, err = tx.Exec(ctx, stmt, bedroom.ID)
	if err != nil {
		return
	}

	return tx.Commit(ctx)
}

func (db *pgDB) ValidateCredentials(name, password string) (*Account, error) {
	a, err := db.GetAccount(name)
	if err != nil {
		return nil, err
	}

	if a.Pwhash == "" {
		return nil, errors.New("this account cannot be logged into")
	}

	// TODO hashing lol

	if a.Pwhash != password {
		return nil, errors.New("invalid credentials")
	}

	return a, nil
}

func (db *pgDB) GetAccount(name string) (a *Account, err error) {
	a = &Account{}
	stmt := "SELECT id, name, pwhash FROM accounts WHERE name = $1"
	err = db.pool.QueryRow(context.Background(), stmt, name).Scan(&a.ID, &a.Name, &a.Pwhash)
	return
}

func (db *pgDB) StartSession(a Account) (sid string, err error) {
	sid = uuid.New().String()
	_, err = db.pool.Exec(context.Background(),
		"INSERT INTO sessions (id, account) VALUES ( $1, $2 )", sid, a.ID)
	if err != nil {
		return
	}

	// Clean up any ghosts to prevent avatar duplication
	// TODO subquery
	stmt := "DELETE FROM contains WHERE contained = (SELECT id FROM objects WHERE objects.avatar = true and objects.owner = $1)"
	_, err = db.pool.Exec(context.Background(), stmt, a.ID)
	if err != nil {
		log.Printf("failed to clean up ghosts for %d: %s", a.ID, err.Error())
		err = nil
	}

	return
}

func (db *pgDB) EndSession(sid string) (err error) {
	if sid == "" {
		log.Println("db.EndSession called with empty session id")
		return
	}

	var o *Object
	if o, err = db.AvatarBySessionID(sid); err == nil {
		if _, err = db.pool.Exec(context.Background(),
			"DELETE FROM contains WHERE contained = $1", o.ID); err != nil {
			log.Printf("failed to remove avatar from room: %s", err.Error())
		}
	} else {
		log.Printf("failed to find avatar for session %s: %s", sid, err.Error())
	}

	_, err = db.pool.Exec(context.Background(), "DELETE FROM sessions WHERE id = $1", sid)

	return
}

func (db *pgDB) AvatarBySessionID(sid string) (avatar *Object, err error) {
	avatar = &Object{}

	// TODO subquery
	stmt := `
	SELECT id, avatar, data, owner, script
	FROM objects WHERE avatar = true AND owner = (
		SELECT a.id FROM sessions s JOIN accounts a ON s.account = a.id WHERE s.id = $1)`
	err = db.pool.QueryRow(context.Background(), stmt, sid).Scan(
		&avatar.ID, &avatar.Avatar, &avatar.Data, &avatar.OwnerID, &avatar.Script)
	return
}

func (db *pgDB) BedroomBySessionID(sid string) (bedroom *Object, err error) {
	bedroom = &Object{}

	// TODO subquery
	stmt := `
	SELECT id, bedroom, data
	FROM objects WHERE bedroom = true AND owner = (
		SELECT a.id FROM sessions s JOIN accounts a ON s.account = a.id WHERE s.id = $1)`
	err = db.pool.QueryRow(context.Background(), stmt, sid).Scan(
		&bedroom.ID, &bedroom.Bedroom, &bedroom.Data)
	return
}

func (db *pgDB) MoveInto(toMove Object, container Object) error {
	ctx := context.Background()
	tx, err := db.pool.Begin(ctx)
	if err != nil {
		return err
	}
	defer tx.Rollback(ctx)

	stmt := "DELETE FROM contains WHERE contained = $1"
	_, err = tx.Exec(ctx, stmt, toMove.ID)
	if err != nil {
		return err
	}

	stmt = "INSERT INTO contains (contained, container) VALUES ($1, $2)"
	_, err = tx.Exec(ctx, stmt, toMove.ID, container.ID)
	if err != nil {
		return err
	}

	return tx.Commit(ctx)
}

func (db *pgDB) Earshot(obj Object) ([]Object, error) {
	stmt := `
	SELECT id, avatar, bedroom, data, owner, script FROM objects
	WHERE id IN (
		SELECT contained FROM contains
		WHERE container = (
			SELECT container FROM contains WHERE contained = $1 LIMIT 1))`
	rows, err := db.pool.Query(context.Background(), stmt, obj.ID)
	if err != nil {
		return nil, err
	}

	out := []Object{}

	for rows.Next() {
		heard := Object{}
		if err = rows.Scan(
			&heard.ID, &heard.Avatar,
			&heard.Bedroom, &heard.Data,
			&heard.OwnerID, &heard.Script); err != nil {
			return nil, err
		}
		out = append(out, heard)
	}

	return out, nil
}

func (db *pgDB) GetObject(owner, name string) (obj *Object, err error) {
	ctx := context.Background()
	obj = &Object{}
	stmt := `
		SELECT id, avatar, data, owner, script
		FROM objects
		WHERE owner = $1 AND data['name'] = $2`
	err = db.pool.QueryRow(ctx, stmt, owner, fmt.Sprintf(`"%s"`, name)).Scan(
		&obj.ID, &obj.Avatar, &obj.Data, &obj.OwnerID, &obj.Script)
	// TODO i think the escaping here is going to create a sadness ^

	return
}

func (db *pgDB) ActiveSessions() (out []Session, err error) {
	stmt := `SELECT id, account FROM sessions`
	rows, err := db.pool.Query(context.Background(), stmt)
	if err != nil {
		return
	}

	for rows.Next() {
		s := Session{}
		if err = rows.Scan(&s.ID, &s.AccountID); err != nil {
			return
		}
		out = append(out, s)
	}

	return
}

func (db *pgDB) ClearSessions() (err error) {
	_, err = db.pool.Exec(context.Background(), "DELETE FROM sessions")
	return
}

func (db *pgDB) CreateObject(owner *Account, obj *Object) error {
	ctx := context.Background()
	stmt := `
		INSERT INTO objects (avatar, bedroom, data, script, owner)
		VALUES ( $1, $2, $3, $4, $5)
		RETURNING id
	`

	err := db.pool.QueryRow(ctx, stmt,
		obj.Avatar, obj.Bedroom, obj.Data, obj.Script, owner.ID).Scan(
		&obj.ID)
	if err != nil {
		return err
	}

	return nil
}

func randSmell() string {
	// TODO seeding
	smells := []string{
		"lavender",
		"wet soil",
		"juniper",
		"pine sap",
		"wood smoke",
	}
	ix := rand.Intn(len(smells))
	return smells[ix]
}