import logging
import os
import sys
from threading import Thread

from i3pystatus.core import io, util
from i3pystatus.core.exceptions import ConfigError
from i3pystatus.core.imputil import ClassFinder
from i3pystatus.core.modules import Module

DEFAULT_LOG_FORMAT = '%(asctime)s [%(levelname)-8s][%(name)s %(lineno)d] %(message)s'
log = logging.getLogger(__name__)


[docs]class CommandEndpoint:
    """
    Endpoint for i3bar click events: http://i3wm.org/docs/i3bar-protocol.html#_click_events

    :param modules: dict-like object with item access semantics via .get()
    :param io_handler_factory: function creating a file-like object returning a JSON generator on .read()
    """

    def __init__(self, modules, io_handler_factory, io):
        self.modules = modules
        self.io_handler_factory = io_handler_factory
        self.io = io
        self.thread = Thread(target=self._command_endpoint)
        self.thread.daemon = True

[docs]    def start(self):
        """Starts the background thread"""
        self.thread.start()

    def _command_endpoint(self):
        for cmd in self.io_handler_factory().read():
            target_module = self.modules.get(cmd["instance"])

            button = cmd["button"]
            kwargs = {"button_id": button}
            try:
                kwargs.update({"pos_x": cmd["x"],
                               "pos_y": cmd["y"]})
            except Exception:
                continue

            if target_module:
                target_module.on_click(button, **kwargs)
                target_module.run()
                self.io.async_refresh()


[docs]class Status:
    """
    The main class used for registering modules and managing I/O

    :param bool standalone: Whether i3pystatus should read i3status-compatible input from `input_stream`.
    :param int interval: Update interval in seconds.
    :param input_stream: A file-like object that provides the input stream, if `standalone` is False.
    :param bool click_events: Enable click events, if `standalone` is True.
    :param str logfile: Path to log file that will be used by i3pystatus.
    :param tuple internet_check: Address of server that will be used to check for internet connection by :py:class:`.internet`.
    :param keep_alive: If True, modules that define the keep_alive flag will not be put to sleep when the status bar is hidden.
    :param dictionary default_hints: Dictionary of default hints to apply to all modules. Can be overridden at a module level.
    """

    def __init__(self, standalone=True, click_events=True, interval=1,
                 input_stream=None, logfile=None, internet_check=None,
                 keep_alive=False, logformat=DEFAULT_LOG_FORMAT,
                 default_hints=None):
        self.standalone = standalone
        self.default_hints = default_hints
        self.click_events = standalone and click_events
        input_stream = input_stream or sys.stdin
        logger = logging.getLogger("i3pystatus")
        if logfile:
            for handler in logger.handlers:
                logger.removeHandler(handler)
            logfile = os.path.expandvars(logfile)
            handler = logging.FileHandler(logfile, delay=True)
            logger.addHandler(handler)
            logger.setLevel(logging.CRITICAL)
        if logformat:
            for index in range(len(logger.handlers)):
                logger.handlers[index].setFormatter(logging.Formatter(logformat))
        if internet_check:
            util.internet.address = internet_check

        self.modules = util.ModuleList(self, ClassFinder(Module))
        if self.standalone:
            self.io = io.StandaloneIO(self.click_events, self.modules, keep_alive, interval)
            if self.click_events:
                self.command_endpoint = CommandEndpoint(
                    self.modules,
                    lambda: io.JSONIO(io=io.IOHandler(sys.stdin, open(os.devnull, "w")), skiplines=1),
                    self.io)
        else:
            self.io = io.IOHandler(input_stream)

[docs]    def register(self, module, *args, **kwargs):
        """
        Register a new module.

        :param module: Either a string module name, or a module class,
                       or a module instance (in which case args and kwargs are
                       invalid).
        :param kwargs: Settings for the module.
        :returns: module instance
        """
        from i3pystatus.text import Text

        if not module:
            return

        # Merge the module's hints with the default hints
        # and overwrite any duplicates with the hint from the module
        hints = self.default_hints.copy() if self.default_hints else {}
        hints.update(kwargs.get('hints', {}))
        if hints:
            kwargs['hints'] = hints

        try:
            return self.modules.append(module, *args, **kwargs)
        except Exception as e:
            log.exception(e)
            return self.modules.append(Text(
                color="#FF0000",
                text="{i3py_mod}: Fatal Error - {ex}({msg})".format(
                    i3py_mod=module,
                    ex=e.__class__.__name__,
                    msg=e
                )
            ))

[docs]    def run(self):
        """
        Run main loop.
        """
        if self.click_events:
            self.command_endpoint.start()
        for j in io.JSONIO(self.io).read():
            for module in self.modules:
                module.inject(j)