controller.go

package main

import (
	"github.com/hoisie/web"
	"log"
	"os"
)

type Controller struct {
	themeEngine     *ThemeEngine
	articlesPerPage int
	log             *log.Logger
}

func (c *Controller) Init(conf *Config, s *web.Server) {
	logFile, err := os.Create("coconut.log")
	if err != nil {
		log.Fatalln("Error creating log file:", err)
	}
	c.log = log.New(logFile, "[coconut]", log.LstdFlags|log.Lshortfile)
	s.Get("/", c.Front)
	for url, filename := range conf.Pages {
		s.Get(url, c.makePageFunc(filename))
	}
	s.Get("/tag/(.*)", c.Tag)
	s.Get("/(.*)", c.Article)
}

func (c *Controller) makePageFunc(filename string) func() string {
	return func() string {
		return c.Page(filename)
	}
}

func (c *Controller) Front(ctx *web.Context) string {
	var content string
	articles, prev, next := paginate(GetArticles(func(a *Article) bool { return true }), c.articlesPerPage, getPageNum(ctx.Params))
	for _, a := range articles {
		a.FullView = false
		content += c.themeEngine.ThemeArticle(a)
	}
	if content == "" {
		content = "No articles found"
	}

	return c.themeEngine.Theme(c.themeEngine.ThemeList(content, prev, next))
}

func (c *Controller) Article(name string) string {
	a, err := GetArticle(name)
	if err != nil {
		return c.themeEngine.Theme("Couldn't find article \"" + name + "\"")
	}
	return c.themeEngine.Theme(c.themeEngine.ThemeArticle(a))
}

func (c *Controller) Page(path string) string {
	p, err := GetPage(path)
	if err != nil {
		return c.themeEngine.Theme("No page at \"" + path + "\"")
	}
	return c.themeEngine.Theme(c.themeEngine.ThemePage(p))
}

func (c *Controller) Tag(ctx *web.Context, tag string) string {
	var content string
	articles, prev, next := paginate(GetArticles(func(a *Article) bool { return a.HasTag(tag) }), c.articlesPerPage, getPageNum(ctx.Params))
	for _, a := range articles {
		a.FullView = false
		content += c.themeEngine.ThemeArticle(a)
	}
	if content == "" {
		content = "No articles found"
	}
	return c.themeEngine.Theme(c.themeEngine.ThemeList(content, prev, next))
}