new: events.stream module enabled by default

session/modules/events_stream.go

@@ -0,0 +1,94 @@
+package session_modules
+
+import (
+	"fmt"
+
+	"github.com/evilsocket/bettercap-ng/core"
+	"github.com/evilsocket/bettercap-ng/session"
+)
+
+type EventsStream struct {
+	session.SessionModule
+	quit chan bool
+}
+
+func NewEventsStream(s *session.Session) *EventsStream {
+	stream := &EventsStream{
+		SessionModule: session.NewSessionModule("events.stream", s),
+		quit:          make(chan bool),
+	}
+
+	stream.AddHandler(session.NewModuleHandler("events.stream on", "",
+		"Start events stream.",
+		func(args []string) error {
+			return stream.Start()
+		}))
+
+	stream.AddHandler(session.NewModuleHandler("events.stream off", "",
+		"Stop events stream.",
+		func(args []string) error {
+			return stream.Stop()
+		}))
+
+	stream.AddHandler(session.NewModuleHandler("events.clear", "",
+		"Clear events stream.",
+		func(args []string) error {
+			stream.Session.Events.Clear()
+			return nil
+		}))
+
+	return stream
+}
+
+func (s EventsStream) Name() string {
+	return "Events Stream"
+}
+
+func (s EventsStream) Description() string {
+	return "Print events as a continuous stream."
+}
+
+func (s EventsStream) Author() string {
+	return "Simone Margaritelli <evilsocket@protonmail.com>"
+}
+
+func (s *EventsStream) Start() error {
+	if s.Running() == false {
+		s.SetRunning(true)
+
+		go func() {
+			for {
+				var e session.Event
+				select {
+				case e = <-s.Session.Events.NewEvents:
+					fmt.Printf("[%s] [%s] %v\n", e.Time.UTC().Format("2006-01-02 15:04:05"), core.Green(e.Tag), e.Data)
+					break
+
+				case <-s.quit:
+					return
+				}
+			}
+		}()
+
+		return nil
+	} else {
+		return fmt.Errorf("Events stream already started.")
+	}
+
+}
+
+func (s *EventsStream) Stop() error {
+	if s.Running() == true {
+		s.SetRunning(false)
+		s.quit <- true
+		return nil
+	} else {
+		return fmt.Errorf("Events stream already stopped.")
+	}
+}
+
+func (p EventsStream) OnSessionEnded(s *session.Session) {
+	if p.Running() {
+		p.Stop()
+	}
+}