From a206fc3270ea6e469b1704f15f2f15a9f6a14bbd Mon Sep 17 00:00:00 2001
From: Wolfy-J <wolfy.jd@gmail.com>
Date: Wed, 26 Sep 2018 22:30:53 +0300
Subject: improved debug handlers

---
 cmd/rr/http/debug.go | 84 +++++++++++++++++++++++++++++++++++++++++++++++-----
 1 file changed, 77 insertions(+), 7 deletions(-)

(limited to 'cmd/rr/http/debug.go')

diff --git a/cmd/rr/http/debug.go b/cmd/rr/http/debug.go
index f69e10a8..53980303 100644
--- a/cmd/rr/http/debug.go
+++ b/cmd/rr/http/debug.go
@@ -1,20 +1,90 @@
 package http
 
 import (
-	rr "github.com/spiral/roadrunner/cmd/rr/cmd"
-
+	"fmt"
+	"github.com/sirupsen/logrus"
 	"github.com/spf13/cobra"
-	"github.com/spiral/roadrunner/cmd/rr/debug"
-	"github.com/spiral/roadrunner/service/http"
+	"github.com/spiral/roadrunner"
+	rr "github.com/spiral/roadrunner/cmd/rr/cmd"
+	"github.com/spiral/roadrunner/cmd/util"
+	rrhttp "github.com/spiral/roadrunner/service/http"
+	"net/http"
 )
 
 func init() {
 	cobra.OnInitialize(func() {
 		if rr.Debug {
-			svc, _ := rr.Container.Get(http.ID)
-			if svc, ok := svc.(*http.Service); ok {
-				svc.AddListener(debug.Listener(rr.Logger))
+			svc, _ := rr.Container.Get(rrhttp.ID)
+			if svc, ok := svc.(*rrhttp.Service); ok {
+				svc.AddListener((&debugger{logger: rr.Logger}).listener)
 			}
 		}
 	})
 }
+
+// listener provide debug callback for system events. With colors!
+type debugger struct{ logger *logrus.Logger }
+
+// listener listens to http events and generates nice looking output.
+func (s *debugger) listener(event int, ctx interface{}) {
+	if util.LogEvent(s.logger, event, ctx) {
+		// handler by default debug package
+		return
+	}
+
+	// http events
+	switch event {
+	case rrhttp.EventResponse:
+		e := ctx.(*rrhttp.ResponseEvent)
+		s.logger.Info(util.Sprintf(
+			"<cyan+h>%s</reset> %s <white+hb>%s</reset> %s",
+			e.Request.RemoteAddr,
+			statusColor(e.Response.Status),
+			e.Request.Method,
+			e.Request.URI,
+		))
+	case rrhttp.EventError:
+		e := ctx.(*rrhttp.ErrorEvent)
+
+		if _, ok := e.Error.(roadrunner.JobError); ok {
+			s.logger.Info(util.Sprintf(
+				"%s <white+hb>%s</reset> %s",
+				statusColor(500),
+				e.Request.Method,
+				uri(e.Request),
+			))
+		} else {
+			s.logger.Info(util.Sprintf(
+				"%s <white+hb>%s</reset> %s <red>%s</reset>",
+				statusColor(500),
+				e.Request.Method,
+				uri(e.Request),
+				e.Error,
+			))
+		}
+	}
+}
+
+func statusColor(status int) string {
+	if status < 300 {
+		return util.Sprintf("<green>%v</reset>", status)
+	}
+
+	if status < 400 {
+		return util.Sprintf("<cyan>%v</reset>", status)
+	}
+
+	if status < 500 {
+		return util.Sprintf("<yellow>%v</reset>", status)
+	}
+
+	return util.Sprintf("<red>%v</reset>", status)
+}
+
+func uri(r *http.Request) string {
+	if r.TLS != nil {
+		return fmt.Sprintf("https://%s%s", r.Host, r.URL.String())
+	}
+
+	return fmt.Sprintf("http://%s%s", r.Host, r.URL.String())
+}
-- 
cgit v1.2.3