gowiki/pkg/db/page.go

package db

import (
	"git.giftfish.de/ston1th/gowiki/pkg/cache"
	"git.giftfish.de/ston1th/gowiki/pkg/permission"
	"git.giftfish.de/ston1th/gowiki/pkg/reneder"
)

type Article struct {
	Title     string        `json:"title"`
	LinkTitle string        `json:"-"`
	Search    string        `json:"search"`
	Index     template.HTML `json:"index"`
	Text      template.HTML `json:"text"`
	MD        string        `json:"md"`
	Created   string        `json:"created"`
	Updated   string        `json:"updated"`
	Public    bool          `json:"public"`
}

type Page struct {
	Title    string                `json:"title"`
	Markdown string                `json:"markdown"`
	Index    template.HTML         `json:"index"`
	Text     template.HTML         `json:"text"`
	Created  string                `json:"created"`
	Updated  string                `json:"updated"`
	Owner    string                `json:"owner"`
	Perm     permission.Permission `json:"perm"`
}

var (
	errPageNotFound      = dbErr("page not found")
	errInvalidPermission = dbErr("invalid permission")
	errPrivateWikiPage   = dbErr("private wiki pages are not allowed")
)

func (bs *BoltStore) GetPage(title, username string) (p *Page, err error) {
	p = bs.cache.Get(title)
	if p == nil {
		p = new(Page)
		err = bs.db.View(func(tx *bolt.Tx) error {
			v := tx.Bucket([]byte(pageTable)).Get([]byte(title))
			if v == nil {
				return errPageNotFound
			}
			return bs.Unmarshal(v, p)
		})
		if err != nil {
			return err
		}
		bs.cache.Add(title, p)
	}
	if !permission.Read(username, p) {
		err = errPageNotFound
	}
	return
}

func (bs *BoltStore) pageExists(title, username string) (page *Page, err error) {
	page, err = bs.GetPage(title)
	if err != nil {
		return
	}
	if !permission.Verify(username, page) {
		err = errPageNotFound
	}
	return
}

func (bs *BoltStore) CreatePage(title, section, markdown, username string, perm int) (page *Page, err error) {
	p := permission.Parse(perm)
	if p == permission.Invalid {
		err = errInvalidPermission
		return
	}
	if section == wikiSection && p == permission.Private {
		err = errPrivateWikiPage
		return
	}
	st := storeTitle(section, title)
	if _, err = bs.pageExists(st, username); err != nil {
		return
	}

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

	err = bs.db.Update(func(tx *bolt.Tx) error {
		v, err := bs.Marshal(page)
		if err != nil {
			return err
		}
		return tx.Bucket([]byte(pageTable)).Put([]byte(st), v)
	})
	if err != nil {
		return
	}

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

func (bs *BoltStore) UpdatePage(title, section, markdown, username string, perm int) (page *Page, err error) {
	p := permission.Parse(perm)
	if p == permission.Invalid {
		err = errInvalidPermission
		return
	}
	if section == wikiSection && p == permission.Private {
		err = errPrivateWikiPage
		return
	}
	st := storeTitle(section, title)
	page, err = bs.pageExists(st, username)
	if err != nil {
		return
	}

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

	err = bs.db.Update(func(tx *bolt.Tx) error {
		v, err := bs.Marshal(page)
		if err != nil {
			return err
		}
		return tx.Bucket([]byte(pageTable)).Put([]byte(st), v)
	})
	if err != nil {
		return
	}

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

func (bs *BoltStore) DeletePage(title, section, username string) error {
	st := storeTitle(section, title)
	if _, err := bs.pageExists(st, username); err != nil {
		return err
	}
	return bs.db.Update(func(tx *bolt.Tx) error {
		return tx.Bucket([]byte(pageTable)).Delete([]byte(st))
	})
}