import tables, sets
import statusgo_backend/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)