88 lines
1.8 KiB
Go
88 lines
1.8 KiB
Go
package db
|
|
|
|
import (
|
|
"git.giftfish.de/ston1th/gowiki/pkg/core"
|
|
"git.giftfish.de/ston1th/gowiki/pkg/log"
|
|
"git.giftfish.de/ston1th/gowiki/pkg/store"
|
|
)
|
|
|
|
const (
|
|
versionKey = "version/version"
|
|
currentVersion = 1
|
|
)
|
|
|
|
type migrator func(*DB) (int, error)
|
|
|
|
var migrators = []migrator{
|
|
func(db *DB) (int, error) {
|
|
return 0, nil
|
|
},
|
|
func(db *DB) (version int, err error) {
|
|
version = 1
|
|
// recreate all pages with new format
|
|
var pages []string
|
|
db.store.ForEach(func(k string, _ []byte) error {
|
|
if trim, ok := hasTrimPrefix(k, pagePrefix); ok {
|
|
pages = append(pages, trim)
|
|
}
|
|
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
|
|
}
|
|
}
|
|
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) 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 %d", v)
|
|
version, err = migrators[v](db)
|
|
if err != nil {
|
|
return
|
|
}
|
|
err = db.store.Set(versionKey, version)
|
|
if err != nil {
|
|
return
|
|
}
|
|
}
|
|
return
|
|
}
|