Edm0nd/edmond/plugin.py

from dataclasses import dataclass
from pathlib import Path


class Plugin:

    REQUIRED_CONFIGS = []

    def __init__(self, bot):
        self.bot = bot
        self.name = self.__class__.__name__.lower()[:-6]  # Remove "Plugin".
        self.config = self.__get_config()
        self.is_ready = self.__check_config()

    @property
    def callbacks(self):
        """List of callback types available for this plugin."""
        return {
            cb[3:]: getattr(self, cb)
            for cb in dir(self)
            if cb.startswith("on_") and callable(getattr(self, cb))
        }

    def __get_config(self):
        """Return the plugin section from the bot config, plus common values."""
        plugins_configs = self.bot.config["plugins"]
        config = plugins_configs["common"].copy()
        config.update(plugins_configs.get(self.name, {}))
        # Load resources as config values.
        resources_dir = self.bot.config["resources_dir"]
        for key, resource_path in config.get("resources", {}).items():
            resource_path = Path(resources_dir) / resource_path
            try:
                with open(resource_path, "rt") as resource_file:
                    config[key] = [l.strip() for l in resource_file.readlines()]
            except OSError:
                self.bot.log_e(f"Could not load resource at {resource_path}.")
        return config

    def __check_config(self):
        """Return True if the plugin config is properly setup."""
        missing = False
        for key in self.REQUIRED_CONFIGS:
            if key not in self.config:
                self.bot.log_e(f"Missing '{key}' in {self.name} configuration.")
                missing = True
        return not missing

    def get_runtime_value(self, key, default=None, ns=None):
        """Get a value from the plugin runtime dict.

        This will get the value from the plugin namespace, but it is possible to
        get runtime values from other plugins using their name as `ns`.
        """
        if ns is None:
            ns = self.name
        return self.bot.values[ns].get(key, default)

    def set_runtime_value(self, key, value):
        """Set a value in the plugin runtime dict."""
        self.bot.values[self.name][key] = value

    def get_storage_value(self, key, default=None):
        """Get a value from the plugin persistent storage."""
        return self.bot.storage.get(self.name, {}).get(key, default)

    def set_storage_value(self, key, value):
        """Set a value in the plugin persistent storage."""
        if self.name not in self.bot.storage:
            self.bot.storage[self.name] = {key: value}
        else:
            self.bot.storage[self.name][key] = value

    def append_storage_list_value(self, key, value):
        """Append a value to a list in the plugin persistent storage."""
        if self.name not in self.bot.storage:
            self.bot.storage[self.name] = {key: [value]}
        elif key not in self.bot.storage[self.name]:
            self.bot.storage[self.name][key] = [value]
        else:
            self.bot.storage[self.name][key].append(value)

    def remove_storage_list_value(self, key, value):
        """Remove a value from a persistent storage list."""
        if self.name in self.bot.storage and key in self.bot.storage[self.name]:
            self.bot.storage[self.name][key].remove(value)

    def should_answer_question(self, message):
        """Store Question in object and return True if it should answer it."""
        words = message.split()
        # Is the message addressed to me?
        if words[0].lower() not in self.bot.names:
            return False

        # Are the config conditions to answer a question valid?
        if not self.__respects_handling_conditions():
            return False

        # Is it a question I can answer?
        question = message[len(words[0]):].strip()
        for q in self.config.get("questions", []):
            if question.startswith(q):
                content = question[len(q):].strip()
                content = content.rstrip("?").rstrip()
                question = Question(q, content)
                self.question = question
                self.bot.log_d(
                    f"Answering question from plugin {self.name}: {question}"
                )
                return True
        return False

    def should_handle_command(
        self,
        message,
        no_content=False,
        exclude_conditions=None,
    ):
        """Store Command in object and return True if it should handle it.

        If no_content is True, the command does not parse command contents and
        put all the command message (without suffix) to the command identifier.
        This is useful for commands that have multiple words in their identifier
        and do not have any content, or when the content parsing should be done
        by the plugin itself.

        If exclude_conditions is a collection of plugin namespaces, the
        conditions of these plugins are not tested. A basic scenario is the
        wakeup command from sleep plugin that requires excluding its own
        condition (i.e. do not handle commands when sleeping).
        """
        # Are the config conditions to handle a command valid?
        if not self.__respects_handling_conditions(exclude_conditions):
            return False

        # Is it a valid command?
        parsed_command = self.__parse_command(message, no_content=no_content)
        if not parsed_command:
            return False

        # Is it a command I can handle?
        available_commands = self.config.get("commands", [])
        aliases = self.config.get("aliases", {})
        for ident in available_commands:
            # Match commands differently according to no_content. If no_content
            # is True, check the parsed command (pc) raw data as a string that
            # may contain the available identifier (ai) at its beginning.
            # If no_content is False (default), simply compare the parsed
            # identifier with available identifiers.
            if no_content:
                match = lambda pc, ai: (
                    pc.raw == ai or pc.raw.startswith(ai + " ")
                )
            else:
                match = lambda pc, ai: pc.ident == ai
            # First case: the command identifier has been used.
            if match(parsed_command, ident):
                parsed_command.ident = ident
                parsed_command.match = ident
                self.__save_command(parsed_command)
                return True
            # Second case: an alias of the identifier has been used.
            ident_aliases = aliases.get(ident, [])
            for alias in ident_aliases:
                if match(parsed_command, alias):
                    parsed_command.ident = ident
                    parsed_command.match = alias
                    self.__save_command(parsed_command)
                    return True
        return False

    def __parse_command(self, message, no_content=False):
        """Return a command ID if this message is a command.

        The command raw field is always set. The ident and content fields are
        not set when no_content is True. The match field is never set by this
        method.
        """
        words = message.split()
        command_suffix = self.config["command_suffix"]
        if words[0].lower() in self.bot.names and words[-1] == command_suffix:
            raw = " ".join(words[1:-1])
            if no_content:
                ident = ""
                content = ""
            else:
                ident = words[1]
                content = " ".join(words[2:-1])
            return Command(ident, content, raw)

    def __respects_handling_conditions(self, exclude_conditions=None):
        """Check if question conditions are valid."""
        conditions = self.config.get("handling_conditions", {})
        for plugin_ns, plugin_conditions in conditions.items():
            if exclude_conditions and plugin_ns in exclude_conditions:
                continue
            for condition_key, condition_value in plugin_conditions.items():
                value = self.get_runtime_value(condition_key, ns=plugin_ns)
                if condition_value != value:
                    self.bot.log_d(
                        f"Handling condition {plugin_ns}.{condition_key} false."
                    )
                    return False
        return True

    def __save_command(self, command):
        """Save command in instance for further processing by the plugin."""
        self.command = command
        self.bot.log_d(f"Processing command from plugin {self.name}: {command}")

    def signal_failure(self, target):
        """Signal a plugin failure to target."""
        self.bot.say(target, self.bot.config["error_message"])


@dataclass
class Question:
    preambule: str
    content: str


@dataclass
class Command:
    # Identifier as set in config. Set even when an alias has been used.
    ident: str
    # Content of the command when it has been parsed, empty str otherwise.
    content: str
    # Raw command content (minus name and suffix), always set.
    raw: str
    # Identifier matched, possibly an alias. Set only when matched.
    match: str = ""