gowiki/pkg/db/page.go

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"
)

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.pages.ForEach(func(title string, _ []byte) error {
		if title == core.IndexURI {
			return nil
		}
		p, err := db.GetPage(title, 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.pages.Get(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
	}
	if section == core.WikiSection && 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:      username,
		Perm:       p,
	}
	//search := render.Render(page)
	render.Render(page)

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

	//TODO
	//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
	//TODO
	//search := render.Render(page)
	render.Render(page)

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

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

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
	}

	if err = db.pages.Delete(st); err != nil {
		return
	}
	db.cache.Delete(st)
	return nil
	//TODO
	//return db.Index.Delete(st)
}