From e5a7accc9c6bfd191f9e8b23203d572de27c05d1 Mon Sep 17 00:00:00 2001
From: evilsocket <evilsocket@gmail.com>
Date: Tue, 20 Feb 2018 03:07:23 +0100
Subject: [PATCH] fix: better rendering of wifi.station.* events

---
 modules/events_view.go | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/modules/events_view.go b/modules/events_view.go
index d8d40d7c..ba6aa677 100644
--- a/modules/events_view.go
+++ b/modules/events_view.go
@@ -19,6 +19,33 @@ func (s EventsStream) viewLogEvent(e session.Event) {
 		e.Data.(session.LogMessage).Message)
 }
 
+func (s EventsStream) viewStationEvent(e session.Event) {
+	st := e.Data.(*network.Station)
+	vend := ""
+	if st.Vendor != "" {
+		vend = fmt.Sprintf(" (%s)", st.Vendor)
+	}
+
+	if e.Tag == "wifi.station.new" {
+		fmt.Printf("[%s] WiFi station %s detected as %s%s.\n",
+			e.Time.Format(eventTimeFormat),
+			core.Bold(st.ESSID()),
+			core.Green(st.BSSID()),
+			vend)
+	} else if e.Tag == "wifi.station.lost" {
+		fmt.Printf("[%s] WiFi station %s (%s) lost.\n",
+			e.Time.Format(eventTimeFormat),
+			core.Red(st.ESSID()),
+			st.BSSID())
+	} else {
+		fmt.Printf("[%s] [%s] %s\n",
+			e.Time.Format(eventTimeFormat),
+			core.Green(e.Tag),
+			st)
+	}
+
+}
+
 func (s EventsStream) viewEndpointEvent(e session.Event) {
 	t := e.Data.(*network.Endpoint)
 	vend := ""
@@ -74,6 +101,8 @@ func (s *EventsStream) View(e session.Event, refresh bool) {
 			s.viewLogEvent(e)
 		} else if strings.HasPrefix(e.Tag, "endpoint.") {
 			s.viewEndpointEvent(e)
+		} else if strings.HasPrefix(e.Tag, "wifi.station.") {
+			s.viewStationEvent(e)
 		} else if strings.HasPrefix(e.Tag, "mod.") {
 			s.viewModuleEvent(e)
 		} else if strings.HasPrefix(e.Tag, "net.sniff.") {