gowiki/pkg/db/page.go

// Copyright (C) 2018 Marius Schellenberger

package db

import (
	"git.giftfish.de/ston1th/gowiki/pkg/core"
	"git.giftfish.de/ston1th/gowiki/pkg/render"
	"git.giftfish.de/ston1th/gowiki/pkg/store"
	"sort"
)

const pagePrefix = "page/"

var (
	errPageNotFound      = store.ErrKeyNotFound
	errPermissionDenied  = dbErr("permission denied")
	errInvalidPermission = dbErr("invalid permission")
	errPrivateWikiPage   = dbErr("private wiki pages are not allowed")
	errDeleteIndexPage   = dbErr("index page can not be deleted")
)

func (db *DB) GetAllPages(username string) (pages core.Pages) {
	db.store.ForEach(func(k string, _ []byte) error {
		if trim, ok := hasTrimPrefix(k, pagePrefix); ok {
			if trim == core.IndexURI {
				return nil
			}
			p, err := db.GetPage(trim, username)
			if err == nil {
				pages = append(pages, *p)
			}
		}
		return nil
	})
	sort.Sort(pages)
	return
}

func (db *DB) GetPage(title, username string) (p *core.Page, err error) {
	p = db.cache.Get(title)
	if p == nil {
		p, err = db.getPage(title, username)
		db.cache.Add(title, p)
		return
	}
	if !core.ReadPerm(username, p) {
		err = errPageNotFound
	}
	return
}

func (db *DB) getPage(title, username string) (p *core.Page, err error) {
	p = new(core.Page)
	err = db.store.Get(pagePrefix+title, p)
	if err != nil {
		return
	}
	if !core.ReadPerm(username, p) {
		err = errPageNotFound
	}
	return
}

func (db *DB) CreatePage(title, section, markdown, username string, p core.Permission) (page *core.Page, err error) {
	if p == core.Invalid {
		err = errInvalidPermission
		return
	}
	owner := username
	if section == core.WikiSection {
		owner = core.WikiSection
		if p == core.Private {
			err = errPrivateWikiPage
			return
		}
	}
	st := render.StoreTitle(section, title)
	if _, err = db.getPage(st, username); err != nil && err != store.ErrKeyNotFound {
		return
	}

	page = &core.Page{
		Title:      title,
		StoreTitle: st,
		Markdown:   markdown,
		Created:    created(username),
		Owner:      owner,
		Perm:       p,
	}
	search := render.Render(page)

	err = db.store.Set(pagePrefix+st, page)
	if err != nil {
		return
	}

	err = db.Index.Add(title, st, search)
	if err != nil {
		return
	}
	db.cache.Add(st, page)
	return
}

func (db *DB) UpdatePage(title, section, markdown, username string, p core.Permission) (err error) {
	if p == core.Invalid {
		err = errInvalidPermission
		return
	}
	if section == core.WikiSection && p == core.Private {
		err = errPrivateWikiPage
		return
	}
	st := render.StoreTitle(section, title)
	page, err := db.getPage(st, username)
	if err != nil {
		return
	}
	if !core.WritePerm(username, section, page) {
		return errPermissionDenied
	}

	page.Markdown = markdown
	page.Updated = created(username)
	page.Perm = p
	search := render.Render(page)

	err = db.store.Set(pagePrefix+st, page)
	if err != nil {
		return
	}

	err = db.Index.Add(title, st, search)
	if err != nil {
		return
	}
	db.cache.Add(st, page)
	return
}

func (db *DB) deletePage(st string) (err error) {
	if err = db.store.Delete(pagePrefix + st); err != nil {
		return
	}
	db.cache.Delete(st)
	return db.Index.Delete(st)
}

func (db *DB) DeletePage(title, section, username string) (err error) {
	var page *core.Page
	if section == core.WikiSection && title == core.IndexPage {
		err = errDeleteIndexPage
		return
	}
	st := render.StoreTitle(section, title)
	if page, err = db.getPage(st, username); err != nil {
		return
	}
	if !core.WritePerm(username, section, page) {
		return errPermissionDenied
	}
	return db.deletePage(st)
}