5 changed files with 379 additions and 97 deletions
|
After Width: | Height: | Size: 102 KiB |
|
Before Width: | Height: | Size: 2.3 KiB |
@ -0,0 +1,132 @@ |
|||
import logging |
|||
|
|||
import argparse |
|||
from sleekxmpp import ClientXMPP |
|||
from sleekxmpp.exceptions import IqError, IqTimeout |
|||
import os, subprocess |
|||
import urllib.request |
|||
import datetime |
|||
import ssl |
|||
from PIL import Image |
|||
from pprint import pprint |
|||
|
|||
# python3 streambot.py -j streambot@vvvvvvaria.org -p ******* -m test@muc.vvvvvvaria.org -o ./stream/ |
|||
|
|||
# rewrite script to work with slixmpp >>> https://slixmpp.readthedocs.io/getting_started/muc.html |
|||
|
|||
parser = argparse.ArgumentParser() |
|||
parser.add_argument("-j", "--jid", help="jabber identifier", type=str, required=True) |
|||
parser.add_argument("-p", "--password", help="password", type=str, required=True) |
|||
parser.add_argument("-m", "--muc", help="destination muc", type=str, required=True) |
|||
parser.add_argument("-n", "--nick", help="nickname of the bot", default="streambot", type=str) |
|||
parser.add_argument("-o", "--output", help="output folder", default="./stream/", type=str) |
|||
args = parser.parse_args() |
|||
|
|||
# This was a useful tutorial: http://sleekxmpp.com/getting_started/echobot.html#echobot |
|||
|
|||
class ArchivistBot(ClientXMPP): |
|||
|
|||
def __init__(self, jid, password, room, nick, output): |
|||
ClientXMPP.__init__(self, jid, password) |
|||
|
|||
self.datadir = output |
|||
self.room = room |
|||
self.nick = nick |
|||
|
|||
self.add_event_handler("session_start", self.session_start) |
|||
self.add_event_handler("message", self.archive_msg) # by using 'message' instead of 'groupchat_message' every message received can be archived (also personal msgs) |
|||
|
|||
# self.add_event_handler("groupchat_message", self.archive_msg2) |
|||
|
|||
self.register_plugin('xep_0045') |
|||
self.register_plugin('xep_0030') |
|||
self.register_plugin('xep_0084') |
|||
self.register_plugin('xep_0096') # Transfer files.. |
|||
self.register_plugin('xep_0066') # Transfer files.. |
|||
|
|||
def session_start(self, event): |
|||
roster = self.get_roster() |
|||
print('\nRoster:', roster) |
|||
|
|||
self.send_presence() |
|||
|
|||
self.plugin['xep_0045'].joinMUC(self.room, self.nick) # MUC plugin |
|||
|
|||
with open('avatar.png', 'rb') as avatar_file: |
|||
avatar = avatar_file.read() |
|||
avatar_id = self['xep_0084'].generate_id(avatar) |
|||
info = { |
|||
'id': avatar_id, |
|||
'type': 'image/png', |
|||
'bytes': len(avatar) |
|||
} |
|||
self['xep_0084'].publish_avatar(avatar) |
|||
self['xep_0084'].publish_avatar_metadata(items=[info]) |
|||
|
|||
# XEP-0153: vCard-Based Avatars |
|||
# Not working ATM |
|||
|
|||
self['xep_0153'].set_avatar(avatar=avatar, mtype='image/png') |
|||
|
|||
def archive_msg(self, msg): |
|||
print('\nMessage:{}'.format(msg)) |
|||
print('\nMessage TYPE:{}'.format(msg['type'])) |
|||
print('\nMessage OOB:{}'.format(msg['oob'])) |
|||
print('\nMessage OOB URL:{}'.format(msg['oob']['url'])) |
|||
print('\nMessage MUCK NICK:{}'.format(msg['mucnick'])) |
|||
|
|||
# Always check that a message is not the bot itself, otherwise you will create an infinite loop responding to your own messages. |
|||
if msg['mucnick'] != self.nick: |
|||
|
|||
if msg['type'] in ('groupchat', 'normal'): |
|||
|
|||
# only works if messages are not encrypted |
|||
if len(msg['oob']['url']) > 0: |
|||
|
|||
# logging.getLogger().debug("received OOB from %s with %s" % (self.nick, msg['oob']['url'])) |
|||
|
|||
filename = os.path.basename(msg['oob']['url']) |
|||
targetDir = self.datadir |
|||
if not os.path.exists(targetDir): |
|||
os.mkdir(targetDir, '0755') |
|||
|
|||
targetFile = os.path.join(targetDir, filename) |
|||
print('\n targetFile', targetFile) |
|||
|
|||
#needed to disable certificate validation: |
|||
ctx = ssl.create_default_context() |
|||
ctx.check_hostname = False |
|||
ctx.verify_mode = ssl.CERT_NONE |
|||
|
|||
# save image to file |
|||
# urllib.request.urlretrieve(msg['oob']['url'], targetFile, context=ctx) |
|||
# with urllib.request.urlopen(msg['oob']['url'], context=ctx) as u, \ |
|||
# open(targetFile, 'w+') as f: |
|||
# f.write(u.read()) |
|||
# f.close() |
|||
|
|||
# Download the file from `url` and save it locally under `file_name`: |
|||
with urllib.request.urlopen(msg['oob']['url']) as response: |
|||
data = response.read() # a `bytes` object |
|||
f = open(targetFile, 'wb') |
|||
f.write(data) |
|||
logging.getLogger().debug("saved to %s" % targetFile) |
|||
|
|||
# subprocess.call(["make","publish"], cwd="/srv/gitea/clones/varia.website/") |
|||
|
|||
msg.reply("Thanks for sending! I will store that for you." % msg).send() # works! also for omemo messages, but it cannot decrypt them. |
|||
|
|||
|
|||
|
|||
if __name__ == '__main__': |
|||
# Ideally use optparse or argparse to get JID, |
|||
# password, and log level. |
|||
|
|||
logging.basicConfig(level=logging.DEBUG, format='%(levelname)-8s %(message)s') |
|||
|
|||
client = ArchivistBot(args.jid, args.password, args.muc, args.nick, args.output) |
|||
|
|||
if client.connect(): |
|||
client.process(block=True) |
|||
else: |
|||
logging.getLogger().error("Can't connect.") |
|||
Loading…
Reference in new issue