hrfee/jfa-go
1
0
Fork 0
mirror of https://github.com/hrfee/jfa-go.git synced 2024-12-22 17:10:10 +00:00
Code Issues Projects Releases Wiki Activity

router: correctly use local FS for custom HTML

Browse Source
This commit is contained in:
Harvey Tindall 2024-05-26 18:08:48 +01:00
parent 11eae035d9
commit 9df2a82b6d
Signed by: hrfee
GPG Key ID: BBC65952848FB1A2
1 changed files with 17 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
router.go
View File

@ -24,20 +24,30 @@ func (app *appContext) loadHTML(router *gin.Engine) {
app.err.Fatalf("Couldn't access template directory: \"%s\"", templatePath) app.err.Fatalf("Couldn't access template directory: \"%s\"", templatePath)
return return
} }
loadFiles := make([]string, len(htmlFiles)) loadInternal := []string{}
for i, f := range htmlFiles { loadExternal := []string{}
for _, f := range htmlFiles {
if _, err := os.Stat(filepath.Join(customPath, f.Name())); os.IsNotExist(err) { if _, err := os.Stat(filepath.Join(customPath, f.Name())); os.IsNotExist(err) {
app.debug.Printf("Using default \"%s\"", f.Name()) app.debug.Printf("Using default \"%s\"", f.Name())
loadFiles[i] = FSJoin(templatePath, f.Name()) loadInternal = append(loadInternal, FSJoin(templatePath, f.Name()))
} else { } else {
app.info.Printf("Using custom \"%s\"", f.Name()) app.info.Printf("Using custom \"%s\"", f.Name())
loadFiles[i] = filepath.Join(filepath.Join(customPath, f.Name())) loadExternal = append(loadExternal, filepath.Join(filepath.Join(customPath, f.Name())))
} }
} }
tmpl, err := template.ParseFS(localFS, loadFiles...) var tmpl *template.Template
if len(loadInternal) != 0 {
tmpl, err = template.ParseFS(localFS, loadInternal...)
if err != nil { if err != nil {
app.err.Fatalf("Failed to load templates: %v", err) app.err.Fatalf("Failed to load templates: %v", err)
} }
}
if len(loadExternal) != 0 {
tmpl, err = tmpl.ParseFiles(loadExternal...)
if err != nil {
app.err.Fatalf("Failed to load external templates: %v", err)
}
}
router.SetHTMLTemplate(tmpl) router.SetHTMLTemplate(tmpl)
} }