103 lines
2.3 KiB
Go
103 lines
2.3 KiB
Go
// Copyright (C) 2019 Marius Schellenberger
|
|
|
|
package db
|
|
|
|
import (
|
|
"errors"
|
|
"git.giftfish.de/ston1th/gowiki/pkg/core"
|
|
"git.giftfish.de/ston1th/gowiki/pkg/log"
|
|
"sort"
|
|
"strings"
|
|
)
|
|
|
|
const sectionPrefix = "section/"
|
|
|
|
var (
|
|
errSectionExists = errors.New("db: section already exist")
|
|
)
|
|
|
|
func (db *DB) GetSections() (secs core.Sections, err error) {
|
|
err = db.store.ForEach(func(k string, _ []byte) error {
|
|
if trim, ok := hasTrimPrefix(k, sectionPrefix); ok {
|
|
s, err := db.GetSection(trim)
|
|
if err == nil {
|
|
secs = append(secs, s)
|
|
}
|
|
}
|
|
return nil
|
|
})
|
|
sort.Sort(secs)
|
|
return
|
|
}
|
|
|
|
func (db *DB) GetSection(section string) (s core.Section, err error) {
|
|
err = db.store.Get(sectionPrefix+section, &s)
|
|
return
|
|
}
|
|
|
|
func (db *DB) SectionExists(section string) error {
|
|
return db.store.Get(sectionPrefix+section, nil)
|
|
}
|
|
|
|
func (db *DB) CreateSection(section string, members []string, user bool) error {
|
|
section = userRe.ReplaceAllString(section, "")
|
|
if err := db.SectionExists(section); err == nil {
|
|
return errSectionExists
|
|
}
|
|
return db.UpdateSection(section, members, user)
|
|
}
|
|
|
|
func (db *DB) UpdateSection(section string, members []string, user bool) error {
|
|
section = userRe.ReplaceAllString(section, "")
|
|
if _, ok := core.Contains(section, reservedNames); ok {
|
|
return errNameReserved
|
|
}
|
|
var m []string
|
|
if user {
|
|
m = append(m, section)
|
|
} else {
|
|
for _, v := range members {
|
|
if v == "" {
|
|
continue
|
|
}
|
|
if err := db.UserExists(v); err == nil {
|
|
m = append(m, v)
|
|
}
|
|
}
|
|
}
|
|
return db.store.Set(sectionPrefix+section, &core.Section{section, user, m})
|
|
}
|
|
|
|
func (db *DB) DeleteSection(section string, user bool) error {
|
|
if section == core.WikiSection {
|
|
return errors.New("section '" + core.WikiSection + "' cannot be deleted")
|
|
}
|
|
s, err := db.GetSection(section)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = db.store.Delete(sectionPrefix + section)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if !s.User && !user {
|
|
return nil
|
|
}
|
|
section += "/"
|
|
var pages []string
|
|
db.store.ForEach(func(k string, _ []byte) error {
|
|
if trim, ok := hasTrimPrefix(k, pagePrefix); ok {
|
|
if strings.HasPrefix(trim, section) {
|
|
pages = append(pages, trim)
|
|
}
|
|
}
|
|
return nil
|
|
})
|
|
for _, p := range pages {
|
|
err := db.deletePage(p)
|
|
if err != nil {
|
|
log.Debugf("DeleteSection %s, %s", section, err)
|
|
}
|
|
}
|
|
return nil
|
|
}
|