pnut-matrix/pnut-matrix-bot.py

119 lines
3.8 KiB
Python
Raw Normal View History

2018-05-09 04:52:53 +00:00
import logging
import yaml
import time
import pnutpy
from matrix_bot_api.matrix_bot_api import MatrixBotAPI
from matrix_bot_api.mregex_handler import MRegexHandler
from matrix_bot_api.mcommand_handler import MCommandHandler
2019-01-04 03:49:38 +00:00
from models import Avatars, Rooms, Events, Users
from database import db_session
from sqlalchemy import and_
2018-05-09 04:52:53 +00:00
def help_cb(room, event):
reply = "This is an admin room for controlling your connection to pnut.io\n"
reply += "The following commands are available.\n\n"
reply += "!auth - Authorize your account on pnut.io\n"
reply += "!save <token> - Save your pnut.io auth token\n"
reply += "!drop - Drop your pnut.io auth token\n"
reply += "!status - Status of your pnut.io auth token\n"
room.send_notice(reply)
def auth_cb(room, event):
reply = "Visit the following URL to authorize your account on pnut.io.\n\n"
reply += "https://pnut.io/oauth/authenticate"
reply += "?client_id=6SeCRCpCZkmZOKFLFGWbcdAeq2fX1M5t"
reply += "&redirect_uri=urn:ietf:wg:oauth:2.0:oob"
reply += "&scope=write_post,presence,messages&response_type=token\n\n"
reply += "You will be provided with a token that you store with the !save command.\n"
room.send_notice(reply)
def save_cb(room, event):
args = event['content']['body'].split(' ', maxsplit=1)
if len(args) < 2:
reply = "You must provide a token with this command.\n"
reply += "!save <token>"
room.send_notice(reply)
return
pnutpy.api.add_authorization_token(args[1])
try:
response, meta = pnutpy.api.get_user('me')
2019-01-04 03:49:38 +00:00
user = Users(
matrix_id=event['sender'],
pnut_user_id=response.id,
pnut_user_token=args[1]
)
db_session.add(user)
db_session.commit()
2018-05-09 04:52:53 +00:00
reply = "Success! You are now authorized as " + response['username']
except pnutpy.errors.PnutAuthAPIException as e:
reply = "Error! Unable to authorize your account."
except Exception as e:
logging.exception('!save')
reply = "Error! Problem saving your token."
room.send_notice(reply)
def drop_cb(room, event):
try:
user = Users.query.filter(Users.matrix_id == event['sender']).one_or_none()
2019-01-04 03:49:38 +00:00
if user is not None:
db_session.delete(user)
db_session.commit()
reply = "Success! Your auth token has been removed."
else:
reply = "You do not appear to be registered."
2018-05-09 04:52:53 +00:00
except Exception as e:
logging.exception('!drop')
reply = "Error! Problem removing your token."
room.send_notice(reply)
def status_cb(room, event):
try:
user = Users.query.filter(Users.matrix_id == event['sender']).one_or_none()
2018-05-09 04:52:53 +00:00
if user is None:
reply = "You are currently not authorized on pnut.io"
else:
2019-01-04 03:49:38 +00:00
pnutpy.api.add_authorization_token(user.pnut_user_token)
2018-05-09 04:52:53 +00:00
response, meta = pnutpy.api.get_user('me')
2019-01-04 03:49:38 +00:00
reply = "You are currently authorized as " + response.username
2018-05-09 04:52:53 +00:00
except pnutpy.errors.PnutAuthAPIException as e:
reply = "You are currently not authorized on pnut.io"
except Exception as e:
logging.exception('!status')
reply = "Error! There was a problem checking your account."
room.send_notice(reply)
if __name__ == "__main__":
logging.basicConfig(level=logging.INFO)
2018-05-09 04:52:53 +00:00
with open("config.yaml", "rb") as config_file:
config = yaml.load(config_file)
bot = MatrixBotAPI(config['TBOT_USER'], config['TBOT_PASS'], config['MATRIX_HOST'])
bot.add_handler(MCommandHandler("help", help_cb))
bot.add_handler(MCommandHandler("auth", auth_cb))
bot.add_handler(MCommandHandler("save", save_cb))
bot.add_handler(MCommandHandler("drop", drop_cb))
bot.add_handler(MCommandHandler("status", status_cb))
bot.start_polling()
while True:
time.sleep(1)