diff options
Diffstat (limited to 'ui/messages/base.go')
-rw-r--r-- | ui/messages/base.go | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/ui/messages/base.go b/ui/messages/base.go index 6049ae3..441aca1 100644 --- a/ui/messages/base.go +++ b/ui/messages/base.go @@ -18,6 +18,7 @@ package messages import ( "encoding/gob" + "maunium.net/go/mautrix" "time" "maunium.net/go/gomuks/config" @@ -33,7 +34,7 @@ func init() { type BaseMessage struct { MsgID string - MsgType string + MsgType mautrix.MessageType MsgSenderID string MsgSender string MsgSenderColor tcell.Color @@ -47,7 +48,7 @@ type BaseMessage struct { prevPrefs config.UserPreferences } -func newBaseMessage(id, sender, displayname, msgtype string, timestamp time.Time) BaseMessage { +func newBaseMessage(id, sender, displayname string, msgtype mautrix.MessageType, timestamp time.Time) BaseMessage { return BaseMessage{ MsgSenderID: sender, MsgSender: displayname, @@ -194,11 +195,11 @@ func (msg *BaseMessage) SetID(id string) { msg.MsgID = id } -func (msg *BaseMessage) Type() string { +func (msg *BaseMessage) Type() mautrix.MessageType { return msg.MsgType } -func (msg *BaseMessage) SetType(msgtype string) { +func (msg *BaseMessage) SetType(msgtype mautrix.MessageType) { msg.MsgType = msgtype } |