Add option to dump request headers to stdout

Author: maetthu

README.md

@@ -15,7 +15,7 @@ $(go env GOPATH)/bin/dirhttps
 
 ## Setup
 
-_dirhttps_ needs a certificate and corresponding key to operate. Easiest option is to use the excellent [mkcert](https://github.com/FiloSottile/mkcert) tool which creates locally trusted, self-signed development certificates.
+_dirhttps_ needs a certificate and corresponding key to operate. Easiest option is to use the excellent [mkcert](https://github.com/FiloSottile/mkcert) tool which creates locally trusted, self-signed development certificates. 
 
 * Install and setup mkcert CA
 * Create dirhttps certificate
@@ -39,6 +39,7 @@ Usage:
 Flags:
       --cache           Enable client side caching
   -c, --cert string     Certificate file (default "/home/maetthu/.config/dirhttps/cert.pem")
+  -d, --dump            Dump client request headers to STDOUT
   -h, --help            help for dirhttps
   -k, --key string      Key file (default "/home/maetthu/.config/dirhttps/key.pem")
   -l, --listen string   Listen address (default ":8443")
@@ -63,4 +64,3 @@ $ dirhttps
 $ dirhttps -l :1234
 $ dirhttps -l 127.0.0.2:8443
 ```
-

cmd/root.go

@@ -8,6 +8,7 @@ import (
 	"github.com/spf13/cobra"
 	"log"
 	"net/http"
+	"net/http/httputil"
 	"os"
 	"path/filepath"
 )
@@ -85,6 +86,11 @@ var rootCmd = &cobra.Command{
 			handler = nocache(handler)
 		}
 
+		// Dump request headers?
+		if d, _ := cmd.Flags().GetBool("dump"); d {
+			handler = dump(handler)
+		}
+
 		log.Fatal(http.ListenAndServeTLS(addr, certFile, keyFile, handler))
 	},
 }
@@ -109,11 +115,13 @@ func init(){
 	certFile := filepath.Join(configDir, certFilename)
 	keyFile := filepath.Join(configDir, keyFilename)
 
-	rootCmd.Flags().StringP("listen", "l", ":8443", "Listen address")
 	rootCmd.Flags().StringP("cert", "c", certFile, "Certificate file")
 	rootCmd.Flags().StringP("key", "k", keyFile, "Key file")
-	rootCmd.Flags().Bool("no-cors", false, "Disable CORS handling")
+	rootCmd.Flags().StringP("listen", "l", ":8443", "Listen address")
+
 	rootCmd.Flags().Bool("cache", false, "Enable client side caching")
+	rootCmd.Flags().BoolP("dump", "d",false, "Dump client request headers to STDOUT")
+	rootCmd.Flags().Bool("no-cors", false, "Disable CORS handling")
 }
 
 func logger(handler http.Handler) http.Handler {
@@ -130,3 +138,12 @@ func nocache(handler http.Handler) http.Handler {
 	})
 }
 
+func dump(handler http.Handler) http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		if dump, err := httputil.DumpRequest(r, true); err == nil {
+			fmt.Printf("---\n%s---\n", dump)
+		}
+
+		handler.ServeHTTP(w, r)
+	})
+}