metamask-extension/app/scripts/lib/message-manager.js

module.exports = new MessageManager()

function MessageManager (opts) {
  this.messages = []
}

MessageManager.prototype.getMsgList = function () {
  return this.messages
}

MessageManager.prototype.unconfirmedMsgs = function () {
  var messages = this.getMsgList()
  return messages.filter(msg => msg.status === 'unconfirmed')
  .reduce((result, msg) => { result[msg.id] = msg; return result }, {})
}

MessageManager.prototype._saveMsgList = function (msgList) {
  this.messages = msgList
}

MessageManager.prototype.addMsg = function (msg) {
  var messages = this.getMsgList()
  messages.push(msg)
  this._saveMsgList(messages)
}

MessageManager.prototype.getMsg = function (msgId) {
  var messages = this.getMsgList()
  var matching = messages.filter(msg => msg.id === msgId)
  return matching.length > 0 ? matching[0] : null
}

MessageManager.prototype.confirmMsg = function (msgId) {
  this._setMsgStatus(msgId, 'confirmed')
}

MessageManager.prototype.rejectMsg = function (msgId) {
  this._setMsgStatus(msgId, 'rejected')
}

MessageManager.prototype._setMsgStatus = function (msgId, status) {
  var msg = this.getMsg(msgId)
  if (msg) msg.status = status
  this.updateMsg(msg)
}

MessageManager.prototype.updateMsg = function (msg) {
  var messages = this.getMsgList()
  var found, index
  messages.forEach((otherMsg, i) => {
    if (otherMsg.id === msg.id) {
      found = true
      index = i
    }
  })
  if (found) {
    messages[index] = msg
  }
  this._saveMsgList(messages)
}