add a simple header printer

Signed-off-by: jet tsang zeon-git <zeon-git@jettsang.com>

1 files changed, 5 insertions, 0 deletions

diff --git a/main.go b/main.go
index a489c51..fc456d4 100644
--- a/main.go
+++ b/main.go
@@ -24,6 +24,7 @@ var blockTxt string
 var shortenMap map[string]string
 var listen string
 var reloadURI = "/reload"
+var clientHeaderURI = "/header"
 
 func init() {
 	const (
@@ -60,6 +61,10 @@ func requestHandler(ctx *fasthttp.RequestCtx) {
 		fmt.Fprint(ctx, "ok")
 		ctx.SetStatusCode(fasthttp.StatusAccepted)
 		return
+	} else if string(ctx.RequestURI()) == clientHeaderURI {
+		fmt.Fprint(ctx, string(ctx.Request.Header.Header()))
+		ctx.SetStatusCode(fasthttp.StatusAccepted)
+		return
 	}
 	if relayUrl, exist := shortenMap[string(ctx.Path())]; exist {
 		var newRequest fasthttp.Request