[V3 Nolinks] Initial commit

README.md

@@ -11,6 +11,8 @@ dungeon - New users with new accounts will be shuffled off to a locked channel o
 
 inspirobot - Fetch "inspirational" messages from inspirobot.me with [p]inspireme.
 
+nolinks - A very blunt hammer to remove anything that looks like a link. Roles can be whitelisted and it can watch multiple channels.
+
 otherbot - Alert a role when bot(s) go offline.
 
 partycrash - A port of Will's partycrash command from the v2 Admin cog. This cog will not generate invites, but will show already-existing invites that the bot has access to view.

nolinks/__init__.py

@@ -0,0 +1,5 @@
+from .nolinks import NoLinks
+
+
+def setup(bot):
+    bot.add_cog(NoLinks(bot))

nolinks/info.json

@@ -0,0 +1,17 @@
+{
+	"author": [
+		"aikaterna"
+	],
+	"description": "Removes all links in specified channels, with the ability to whitelist roles.",
+    "install_msg": "Thanks for installing. Use `[p]nolinks` to get started.",
+    "min_python_version": [3, 6, 0],
+    "permissions" : [
+		"manage_messages"
+	],
+	"short": "No links allowed.",
+	"tags": [
+		"links",
+        "automod"
+	],
+	"type": "COG"
+}

nolinks/nolinks.py

@@ -0,0 +1,136 @@
+import discord
+from redbot.core import Config, commands, checks
+from urllib.parse import urlparse
+
+
+class NoLinks:
+    def __init__(self, bot):
+        self.bot = bot
+        self.config = Config.get_conf(self, 2740131001, force_registration=True)
+
+        default_guild = {
+            "report_channel": None,
+            "role": [],
+            "watching": [],
+        }
+
+        self.config.register_guild(**default_guild)
+
+    @commands.group()
+    @checks.mod_or_permissions(administrator=True)
+    @commands.guild_only()
+    async def nolinks(self, ctx):
+        """Configuration options."""
+        pass
+
+    @nolinks.command()
+    async def channel(self, ctx, channel: discord.TextChannel=None):
+        """Set the message transfer channel. Leave the channel blank to turn it off."""
+        if not channel:
+            await self.config.guild(ctx.guild).report_channel.clear()
+            return await ctx.send("Message transfer channel turned off.")
+        await self.config.guild(ctx.guild).report_channel.set(channel.id)
+        await ctx.send(f'Message transfer channel set to: {channel.mention}.')
+
+    @nolinks.command()
+    async def rolelist(self, ctx):
+        """List whitelisted roles."""
+        role_list = await self.config.guild(ctx.guild).role()
+        role_msg = "Whitelisted Roles:\n"
+        for role in role_list:
+            role_obj = discord.utils.get(ctx.guild.roles, id=role)
+            role_msg += f'{role_obj.name}\n'
+        await ctx.send(role_msg)
+
+    @nolinks.command()
+    async def removerole(self, ctx, role_name: discord.Role):
+        """Remove a whitelisted role."""
+        role_list = await self.config.guild(ctx.guild).role()
+        if role_name.id in role_list:
+            role_list.remove(role_name.id)
+        else:
+            return await ctx.send("Role not in whitelist.")
+        await self.config.guild(ctx.guild).role.set(role_list)
+        role_obj = discord.utils.get(ctx.guild.roles, id=role_name.id)
+        await ctx.send(f'{role_obj.na} removed from the link whitelist.')
+
+    @nolinks.command()
+    async def role(self, ctx, role_name: discord.Role):
+        """Add a whitelisted role."""
+        role_list = await self.config.guild(ctx.guild).role()
+        if role_name.id not in role_list:
+            role_list.append(role_name.id)
+        await self.config.guild(ctx.guild).role.set(role_list)
+        role_obj = discord.utils.get(ctx.guild.roles, id=role_name.id)
+        await ctx.send(f'{role_obj.name} appended to the link whitelist.')
+
+    @nolinks.command()
+    async def watch(self, ctx, channel: discord.TextChannel):
+        """Add a channel to watch. Links will be removed in these channels."""
+        channel_list = await self.config.guild(ctx.guild).watching()
+        if channel.id not in channel_list:
+            channel_list.append(channel.id)
+        await self.config.guild(ctx.guild).watching.set(channel_list)
+        await ctx.send(f'{self.bot.get_channel(channel.id).mention} will have links removed.')
+
+    @nolinks.command()
+    async def watchlist(self, ctx):
+        """List the channels being watched."""
+        channel_list = await self.config.guild(ctx.guild).watching()
+        msg = "Links will be removed in:\n"
+        for channel in channel_list:
+            channel_obj = self.bot.get_channel(channel)
+            msg += f'{channel_obj.mention}\n'
+        await ctx.send(msg)
+
+    @nolinks.command()
+    async def unwatch(self, ctx, channel: discord.TextChannel):
+        """Remove a channel from the watch list."""
+        channel_list = await self.config.guild(ctx.guild).watching()
+        if channel.id in channel_list:
+            channel_list.remove(channel.id)
+        else:
+            return await ctx.send("Channel is not being watched.")
+        await self.config.guild(ctx.guild).watching.set(channel_list)
+        await ctx.send(f'{self.bot.get_channel(channel.id).mention} will not have links removed.')
+
+    async def on_message(self, message):
+        if isinstance(message.channel, discord.abc.PrivateChannel):
+            return
+        if message.author.bot:
+            return
+        data = await self.config.guild(message.guild).all()
+        watch_channel_list = data["watching"]
+        if not watch_channel_list:
+            return
+        if message.channel.id not in watch_channel_list:
+            return
+        allowed_roles = []
+        for role in data["role"]:
+            whitelist_role = discord.utils.get(message.author.roles, id=role)
+            if whitelist_role:
+                allowed_roles.append(whitelist_role)
+        message_channel = self.bot.get_channel(data["report_channel"])
+        if not allowed_roles:
+            try:
+                sentence = message.content.split()
+                for word in sentence:
+                    if self._match_url(word):
+                        msg = "**Message Removed in** {} ({})\n".format(message.channel.mention, message.channel.id)
+                        msg += "**Message sent by**: {} ({})\n".format(message.author.name, message.author.id)
+                        msg += "**Message content**:\n{}".format(message.content)
+                        if message_channel:
+                            await message_channel.send(msg)
+                        await message.delete()
+            except Exception as e:
+                if message_channel:
+                    await message_channel.send(e)
+                pass
+
+    @staticmethod
+    def _match_url(word):
+        try:
+            query_url = urlparse(word)
+            return any([query_url.scheme, query_url.netloc, query_url.path])
+        except:
+            return False