#!/usr/bin/env python
import logging
import requests

"""
Communicator module
"""

__author__ = "@tormakris"
__copyright__ = "Copyright 2020, GoldenPogácsa Team"
__module_name__ = "messagesender"
__version__text__ = "1"

logging.basicConfig(level=logging.INFO)
LOGGER = logging.getLogger(__name__)

class Communicator:
    """
    Class handling low level communication with consumers.
    """

    def __init__(self, currentconsumer: str, uuid: str):
        """Initialize object
        :param consumerlocator: the current consumer's IP address as a string
        :param uuid: string typed UUID.
        """
        self.currenctconsumer=currentconsumer
        self.uuid = uuid

    def sendmessage(self, message: str) -> None:
        """Send message to the current consumer. Logs the process.
        :param message: the message of type string that will be sent.
        :return: None
        """
        currentconsumer=self.currenctconsumer
        LOGGER.info(f"Sending message to {currentconsumer}")
        postresponse=requests.post(f'http://{currentconsumer}/log', json={'uuid': self.uuid, 'message': message})
        LOGGER.debug(f"Message status code is:{postresponse.status_code}")

    def discoveravailableconsumers(self) -> list:
        """Get the list of available consumer from the current primary consumer. Logs the received list.
        :return: list of consumers' IP addresses
        """
        try:
            currentconsumer = self.currenctconsumer
            response = requests.get(f'http://{currentconsumer}/consumers')
            json = response.json()
            LOGGER.info(f"List of currently available consumers: {json}")
            return json
        except Exception as e:
            LOGGER.exception(e)
            return []

    def isconsumeravailable(self) -> bool:
        """Readiness probe current consumer. Logs the result.
        :return: True if available, False otherwise
        """
        currentconsumer = self.currenctconsumer
        try:
            response = requests.get(f'http://{currentconsumer}/consumers')
            isavailable = response.status_code == 200
        except Exception as e:
            LOGGER.exception(e)
            isavailable = False
        LOGGER.info(f"Current consumer availability: {isavailable}")
        return isavailable

    def checkconsumer(self, consumer: str) -> bool:
        """Readiness probe of a particular consumer. Logs the result.
        :param consumer: the consumer's IP address
        :return: True if available, False otherwise
        """
        try:
            response = requests.get(f'http://{consumer}/consumers')
            isavailable = response.status_code == 200
        except Exception as e:
            LOGGER.exception(e)
            isavailable = False
        LOGGER.info(f"Consumer {consumer} availability: {isavailable}")
        return isavailable

    def set_currentconsumer(self,currenctconsumer) -> None:
        """Set current consumer
        :param currenctconsumer: the consumer's IP address
        :return: None
        """
        self.currenctconsumer=currenctconsumer