mirror of
https://github.com/dnote/dnote
synced 2026-03-15 15:05:51 +01:00
139 lines
3.1 KiB
Go
139 lines
3.1 KiB
Go
/* Copyright (C) 2019, 2020, 2021, 2022, 2023, 2024, 2025 Dnote contributors
|
|
*
|
|
* This file is part of Dnote.
|
|
*
|
|
* Dnote is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* Dnote is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU Affero General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Affero General Public License
|
|
* along with Dnote. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
package views
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"html/template"
|
|
"io"
|
|
"net/http"
|
|
|
|
"github.com/dnote/dnote/pkg/clock"
|
|
"github.com/dnote/dnote/pkg/server/app"
|
|
"github.com/dnote/dnote/pkg/server/buildinfo"
|
|
"github.com/dnote/dnote/pkg/server/context"
|
|
"github.com/dnote/dnote/pkg/server/log"
|
|
"github.com/gorilla/csrf"
|
|
)
|
|
|
|
const (
|
|
// TemplateExt is the template extension
|
|
TemplateExt string = ".gohtml"
|
|
)
|
|
|
|
const (
|
|
siteTitle = "Dnote"
|
|
)
|
|
|
|
// Config is a view config
|
|
type Config struct {
|
|
Title string
|
|
Layout string
|
|
HeaderTemplate string
|
|
HelperFuncs map[string]interface{}
|
|
AlertInBody bool
|
|
Clock clock.Clock
|
|
}
|
|
|
|
type viewCtx struct {
|
|
Clock clock.Clock
|
|
Config Config
|
|
}
|
|
|
|
func newViewCtx(c Config) viewCtx {
|
|
return viewCtx{
|
|
Clock: c.getClock(),
|
|
Config: c,
|
|
}
|
|
}
|
|
|
|
func (c Config) getLayout() string {
|
|
if c.Layout == "" {
|
|
return "base"
|
|
}
|
|
|
|
return c.Layout
|
|
}
|
|
|
|
func (c Config) getClock() clock.Clock {
|
|
if c.Clock != nil {
|
|
return c.Clock
|
|
}
|
|
|
|
return clock.New()
|
|
}
|
|
|
|
// View holds the information about a view
|
|
type View struct {
|
|
Template *template.Template
|
|
Layout string
|
|
// AlertInBody specifies if alert should be set in the body instead of the header
|
|
AlertInBody bool
|
|
App *app.App
|
|
}
|
|
|
|
func (v *View) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
|
v.Render(w, r, nil, http.StatusOK)
|
|
}
|
|
|
|
// Render is used to render the view with the predefined layout
|
|
func (v *View) Render(w http.ResponseWriter, r *http.Request, data *Data, statusCode int) {
|
|
w.Header().Set("Content-Type", "text/html")
|
|
|
|
var vd Data
|
|
if data != nil {
|
|
vd = *data
|
|
}
|
|
|
|
if alert := getAlert(r); alert != nil {
|
|
vd.PutAlert(*alert, v.AlertInBody)
|
|
clearAlert(w)
|
|
}
|
|
|
|
vd.User = context.User(r.Context())
|
|
|
|
// Put user data in Yield
|
|
if vd.Yield == nil {
|
|
vd.Yield = map[string]interface{}{}
|
|
}
|
|
if vd.User != nil {
|
|
vd.Yield["Email"] = vd.User.Email.String
|
|
}
|
|
vd.Yield["CurrentPath"] = r.URL.Path
|
|
vd.Yield["Standalone"] = buildinfo.Standalone
|
|
|
|
var buf bytes.Buffer
|
|
csrfField := csrf.TemplateField(r)
|
|
tpl := v.Template.Funcs(template.FuncMap{
|
|
"csrfField": func() template.HTML {
|
|
return csrfField
|
|
},
|
|
})
|
|
|
|
if err := tpl.ExecuteTemplate(&buf, v.Layout, vd); err != nil {
|
|
log.ErrorWrap(err, fmt.Sprintf("executing template for URI '%s'", r.RequestURI))
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
w.Write(v.App.HTTP500Page)
|
|
return
|
|
}
|
|
|
|
w.WriteHeader(statusCode)
|
|
io.Copy(w, &buf)
|
|
}
|