plugin: add question-type plugins

edmond/__main__.py

@@ -1,4 +1,5 @@
 import argparse
+import random
 
 import edmond.bot
 import edmond.config
@@ -10,6 +11,8 @@ def main():
     argparser.add_argument("-c", "--config", default="config.json")
     args = argparser.parse_args()
 
+    random.seed()
+
     logger = edmond.log.get_logger(name="edmond")
     config = edmond.config.load_config(args.config, logger=logger)
     bot = edmond.bot.Bot(config, logger)

edmond/bot.py

@@ -16,6 +16,7 @@ class Bot(irc.client.SimpleIRCClient, Logger):
         self.config = config
         self.logger = logger
         self.plugins = []
+        self.values = {}
 
     @property
     def nick(self):
@@ -27,6 +28,7 @@ class Bot(irc.client.SimpleIRCClient, Logger):
 
     def on_welcome(self, connection, event):
         self.log_i(f"Connected to server {event.source}.")
+        self.run_plugin_callbacks(event)
         for channel in self.config["channels"]:
             connection.join(channel)
 
@@ -74,6 +76,7 @@ class Bot(irc.client.SimpleIRCClient, Logger):
             class_name = plugin_name.capitalize() + "Plugin"
             plugin_class = getattr(module, class_name)
             self.plugins.append(plugin_class(self))
+            self.values[plugin_name] = {}
             self.log_d(f"Loaded {class_name}.")
 
     def say(self, target, message):

edmond/plugin.py

@@ -5,6 +5,7 @@ class Plugin:
 
     def __init__(self, bot):
         self.bot = bot
+        self.name = self.__class__.__name__.lower()[:-6]  # Remove "Plugin".
         self.config = self.get_config()
 
     @property
@@ -18,12 +19,34 @@ class Plugin:
 
     def get_config(self):
         """Return the plugin section from the bot config."""
-        name = self.__class__.__name__.lower()[:-6]  # Remove Plugin suffix.
         plugins_configs = self.bot.config["plugins"]
-        return plugins_configs.get(name, {})
+        return plugins_configs.get(self.name, {})
+
+    def get_runtime_value(self, key, plugin_name=None):
+        """Get a value from the plugin runtime dict."""
+        if plugin_name is None:
+            plugin_name = self.name
+        return self.bot.values[self.name].get(key)
+
+    def set_runtime_value(self, key, value):
+        """Set a value in the plugin runtime dict."""
+        self.bot.values[self.name][key] = value
+
+    def should_answer_question(self, message):
+        """Store Question in object and return True if it should answer it."""
+        words = message.split()
+        if words[0].lower() not in self.bot.names:
+            return False
+        question = message[len(words[0]):].strip()
+
+        for q in self.QUESTIONS:
+            if question.startswith(q):
+                self.question = Question(q, question[len(q):].strip())
+                return True
+        return False
 
     def should_handle_command(self, message):
-        """Store Command in object and return True if it should handle it. """
+        """Store Command in object and return True if it should handle it."""
         command = self.parse_command(message)
         if command and any(c == command.ident for c in self.COMMANDS):
             self.command = command
@@ -39,6 +62,12 @@ class Plugin:
             return Command(ident, content)
 
 
+@dataclass
+class Question:
+    preambule: str
+    content: str
+
+
 @dataclass
 class Command:
     ident: str

edmond/plugins/horoscope.py

@@ -19,3 +19,4 @@ class HoroscopePlugin(Plugin):
         text = http_get(self.config["url"])
         if text:
             self.bot.say(event.target, text)
+        return True

edmond/plugins/mood.py

@@ -0,0 +1,31 @@
+import random
+from enum import Enum
+
+from edmond.plugin import Plugin
+
+
+class Mood(Enum):
+    CALM = 0
+    PISSED = 1
+
+
+class MoodPlugin(Plugin):
+
+    QUESTIONS = ["how are you?"]
+
+    def __init__(self, bot):
+        super().__init__(bot)
+
+    def on_welcome(self, event):
+        mood = random.choice(list(Mood))
+        self.set_runtime_value("mood", mood)
+
+    def on_pubmsg(self, event):
+        if not self.should_answer_question(event.arguments[0]):
+            return False
+        mood = self.get_runtime_value("mood")
+        if mood == Mood.CALM:
+            self.bot.say(event.target, "I'm calm.")
+        elif mood == Mood.PISSED:
+            self.bot.say(event.target, "I'm pissed off.")
+        return True

edmond/plugins/opinion.py

@@ -1,8 +0,0 @@
-from edmond.plugin import Plugin
-
-
-class OpinionPlugin(Plugin):
-
-    def __init__(self, bot):
-        super().__init__(bot)
-