Experimental Discord bot written in Python
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761
  1. """
  2. Cog for detecting large numbers of guild joins in a short period of time.
  3. """
  4. import weakref
  5. from collections.abc import Sequence
  6. from datetime import datetime, timedelta
  7. from discord import AuditLogAction, AuditLogEntry, Emoji, Guild, GuildSticker, Invite, Member, Message, RawBulkMessageDeleteEvent, RawMessageDeleteEvent, RawMessageUpdateEvent, Role, Thread, User
  8. from discord.abc import GuildChannel
  9. from discord.ext import commands
  10. from discord.utils import escape_markdown
  11. from typing import List, Optional, Tuple, Union
  12. import difflib
  13. import traceback
  14. from config import CONFIG
  15. from rocketbot.cogs.basecog import BaseCog, BotMessage, BotMessageReaction, CogSetting
  16. from rocketbot.collections import AgeBoundList
  17. from rocketbot.storage import Storage
  18. class LoggingCog(BaseCog, name='Logging'):
  19. """
  20. Cog for logging notable events to a designated logging channel.
  21. """
  22. SETTING_ENABLED = CogSetting('enabled', bool,
  23. brief='logging',
  24. description='Whether this cog is enabled for a guild.')
  25. def __init__(self, bot):
  26. super().__init__(bot)
  27. self.add_setting(LoggingCog.SETTING_ENABLED)
  28. @commands.group(
  29. brief='Manages event logging',
  30. )
  31. @commands.has_permissions(ban_members=True)
  32. @commands.guild_only()
  33. async def logging(self, context: commands.Context):
  34. 'Logging command group'
  35. if context.invoked_subcommand is None:
  36. await context.send_help()
  37. # Events - Channels
  38. @commands.Cog.listener()
  39. async def on_guild_channel_delete(self, channel: GuildChannel) -> None:
  40. """
  41. Called whenever a guild channel is deleted or created.
  42. Note that you can get the guild from guild.
  43. https://discordpy.readthedocs.io/en/stable/api.html#discord.on_guild_channel_delete
  44. """
  45. guild = channel.guild
  46. if not self.get_guild_setting(guild, self.SETTING_ENABLED):
  47. return
  48. text = f'Channel **{channel.name}** deleted.'
  49. bot_message = BotMessage(guild, text, BotMessage.TYPE_LOG)
  50. await bot_message.update()
  51. @commands.Cog.listener()
  52. async def on_guild_channel_create(self, channel: GuildChannel) -> None:
  53. """
  54. Called whenever a guild channel is deleted or created.
  55. Note that you can get the guild from guild.
  56. https://discordpy.readthedocs.io/en/stable/api.html#discord.on_guild_channel_create
  57. """
  58. guild = channel.guild
  59. if not self.get_guild_setting(guild, self.SETTING_ENABLED):
  60. return
  61. text = f'Channel **{channel.name}** created. {channel.mention}'
  62. bot_message = BotMessage(guild, text, BotMessage.TYPE_LOG)
  63. await bot_message.update()
  64. @commands.Cog.listener()
  65. async def on_guild_channel_update(self, before: GuildChannel, after: GuildChannel) -> None:
  66. """
  67. Called whenever a guild channel is updated. e.g. changed name, topic,
  68. permissions.
  69. https://discordpy.readthedocs.io/en/stable/api.html#discord.on_guild_channel_update
  70. """
  71. guild = after.guild
  72. if not self.get_guild_setting(guild, self.SETTING_ENABLED):
  73. return
  74. changes = []
  75. if after.name != before.name:
  76. changes.append(f'Name: `{before.name}` -> `{after.name}`')
  77. if after.category != before.category:
  78. changes.append(f'Category: {before.category.name if before.category else None}' + \
  79. f' -> {after.category.name if after.category else None}')
  80. if after.changed_roles != before.changed_roles:
  81. changes.append('Roles changed')
  82. if after.overwrites != before.overwrites:
  83. changes.append('Permission overwrites changed')
  84. if after.position != before.position:
  85. changes.append(f'Position: {before.position} -> {after.position}')
  86. if len(changes) == 0:
  87. return
  88. text = f'Channel **{before.name}** updated. Changes:\n'
  89. text += '* ' + '\n* '.join(changes)
  90. bot_message = BotMessage(guild, text, BotMessage.TYPE_LOG)
  91. await bot_message.update()
  92. # Events - Guilds
  93. @commands.Cog.listener()
  94. async def on_guild_available(self, guild: Guild) -> None:
  95. pass
  96. @commands.Cog.listener()
  97. async def on_guild_unavailable(self, guild: Guild) -> None:
  98. pass
  99. @commands.Cog.listener()
  100. async def on_guild_update(self, before: Guild, after: Guild) -> None:
  101. pass
  102. @commands.Cog.listener()
  103. async def on_guild_emojis_update(self, guild: Guild, before: Sequence[Emoji], after: Sequence[Emoji]) -> None:
  104. pass
  105. @commands.Cog.listener()
  106. async def on_guild_stickers_update(self, guild: Guild, before: Sequence[GuildSticker], after: Sequence[GuildSticker]) -> None:
  107. pass
  108. @commands.Cog.listener()
  109. async def on_invite_create(self, invite: Invite) -> None:
  110. """
  111. Called when an Invite is created. You must have manage_channels to receive this.
  112. https://discordpy.readthedocs.io/en/stable/api.html#discord.on_invite_create
  113. """
  114. guild = invite.guild
  115. if not self.get_guild_setting(guild, self.SETTING_ENABLED):
  116. return
  117. text = f'Invite code `{invite.code}` created by {self.__describe_user(invite.inviter)}. '
  118. if invite.max_age == 0:
  119. text += "Doesn't expire."
  120. else:
  121. text += f'Expires in {invite.max_age} seconds.'
  122. bot_message = BotMessage(guild, text, BotMessage.TYPE_LOG)
  123. await bot_message.update()
  124. @commands.Cog.listener()
  125. async def on_invite_delete(self, invite: Invite) -> None:
  126. """
  127. Called when an Invite is deleted. You must have manage_channels to receive this.
  128. https://discordpy.readthedocs.io/en/stable/api.html#discord.on_invite_delete
  129. """
  130. guild = invite.guild
  131. if not self.get_guild_setting(guild, self.SETTING_ENABLED):
  132. return
  133. if invite.inviter:
  134. text = f'Invite code `{invite.code}` deleted. Originally created by {self.__describe_user(invite.inviter)}.'
  135. else:
  136. text = f'Invite code `{invite.code}` deleted.'
  137. bot_message = BotMessage(guild, text, BotMessage.TYPE_LOG)
  138. await bot_message.update()
  139. # Events - Members
  140. @commands.Cog.listener()
  141. async def on_member_join(self, member: Member) -> None:
  142. """
  143. Called when a Member joins a Guild.
  144. https://discordpy.readthedocs.io/en/stable/api.html#discord.on_member_join
  145. """
  146. guild = member.guild
  147. if not self.get_guild_setting(guild, self.SETTING_ENABLED):
  148. return
  149. text = f'Member joined server: {self.__describe_user(member)}.'
  150. flags = []
  151. noteworthy = False
  152. if member.flags.did_rejoin:
  153. flags.append('Rejoined this server')
  154. if member.public_flags.active_developer:
  155. flags.append('Is an active developer')
  156. if member.public_flags.hypesquad:
  157. flags.append('Is a HypeSquad Events member')
  158. if member.public_flags.hypesquad_bravery:
  159. flags.append('Is a HypeSquad Bravery member')
  160. if member.public_flags.hypesquad_brilliance:
  161. flags.append('Is a HypeSquad Brilliance member')
  162. if member.public_flags.hypesquad_balance:
  163. flags.append('Is a HypeSquad Balance member')
  164. if member.public_flags.early_supporter:
  165. flags.append('Is an early supporter')
  166. if member.public_flags.spammer:
  167. flags.append('**Is flagged as a spammer**')
  168. noteworthy = True
  169. if member.public_flags.discord_certified_moderator:
  170. flags.append('**Is a Discord Certified Moderator**')
  171. noteworthy = True
  172. if member.public_flags.early_verified_bot_developer:
  173. flags.append('**Is a verified bot developer**')
  174. noteworthy = True
  175. if member.public_flags.verified_bot:
  176. flags.append('**Is a verified bot**')
  177. noteworthy = True
  178. if member.public_flags.bug_hunter or member.public_flags.bug_hunter_level_2:
  179. flags.append('**Is a bug hunter**')
  180. noteworthy = True
  181. if member.public_flags.system:
  182. flags.append('**Is a Discord system user**')
  183. noteworthy = True
  184. if member.public_flags.staff:
  185. flags.append('**Is Discord staff**')
  186. noteworthy = True
  187. if member.public_flags.partner:
  188. flags.append('**Is a Discord partner**')
  189. noteworthy = True
  190. if len(flags) > 0:
  191. text += '\n* ' + '\n* '.join(flags)
  192. if noteworthy:
  193. text += f'\n\nLink: {member.mention}'
  194. bot_message = BotMessage(guild, text, BotMessage.TYPE_MOD_WARNING)
  195. else:
  196. bot_message = BotMessage(guild, text, BotMessage.TYPE_LOG)
  197. await bot_message.update()
  198. @commands.Cog.listener()
  199. async def on_member_remove(self, member: Member) -> None:
  200. """
  201. Called when a Member leaves a Guild.
  202. If the guild or member could not be found in the internal cache this event
  203. will not be called, you may use on_raw_member_remove() instead.
  204. https://discordpy.readthedocs.io/en/stable/api.html#discord.on_member_remove
  205. """
  206. guild = member.guild
  207. if not self.get_guild_setting(guild, self.SETTING_ENABLED):
  208. return
  209. is_kick = False
  210. kicker = None
  211. kick_reason = None
  212. entry = await self.__find_audit_entry(member, AuditLogAction.kick)
  213. if entry:
  214. is_kick = True
  215. kicker = entry.user
  216. kick_reason = entry.reason
  217. if is_kick:
  218. if kicker and kicker != member:
  219. text = f'Member kicked from the server: {self.__describe_user(member)} by **{kicker.name}**'
  220. else:
  221. text = f'Member kicked from the server: {self.__describe_user(member)}'
  222. else:
  223. text = f'Member left server: {self.__describe_user(member)}'
  224. if kick_reason:
  225. text += f'\nReason: "{kick_reason}"'
  226. bot_message = BotMessage(guild, text, BotMessage.TYPE_LOG)
  227. await bot_message.update()
  228. @commands.Cog.listener()
  229. async def on_member_update(self, before: Member, after: Member) -> None:
  230. """
  231. Called when a Member updates their profile.
  232. This is called when one or more of the following things change:
  233. * nickname
  234. * roles
  235. * pending
  236. * timeout
  237. * guild avatar
  238. * flags
  239. https://discordpy.readthedocs.io/en/stable/api.html#discord.on_member_update
  240. """
  241. guild = after.guild
  242. if not self.get_guild_setting(guild, self.SETTING_ENABLED):
  243. return
  244. changes = []
  245. if after.nick != before.nick:
  246. changes.append(f'Nick: `{before.nick}` -> `{after.nick}`')
  247. if after.roles != before.roles:
  248. added_role_names = []
  249. removed_role_names = []
  250. for role in before.roles:
  251. if role not in after.roles:
  252. removed_role_names.append(role.name)
  253. for role in after.roles:
  254. if role not in before.roles:
  255. added_role_names.append(role.name)
  256. if len(removed_role_names) > 0:
  257. changes.append(f'Removed roles: ~~**{"**~~, ~~**".join(removed_role_names)}**~~')
  258. if len(added_role_names) > 0:
  259. changes.append(f'Added roles: **{"**, **".join(added_role_names)}**')
  260. if after.pending != before.pending:
  261. pass # not that interesting and probably noisy
  262. if after.timed_out_until != before.timed_out_until:
  263. if after.timed_out_until:
  264. delta = after.timed_out_until - datetime.now()
  265. changes.append(f'Timed out for `{delta}`')
  266. elif before.timed_out_until:
  267. changes.append('Timeout cleared')
  268. before_guild_avatar = before.guild_avatar.url if before.guild_avatar else None
  269. after_guild_avatar = after.guild_avatar.url if after.guild_avatar else None
  270. if after_guild_avatar != before_guild_avatar:
  271. changes.append(f'Guild avatar: <{before_guild_avatar}> -> <{after_guild_avatar}>')
  272. if after.flags != before.flags:
  273. flag_changes = []
  274. for (name, after_value) in after.iter():
  275. before_value = getattr(before, name)
  276. if after_value != before_value:
  277. flag_changes.append(f'`{name}` = `{before_value}` -> `{after_value}`')
  278. if len(flag_changes) > 0:
  279. changes.append(f'Flag changes: {", ".join(flag_changes)}')
  280. if len(changes) == 0:
  281. return
  282. text = f'Details for member {self.__describe_user(before)} changed:\n'
  283. text += '* ' + '\n* '.join(changes)
  284. bot_message = BotMessage(guild, text, BotMessage.TYPE_LOG)
  285. await bot_message.update()
  286. @commands.Cog.listener()
  287. async def on_user_update(self, before: User, after: User) -> None:
  288. """
  289. Called when a User updates their profile.
  290. This is called when one or more of the following things change:
  291. * avatar
  292. * username
  293. * discriminator
  294. https://discordpy.readthedocs.io/en/stable/api.html#discord.on_user_update
  295. """
  296. if hasattr(after, 'guild'):
  297. guild = after.guild
  298. else:
  299. return
  300. if not self.get_guild_setting(guild, self.SETTING_ENABLED):
  301. return
  302. changes = []
  303. before_avatar_url = before.avatar.url if before.avatar else None
  304. after_avatar_url = after.avatar.url if after.avatar else None
  305. if after_avatar_url != before_avatar_url:
  306. changes.append(f'Avatar URL: <{before_avatar_url}> -> <{after_avatar_url}>')
  307. if after.name != before.name:
  308. changes.append(f'Username: `{before.name}` -> `{after.name}`')
  309. if len(changes) == 0:
  310. return
  311. text = f'Details for user {self.__describe_user(before)} changed:\n'
  312. text += '* ' + '\n* '.join(changes)
  313. bot_message = BotMessage(guild, text, BotMessage.TYPE_LOG)
  314. await bot_message.update()
  315. @commands.Cog.listener()
  316. async def on_member_ban(self, guild: Guild, user: Union[User, Member]) -> None:
  317. """
  318. Called when user gets banned from a Guild.
  319. https://discordpy.readthedocs.io/en/stable/api.html#discord.on_member_ban
  320. """
  321. if not self.get_guild_setting(guild, self.SETTING_ENABLED):
  322. return
  323. banner = None
  324. ban_reason = None
  325. entry = await self.__find_audit_entry(user, AuditLogAction.ban)
  326. if entry:
  327. banner = entry.user
  328. ban_reason = entry.reason
  329. if banner:
  330. text = f'Member {self.__describe_user(user)} banned by **{banner.name}**.'
  331. if ban_reason:
  332. text += f'\nReason: "{ban_reason}"'
  333. else:
  334. text = f'Member {self.__describe_user(user)} banned.'
  335. bot_message = BotMessage(guild, text, BotMessage.TYPE_LOG)
  336. await bot_message.update()
  337. async def __find_audit_entry(self, user: Union[User, Member], action: AuditLogAction, max_age: int = 10) -> Optional[AuditLogEntry]:
  338. """
  339. Searches the audit log for the most recent entry of a given type for a
  340. given user. Intended for finding the relevant entry for a ban/kick that
  341. just occurred.
  342. """
  343. if hasattr(user, 'guild') and user.guild:
  344. guild = user.guild
  345. else:
  346. return None
  347. now = datetime.now()
  348. async for entry in guild.audit_logs():
  349. age_seconds = now.timestamp() - entry.created_at.timestamp()
  350. if entry.action == action and entry.target == user and age_seconds <= max_age:
  351. return entry
  352. return None
  353. @commands.Cog.listener()
  354. async def on_member_unban(self, guild: Guild, user: Union[User, Member]) -> None:
  355. """
  356. Called when a User gets unbanned from a Guild.
  357. https://discordpy.readthedocs.io/en/stable/api.html#discord.on_member_unban
  358. """
  359. if not self.get_guild_setting(guild, self.SETTING_ENABLED):
  360. return
  361. text = f'Member {self.__describe_user(user)} unbanned'
  362. bot_message = BotMessage(guild, text, BotMessage.TYPE_LOG)
  363. await bot_message.update()
  364. # Events - Messages
  365. @commands.Cog.listener()
  366. async def on_message(self, message: Message) -> None:
  367. """
  368. Called when a Message is created and sent.
  369. https://discordpy.readthedocs.io/en/stable/api.html#discord.on_message
  370. """
  371. # print(f"Saw message {message.id} \"{message.content}\"")
  372. @commands.Cog.listener()
  373. async def on_message_edit(self, before: Message, after: Message) -> None:
  374. """
  375. Called when a Message receives an update event. If the message is not
  376. found in the internal message cache, then these events will not be
  377. called. Messages might not be in cache if the message is too old or the
  378. client is participating in high traffic guilds.
  379. If this occurs increase the max_messages parameter or use the
  380. on_raw_message_edit() event instead.
  381. The following non-exhaustive cases trigger this event:
  382. * A message has been pinned or unpinned.
  383. * The message content has been changed.
  384. * The message has received an embed.
  385. * For performance reasons, the embed server does not do this in a
  386. “consistent” manner.
  387. * The message’s embeds were suppressed or unsuppressed.
  388. * A call message has received an update to its participants or ending time.
  389. https://discordpy.readthedocs.io/en/stable/api.html#discord.on_message_edit
  390. """
  391. guild = after.guild
  392. if not self.get_guild_setting(guild, self.SETTING_ENABLED):
  393. return
  394. if after.author.id == self.bot.user.id:
  395. return
  396. content_changed = (after.content != before.content)
  397. attachments_changed = (after.attachments != before.attachments)
  398. embeds_changed = (after.embeds != before.embeds)
  399. embeds_add_only = len(before.embeds or []) == 0 and len(after.embeds or []) > 0
  400. if not content_changed and not attachments_changed and (not embeds_changed or embeds_add_only):
  401. # Most likely an embed being asynchronously populated by server
  402. return
  403. if content_changed:
  404. (before_markdown, after_markdown) = self.__diff(self.__quote_markdown(before.content), \
  405. self.__quote_markdown(after.content))
  406. else:
  407. before_markdown = self.__quote_markdown(before.content)
  408. after_markdown = before_markdown if len(before.content.strip()) == 0 else '> _<content unchanged>_'
  409. if attachments_changed:
  410. if len(before.attachments or []) > 0:
  411. for attachment in before.attachments:
  412. before_markdown += f'\n> * 📎 {attachment.url}'
  413. if attachment not in after.attachments or []:
  414. before_markdown += ' (removed)'
  415. else:
  416. before_markdown += '\n> * _<no attachments>_'
  417. if len(after.attachments or []) > 0:
  418. for attachment in after.attachments:
  419. after_markdown += f'\n> * 📎 {attachment.url}'
  420. if attachment not in before.attachments or []:
  421. after_markdown += ' (added)'
  422. else:
  423. after_markdown += '\n> * _<no attachments>_'
  424. if embeds_changed:
  425. if len(before.embeds or []) > 0:
  426. for embed in before.embeds:
  427. before_markdown += f'\n> * 🔗 {embed.url}'
  428. if embed not in after.embeds or []:
  429. before_markdown += ' (removed)'
  430. else:
  431. before_markdown += '\n> * _<no embeds>_'
  432. if len(after.embeds or []) > 0:
  433. for embed in after.embeds:
  434. after_markdown += f'\n> * 🔗 {embed.url}'
  435. if embed not in before.embeds or []:
  436. after_markdown += ' (added)'
  437. else:
  438. after_markdown += '\n> * _<no embeds>_'
  439. text = f'Message {after.jump_url} edited by {self.__describe_user(after.author)}.\n' + \
  440. f'Original:\n{before_markdown}\n' + \
  441. f'Updated:\n{after_markdown}'
  442. bot_message = BotMessage(guild, text, BotMessage.TYPE_LOG, suppress_embeds=True)
  443. await bot_message.update()
  444. @commands.Cog.listener()
  445. async def on_raw_message_edit(self, payload: RawMessageUpdateEvent) -> None:
  446. """
  447. Called when a message is edited. Unlike on_message_edit(), this is called
  448. regardless of the state of the internal message cache.
  449. If the message is found in the message cache, it can be accessed via
  450. RawMessageUpdateEvent.cached_message. The cached message represents the
  451. message before it has been edited. For example, if the content of a
  452. message is modified and triggers the on_raw_message_edit() coroutine,
  453. the RawMessageUpdateEvent.cached_message will return a Message object
  454. that represents the message before the content was modified.
  455. Due to the inherently raw nature of this event, the data parameter
  456. coincides with the raw data given by the gateway.
  457. Since the data payload can be partial, care must be taken when accessing
  458. stuff in the dictionary. One example of a common case of partial data is
  459. when the 'content' key is inaccessible. This denotes an “embed” only
  460. edit, which is an edit in which only the embeds are updated by the
  461. Discord embed server.
  462. https://discordpy.readthedocs.io/en/stable/api.html#discord.on_raw_message_edit
  463. """
  464. if payload.cached_message:
  465. return # already handled by on_message_edit
  466. guild = self.bot.get_guild(payload.guild_id) or await self.bot.fetch_guild(payload.guild_id)
  467. if not guild:
  468. return
  469. if not self.get_guild_setting(guild, self.SETTING_ENABLED):
  470. return
  471. channel = guild.get_channel(payload.channel_id) or await guild.fetch_channel(payload.channel_id)
  472. if not channel:
  473. return
  474. message = await channel.fetch_message(payload.message_id)
  475. if not message:
  476. return
  477. text = f'Message {message.jump_url} edited by {self.__describe_user(message.author)}.\n' + \
  478. 'Original markdown unavailable in cache.\n' + \
  479. f'Updated markdown:\n{self.__quote_markdown(message.content)}'
  480. bot_message = BotMessage(guild, text, BotMessage.TYPE_LOG, suppress_embeds=True)
  481. await bot_message.update()
  482. @commands.Cog.listener()
  483. async def on_raw_message_delete(self, payload: RawMessageDeleteEvent) -> None:
  484. """
  485. Called when a message is deleted. Unlike on_message_delete(), this is
  486. called regardless of the message being in the internal message cache or not.
  487. If the message is found in the message cache, it can be accessed via
  488. RawMessageDeleteEvent.cached_message
  489. https://discordpy.readthedocs.io/en/stable/api.html#discord.on_raw_message_delete
  490. """
  491. if payload.cached_message:
  492. message = payload.cached_message
  493. guild = message.guild
  494. if not self.get_guild_setting(guild, self.SETTING_ENABLED):
  495. return
  496. if message.author.id == self.bot.user.id:
  497. return
  498. text = f'Message by {self.__describe_user(message.author)} deleted from {message.channel.mention}. ' + \
  499. f'Markdown:\n{self.__quote_markdown(message.content)}'
  500. for attachment in message.attachments or []:
  501. text += f'\n> * 📎 {attachment.url}'
  502. for embed in message.embeds or []:
  503. text += f'\n> * 🔗 {embed.url}'
  504. bot_message = BotMessage(message.guild, text, BotMessage.TYPE_LOG, suppress_embeds=True)
  505. await bot_message.update()
  506. else:
  507. guild = self.bot.get_guild(payload.guild_id) or await self.bot.fetch_guild(payload.guild_id)
  508. if not guild:
  509. return
  510. if not self.get_guild_setting(guild, self.SETTING_ENABLED):
  511. return
  512. channel = guild.get_channel(payload.channel_id) or await guild.fetch_channel(payload.channel_id)
  513. if not channel:
  514. return
  515. text = f'Message {payload.message_id} deleted in ' + channel.mention + ' but content and author not available in cache.'
  516. bot_message = BotMessage(guild, text, BotMessage.TYPE_LOG, suppress_embeds=True)
  517. await bot_message.update()
  518. @commands.Cog.listener()
  519. async def on_raw_bulk_message_delete(self, payload: RawBulkMessageDeleteEvent) -> None:
  520. """
  521. Called when a bulk delete is triggered. Unlike on_bulk_message_delete(),
  522. this is called regardless of the messages being in the internal message
  523. cache or not.
  524. If the messages are found in the message cache, they can be accessed via
  525. RawBulkMessageDeleteEvent.cached_messages
  526. https://discordpy.readthedocs.io/en/stable/api.html#discord.on_raw_bulk_message_delete
  527. """
  528. guild = self.bot.get_guild(payload.guild_id) or await self.bot.fetch_guild(payload.guild_id)
  529. if not guild:
  530. return
  531. if not self.get_guild_setting(guild, self.SETTING_ENABLED):
  532. return
  533. channel = guild.get_channel(payload.channel_id) or await guild.fetch_channel(payload.channel_id)
  534. count = len(payload.message_ids)
  535. cached_count = len(payload.cached_messages)
  536. uncached_count = count - cached_count
  537. text = f'Bulk deletion of {count} message(s) from {channel.mention}.'
  538. if uncached_count == count:
  539. text += f' No cached content available for any of them.'
  540. elif uncached_count > 0:
  541. text += f' No cached content available for {uncached_count} of them.'
  542. bot_message = BotMessage(guild, text, BotMessage.TYPE_LOG, suppress_embeds=True)
  543. await bot_message.update()
  544. for message in payload.cached_messages:
  545. text = f'Message by {self.__describe_user(message.author)} bulk deleted from {message.channel.mention}. ' + \
  546. f'Markdown:\n{self.__quote_markdown(message.content)}'
  547. for attachment in message.attachments or []:
  548. text += f'\n> * 📎 {attachment.url}'
  549. for embed in message.embeds or []:
  550. text += f'\n> * 🔗 {embed.url}'
  551. bot_message = BotMessage(guild, text, BotMessage.TYPE_LOG, suppress_embeds=True)
  552. await bot_message.update()
  553. # Events - Roles
  554. @commands.Cog.listener()
  555. async def on_guild_role_create(self, role: Role) -> None:
  556. """
  557. Called when a Guild creates or deletes a new Role.
  558. To get the guild it belongs to, use Role.guild.
  559. https://discordpy.readthedocs.io/en/stable/api.html#discord.on_guild_role_create
  560. """
  561. guild = role.guild
  562. if not self.get_guild_setting(guild, self.SETTING_ENABLED):
  563. return
  564. text = f'Role created: **{role.name}**'
  565. bot_message = BotMessage(guild, text, BotMessage.TYPE_LOG)
  566. await bot_message.update()
  567. @commands.Cog.listener()
  568. async def on_guild_role_delete(self, role: Role) -> None:
  569. """
  570. Called when a Guild creates or deletes a new Role.
  571. To get the guild it belongs to, use Role.guild.
  572. https://discordpy.readthedocs.io/en/stable/api.html#discord.on_guild_role_delete
  573. """
  574. guild = role.guild
  575. if not self.get_guild_setting(guild, self.SETTING_ENABLED):
  576. return
  577. text = f'Role removed: **{role.name}**'
  578. bot_message = BotMessage(guild, text, BotMessage.TYPE_LOG)
  579. await bot_message.update()
  580. @commands.Cog.listener()
  581. async def on_guild_role_update(self, before: Role, after: Role) -> None:
  582. """
  583. Called when a Role is changed guild-wide.
  584. https://discordpy.readthedocs.io/en/stable/api.html#discord.on_guild_role_update
  585. """
  586. guild = after.guild
  587. if not self.get_guild_setting(guild, self.SETTING_ENABLED):
  588. return
  589. changes = []
  590. if after.name != before.name:
  591. changes.append(f'Name: `{before.name}` -> `{after.name}`')
  592. if after.hoist != before.hoist:
  593. changes.append(f'Hoisted: `{before.hoist}` -> `{after.hoist}`')
  594. if after.position != before.position:
  595. changes.append(f'Position: `{before.position}` -> `{after.position}`')
  596. if after.unicode_emoji != before.unicode_emoji:
  597. changes.append(f'Emoji: {before.unicode_emoji} -> {after.unicode_emoji}')
  598. if after.mentionable != before.mentionable:
  599. changes.append(f'Mentionable: `{before.mentionable}` -> `{after.mentionable}`')
  600. if after.permissions != before.permissions:
  601. changes.append('Permissions edited')
  602. if after.color != before.color:
  603. changes.append('Color edited')
  604. before_icon_url = before.icon.url if before.icon else None
  605. after_icon_url = after.icon.url if after.icon else None
  606. if after_icon_url != before_icon_url:
  607. changes.append(f'Icon: <{before_icon_url}> -> <{after_icon_url}>')
  608. before_icon_url = before.display_icon.url if before.display_icon else None
  609. after_icon_url = after.display_icon.url if after.display_icon else None
  610. if after_icon_url != before_icon_url:
  611. changes.append(f'Display icon: <{before_icon_url}> -> <{after_icon_url}>')
  612. if len(changes) == 0:
  613. return
  614. text = f'Role **{after.name}** updated. Changes:\n'
  615. text += '* ' + '\n* '.join(changes)
  616. bot_message = BotMessage(guild, text, BotMessage.TYPE_LOG)
  617. await bot_message.update()
  618. # Events - Threads
  619. @commands.Cog.listener()
  620. async def on_thread_create(self, thread: Thread) -> None:
  621. pass
  622. @commands.Cog.listener()
  623. async def on_thread_update(self, before: Thread, after: Thread) -> None:
  624. pass
  625. @commands.Cog.listener()
  626. async def on_thread_delete(self, thread: Thread) -> None:
  627. pass
  628. # ------------------------------------------------------------------------
  629. def __quote_markdown(self, s: str) -> str:
  630. if len(s.strip()) == 0:
  631. return '> _<no content>_'
  632. return '> ' + escape_markdown(s).replace('\n', '\n> ')
  633. def __describe_user(self, user: Union[User, Member]) -> str:
  634. """
  635. Standardized markdown describing a user or member.
  636. """
  637. return f'**{user.name}** ({user.display_name} {user.id})'
  638. def __diff(self, a: str, b: str) -> Tuple[str, str]:
  639. deletion_start = '~~'
  640. deletion_end = '~~'
  641. addition_start = '**'
  642. addition_end = '**'
  643. markdown_a = ''
  644. markdown_b = ''
  645. a_open = False
  646. b_open = False
  647. # FIXME: Handle URLs better. They get mangled.
  648. # URL regex: http[s]?://(?:[a-zA-Z]|[0-9]|[$-_@.&+]|[!*\(\),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+
  649. for i, s in enumerate(difflib.ndiff(a, b)):
  650. operation = s[0]
  651. content = s[2:]
  652. if operation != '-' and a_open:
  653. markdown_a += deletion_end
  654. a_open = False
  655. if operation != '+' and b_open:
  656. markdown_b += addition_end
  657. b_open = False
  658. if operation == ' ':
  659. markdown_a += content
  660. markdown_b += content
  661. elif operation == '-':
  662. if not a_open:
  663. markdown_a += deletion_start
  664. a_open = True
  665. markdown_a += content
  666. elif operation == '+':
  667. if not b_open:
  668. markdown_b += addition_start
  669. b_open = True
  670. markdown_b += content
  671. if a_open:
  672. markdown_a += deletion_end
  673. if b_open:
  674. markdown_b += addition_end
  675. return (markdown_a, markdown_b)