92 lines
3.9 KiB
Python
92 lines
3.9 KiB
Python
from mastodon import Mastodon
|
|
|
|
from modules.common.module import BotModule
|
|
|
|
|
|
class MatrixModule(BotModule):
|
|
apps = dict() # instance url <-> [app_id, app_secret]
|
|
logins = dict() # mxid <-> [username, accesstoken, instanceurl]
|
|
public = False
|
|
|
|
async def matrix_message(self, bot, room, event):
|
|
args = event.body.split()
|
|
args.pop(0)
|
|
if len(args) >= 1:
|
|
if args[0] == "toot":
|
|
if not self.public:
|
|
bot.must_be_owner(event)
|
|
toot_body = " ".join(args[1:])
|
|
if event.sender in self.logins.keys():
|
|
username = self.logins[event.sender][0]
|
|
accesstoken = self.logins[event.sender][1]
|
|
instanceurl = self.logins[event.sender][2]
|
|
toottodon = Mastodon(
|
|
access_token = accesstoken,
|
|
api_base_url = instanceurl
|
|
)
|
|
tootdict = toottodon.toot(toot_body)
|
|
print(tootdict)
|
|
await bot.send_text(room, tootdict['url'])
|
|
else:
|
|
await bot.send_text(room, f'{event.sender} has not logged in yet with the bot. Please do so.')
|
|
return
|
|
|
|
if len(args) == 4:
|
|
if args[0] == "login":
|
|
if not self.public:
|
|
bot.must_be_owner(event)
|
|
mxid = event.sender
|
|
instanceurl = args[1]
|
|
username = args[2]
|
|
password = args[3]
|
|
if not instanceurl in self.apps.keys():
|
|
app = Mastodon.create_app(f'Hemppa The Bot - {bot.client.user}', api_base_url = instanceurl)
|
|
self.apps[instanceurl] = [app[0], app[1]]
|
|
bot.save_settings()
|
|
await bot.send_text(room, f'Registered Mastodon app on {instanceurl}')
|
|
|
|
mastodon = Mastodon(client_id = self.apps[instanceurl][0], client_secret = self.apps[instanceurl][1], api_base_url = instanceurl)
|
|
access_token = mastodon.log_in(username, password)
|
|
self.logins[mxid] = [username, access_token, instanceurl]
|
|
bot.save_settings()
|
|
await bot.send_text(room, f'Logged into {instanceurl} as {username}')
|
|
return
|
|
if len(args) == 1:
|
|
if args[0] == "status":
|
|
await bot.send_text(room, f'{len(self.logins)} users logged in, app registered on {len(self.apps)} instances, public use enabled: {self.public}')
|
|
if args[0] == "logout":
|
|
if event.sender in self.logins.keys():
|
|
# TODO: Is there a way to invalidate the access token with API?
|
|
del self.logins[event.sender]
|
|
bot.save_settings()
|
|
await bot.send_text(room, f'{event.sender} login data removed from the bot.')
|
|
if args[0] == "setpublic":
|
|
bot.must_be_owner(event)
|
|
self.public = True
|
|
bot.save_settings()
|
|
await bot.send_text(room, f'Mastodon usage is now public use')
|
|
if args[0] == "setprivate":
|
|
bot.must_be_owner(event)
|
|
self.public = False
|
|
bot.save_settings()
|
|
await bot.send_text(room, f'Mastodon usage is now restricted to bot owners')
|
|
|
|
def get_settings(self):
|
|
data = super().get_settings()
|
|
data['apps'] = self.apps
|
|
data['logins'] = self.logins
|
|
data['public'] = self.public
|
|
return data
|
|
|
|
def set_settings(self, data):
|
|
super().set_settings(data)
|
|
if data.get("apps"):
|
|
self.apps = data["apps"]
|
|
if data.get("logins"):
|
|
self.logins = data["logins"]
|
|
if data.get("public"):
|
|
self.public = data["public"]
|
|
|
|
def help(self):
|
|
return ('Mastodon')
|