mirror of
https://codeberg.org/VnPower/PixivFE
synced 2024-12-06 19:16:23 +01:00
iacore
94 lines
2.2 KiB
Go
94 lines
2.2 KiB
Go
package routes
|
|
|
|
import (
|
|
"io"
|
|
"log"
|
|
"net/http"
|
|
"net/url"
|
|
"reflect"
|
|
"strings"
|
|
|
|
"codeberg.org/vnpower/pixivfe/v2/session"
|
|
"codeberg.org/vnpower/pixivfe/v2/utils"
|
|
|
|
"github.com/CloudyKit/jet/v6"
|
|
"net/http"
|
|
)
|
|
|
|
// global variable, yes.
|
|
var views *jet.Set
|
|
|
|
func InitTemplatingEngine(DisableCache bool) {
|
|
if DisableCache {
|
|
views = jet.NewSet(
|
|
jet.NewOSFileSystemLoader("assets/views"),
|
|
jet.InDevelopmentMode(), // disable cache
|
|
)
|
|
} else {
|
|
views = jet.NewSet(
|
|
jet.NewOSFileSystemLoader("assets/views"),
|
|
)
|
|
}
|
|
for fn_name, fn := range utils.GetTemplateFunctions() {
|
|
views.AddGlobal(fn_name, fn)
|
|
}
|
|
}
|
|
|
|
func Render[T any](w http.ResponseWriter, r *http.Request, data T) error {
|
|
// Pass in values that we want to be available to all pages here
|
|
|
|
r.Response.Header.Set("content-type", "text/html; charset=utf-8")
|
|
return RenderInner(w, GetTemplatingVariables(r), data)
|
|
}
|
|
|
|
func RenderInner[T any](w io.Writer, variables jet.VarMap, data T) error {
|
|
template_name, found := strings.CutPrefix(reflect.TypeFor[T]().Name(), "Data_")
|
|
if !found {
|
|
log.Panicf("struct name does not start with 'Data_': %s", template_name)
|
|
}
|
|
|
|
template, err := views.GetTemplate(template_name + ".jet.html")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
views.Parse(template_name+".jet.html", template.String())
|
|
|
|
return template.Execute(w, variables, data)
|
|
}
|
|
|
|
func GetTemplatingVariables(r *http.Request) jet.VarMap {
|
|
// Pass in values that we want to be available to all pages here
|
|
token := session.GetPixivToken(r)
|
|
baseURL := (&url.URL{
|
|
Scheme: r.URL.Scheme,
|
|
Opaque: r.URL.Opaque,
|
|
User: r.URL.User,
|
|
Host: r.URL.Host,
|
|
}).String()
|
|
originalURL := (&url.URL{
|
|
Path: r.URL.Path,
|
|
RawPath: r.URL.RawPath,
|
|
OmitHost: r.URL.OmitHost,
|
|
ForceQuery: r.URL.ForceQuery,
|
|
RawQuery: r.URL.RawQuery,
|
|
Fragment: r.URL.Fragment,
|
|
RawFragment: r.URL.RawFragment,
|
|
}).String()
|
|
pageURL := r.URL.String()
|
|
|
|
cookies := map[string]string{}
|
|
for _, name := range session.AllCookieNames {
|
|
value := session.GetCookie(r, name)
|
|
cookies[string(name)] = value
|
|
}
|
|
|
|
return jet.VarMap{}.
|
|
Set("BaseURL", baseURL).
|
|
Set("OriginalURL", originalURL).
|
|
Set("PageURL", pageURL).
|
|
Set("LoggedIn", token != "").
|
|
Set("Queries", r.URL.Query().Encode()).
|
|
Set("CookieList", cookies)
|
|
}
|