add header/KEY request

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

1 files changed, 6 insertions, 0 deletions

diff --git a/main.go b/main.go
index 66f5d6d..06552ab 100644
--- a/main.go
+++ b/main.go
@@ -72,6 +72,12 @@ func requestHandler(ctx *fasthttp.RequestCtx) {
 		fmt.Fprint(ctx, string(ctx.Request.Header.Header()))
 		ctx.SetStatusCode(fasthttp.StatusAccepted)
 		return
+	} else if strings.HasPrefix(strr, clientHeaderURI+"/") {
+		headerkey := strings.TrimLeft(strr, clientHeaderURI+"/")
+		log.Println("headerkey", headerkey)
+		fmt.Fprint(ctx, string(ctx.Request.Header.Peek(headerkey)))
+		ctx.SetStatusCode(fasthttp.StatusAccepted)
+		return
 	} else if strings.HasPrefix(strr, "/yt/") {
 		var req fasthttp.Request
 		var resp fasthttp.Response