Add Favicon

2 files changed, 18 insertions, 0 deletions

diff --git a/assets/favicon.ico b/assets/favicon.ico
new file mode 100644
index 0000000..b6d0998
--- /dev/null
+++ b/assets/favicon.ico
diff --git a/gopherproxy.go b/gopherproxy.go
index b65418d..f034cb9 100644
--- a/gopherproxy.go
+++ b/gopherproxy.go
@@ -166,6 +166,18 @@ func RobotsTxtHandler(robotstxtdata []byte) http.HandlerFunc {
         }
 }
 
+func FaviconHandler(favicondata []byte) http.HandlerFunc {
+        return func(w http.ResponseWriter, req *http.Request) {
+                if favicondata == nil {
+                        http.Error(w, "Not Found", http.StatusNotFound)
+                        return
+                }
+
+                w.Header().Set("Content-Type", "image/vnd.microsoft.icon")
+                w.Write(favicondata)
+        }
+}
+
 func FontHandler(woff2 bool, fontdata []byte) http.HandlerFunc {
         return func(w http.ResponseWriter, req *http.Request) {
                 if fontdata == nil {
@@ -225,6 +237,11 @@ func ListenAndServe(bind, robotsfile, uri string) error {
                 styletext = ""
         }
 
+        favicondata, err := box.Find("favicon.ico")
+        if err != nil {
+                favicondata = []byte{}
+        }
+
         tpldata, err := ioutil.ReadFile(".template")
         if err == nil {
                 tpltext = string(tpldata)
@@ -246,6 +263,7 @@ func ListenAndServe(bind, robotsfile, uri string) error {
 
         http.HandleFunc("/", GopherHandler(tpl, robotsdata, styletext, uri))
         http.HandleFunc("/robots.txt", RobotsTxtHandler(robotstxtdata))
+        http.HandleFunc("/favicon.ico", FaviconHandler(favicondata))
         http.HandleFunc("/iosevka-term-ss03-regular.woff", FontHandler(false, fontdataw))
         http.HandleFunc("/iosevka-term-ss03-regular.woff2", FontHandler(true, fontdataw2))
         //http.Handle("/assets/", http.StripPrefix("/assets/", http.FileServer(http.Dir("assets/"))))