data-expunged/bot.py
2022-10-29 15:53:29 -04:00

90 lines
3.9 KiB
Python

import discord
import io
import re
import sys
SPOILER_REGEX = re.compile(r'spoil|\bcw\b', re.IGNORECASE)
intents = discord.Intents.default()
intents.message_content = True
client = discord.Client(intents=intents)
def mentions_me(message):
return client.user in message.mentions or [client.user] in [role.members for role in message.role_mentions]
@client.event
async def on_ready():
print('logged in as {0.user}'.format(client), file=sys.stderr)
@client.event
async def on_message(message):
if message.author == client.user:
return
if ((SPOILER_REGEX.search(message.content) or mentions_me(message))
and message.attachments
and not any(a.is_spoiler() for a in message.attachments)):
pluralkit = False
if message.webhook_id is not None:
try:
webhook = await client.fetch_webhook(message.webhook_id)
except discord.Forbidden as e:
print('error fetching message webhook: {0}'.format(e), file=sys.stderr)
await message.channel.send("I can't inspect the webhook on this message because I don't have the permissions I need. This is important so I can interact properly with messages sent by PluralKit. Admins, please make sure I have the Manage Webhooks permission in all channels where people want to use me.")
return
except discord.DiscordException as e:
print('error fetching message webhook: {0}'.format(e), file=sys.stderr)
return
if webhook.user is not None and webhook.user.name == "PluralKit":
pluralkit = True
# If this is a message that's being reposted by PluralKit,
# assume we already caught and reposted the original, and
# just delete this one without reposting a spoilered version
if not pluralkit:
attachments = []
try:
for a in message.attachments:
f = await a.to_file()
f.filename = 'SPOILER_' + f.filename
attachments.append(f)
except discord.DiscordException as e:
print('error reading attachments: {0}'.format(e), file=sys.stderr)
return
try:
spoilered_message = await message.channel.send("{0} says: {1}".format(message.author.mention, message.content), files=attachments)
try:
await spoilered_message.add_reaction('')
except discord.DiscordException as e:
print('error reacting to message: {0}'.format(e))
except discord.DiscordException as e:
print('error sending message: {0}'.format(e), file=sys.stderr)
return
try:
await message.delete()
except discord.Forbidden as e:
print('error deleting message: {0}'.format(e))
await message.channel.send("I can't delete the original unspoiled message because I don't have the permissions I need. Admins, please make sure I have the Manage Messages permission in all channels where people want to use me.")
return
except discord.DiscordException as e:
print('error deleting message: {0}'.format(e))
return
elif mentions_me(message):
await message.channel.send("upload an image and put 'spoil' or 'cw' (doesn't have to be lowercase) anywhere in the description, and I'll spoiler it for you. to delete your post, add a :x: react to it.")
@client.event
async def on_reaction_add(reaction, user):
if (reaction.message.author == client.user
and str(reaction.emoji) == ''
and reaction.message.content.startswith('{0} says: '.format(user.mention))):
try:
await reaction.message.delete()
except discord.DiscordException as e:
print('error deleting my own message: {0}'.format(e))
with open('token') as f:
token = f.read().strip()
client.run(token)