diff --git a/main.go b/main.go index 7da215a..0b11463 100644 --- a/main.go +++ b/main.go @@ -236,6 +236,18 @@ func renderIndex(w gemini.ResponseWriter, client TreeManagerClient) { return } + lastRebuild, found := db.Get("last_rebuild") + if !found { + w.WriteHeader(gemini.StatusTemporaryFailure, "Internal server error") + return + } + + pageCounter, found := db.Get("page_counter") + if !found { + w.WriteHeader(gemini.StatusTemporaryFailure, "Internal server error") + return + } + w.SetMediaType("text/gemini") tpl := pongo2.Must(pongo2.FromFile(fmt.Sprintf("templates/%s/index.gmi", defaultLang))) page, err := tpl.Execute( @@ -243,8 +255,8 @@ func renderIndex(w gemini.ResponseWriter, client TreeManagerClient) { "tree": tree, "lang": defaultLang, "lastFiles": GetLastFiles(tree.Files), - "lastRebuild": db.Get("last_rebuild"), - "pageCounter": db.Get("page_counter"), + "lastRebuild": lastRebuild, + "pageCounter": pageCounter, }, ) if err != nil {