52 lines
1.8 KiB
Nim
52 lines
1.8 KiB
Nim
|
import tables, sets
|
||
|
|
import libstatus/chat
|
||
|
|
import ../eventemitter
|
||
|
|
|
||
|
|
type
|
||
|
|
MessageDetails* = object
|
||
|
|
status*: string
|
||
|
|
chatId*: string
|
||
|
|
|
||
|
|
MessagesModel* = ref object
|
||
|
|
events*: EventEmitter
|
||
|
|
messages*: Table[string, MessageDetails]
|
||
|
|
confirmations*: HashSet[string]
|
||
|
|
|
||
|
|
MessageSentArgs* = ref object of Args
|
||
|
|
messageId*: string
|
||
|
|
chatId*: string
|
||
|
|
|
||
|
|
proc newMessagesModel*(events: EventEmitter): MessagesModel =
|
||
|
|
result = MessagesModel()
|
||
|
|
result.events = events
|
||
|
|
result.messages = initTable[string, MessageDetails]()
|
||
|
|
result.confirmations = initHashSet[string]()
|
||
|
|
|
||
|
|
proc delete*(self: MessagesModel) =
|
||
|
|
discard
|
||
|
|
|
||
|
|
# For each message sent we call trackMessage to register the message id,
|
||
|
|
# and wait until an EnvelopeSent signals is emitted for that message. However
|
||
|
|
# due to communication being async, it's possible that the signal arrives
|
||
|
|
# first, hence why we check if there's a confirmation (an envelope.sent)
|
||
|
|
# inside trackMessage to emit the "messageSent" event
|
||
|
|
|
||
|
|
proc trackMessage*(self: MessagesModel, messageId: string, chatId: string) =
|
||
|
|
if self.messages.hasKey(messageId): return
|
||
|
|
|
||
|
|
self.messages[messageId] = MessageDetails(status: "sending", chatId: chatId)
|
||
|
|
if self.confirmations.contains(messageId):
|
||
|
|
self.confirmations.excl(messageId)
|
||
|
|
self.messages[messageId].status = "sent"
|
||
|
|
discard updateOutgoingMessageStatus(messageId, "sent")
|
||
|
|
self.events.emit("messageSent", MessageSentArgs(messageId: messageId, chatId: chatId))
|
||
|
|
|
||
|
|
proc updateStatus*(self: MessagesModel, messageIds: seq[string]) =
|
||
|
|
for messageId in messageIds:
|
||
|
|
if self.messages.hasKey(messageId):
|
||
|
|
self.messages[messageId].status = "sent"
|
||
|
|
discard updateOutgoingMessageStatus(messageId, "sent")
|
||
|
|
self.events.emit("messageSent", MessageSentArgs(messageId: messageId, chatId: self.messages[messageId].chatId))
|
||
|
|
else:
|
||
|
|
self.confirmations.incl(messageId)
|