#    Copyright 2016 Mirantis, Inc.
#
#    Licensed under the Apache License, Version 2.0 (the "License"); you may
#    not use this file except in compliance with the License. You may obtain
#    a copy of the License at
#
#         http://www.apache.org/licenses/LICENSE-2.0
#
#    Unless required by applicable law or agreed to in writing, software
#    distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
#    WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
#    License for the specific language governing permissions and limitations
#    under the License.

import abc
import logging

import six

from oslo_messaging._drivers.zmq_driver import zmq_async
from oslo_messaging._drivers.zmq_driver import zmq_names

zmq = zmq_async.import_zmq()
LOG = logging.getLogger(__name__)


@six.add_metaclass(abc.ABCMeta)
class Sender(object):

    @abc.abstractmethod
    def send_message(self, socket, multipart_message):
        """Send message to a socket from multipart list"""


class CentralRouterSender(Sender):

    def send_message(self, socket, multipart_message):
        message_type = int(multipart_message[zmq_names.MESSAGE_TYPE_IDX])
        routing_key = multipart_message[zmq_names.ROUTING_KEY_IDX]
        reply_id = multipart_message[zmq_names.REPLY_ID_IDX]
        message_id = multipart_message[zmq_names.MESSAGE_ID_IDX]
        socket.send(routing_key, zmq.SNDMORE)
        socket.send(b'', zmq.SNDMORE)
        socket.send(reply_id, zmq.SNDMORE)
        socket.send(multipart_message[zmq_names.MESSAGE_TYPE_IDX], zmq.SNDMORE)
        LOG.debug("Dispatching %(msg_type)s message %(msg_id)s - from %(rid)s "
                  "to -> %(rkey)s" %
                  {"msg_type": zmq_names.message_type_str(message_type),
                   "msg_id": message_id,
                   "rkey": routing_key,
                   "rid": reply_id})
        socket.send_multipart(multipart_message[zmq_names.MESSAGE_ID_IDX:])


class CentralPublisherSender(Sender):

    def send_message(self, socket, multipart_message):
        message_type = int(multipart_message[zmq_names.MESSAGE_TYPE_IDX])
        assert message_type in (zmq_names.CAST_FANOUT_TYPE,
                                zmq_names.NOTIFY_TYPE), "Fanout expected!"
        topic_filter = multipart_message[zmq_names.ROUTING_KEY_IDX]
        message_id = multipart_message[zmq_names.MESSAGE_ID_IDX]

        socket.send(topic_filter, zmq.SNDMORE)
        socket.send_multipart(multipart_message[zmq_names.MESSAGE_ID_IDX:])

        LOG.debug("Publishing message %(message_id)s on [%(topic)s]",
                  {"topic": topic_filter,
                   "message_id": message_id})