// MIT License
//
// Copyright 2016-2017 Electric Imp
//
// SPDX-License-Identifier: MIT
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be
// included in all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
// MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO
// EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES
// OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
// ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
// OTHER DEALINGS IN THE SOFTWARE.


// Default configuration values
const MM_DEFAULT_DEBUG                  = 0;
const MM_DEFAULT_MSG_TIMEOUT            = 10;   // sec
const MM_DEFAULT_RETRY_INTERVAL         = 10;   // sec
const MM_DEFAULT_AUTO_RETRY             = 0;
const MM_DEFAULT_MAX_AUTO_RETRIES       = 0;
const MM_DEFAULT_MAX_MESSAGE_RATE       = 10;   // max 10 messages per second
const MM_DEFAULT_FIRST_MESSAGE_ID       = 0;

// Other configuration constants
const MM_QUEUE_CHECK_INTERVAL           = 0.5;  // sec
const MM_START_UP_DELAY                 = 0.5;  // sec

// Message types
const MM_MESSAGE_TYPE_DATA              = "MM_DATA";
const MM_MESSAGE_TYPE_REPLY             = "MM_REPLY";
const MM_MESSAGE_TYPE_ACK               = "MM_ACK";
const MM_MESSAGE_TYPE_NACK              = "MM_NACK";
const MM_MESSAGE_TYPE_CONNECTED         = "MM_CONNECT";
const MM_MESSAGE_TYPE_CONNECTED_REPLY   = "MM_CONNECT_REPLY";

// Error messages
const MM_ERR_NO_HANDLER                 = "No handler error";
const MM_ERR_NO_CONNECTION              = "No connection error";
const MM_ERR_USER_DROPPED_MESSAGE       = "User dropped the message";
const MM_ERR_USER_CALLED_FAIL           = "User called fail";
const MM_ERR_RATE_LIMIT_EXCEEDED        = "Maximum sending rate exceeded";

// Message names for handlers
const MM_HANDLER_NAME_ON_ACK            = "onAck";
const MM_HANDLER_NAME_ON_FAIL           = "onFail";
const MM_HANDLER_NAME_ON_REPLY          = "onReply";
const MM_HANDLER_NAME_ON_TIMEOUT        = "onTimeout";

// ConnectionManager handler ids
const MM_CM_HANDLERS_ID                 = "mmConnectionHandlers";

class MessageManager {

    static VERSION = "2.4.0";

    // Queue of messages that are pending for acknowledgement
    _sentQueue = null;

    // Queue of messages that are pending for retry
    _retryQueue = null;

    // Current message id
    _nextId = null;

    // The device or agent object
    _partner = null;

    // Timer for checking the queues
    _queueTimer = null;

    // Message timeout
    _msgTimeout = null;

    // The flag indicating that the debug mode enabled
    _debug = null;

    // ConnectionManager instance (for device only).
    // Optional parameter for MessageManager
    _cm = null;

    // Retry interval
    _retryInterval = null;

    // Flag indicating if the autoretry is enabled or not
    _autoRetry = null;

    //
    // Rate measurement variables
    //

    // Max message send rate (messages per second)
    _maxRate = null;

    // Message sent counter for rate measurement
    _rateCounter = null;

    // Last time the message send rate was measured
    _lastRateMeasured = null;

    //
    // Callback definitions
    //

    // Max number of auto retries
    _maxAutoRetries = null;

    // Name-specific handlers to be called on a message received
    _on = null;

    // Generic handler to be called on a message received without name-specific handler.
    _defaultOn = null;
    
     // Handler to be called right before a message is sent
    _beforeSend = null;

    // Handler to be called before the message is being retried
    _beforeRetry = null;

    // Global handler to be called when a message delivery failed
    _onFail = null;

    // Global handler to be called when message sits in the awaiting for ACK queue for too long
    _onTimeout = null;

    // Global handler to be called when a message is acknowledged
    _onAck = null;

    // Global handler to be called when a message is replied
    _onReply = null;

    // User defined callback to generate next message id
    _nextIdGenerator = null;

    // Callback to be executed when a partner is connected
    _onPartnerCon = null;

    // Callback to be executed when a partner responds to the "connected" notification
    _onConReply = null;

    // Data message class definition. Any message being sent by the user
    // is considered to be data message.
    //
    // The class defines the data message structure and handlers.
    DataMessage = class {

        // Message payload to be sent
        payload = null;

        // Message metadata that can be used for application specific purposes
        metadata = null;

        // Number of attempts to send the message
        tries = null;

        // Individual message timeout
        _timeout = null;

        // Message sent time
        _sent = null;

        // Time of the next retry
        _nextRetry = null;

        // Handler to be called when the message delivery failed
        _onFail = null;

        // Handler to be called when the message sits in the sent (waiting for ACK queue) for too long
        _onTimeout = null;

        // Handler to be called when the message is acknowledged
        _onAck = null;

        // Handler to be called when the message is replied
        _onReply = null;

        // Flag indicating that the message was acknowledged
        _acked = null;

        // Data message constructor
        // Constructor is not going to be called from the user code
        //
        // Parameters:
        //      id              Data message unique identifier
        //      name            Data message name
        //      data            The actual data being sent
        //      timeout         Individual message timeout
        //      metadata        Data message metadata
        //
        // Returns:             Data message object created
        function constructor(id, name, data, timeout, metadata) {
            payload = {
                "id": id,
                "type": MM_MESSAGE_TYPE_DATA,
                "name": name,
                "data": data,
                "created": time()
            };
            this.tries = 0;
            this._acked = false;
            this.metadata = metadata;
            this._timeout = timeout;
            this._nextRetry = 0;
        }

        // Sets the message-local handler to be called when an error occurs
        //
        // Parameters:
        //      handler         The handler to be called. It has signature:
        //                      handler(message, reason, retry)
        //                      Paremeters:
        //                          message         The message that received an error
        //                          reason          The error reason details
        //                          retry(interval) Retry handler, which moves the message
        //                                          to the retry queue for further processing
        //
        // Returns:             Nothing
        function onFail(handler) {
            _onFail = handler;
        }

        // Sets the message-local handler to be called when message timeout error occurs
        //
        // Parameters:
        //      handler         The handler to be called. It has signature:
        //                      handler(message, wait)
        //                      Paremeters:
        //                          message         The message that received an error
        //                          wait(duration)  Handler that returns the message back to the
        //                                          sent (waiting for ACK) queue for the
        //                                          specified period of time
        //                          fail()          Handler that makes the onFail
        //                                          to be called for the message
        //
        // Returns:             Nothing
        function onTimeout(handler) {
            _onTimeout = handler;
        }

        // Sets the message-local handler to be called on the message acknowledgement
        //
        // Parameters:
        //      handler         The handler to be called. It has signature:
        //                      handler(message)
        //                      Paremeters:
        //                          message         The message that was acked
        //
        // Returns:             Nothing
        function onAck(handler) {
            _onAck = handler;
        }

        // Sets the message-local handler to be called when the message is replied
        //
        // Parameters:
        //      handler         The handler to be called. It has signature:
        //                      handler(message, response), where
        //                          message         The message that received a reply
        //                          response        Response received as reply
        //
        // Returns:             Nothing
        function onReply(handler) {
            _onReply = handler;
        }
    }

    // MessageManager constructor
    //
    // Parameters:
    //      config          Configuration of the message manager
    //
    // Returns:             MessageManager object created
    function constructor(config = null) {

        if (!config) {
            config = {}
        }

        _sentQueue  = {};
        _retryQueue = {};
        _rateCounter = 0;
        _lastRateMeasured = 0;

        // Handlers
        _on = {};

        // Partner initialization
        _partner = _isAgent() ? device : agent;
        _partner.on(MM_MESSAGE_TYPE_ACK, _onAckReceived.bindenv(this));
        _partner.on(MM_MESSAGE_TYPE_DATA, _onDataReceived.bindenv(this));
        _partner.on(MM_MESSAGE_TYPE_NACK, _onNackReceived.bindenv(this));
        _partner.on(MM_MESSAGE_TYPE_REPLY, _onReplyReceived.bindenv(this));
        _partner.on(MM_MESSAGE_TYPE_CONNECTED, _onConReceived.bindenv(this));
        _partner.on(MM_MESSAGE_TYPE_CONNECTED_REPLY, _onConReplyReceived.bindenv(this));
        
        // This is a really dirty hack to address the "no handler" error.
        // We just give the partner some time to register its message handlers.
        // The "no handler" error is raised by the server when a message is sent between
        // agent and device and there is no callback registered for
        // the specified message name on the other side.
        // We have to do this workaround, because unfortunately until impOS natively
        // supports the "partner connected" notifications that don't raise errors,
        // there is no other way to properly handle this.
        // We understand that this is an ugly hacky way to implement things but it helps in
        // most of cases that we are aware of. And this is definitely not the approach we
        // recommend anyone to follow.
        imp.sleep(MM_START_UP_DELAY);        

        // Read configuration
        _cm              = "connectionManager"  in config ? config["connectionManager"]  : null;
        _nextIdGenerator = "nextIdGenerator"    in config ? config["nextIdGenerator"]    : null;
        _onConReply      = "onConnectedReply"   in config ? config["onConnectedReply"]   : null;
        _onPartnerCon    = "onPartnerConnected" in config ? config["onPartnerConnected"] : null;
        _debug           = "debug"              in config ? config["debug"]              : MM_DEFAULT_DEBUG;
        _retryInterval   = "retryInterval"      in config ? config["retryInterval"]      : MM_DEFAULT_RETRY_INTERVAL;
        _msgTimeout      = "messageTimeout"     in config ? config["messageTimeout"]     : MM_DEFAULT_MSG_TIMEOUT;
        _autoRetry       = "autoRetry"          in config ? config["autoRetry"]          : MM_DEFAULT_AUTO_RETRY;
        _maxAutoRetries  = "maxAutoRetries"     in config ? config["maxAutoRetries"]     : MM_DEFAULT_MAX_AUTO_RETRIES;
        _maxRate         = "maxMessageRate"     in config ? config["maxMessageRate"]     : MM_DEFAULT_MAX_MESSAGE_RATE;
        _nextId          = "firstMessageId"     in config ? config["firstMessageId"]     : MM_DEFAULT_FIRST_MESSAGE_ID;

        if (_cm) {
            local cmVersion = split(_cm.VERSION, ".");
            if (cmVersion[0].tointeger() < 3 || cmVersion[1].tointeger() < 1) {
                throw "MessageManager requires ConnectionManager version to be not less than v3.1.0";
            }
            _cm.onConnect(_onConnect.bindenv(this), MM_CM_HANDLERS_ID);
            _cm.onDisconnect(_onDisconnect.bindenv(this), MM_CM_HANDLERS_ID);
        }
    }

    // Sends data message
    //
    // Parameters:
    //      name            Name of the message to be sent
    //      data            Data to be sent
    //      handlers        onAck, onFail and onReply handlers for this message
    //                         for more details, please see DataMessage
    //      timeout         Individual message timeout
    //      metadata        Data message metadata
    //
    // Returns:             The data message object created
    function send(name, data = null, handlers = null, timeout = null, metadata = null) {
        local msg = DataMessage(_getNextId(), name, data, timeout, metadata);
        // Process per-message handlers
        if (handlers && handlers.len() > 0) {
            local onAck     = MM_HANDLER_NAME_ON_ACK     in handlers ? handlers[MM_HANDLER_NAME_ON_ACK]     : null;
            local onFail    = MM_HANDLER_NAME_ON_FAIL    in handlers ? handlers[MM_HANDLER_NAME_ON_FAIL]    : null;
            local onReply   = MM_HANDLER_NAME_ON_REPLY   in handlers ? handlers[MM_HANDLER_NAME_ON_REPLY]   : null;
            local onTimeout = MM_HANDLER_NAME_ON_TIMEOUT in handlers ? handlers[MM_HANDLER_NAME_ON_TIMEOUT] : null;

            onAck     && msg.onAck(onAck);
            onFail    && msg.onFail(onFail);
            onReply   && msg.onReply(onReply);
            onTimeout && msg.onTimeout(onTimeout);
        }
        return _send(msg);
    }

    // Sets the handler, which will be called when a message is received
    //
    // Parameters:
    //      name            The name of the message to register the handler for,
    //                      or null to register generic handler
    //      handler         The handler to be called. The handler's signature:
    //                          handler(message, reply), where
    //                              message         The message received
    //                              reply(data)     The function that can be used to reply
    //                                              to the received message:
    //                                              reply(data)
    //
    // Returns:             Nothing
    function on(name, handler) {
        if (name != null) {
            // set name-specific handler
            _on[name] <- handler;
        } else {
            // set generic handler
            _defaultOn = handler;
        }
    }

    // Sets the specified default handler, to be called when a message doesn't match any of the name-specific handlers
    //
    // Parameters:
    //      handler         The handler to be called. The handler's signature:
    //                          handler(message, reply), where
    //                              message         The message received
    //                              reply(data)     The function that can be used to reply
    //                                              to the received message:
    //                                              reply(data)
    //
    // Returns:             Nothing
    function defaultOn(handler) {
        on(null, handler);
    }

    // Sets the handler which will be called before a message is sent
    //
    // Parameters:
    //      handler         The handler to be called before send. The handler's signature:
    //                          handler(message, enqueue, drop), where
    //                              message                          The message to be sent
    //                              enqueue()                        Callback which when called
    //                                                               makes the message appended to the
    //                                                               retry queue for later processing
    //                                                               enqueue() has no arguments
    //
    //                              drop(silently, error)            Callback which when called
    //                                                               drops the message
    //                                                               drop() has two optional arguments
    //                                                                 - silently - boolean (default: true)  - if true, the onFail handlers (MessageManager.onFail and MessageManager.DataMessage.onFail) do not get called
    //                                                                 - error    - string  (default: null)  - if `silently` is false, this error is provided to the onFail handler
    //
    // Returns:             Nothing
    function beforeSend(handler) {
        _beforeSend = handler;
    }


    // Sets the handler for retry operation, which will
    // be called before the message is retried
    //
    // Parameters:
    //      handler         The handler to be called before retry. The handler's signature:
    //                          handler(message, skip, drop), where
    //                              message                          The message that was replied to
    //                              skip(duration)                   The callback which when called postpones the retry
    //                                                               attempt and leaves the message
    //                                                               in the retry queue for the specified amount of time.
    //                              drop(silently, error)            Callback which when called
    //                                                               drops the message
    //                                                               drop() has two optional arguments
    //                                                                 - silently - boolean (default: true)  - if true, the onFail handlers (MessageManager.onFail and MessageManager.DataMessage.onFail) do not get called
    //                                                                 - error    - string  (default: null)  - if `silently` is false, this error is provided to the onFail handler
    //
    // Returns:             Nothing
    function beforeRetry(handler) {
        _beforeRetry = handler;
    }

    // Sets the handler to be called when an error occurs
    //
    // Parameters:
    //      handler         The handler to be called. It has signature:
    //                      handler(message, reason, retry)
    //                      Paremeters:
    //                          message         The message that received an error
    //                          error           The string with the error description
    //                          retry           The callback to be invoked if the message should be retried
    //
    // Returns:             Nothing
    function onFail(handler) {
        _onFail = handler;
    }

    // Sets the handler to be called when message timeout error occurs
    //
    // Parameters:
    //      handler         The handler to be called. It has signature:
    //                      handler(message, wait, fail)
    //                      Paremeters:
    //                          message         The message that received an error
    //                          wait(duration)  Returns the message back to the
    //                                          sent (waiting for ACK) queue for the
    //                                          specified period of time
    //                          fail(message)   Handler that makes the onFail
    //                                          to be called for the message
    //
    // Returns:             Nothing
    function onTimeout(handler) {
        _onTimeout = handler;
    }

    // Sets the handler to be called on the message acknowledgement
    //
    // Parameters:
    //      handler         The handler to be called. It has signature:
    //                      handler(message)
    //                      Paremeters:
    //                          message         The message that was acked
    //
    // Returns:             Nothing
    function onAck(handler) {
        _onAck = handler;
    }

    // Sets the handler to be called when the message is replied
    //
    // Parameters:
    //      handler         The handler to be called. It has signature:
    //                      handler(message, response), where
    //                          message         The message that received a reply
    //                          response        Response received as reply
    //
    // Returns:             Nothing
    function onReply(handler) {
        _onReply = handler;
    }

    // Returns the overall number of pending messages
    // (either waiting for acknowledgement or hanging in the retry queue)
    //
    // Parameters:
    //
    // Returns:             The number of all the pending messages
    function getPendingCount() {
        return _sentQueue.len() + _retryQueue.len();
    }

    // Returns the current value of the "monotonic" millisecond timer.
    //
    // NOTE: the timer can overflow!!!
    // The primary purpose is rate measurements, where the above
    // limitation is not critical.
    //
    // Parameters:
    //
    // Returns:             The current value of the monotonic
    function _monotonicMillis() {
        return _isAgent() ? time() * 1000 + date().usec / 1000 : hardware.millis();
    }

    // Enqueues the message
    //
    // Parameters:
    //      msg             The message to be queued
    //
    // Returns:             Nothing
    function _enqueue(msg) {
        _log("Adding for retry: " + msg.payload.data);
        _retryQueue[msg.payload["id"]] <- msg;
        _setTimer();
    }

    // Returns true if the code is running on the agent side, false otherwise
    //
    // Parameters:
    //
    // Returns:             true for agent and false for device
    function _isAgent() {
        return imp.environment() == ENVIRONMENT_AGENT;
    }

    // Returns true if the imp and the agent are connected, false otherwise
    //
    // Parameters:
    //
    // Returns:             true if imp and agent are
    //                      connected, false otherwise
    function _isConnected() {
        local connected = false;
        if (_isAgent()) {
            connected = device.isconnected();
        } else {
            connected = _cm ? _cm.isConnected() : server.isconnected();
        }
        _log("_isConnected returns: " + connected);
        return connected;
    }

    // On connect handler, which initiates the retry queue processing
    //
    // Parameters:          None
    //
    // Returns:             Nothing
    function _onConnect() {
        _processRetryQueue();
        _partner.send(MM_MESSAGE_TYPE_CONNECTED, null);
    }

    // On disconnect handler
    //
    // Parameters:
    //      expected        The flag indicating whether it's an expected disconnect
    //                      (we ignore it for now)
    //
    // Returns:             Nothing
    function _onDisconnect(expected) {
        foreach (id, msg in _sentQueue) {
            _callOnFail(msg, MM_ERR_NO_CONNECTION);
        }
    }

    // The sent and reply queues processor
    // Handles timeouts, does resent, etc.
    //
    // Parameters:
    //
    // Returns:             Nothing
    function _processQueues() {
        // Clean up the timer
        _queueTimer = null;

        local now   = time();
        local drop  = true;

        // Process timed out messages from the sent (waiting for ack) queue
        foreach (id, msg in _sentQueue) {
            local timeout = msg._timeout ? msg._timeout : _msgTimeout;
            if (now - msg._sent >= timeout) {
                local wait = function(duration = null) {
                    local delay = duration != null ? duration : timeout;
                    msg._timeout = now - msg._sent + delay;
                    drop = false;
                }.bindenv(this);

                local fail = function() {
                    _callOnFail(msg, MM_ERR_USER_CALLED_FAIL);
                }.bindenv(this);

                if (!msg._acked) {
                    if (_isFunc(msg._onTimeout)) {
                        msg._onTimeout(msg, wait, fail);
                    } else if(_isFunc(_onTimeout)) {
                        _onTimeout(msg, wait, fail);
                    }
                }

                if (drop) {
                    delete _sentQueue[id];
                }
            }
        }

        _processRetryQueue();

        // Restart the timer if there is something pending in the queues
        if (getPendingCount()) {
            _setTimer();
        }
    }

    // Processes the retry queue
    //
    // Parameters:
    //
    // Returns:             Nothing
    function _processRetryQueue() {
        local now = time();
        // Process retry message queue
        foreach (id, msg in _retryQueue) {
            if (now >= msg._nextRetry) {
                _retry(msg);
            }
        }
    }

    // The sent and reply queues processor
    // Handles timeouts, tries, etc.
    //
    // Parameters:
    //
    // Returns:             Nothing
    function _setTimer() {
        if (_queueTimer) {
            // The timer is running already
            return;
        }
        _queueTimer = imp.wakeup(MM_QUEUE_CHECK_INTERVAL,
                                _processQueues.bindenv(this));
    }

    // Returns true if the argument is function and false otherwise
    //
    // Parameters:
    //
    // Returns:             true if the argument is function and false otherwise
    function _isFunc(f) {
        return f && typeof f == "function";
    }

    // Sends the message and restarts the queue timer
    //
    // Parameters:          Message to be sent
    //
    // Returns:             Nothing
    function _sendMessage(msg) {
        _log("Trying to send: " + msg.payload.data);

        if (!_isConnected()) {
            // Not connected, just raise the error
            _callOnFail(msg, MM_ERR_NO_CONNECTION);
            return;
        }

        local now = _monotonicMillis();
        if (now - 1000 > _lastRateMeasured || now < _lastRateMeasured) {
            // Resent the counters, if the timer's overflowen or
            // more than a second passed from the last measurement
            _rateCounter = 0;
            _lastRateMeasured = now;
        } else if (_rateCounter >= _maxRate) {
            // Rate limit exeeded, raise the error
            _callOnFail(msg, MM_ERR_RATE_LIMIT_EXCEEDED);
            return;
        }

        local payload = msg.payload;
        if (!_partner.send(MM_MESSAGE_TYPE_DATA, payload)) {
            // Message was successfully sent, increment the timer
            _rateCounter++;

            // The message was successfully sent
            // Update the sent time
            msg._sent = time();
            _sentQueue[payload["id"]] <- msg;
            // Make sure the timer is running
            _setTimer();
        } else {
            // Connectivity issue
            _callOnFail(msg, MM_ERR_NO_CONNECTION);
        }
    }

    // tries to send the message, and executes the beforeRerty handler
    //
    // Parameters:          Message to be resent
    //
    // Returns:             Nothing
    function _retry(msg) {
        _log("Retrying to send: " + msg.payload.data);

        local send = true;
        local payload = msg.payload;

        if (_isFunc(_beforeRetry)) {
            _beforeRetry(msg,
                function/*skip*/(duration = null) {
                    msg._nextRetry = time() + (duration ? duration : _retryInterval);
                    send = false;
                }.bindenv(this),
                function/*drop*/(silently = true, error = null) {
                    // User requests to dispose the message, so drop it on the floor
                    delete _retryQueue[payload["id"]];
                    send = false;
                    if (!silently) {
                        _callOnFail(msg, (error == null ? MM_ERR_USER_DROPPED_MESSAGE : error));
                    }
                }.bindenv(this)
            )
        }

        if (send) {
            msg.tries++;
            delete _retryQueue[payload["id"]];
            _sendMessage(msg);
        }
    }

    // Calls the before send handler and sends the message
    //
    // Parameters:          Message to be resent
    //
    // Returns:             The message
    function _send(msg) {
        local send = true;
        if (_isFunc(_beforeSend)) {
            _beforeSend(msg,
                function/*enqueue*/() {
                    _enqueue(msg)
                    send = false
                }.bindenv(this),
                function/*drop*/(silently = true, error = null) {
                    send = false
                    if (!silently) {
                        _callOnFail(msg, (error == null ? MM_ERR_USER_DROPPED_MESSAGE : error));
                    }
                }.bindenv(this)
            )
        }

        if (send) {
            _sendMessage(msg);
        }
        return msg;
    }

    // A handler for message received (agent/device.on) events
    //
    // Parameters:          Message payload received
    //
    // Returns:             Nothing
    function _onDataReceived(payload) {

        // Call the on callback
        local name = payload["name"];
        local data = payload["data"];
        local replied = false;
        local error = 0;

        // Name-specific handler if exists and valid, otherwise - generic 
        local handler = name in _on && _isFunc(_on[name]) ? _on[name] : _defaultOn;
                
        if (_isFunc(handler)) {
            handler(payload, function/*reply*/(data = null) {
                replied = true;
                error = _partner.send(MM_MESSAGE_TYPE_REPLY, {
                    "id"   : payload["id"],
                    "data" : data
                });
            }.bindenv(this));

            // If message was not replied, send the ACK
            if (!replied) {
                error = _partner.send(MM_MESSAGE_TYPE_ACK, {
                    "id" : payload["id"]
                });
            }
        } else {
            // No valid handler - send NACK
            error = _partner.send(MM_MESSAGE_TYPE_NACK, {
                "id" : payload["id"]
            });
        }

        if (error) {
            // Responding to a message failed
            // though this is a valid case
            _log("Responding to a message failed");
        }
    }

    // A handler for message acknowledgement
    //
    // Parameters:          Acknowledgement message containing original message id
    //
    // Returns:             Nothing
    function _onAckReceived(payload) {
        local id = payload["id"];
        if (id in _sentQueue) {
            local msg = _sentQueue[id];

            if (_isFunc(msg._onAck)) {
                msg._onAck(msg);
            } else if (_isFunc(_onAck)) {
                _onAck(msg);
            }

            // Delete the acked message from the queue if there is no _onReply handler set (either global or message-specific)
            if (!_isFunc(msg._onReply) && !_isFunc(_onReply)) {
                delete _sentQueue[id]
            } else {
                msg._acked = true;
            }
        }
    }

    // Calls an error handler if set for the message
    // with the error specified
    //
    // Parameters:
    //      msg             The message that triggered the error
    //      error           The error message
    //
    // Returns:             Nothing
    function _callOnFail(msg, error) {
        local hasHandler = false;
        local checkAndCall = function(f) {
            if (_isFunc(f)) {
                hasHandler = true;
                f(msg, error,
                    function/*retry*/(interval = null) {
                        msg._nextRetry = time() + (interval ? interval : _retryInterval);
                        _enqueue(msg);
                    }.bindenv(this)
                )
            }
        }

        checkAndCall(msg._onFail);
        !hasHandler && checkAndCall(_onFail);

        if (!hasHandler) {
            // Error handler is not set. Let's check the autoretry.
            if (_autoRetry && (!_maxAutoRetries || msg.tries < _maxAutoRetries)) {
                msg._nextRetry = time() + _retryInterval;
                _enqueue(msg);
            }
        }
    }

    // A handler for message nack events
    //
    // Parameters:          Payload containing id of the message that failed to be acknowledged
    //
    // Returns:             Nothing
    function _onNackReceived(payload) {
        local id = payload["id"];
        if (id in _sentQueue) {
            _callOnFail(_sentQueue[id], MM_ERR_NO_HANDLER);
        }
    }

    // A handler for message replies
    // Calls both user provided onAck and onReply handlers
    //
    // Parameters:          Payload containing id and response data for reply
    //
    // Returns:             Nothing
    function _onReplyReceived(payload) {
        local id = payload["id"];
        if (id in _sentQueue) {
            local msg = _sentQueue[id];

            // Make sure to call acknowledgement handlers first
            if (_isFunc(msg._onAck)) {
                msg._onAck(msg);
            } else if (_isFunc(_onAck)) {
                _onAck(msg);
            }

            // Then call the global handlers
            if (_isFunc(msg._onReply))  {
                msg._onReply(msg, payload["data"]);
            } else if (_isFunc(_onReply)) {
                _onReply(msg, payload["data"]);
            }

            delete _sentQueue[id];
        }
    }

    // Handler to be called on connected notification received
    //
    // Parameters:
    //      payload         The payload message received
    //
    // Returns:             Nothing
    function _onConReceived(payload) {
        // Call the onPartnerConnected handler
        if (_isFunc(_onPartnerCon)) {
            _onPartnerCon(
                function/*reply*/(data = null) {
                    _partner.send(MM_MESSAGE_TYPE_CONNECTED_REPLY, {
                            "data" : data
                        }
                    );
                }.bindenv(this)
            );
        }

        // Now process the retry queue
        _processRetryQueue();
    }

    // Handler to be called on receiving the connection reply from the partner
    //
    // Parameters:
    //      payload         The payload received
    //
    // Returns:             Nothing
    function _onConReplyReceived(payload) {
        // Call the onPartnerConnectedResponse handler
        if (_isFunc(_onConReply)) {
            _onConReply("data" in payload ? payload["data"] : null);
        }
    }

    // Incremental message id generator
    //
    // Parameters:
    //
    // Returns:             Next message id
    function _getNextId() {
        if (_isFunc(_nextIdGenerator)) {
            _nextId = _nextIdGenerator();
        } else {
            _nextId = (_nextId + 1) % RAND_MAX;
        }
        return _nextId;
    }

    // Implements debug logging. Sends the log message
    // to the console output if "debug" configuration
    // flag is set
    //
    // Parameters:
    //
    // Returns:             Nothing
    function _log(message) {
        if (_debug) {
            server.log("[MM] " + message);
        }
    }
}