160 lines
7.3 KiB
Python
160 lines
7.3 KiB
Python
from discord.ext import commands
|
|
from .utils import checks
|
|
from .utils import config
|
|
import discord
|
|
import re
|
|
|
|
valid_perms = list(dict(discord.Permissions.none()))
|
|
|
|
|
|
class Mod:
|
|
"""Commands that can be used by a or an admin, depending on the command"""
|
|
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
|
|
@commands.group(pass_context=True, no_pm=True)
|
|
async def nsfw(self, ctx):
|
|
"""Handles adding or removing a channel as a nsfw channel"""
|
|
if ctx.invoked_subcommand is None:
|
|
await self.bot.say('Invalid subcommand passed: {0.subcommand_passed}'.format(ctx))
|
|
|
|
@nsfw.command(name="add", pass_context=True, no_pm=True)
|
|
@checks.customPermsOrRole("kick_members")
|
|
async def nsfw_add(self, ctx):
|
|
"""Registers this channel as a 'nsfw' channel"""
|
|
nsfw_channels = config.getContent('nsfw_channels')
|
|
if ctx.message.channel.id in nsfw_channels:
|
|
await self.bot.say("This channel is already registered as 'nsfw'!")
|
|
else:
|
|
nsfw_channels.append(ctx.message.channel.id)
|
|
if config.saveContent('nsfw_channels', nsfw_channels):
|
|
await self.bot.say("This channel has just been registered as 'nsfw'! Have fun you naughties ;)")
|
|
else:
|
|
await self.bot.say("I was unable to save this data")
|
|
|
|
@nsfw.command(name="remove", aliases=["delete"], pass_context=True, no_pm=True)
|
|
@checks.customPermsOrRole("kick_members")
|
|
async def nsfw_remove(self, ctx):
|
|
"""Removes this channel as a 'nsfw' channel"""
|
|
nsfw_channels = config.getContent('nsfw_channels')
|
|
if ctx.message.channel.id not in nsfw_channels:
|
|
await self.bot.say("This channel is not registered as a ''nsfw' channel!")
|
|
else:
|
|
nsfw_channels.remove(ctx.message.channel.id)
|
|
if config.saveContent('nsfw_channels', nsfw_channels):
|
|
await self.bot.say("This channel has just been unregistered as a nsfw channel")
|
|
else:
|
|
await self.bot.say("I was unable to save this data")
|
|
|
|
@commands.command(pass_context=True, no_pm=True)
|
|
@checks.customPermsOrRole("manage_server")
|
|
async def leave(self, ctx):
|
|
"""Forces the bot to leave the server"""
|
|
await self.bot.say('Why must I leave? Hopefully I can come back :c')
|
|
await self.bot.leave_server(ctx.message.server)
|
|
|
|
@commands.command(pass_context=True, no_pm=True)
|
|
@checks.customPermsOrRole("kick_members")
|
|
async def say(self, ctx, *msg: str):
|
|
"""Tells the bot to repeat what you say"""
|
|
msg = ' '.join(msg)
|
|
await self.bot.say(msg)
|
|
await self.bot.delete_message(ctx.message)
|
|
|
|
@commands.group(pass_context=True, invoke_without_command=True, no_pm=True)
|
|
@checks.customPermsOrRole("send_messages")
|
|
async def perms(self, ctx, *command: str):
|
|
"""This command can be used to print the current allowed permissions on a specific command
|
|
This supports groups as well as subcommands; pass no argument to print a list of available permissions"""
|
|
if command is None or len(command) == 0:
|
|
await self.bot.say("Valid permissions are: ```{}```".format("\n".join("{}".format(i) for i in valid_perms)))
|
|
return
|
|
command = " ".join(command)
|
|
|
|
custom_perms = config.getContent('custom_permissions')
|
|
if custom_perms is None:
|
|
await self.bot.say("There are no custom permissions setup on this server yet!")
|
|
return
|
|
server_perms = custom_perms.get(ctx.message.server.id)
|
|
if server_perms is None:
|
|
await self.bot.say("There are no custom permissions setup on this server yet!")
|
|
return
|
|
command_perms = server_perms.get(command)
|
|
if command_perms is None:
|
|
await self.bot.say("That command has no custom permissions setup on it!")
|
|
else:
|
|
await self.bot.say("You need to have the permission `{}` "
|
|
"to use the command `{}` in this server".format(command_perms, command))
|
|
|
|
@perms.command(name="add", aliases=["setup,create"], pass_context=True, no_pm=True)
|
|
@commands.has_permissions(manage_server=True)
|
|
async def add_perms(self, ctx, *msg: str):
|
|
"""Sets up custom permissions on the provided command
|
|
Format must be 'perms add <command> <permission>'
|
|
If you want to open the command to everyone, provide 'none' as the permission"""
|
|
command = " ".join(msg[0:len(msg) - 1])
|
|
permissions = msg[len(msg) - 1]
|
|
if permissions.lower() == "none":
|
|
permissions = "send_messages"
|
|
msg = msg[0:len(msg) - 1]
|
|
count = 0
|
|
cmd = self.bot.commands.get(msg[count])
|
|
while isinstance(cmd, commands.Group):
|
|
count += 1
|
|
try:
|
|
cmd = cmd.commands.get(msg[count])
|
|
except:
|
|
break
|
|
|
|
for check in cmd.checks:
|
|
if "isOwner" == check.__name__ or re.search("has_permissions", str(check)) is not None:
|
|
await self.bot.say("This command cannot have custom permissions setup!")
|
|
return
|
|
|
|
if getattr(discord.Permissions, permissions, None) is None and not permissions.lower() == "none":
|
|
await self.bot.say("{} does not appear to be a valid permission! Valid permissions are: ```{}```"
|
|
.format(permissions, "\n".join(valid_perms)))
|
|
return
|
|
|
|
custom_perms = config.getContent('custom_permissions')
|
|
if custom_perms is None:
|
|
custom_perms = {}
|
|
server_perms = custom_perms.get(ctx.message.server.id)
|
|
if server_perms is None:
|
|
custom_perms[ctx.message.server.id] = {command: permissions}
|
|
else:
|
|
server_perms[command] = permissions
|
|
custom_perms[ctx.message.server.id] = server_perms
|
|
if config.saveContent('custom_permissions', custom_perms):
|
|
await self.bot.say("I have just added your custom permissions; "
|
|
"you now need to have `{}` permissions to use the command `{}`".format(permissions, command))
|
|
else:
|
|
await self.bot.say("I was unable to save this data")
|
|
|
|
@perms.command(name="remove", aliases=["delete"], pass_context=True, no_pm=True)
|
|
@commands.has_permissions(manage_server=True)
|
|
async def remove_perms(self, ctx, *command: str):
|
|
"""Removes the custom permissions setup on the command specified"""
|
|
cmd = " ".join(command)
|
|
custom_perms = config.getContent('custom_permissions')
|
|
if custom_perms is None:
|
|
await self.bot.say("You do not have custom permissions setup on this server yet!")
|
|
return
|
|
server_perms = custom_perms.get(ctx.message.server.id)
|
|
if server_perms is None:
|
|
await self.bot.say("There are no custom permissions setup on this server yet!")
|
|
return
|
|
command_perms = server_perms.get(cmd)
|
|
if command_perms is None:
|
|
await self.bot.say("You do not have custom permissions setup on this command yet!")
|
|
return
|
|
del custom_perms[ctx.message.server.id][cmd]
|
|
if config.saveContent('custom_permissions', custom_perms):
|
|
await self.bot.say("I have just removed the custom permissions for {}!".format(cmd))
|
|
else:
|
|
await self.bot.say("I was unable to save this data")
|
|
|
|
|
|
def setup(bot):
|
|
bot.add_cog(Mod(bot))
|