From dcd12a4a6d22ca4af4e21e35ad4353b983d72511 Mon Sep 17 00:00:00 2001 From: bouncepaw Date: Tue, 2 Feb 2021 22:28:26 +0500 Subject: [PATCH] Allow adding styles in addition to default to /static/custom.css --- main.go | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/main.go b/main.go index 068f706..cf4559f 100644 --- a/main.go +++ b/main.go @@ -119,12 +119,15 @@ func handlerRandom(w http.ResponseWriter, rq *http.Request) { func handlerStyle(w http.ResponseWriter, rq *http.Request) { log.Println(rq.URL) - if _, err := os.Stat(WikiDir + "/static/common.css"); err == nil { - http.ServeFile(w, rq, WikiDir+"/static/common.css") + if _, err := os.Stat(util.WikiDir + "/static/common.css"); err == nil { + http.ServeFile(w, rq, util.WikiDir+"/static/common.css") } else { w.Header().Set("Content-Type", "text/css;charset=utf-8") w.Write([]byte(templates.DefaultCSS())) } + if bytes, err := ioutil.ReadFile(util.WikiDir + "/static/custom.css"); err == nil { + w.Write(bytes) + } } func handlerIcon(w http.ResponseWriter, rq *http.Request) {