115 lines
2.5 KiB
Go
115 lines
2.5 KiB
Go
// Copyright (C) 2021 Marius Schellenberger
|
|
|
|
package db
|
|
|
|
import (
|
|
"git.giftfish.de/ston1th/gowiki/pkg/core"
|
|
"git.giftfish.de/ston1th/gowiki/pkg/log"
|
|
"git.giftfish.de/ston1th/gowiki/pkg/render"
|
|
"git.giftfish.de/ston1th/gowiki/pkg/store"
|
|
)
|
|
|
|
const (
|
|
versionKey = "version/version"
|
|
currentVersion = 2
|
|
)
|
|
|
|
type migrator func(*DB, int) error
|
|
|
|
var migrators = []migrator{
|
|
func(_ *DB, _ int) error {
|
|
return nil
|
|
},
|
|
func(db *DB, version int) (err error) {
|
|
// recreate all pages with new format
|
|
var pages []string
|
|
db.store.ForEachPrefix(pagePrefix, func(k string, _ []byte) error {
|
|
pages = append(pages, k)
|
|
return nil
|
|
})
|
|
var p core.Page
|
|
for _, v := range pages {
|
|
k := pagePrefix + v
|
|
err := db.store.Get(k, &p)
|
|
if err != nil {
|
|
log.Printf("db: migrator[%d]: pageGet %s: %s", version, k, err)
|
|
continue
|
|
}
|
|
if p.Perm == 3 {
|
|
p.Perm = core.Private
|
|
}
|
|
err = db.store.Set(k, &p)
|
|
if err != nil {
|
|
log.Printf("db: migrator[%d]: pageSet %s: %s", version, k, err)
|
|
continue
|
|
}
|
|
}
|
|
// create user sections
|
|
users, err := db.GetUsers()
|
|
if err != nil {
|
|
return
|
|
}
|
|
for _, u := range users {
|
|
err = db.CreateSection(u.Username, []string{u.Username}, true)
|
|
if err != nil {
|
|
log.Printf("db: migrator[%d]: userSection %s: %s", version, u.Username, err)
|
|
}
|
|
}
|
|
return
|
|
},
|
|
func(db *DB, version int) (err error) {
|
|
// re-render all pages
|
|
var pages []string
|
|
db.store.ForEachPrefix(pagePrefix, func(k string, _ []byte) error {
|
|
pages = append(pages, k)
|
|
return nil
|
|
})
|
|
var p core.Page
|
|
for _, v := range pages {
|
|
k := pagePrefix + v
|
|
err := db.store.Get(k, &p)
|
|
if err != nil {
|
|
log.Printf("db: migrator[%d]: pageGet %s: %s", version, k, err)
|
|
continue
|
|
}
|
|
render.Render(&p)
|
|
err = db.store.Set(k, &p)
|
|
if err != nil {
|
|
log.Printf("db: migrator[%d]: pageSet %s: %s", version, k, err)
|
|
continue
|
|
}
|
|
}
|
|
return
|
|
},
|
|
}
|
|
|
|
func (db *DB) RunMigrations() (err error) {
|
|
var version int
|
|
err = db.store.Get(versionKey, &version)
|
|
if err == store.ErrKeyNotFound {
|
|
err = db.store.Set(versionKey, currentVersion)
|
|
if err != nil {
|
|
return
|
|
}
|
|
err = db.store.Get(versionKey, &version)
|
|
if err != nil {
|
|
return
|
|
}
|
|
}
|
|
log.Printf("db: database version %d", version)
|
|
for version < currentVersion {
|
|
v := version + 1
|
|
log.Printf("db: running migration version %d", v)
|
|
err = migrators[v](db, v)
|
|
if err != nil {
|
|
return
|
|
}
|
|
err = db.store.Set(versionKey, v)
|
|
if err != nil {
|
|
return
|
|
}
|
|
version = v
|
|
log.Printf("db: database version %d", version)
|
|
}
|
|
return
|
|
}
|