from pubnub import utils
from pubnub.endpoints.endpoint import Endpoint
from pubnub.endpoints.mixins import TimeTokenOverrideMixin
from pubnub.enums import HttpMethod, PNOperationType
from pubnub.errors import PNERR_MESSAGE_MISSING
from pubnub.exceptions import PubNubException
from pubnub.models.consumer.pubsub import PNPublishResult


class Publish(Endpoint, TimeTokenOverrideMixin):
    # /publish/<pub_key>/<sub_key>/<signature>/<channel>/<callback>/<message>[?argument(s)]
    PUBLISH_GET_PATH = "/publish/%s/%s/0/%s/%s/%s"
    PUBLISH_POST_PATH = "/publish/%s/%s/0/%s/%s"

    def __init__(self, pubnub):
        super().__init__(pubnub)
        self._channel = None
        self._message = None
        self._should_store = None
        self._use_post = None
        self._meta = None
        self._replicate = None
        self._ptto = None
        self._ttl = None

    def channel(self, channel):
        self._channel = str(channel)
        return self

    def message(self, message):
        self._message = message
        return self

    def use_post(self, use_post):
        self._use_post = bool(use_post)
        return self

    def use_compression(self, compress=True):
        self._use_compression = bool(compress)
        return self

    def is_compressable(self):
        return True

    def should_store(self, should_store):
        self._should_store = bool(should_store)
        return self

    def meta(self, meta):
        self._meta = meta
        return self

    def ttl(self, ttl):
        self._ttl = ttl
        return self

    def build_data(self):
        if self._use_post is True:
            cipher = self.pubnub.config.cipher_key
            if cipher is not None:
                return (
                    '"'
                    + self.pubnub.config.crypto.encrypt(
                        cipher, utils.write_value_as_string(self._message)
                    )
                    + '"'
                )
            else:
                return utils.write_value_as_string(self._message)
        else:
            return None

    def encoded_params(self):
        if self._meta:
            return {"meta": utils.url_write(self._meta)}
        else:
            return {}

    def custom_params(self):
        params = TimeTokenOverrideMixin.custom_params(self)

        if self._ttl:
            params["ttl"] = self._ttl

        if self._meta:
            params["meta"] = utils.write_value_as_string(self._meta)

        if self._should_store is not None:
            if self._should_store:
                params["store"] = "1"
            else:
                params["store"] = "0"

        # REVIEW: should auth key be assigned here?
        if self.pubnub.config.auth_key is not None:
            params["auth"] = utils.url_encode(self.pubnub.config.auth_key)

        return params

    def build_path(self):
        if self._use_post:
            return Publish.PUBLISH_POST_PATH % (
                self.pubnub.config.publish_key,
                self.pubnub.config.subscribe_key,
                utils.url_encode(self._channel),
                0,
            )
        else:
            cipher = self.pubnub.config.cipher_key
            stringified_message = utils.write_value_as_string(self._message)

            if cipher is not None:
                stringified_message = (
                    '"'
                    + self.pubnub.config.crypto.encrypt(cipher, stringified_message)
                    + '"'
                )

            stringified_message = utils.url_encode(stringified_message)

            return Publish.PUBLISH_GET_PATH % (
                self.pubnub.config.publish_key,
                self.pubnub.config.subscribe_key,
                utils.url_encode(self._channel),
                0,
                stringified_message,
            )

    def http_method(self):
        if self._use_post is True:
            return HttpMethod.POST
        else:
            return HttpMethod.GET

    def validate_params(self):
        self.validate_channel()

        if self._message is None:
            raise PubNubException(pn_error=PNERR_MESSAGE_MISSING)

        self.validate_subscribe_key()
        self.validate_publish_key()

    def create_response(self, envelope):
        """
        :param envelope: an already serialized json response
        :return:
        """
        if envelope is None:
            return None

        timetoken = int(envelope[2])

        res = PNPublishResult(envelope, timetoken)

        return res

    def is_auth_required(self):
        return True

    def affected_channels(self):
        return None

    def affected_channels_groups(self):
        return None

    def request_timeout(self):
        return self.pubnub.config.non_subscribe_request_timeout

    def connect_timeout(self):
        return self.pubnub.config.connect_timeout

    def operation_type(self):
        return PNOperationType.PNPublishOperation

    def name(self):
        return "Publish"