82 lines
2.9 KiB
Python
82 lines
2.9 KiB
Python
|
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)
|