debug is not service anymore

1 files changed, 14 insertions, 2 deletions

diff --git a/cmd/rr/main.go b/cmd/rr/main.go
index 6065d3d1..945b8fa0 100644
--- a/cmd/rr/main.go
+++ b/cmd/rr/main.go
@@ -32,9 +32,13 @@ import (
 
 	// cli plugins
 	_ "github.com/spiral/roadrunner/cmd/rr/http"
-	"github.com/spiral/roadrunner/debug"
+	"github.com/spiral/roadrunner/cmd/rr/debug"
+
+	"github.com/spf13/cobra"
 )
 
+var debugMode bool
+
 func main() {
 	// provides ability to make local connection to services
 	rr.Container.Register(rpc.Name, &rpc.Service{})
@@ -46,7 +50,15 @@ func main() {
 	rr.Container.Register(static.Name, &static.Service{})
 
 	// provides additional verbosity
-	rr.Container.Register(debug.Name, &debug.Service{Logger: rr.Logger})
+
+	// debug mode
+	rr.CLI.PersistentFlags().BoolVarP(&debugMode, "debug", "d", false, "debug mode", )
+	cobra.OnInitialize(func() {
+		if debugMode {
+			service, _ := rr.Container.Get(http.Name)
+			service.(*http.Service).AddListener(debug.NewListener(rr.Logger).Listener)
+		}
+	})
 
 	// you can register additional commands using cmd.CLI
 	rr.Execute()