context.go

package pages

import (
	"context"
	"html/template"
	"net/http"
	"sync"
)

type contextKey struct{}

func FromContext(ctx context.Context) *Context {
	c, _ := ctx.Value(contextKey{}).(*Context)
	return c
}

func MustContext(ctx context.Context) *Context {
	if c := FromContext(ctx); c != nil {
		return c
	}
	panic("Pages context is nil, you should use pages.NewContext before pages.MustContext or use pages.FromContext directly.")
}

var ctxPool = &sync.Pool{
	New: func() any {
		c := new(Context)
		c.Reset()
		return c
	},
}

func NewContext(parent context.Context) (context.Context, context.CancelFunc) {
	c := ctxPool.Get().(*Context)

	cancel := func() {
		c.Reset()
		ctxPool.Put(c)
	}

	return context.WithValue(parent, contextKey{}, c), cancel
}

type Context struct {
	dom      *DOM
	site     *Site
	page     *Page
	err      error
	debug    bool
	guest    bool
	status   int
	content  template.HTML
	template string
}

func (c *Context) Reset() {
	c.status = http.StatusOK
	c.template = ""
	c.content = ""
	c.debug = false
	c.guest = true
	c.dom = nil
	c.site = nil
	c.page = nil
	c.err = nil
}

func (c *Context) DOM() *DOM {
	if c.dom == nil {
		c.dom = new(DOM)
	}
	return c.dom
}

func (c *Context) Site() *Site {
	return c.site
}

func (c *Context) SetSite(site *Site) {
	c.site = site
}

func (c *Context) HasSite() bool {
	return c.site != nil
}

func (c *Context) Page() *Page {
	return c.page
}

func (c *Context) SetPage(page *Page) {
	if page != nil {
		if page.Site == nil && c.HasSite() {
			page.SiteID = c.site.ID
			page.Site = c.site
		}
	}
	c.page = page
}

func (c *Context) HasPage() bool {
	return c.page != nil
}

func (c *Context) Error() error {
	return c.err
}

func (c *Context) SetError(err error) {
	c.err = err
}

func (c *Context) HasError() bool {
	return c.err != nil
}

func (c *Context) Debug() bool {
	return c.debug
}

func (c *Context) SetDebug(debug bool) {
	c.debug = debug
}

func (c *Context) Guest() bool {
	return c.guest
}

func (c *Context) SetGuest(guest bool) {
	c.guest = guest
}

func (c *Context) Status() int {
	if c.status <= 0 {
		return http.StatusOK
	}
	if c.HasError() && c.status < http.StatusBadRequest {
		return http.StatusInternalServerError
	}
	return c.status
}

func (c *Context) SetStatus(status int) {
	c.status = status
}

func (c *Context) Content() template.HTML {
	return c.content
}

func (c *Context) SetContent(content template.HTML) {
	c.content = content
}

func (c *Context) HasContent() bool {
	return c.content != ""
}

func (c *Context) Template() string {
	if c.template != "" {
		return c.template
	}
	if c.HasPage() {
		return c.page.Template
	}
	return ""
}

func (c *Context) SetTemplate(template string) {
	c.template = template
}

func (c *Context) HasTemplate() bool {
	return c.Template() != ""
}