From a21420cb7b0670a2a4167283326c2b39ec35fdfe Mon Sep 17 00:00:00 2001
From: evilsocket <evilsocket@gmail.com>
Date: Thu, 8 Feb 2018 08:45:22 +0100
Subject: [PATCH] fix: better endpoints related events visualization

---
 modules/events_view.go | 27 +++++++++++++++++++++++----
 1 file changed, 23 insertions(+), 4 deletions(-)

diff --git a/modules/events_view.go b/modules/events_view.go
index 3673b974..104caabb 100644
--- a/modules/events_view.go
+++ b/modules/events_view.go
@@ -22,10 +22,29 @@ func (s EventsStream) viewLogEvent(e session.Event) {
 
 func (s EventsStream) viewEndpointEvent(e session.Event) {
 	t := e.Data.(*net.Endpoint)
-	fmt.Printf("[%s] [%s] %s\n",
-		e.Time.Format(eventTimeFormat),
-		core.Green(e.Tag),
-		t)
+	if e.Tag == "endpoint.new" {
+		fmt.Printf("[%s] [%s] Endpoint %s detected as %s\n",
+			e.Time.Format(eventTimeFormat),
+			core.Green(e.Tag),
+			core.Bold(t.IpAddress),
+			t.HwAddress)
+	} else if e.Tag == "endpoint.resolved" {
+		fmt.Printf("[%s] [%s] Endpoint %s resolved as %s\n",
+			e.Time.Format(eventTimeFormat),
+			core.Green(e.Tag),
+			core.Bold(t.IpAddress),
+			core.Yellow(t.Hostname))
+	} else if e.Tag == "endpoint.lost" {
+		fmt.Printf("[%s] [%s] Endpoint %s lost\n",
+			e.Time.Format(eventTimeFormat),
+			core.Green(e.Tag),
+			core.Red(t.IpAddress))
+	} else {
+		fmt.Printf("[%s] [%s] %s\n",
+			e.Time.Format(eventTimeFormat),
+			core.Green(e.Tag),
+			t)
+	}
 }
 
 func (s EventsStream) viewModuleEvent(e session.Event) {