From 149cbe7681607eb48f47df14c7a39e6f289fb7b6 Mon Sep 17 00:00:00 2001
From: Alan Pearce
Date: Wed, 15 May 2024 12:35:40 +0200
Subject: refactor(templates): render partials with less hackiness

---
 internal/server/mux.go       |  6 +++---
 internal/server/templates.go | 13 ++++++++-----
 2 files changed, 11 insertions(+), 8 deletions(-)

(limited to 'internal/server')

diff --git a/internal/server/mux.go b/internal/server/mux.go
index f37bd6e..aa2c1a6 100644
--- a/internal/server/mux.go
+++ b/internal/server/mux.go
@@ -106,7 +106,7 @@ func NewMux(
 			Sources:       config.Importer.Sources,
 			Version:       *versionInfo,
 		}
-		err := templates["index"].ExecuteTemplate(w, "index.gotmpl", indexData)
+		err := templates["index"].Execute(w, indexData)
 		if err != nil {
 			http.Error(w, err.Error(), http.StatusInternalServerError)
 		}
@@ -192,9 +192,9 @@ func NewMux(
 
 			if r.Header.Get("Fetch") == "true" {
 				w.Header().Add("Content-Type", "text/html; charset=utf-8")
-				err = templates["options"].ExecuteTemplate(w, "options.gotmpl", tdata)
+				err = templates["options"].ExecuteTemplate(w, "results", tdata)
 			} else {
-				err = templates["options"].ExecuteTemplate(w, "index.gotmpl", tdata)
+				err = templates["options"].Execute(w, tdata)
 			}
 			if err != nil {
 				slog.Error("template error", "template", "options", "error", err)
diff --git a/internal/server/templates.go b/internal/server/templates.go
index bd3ad90..d8ffcf5 100644
--- a/internal/server/templates.go
+++ b/internal/server/templates.go
@@ -47,7 +47,10 @@ var templateFuncs = template.FuncMap{
 	},
 }
 
-func loadTemplate(layoutFile string, filename string) (*template.Template, error) {
+func loadTemplate(
+	layoutFile string,
+	filenames ...string,
+) (*template.Template, error) {
 	tpl := template.New("index.gotmpl")
 
 	tpl.Funcs(templateFuncs)
@@ -56,8 +59,8 @@ func loadTemplate(layoutFile string, filename string) (*template.Template, error
 		return nil, errors.WithMessage(err, "could not parse layout template")
 	}
 
-	if filename != "" {
-		_, err = tpl.ParseFS(frontend.Files, filename)
+	if len(filenames) > 0 {
+		_, err = tpl.ParseFS(frontend.Files, filenames...)
 		if err != nil {
 			return nil, errors.WithMessage(err, "could not parse template")
 		}
@@ -72,7 +75,7 @@ func loadTemplates() (TemplateCollection, error) {
 
 	layoutFile := path.Join(templateDir, "index.gotmpl")
 
-	index, err := loadTemplate(layoutFile, "")
+	index, err := loadTemplate(layoutFile)
 	if err != nil {
 		return nil, err
 	}
@@ -84,7 +87,7 @@ func loadTemplates() (TemplateCollection, error) {
 		return nil, errors.WithMessage(err, "could not glob block templates")
 	}
 	for _, fullname := range templatePaths {
-		tpl, err := loadTemplate(layoutFile, glob)
+		tpl, err := loadTemplate(layoutFile, glob, fullname)
 		if err != nil {
 			return nil, err
 		}
-- 
cgit 1.4.1