Initial database setup including migrations

database/database.go

@@ -0,0 +1,46 @@
+package database
+
+import (
+	"database/sql"
+
+	_ "github.com/lib/pq"
+	_ "github.com/mattn/go-sqlite3"
+
+	log "maunium.net/go/maulogger/v2"
+
+	"gitlab.com/beeper/discord/database/migrations"
+)
+
+type Database struct {
+	*sql.DB
+	log     log.Logger
+	dialect string
+}
+
+func New(dbType, uri string, maxOpenConns, maxIdleConns int, baseLog log.Logger) (*Database, error) {
+	conn, err := sql.Open(dbType, uri)
+	if err != nil {
+		return nil, err
+	}
+
+	if dbType == "sqlite3" {
+		conn.Exec("PRAGMA foreign_keys = ON")
+	}
+
+	conn.SetMaxOpenConns(maxOpenConns)
+	conn.SetMaxIdleConns(maxIdleConns)
+
+	dbLog := baseLog.Sub("Database")
+
+	if err := migrations.Run(conn, dbLog); err != nil {
+		return nil, err
+	}
+
+	db := &Database{
+		DB:      conn,
+		log:     dbLog,
+		dialect: dbType,
+	}
+
+	return db, nil
+}

database/migrations/01-initial.sql

@@ -0,0 +1,11 @@
+CREATE TABLE IF NOT EXISTS portal (
+	did      text,
+	receiver text,
+	mxid     text UNIQUE,
+
+	name   text NOT NULL,
+	topic  text NOT NULL,
+	avatar text NOT NULL,
+
+	PRIMARY KEY (did, receiver)
+);

database/migrations/migrations.go

@@ -0,0 +1,54 @@
+package migrations
+
+import (
+	"database/sql"
+	"embed"
+
+	"github.com/lopezator/migrator"
+	log "maunium.net/go/maulogger/v2"
+)
+
+//go:embed *.sql
+var migrations embed.FS
+
+func migrationFromFile(filename string) *migrator.Migration {
+	return &migrator.Migration{
+		Name: filename,
+		Func: func(tx *sql.Tx) error {
+			data, err := migrations.ReadFile(filename)
+			if err != nil {
+				return err
+			}
+
+			if _, err := tx.Exec(string(data)); err != nil {
+				return err
+			}
+
+			return nil
+		},
+	}
+}
+
+func Run(db *sql.DB, baseLog log.Logger) error {
+	subLogger := baseLog.Sub("Migrations")
+	logger := migrator.LoggerFunc(func(msg string, args ...interface{}) {
+		subLogger.Infof(msg, args...)
+	})
+
+	m, err := migrator.New(
+		migrator.TableName("version"),
+		migrator.WithLogger(logger),
+		migrator.Migrations(
+			migrationFromFile("01-initial.sql"),
+		),
+	)
+	if err != nil {
+		return err
+	}
+
+	if err := m.Migrate(db); err != nil {
+		return err
+	}
+
+	return nil
+}