felt/mongodb/adapter.go

package dbengine

import (
	"context"
	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
	"nilfm.cc/git/felt/models"
	"time"
)

type DbAdapter interface {
	Init(mongoUri string) error

	CreateTable(table models.TableKey) error
	DestroyTable(table models.TableKey) error

	InsertDiceRoll(table models.TableKey, diceRoll models.DiceRoll) error
	GetDiceRolls(table models.TableKey) ([]models.DiceRoll, error)

	SetMapImageUrl(table models.TableKey, url string) error
	GetMapImageUrl(table models.TableKey) (string, error)

	AddToken(table models.TableKey, token models.Token) error
	RemoveToken(table models.TableKey, tokenId string) error
	ModifyToken(table models.TableKey, token models.Token) error
	GetTokens(table models.TableKey) ([]models.Token, error)
}

type DbEngine struct {
	client mongo.Client
}

func (self *DbEngine) mkCtx(timeoutSec int) context.Context {
	return context.WithTimeout(context.Background(), 10*time.Second)
}

func (self *DbEngine) Init(mongoUri string) error {
	client, err := mongo.NewClient(options.Client().ApplyURI(mongoUri))
	if err != nil {
		return err
	}
	self.client = client
	ctx, _ := self.mkCtx(10)

	err = client.Connect(ctx)
	if err != nil {
		return err
	}
	defer client.Disconnect(ctx)

	db := client.Database("felt")

	err = self.ensureCollections(db)
	return err
}

func (self *DbEngine) ensureCollections(db mongo.Database) error {
	tables := db.Collection("tables")
	if tables == nil {
		createCmd := bson.D{
			{"create", "tables"},
			{"clusteredIndex", {
				{"key", {"name"}},
				{"unique", true},
				{"name", "idx_tables_unique_names"},
			}},
		}

		var createResult bson.M
		err := db.RunCommand(
			self.mkCtx(10),
			createCmd).Decode(&createResult)

		if err != nil {
			return err
		}
	}
	return nil
}

func (self *DbEngine) CreateTable(table models.TableKey) error {
	tables := self.db.Collection("tables")
	if tables != nil {
		_, err := tables.insertOne(self.mkCtx(10), bson.D{
			{"name", table.Name},
			{"passcode", table.Passcode},
			{"mapUri", ""},
			{"diceRolls", bson.A{}},
			{"tokens", bson.A{}},
		})
	}

	return err
}