YuntechNet/AtomicP2P

from ...communication.command import Command
from ..entity.peer_status import StatusType
from .communication import CheckHandler


class HelpCmd(Command):
    """HelpCmd
        show the help for monitor.
        Usage in prompt: monitor help [cmd]
    """

    def __init__(self, monitor):
        super().__init__("help")
        self.monitor = monitor
        self.peer = monitor.peer

    def _execute(self, msg_arr):
        if msg_arr != [] and msg_arr[0] in self.monitor.commands:
            return self.monitor.commands[msg_arr[0]].__doc__
        else:
            return (
                "monitor [cmd] <options>\n"
                " - pause                                          "
                "pause monitor's main loop thread.\n"
                " - period [seconds]                               "
                "change monitor's loop period to another second.\n"
                " - list                                           "
                "list each statuses in list.\n"
                " - reset [peer name/role/all]                     "
                "reset all or specific name or role's peer status t"
                "o PENDING."
            )


class PauseCmd(Command):
    """PauseCmd
        pause monitor keep checking peers.
        Usage in prompt: monitor pause
    """

    def __init__(self, monitor):
        super().__init__("pause")
        self.monitor = monitor
        self.peer = monitor.peer

    def _execute(self, msg_arr):
        self.monitor.pause = not self.monitor.pause
        return "Monitor pause: {}".format(self.monitor.pause)


class PeriodCmd(Command):
    """PeriodCmd
        adjust monitor's checking delay in seconds.
        Usage in prompt: monitor period [seconds]
    """

    def __init__(self, monitor):
        super().__init__("period")
        self.monitor = monitor
        self.peer = monitor.peer

    def _execute(self, msg_arr):
        try:
            period = int(msg_arr[0])
            self.monitor.loopDelay = period
            return "Monitor check sending period \
                       change to: {} seconds.".format(
                period
            )
        except ValueError:
            return "Please input a integer: {}".format(msg_arr[0])


class ListCmd(Command):
    """ListCmd
        list all peer status in list
        Usage in prompt: monitor list
    """

    def __init__(self, monitor):
        super().__init__("list")
        self.monitor = monitor
        self.peer = monitor.peer

    def _execute(self, msg_arr):
        if len(self.monitor.peer.connectlist) == 0:
            return "There is no peer's info in current list"
        else:
            output_text = "Current peers status:"
            for each in self.monitor.peer.connectlist:
                output_text += " - " + str(each.status) + "\n"
            output_text += "[---End of list---]"
            return output_text


class ResetCmd(Command):
    """ResetCmd
        reset specific or every peer's status in monitor's list.
        Usage in prompt: monitor reset all
    """

    def __init__(self, monitor):
        super().__init__("reset")
        self.monitor = monitor
        self.peer = monitor.peer

    def _execute(self, msg_arr):
        if msg_arr == []:
            for each in self.peer.connectlist:
                each.status.update(status_type=StatusType.PENDING)
                self.monitor.logger.info(each.status)
            self.monitor.logger.info("Reset success")
        else:
            pass


class VerboseCmd(Command):
    """VerboseCmd
        toggle verbose flag in monitor to output more detail or not.
        Usage in prompt: monitor verbose
    """

    def __init__(self, monitor):
        super().__init__("verbose")
        self.monitor = monitor
        self.peer = monitor.peer

    def _execute(self, msg_arr):
        self.monitor.verbose = not self.monitor.verbose
        return "Monitor verbose toggle to: {}".format(self.monitor.verbose)


class ManualCmd(Command):
    """ManualCmd
        manually to send a check pkt to specific peer.
        Usage in prompt: monitor manuadl [ip:port]
    """

    def __init__(self, monitor):
        super().__init__("manual")
        self.monitor = monitor
        self.peer = monitor.peer

    def _execute(self, msg_arr):
        host = msg_arr[0].split(":")
        try:
            host[1] = int(host[1])
            self.peer.handler_unicast_packet(
                host=(host[0], host[1]), pkt_type=CheckHandler.pkt_type
            )
        except ValueError:
            self.peer.handler_broadcast_packet(
                host=(host[0], host[1]), pkt_type=CheckHandler.pkt_type
            )
        if self.monitor.verbose:
            return "Sended a monitor check to: {}".format(host)