Add support for editing messages

1 files changed, 2 insertions, 3 deletions

diff --git a/ui/messages/base.go b/ui/messages/base.go
index 567e614..b54691e 100644
--- a/ui/messages/base.go
+++ b/ui/messages/base.go
@@ -17,7 +17,6 @@
 package messages
 
 import (
-	"encoding/json"
 	"fmt"
 	"time"
 
@@ -55,7 +54,7 @@ type UIMessage struct {
 	IsHighlight        bool
 	IsService          bool
 	Edited             bool
-	Source             json.RawMessage
+	Event              *event.Event
 	ReplyTo            *UIMessage
 	Renderer           MessageRenderer
 }
@@ -82,7 +81,7 @@ func newUIMessage(evt *event.Event, displayname string, renderer MessageRenderer
 		IsHighlight:        false,
 		IsService:          false,
 		Edited:             len(evt.Gomuks.Edits) > 0,
-		Source:             evt.Content.VeryRaw,
+		Event:              evt,
 		Renderer:           renderer,
 	}
 }