From f884386a5487b8ac757e743bb8945f0af58270b9 Mon Sep 17 00:00:00 2001
From: Maddie Zhan <maddie.zhan@cynovan.com>
Date: Mon, 17 Aug 2020 15:23:54 +0800
Subject: [PATCH] Support HEAD method for all endpoints via middleware

---
 web/web.go | 13 +++----------
 1 file changed, 3 insertions(+), 10 deletions(-)

diff --git a/web/web.go b/web/web.go
index a5ff8b1..fa548b6 100644
--- a/web/web.go
+++ b/web/web.go
@@ -31,8 +31,9 @@ var (
 )
 
 func ListenAndServe(conf *config.Config) error {
-	r := chi.NewMux()
+	r := chi.NewRouter()
 	r.Use(middleware.RealIP)
+	r.Use(middleware.GetHead)
 
 	cs := cors.New(cors.Options{
 		AllowedOrigins: []string{"*"},
@@ -43,20 +44,16 @@ func ListenAndServe(conf *config.Config) error {
 	r.Use(cs.Handler)
 	r.Use(middleware.NoCache)
 	r.Use(middleware.Logger)
+	r.Use(middleware.Recoverer)
 
 	log.Infof("Starting backend server on %s", net.JoinHostPort(conf.BindAddress, conf.Port))
 	r.Get("/*", pages)
-	r.Head("/*", pages)
 	r.HandleFunc("/empty", empty)
 	r.HandleFunc("/backend/empty", empty)
 	r.Get("/garbage", garbage)
-	r.Head("/garbage", garbage)
 	r.Get("/backend/garbage", garbage)
-	r.Head("/backend/garbage", garbage)
 	r.Get("/getIP", getIP)
-	r.Head("/getIP", getIP)
 	r.Get("/backend/getIP", getIP)
-	r.Head("/backend/getIP", getIP)
 	r.Get("/results", results.DrawPNG)
 	r.Get("/results/", results.DrawPNG)
 	r.Get("/backend/results", results.DrawPNG)
@@ -70,13 +67,9 @@ func ListenAndServe(conf *config.Config) error {
 	r.HandleFunc("/empty.php", empty)
 	r.HandleFunc("/backend/empty.php", empty)
 	r.Get("/garbage.php", garbage)
-	r.Head("/garbage.php", garbage)
 	r.Get("/backend/garbage.php", garbage)
-	r.Head("/backend/garbage.php", garbage)
 	r.Get("/getIP.php", getIP)
-	r.Head("/getIP.php", getIP)
 	r.Get("/backend/getIP.php", getIP)
-	r.Head("/backend/getIP.php", getIP)
 	r.Post("/results/telemetry.php", results.Record)
 	r.Post("/backend/results/telemetry.php", results.Record)
 	r.HandleFunc("/stats.php", results.Stats)