79 lines
1.4 KiB
Go
79 lines
1.4 KiB
Go
|
package db
|
||
|
|
|
||
|
|
import (
|
||
|
|
"database/sql"
|
||
|
|
"embed"
|
||
|
|
"os"
|
||
|
|
|
||
|
|
"codeberg.org/u1f320/filer/db/queries"
|
||
|
|
"codeberg.org/u1f320/filer/store"
|
||
|
|
"codeberg.org/u1f320/filer/store/local"
|
||
|
|
"emperror.dev/errors"
|
||
|
|
"github.com/joho/godotenv"
|
||
|
|
"github.com/rs/zerolog/log"
|
||
|
|
migrate "github.com/rubenv/sql-migrate"
|
||
|
|
|
||
|
|
// sqlite driver
|
||
|
|
_ "modernc.org/sqlite"
|
||
|
|
)
|
||
|
|
|
||
|
|
func init() {
|
||
|
|
godotenv.Load()
|
||
|
|
}
|
||
|
|
|
||
|
|
const defaultDBName = "filer.db?_pragma=foreign_keys(1)"
|
||
|
|
|
||
|
|
type DB struct {
|
||
|
|
*queries.Queries
|
||
|
|
|
||
|
|
db *sql.DB
|
||
|
|
Store store.Store
|
||
|
|
}
|
||
|
|
|
||
|
|
func New() (*DB, error) {
|
||
|
|
sqldb, err := sql.Open("sqlite", defaultDBName)
|
||
|
|
if err != nil {
|
||
|
|
return nil, errors.Wrap(err, "creating database")
|
||
|
|
}
|
||
|
|
|
||
|
|
err = migrateDB(sqldb, defaultDBName)
|
||
|
|
if err != nil {
|
||
|
|
return nil, errors.Wrap(err, "migrating database")
|
||
|
|
}
|
||
|
|
|
||
|
|
db := &DB{
|
||
|
|
Queries: queries.New(sqldb),
|
||
|
|
db: sqldb,
|
||
|
|
}
|
||
|
|
|
||
|
|
switch os.Getenv("STORAGE") {
|
||
|
|
case local.StoreKey:
|
||
|
|
db.Store, err = local.New(os.Getenv("LOCAL_PATH"))
|
||
|
|
default:
|
||
|
|
err = store.ErrInvalidStoreKey
|
||
|
|
}
|
||
|
|
if err != nil {
|
||
|
|
return nil, errors.Wrap(err, "creating store")
|
||
|
|
}
|
||
|
|
|
||
|
|
return db, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
//go:embed migrations
|
||
|
|
var migrateFS embed.FS
|
||
|
|
|
||
|
|
func migrateDB(db *sql.DB, dbName string) error {
|
||
|
|
migrations := &migrate.EmbedFileSystemMigrationSource{
|
||
|
|
FileSystem: migrateFS,
|
||
|
|
Root: "migrations",
|
||
|
|
}
|
||
|
|
|
||
|
|
n, err := migrate.Exec(db, "sqlite3", migrations, migrate.Up)
|
||
|
|
if err != nil {
|
||
|
|
return errors.Wrap(err, "executing migrations")
|
||
|
|
}
|
||
|
|
|
||
|
|
log.Info().Int("count", n).Msg("Performed migrations!")
|
||
|
|
return nil
|
||
|
|
}
|