Fix bugs and compile errors

2 files changed, 1 insertions, 5 deletions

diff --git a/ui/widget/message-view.go b/ui/widget/message-view.go
index 444aa03..dc0cde5 100644
--- a/ui/widget/message-view.go
+++ b/ui/widget/message-view.go
@@ -152,7 +152,6 @@ func (view *MessageView) AddMessage(message *types.Message, direction MessageDir
 	if msg != nil && messageExists {
 		message.CopyTo(msg)
 		message = msg
-		message.SetIsSending(false)
 		direction = IgnoreMessage
 	}
 
diff --git a/ui/widget/room-view.go b/ui/widget/room-view.go
index 83073c2..69216fb 100644
--- a/ui/widget/room-view.go
+++ b/ui/widget/room-view.go
@@ -253,10 +253,7 @@ func (view *RoomView) NewTempMessage(msgtype, text string) *types.Message {
 	id := strconv.FormatInt(now.UnixNano(), 10)
 	sender := view.Room.GetSessionOwner().DisplayName
 	message := view.NewMessage(id, sender, msgtype, text, now)
-	message.SetIsSending(true)
-	message.TimestampColor = tcell.ColorGray
-	message.TextColor = tcell.ColorGray
-	message.SenderColor = tcell.ColorGray
+	message.State = types.MessageStateSending
 	view.AddMessage(message, AppendMessage)
 	return message
 }