Add chatbot

requirements.txt

@@ -3,3 +3,4 @@ flask
 frozen-flask
 gunicorn
 supervisor
+RocketChatAPIBot

scripts/chatbot.py

+#!/usr/bin/env python
+# coding=utf-8
+# Filename: chatbot.py
+# Author: Tamas Gal <tgal@km3net.de>
+"""
+The monitoring chatbot.
+
+Usage:
+    chatbot.py
+    chatbot.py (-h | --help)
+
+Options:
+    -h --help   Show this screen.
+
+"""
+import re
+import toml
+from RocketChatBot import RocketChatBot
+
+URL = "https://chat.km3net.de"
+CONFIG = "pipeline.toml"
+
+with open(CONFIG, 'r') as fobj:
+    config = toml.load(fobj)
+    BOTNAME = config['Alerts']['botname']
+    PASSWORD = config['Alerts']['password']
+
+
+def run():
+    bot = spawn_bot()
+    register_handlers(bot)
+    bot.run()
+
+
+def spawn_bot():
+    return RocketChatBot(BOTNAME, PASSWORD, URL)
+
+
+def register_handlers(bot):
+    def greet(msg, user, channel_id):
+        bot.send_message('hello @' + user, channel_id)
+
+    def status(msg, user, channel_id):
+        bot.send_message('erm... smooth datataking... for sure', channel_id)
+
+    def shifters(msg, user, channel_id):
+        try:
+            with open(CONFIG, 'r') as fobj:
+                config = toml.load(fobj)
+            shifters = msg.split("shifters are now ")[1].strip()
+            config['Alerts']['shifters'] = shifters
+            with open(CONFIG, 'w') as fobj:
+                toml.dump(config, fobj)
+            bot.send_message(
+                f'Alright, the new shifters are {shifters}, welcome!',
+                channel_id)
+        except Exception as e:
+            bot.send_message(f'something went horribly wrong... {e}',
+                             channel_id)
+
+    def help(msg, user, channel_id):
+        help_str = f"""
+        Hi @{user} I was built to take care of the monitoring alerst.
+        Here is how you can use me:
+        - `@{BOTNAME} shifters are now cnorris and bspencer`
+          -> set the new shifters who I may annoy with chat messages and
+          emails.
+        - `@{BOTNAME} status` -> show some status
+        - `@{BOTNAME} help` -> show this message
+        """
+
+        bot.send_message(help_str, channel_id)
+
+    handlers = [(['hello', 'hey', 'hi', 'ciao'], greet), (['status'], status),
+                (['help'], help), (['shifters'], shifters)]
+    for trigger, handler in handlers:
+        bot.add_dm_handler(trigger, handler)
+
+
+def main():
+    from docopt import docopt
+    args = docopt(__doc__)
+
+    run()
+
+
+if __name__ == '__main__':
+    main()