weiro/providers/db/provider.go

package db

import (
	"context"
	"database/sql"

	"github.com/Southclaws/fault"
	"lmika.dev/lmika/weiro/providers/db/gen/sqlgen"
	"lmika.dev/lmika/weiro/sql/schema"
	migration "lmika.dev/pkg/litemigrate"
)

type Provider struct {
	drvr    *sql.DB
	queries *sqlgen.Queries
}

func New(dbFile string) (*Provider, error) {
	drvr, err := sql.Open("sqlite", dbFile)
	if err != nil {
		return nil, fault.Wrap(err)
	}

	if err := migration.New(schema.FS, drvr).MigrateUp(context.Background()); err != nil {
		return nil, fault.Wrap(err)
	}

	if _, err := drvr.Exec(`PRAGMA foreign_keys = 1;`); err != nil {
		return nil, fault.Wrap(err)
	}

	return &Provider{
		drvr:    drvr,
		queries: sqlgen.New(drvr),
	}, nil
}

func (db *Provider) Close() error {
	return db.drvr.Close()
}