Merge pull request #15 from xBelladonna/discord

Add support for Discord (using webhooks)

discord.py

@@ -0,0 +1,81 @@
+import logging
+from pwnagotchi.voice import Voice
+import pwnagotchi.plugins as plugins
+import os
+import requests
+import subprocess
+
+
+class Discord(plugins.Plugin):
+    __author__ = 'isabelladonnamoore@outlook.com'
+    __version__ = '1.0.0'
+    __license__ = 'GPL3'
+    __description__ = 'Post recent activity to a Discord channel using webhooks. Requires discord.py module.'
+
+    def __init__(self):
+        self.ready = False
+
+    def on_loaded(self):
+        try:
+            import discord
+        except ImportError as e:
+            logging.error("Discord: discord.py module is not installed, cannot post to Discord")
+            logging.debug(e)
+            return
+
+        if 'webhook_url' not in self.options or not self.options['webhook_url']:
+            logging.error("Discord: Webhook URL is not set, cannot post to Discord")
+            return
+
+        if 'username' not in self.options or not self.options['username']:
+            with open('/etc/hostname') as fp:
+                self.options['username'] = fp.read().strip()
+
+        self.ready = True
+        logging.info("Discord: plugin loaded")
+
+    # called when there's available internet
+    def on_internet_available(self, agent):
+        if not self.ready:
+            return
+
+        config = agent.config()
+        display = agent.view()
+        last_session = agent.last_session
+
+        if last_session.is_new() and last_session.handshakes > 0:
+            try:
+                from discord import Webhook, RequestsWebhookAdapter, File
+            except ImportError as e:
+                logging.error("Discord: couldn't import discord.py")
+                logging.debug(e)
+                return
+
+            logging.info("Discord: detected new activity and internet, time to send a message!")
+
+            picture = '/var/tmp/pwnagotchi/pwnagotchi.png' if os.path.exists(
+                "/var/tmp/pwnagotchi/pwnagotchi.png") else '/root/pwnagotchi.png'
+            display.on_manual_mode(last_session)
+            display.image().save(picture, 'png')
+            display.update(force=True)
+
+            try:
+                logging.info("Discord: sending message...")
+
+                message = Voice(lang=config['main']['lang']).on_last_session_tweet(
+                    last_session)
+                url = self.options['webhook_url']
+                username = self.options['username']
+
+                webhook = Webhook.from_url(
+                    url, adapter=RequestsWebhookAdapter())
+                webhook.send(
+                    message, username=username, file=File(picture))
+                logging.info("Discord: message sent: %s" % message)
+
+                last_session.save_session_id()
+                display.set('status', 'Discord notification sent!')
+                display.update(force=True)
+            except Exception as e:
+                logging.exception("Discord: error while sending message")
+                logging.debug(e)

discord.yml

@@ -0,0 +1,4 @@
+discord:
+  enabled: false # requires discord.py module to work (pip3 install discord.py)
+  webhook_url: ~
+  username: 'pwnagotchi' # defaults to hostname if not set