mirror of
https://github.com/AsgardEternal/DiscordWhitelist.git
synced 2024-09-29 12:14:27 -05:00
108 lines
3.6 KiB
Python
108 lines
3.6 KiB
Python
import discord
|
|
import random
|
|
import logging
|
|
from autowl import config
|
|
from autowl.bot import Bot
|
|
from discord.ext import commands
|
|
from discord import app_commands
|
|
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
class Group(commands.Cog, name="group"):
|
|
def __init__(self, client: Bot):
|
|
self.client = client
|
|
|
|
async def baseperm(self, interaction: discord.Interaction, role: discord.Role, perms: str):
|
|
if role.id in self.client.whitelistGrps.keys():
|
|
await interaction.response.send_message(
|
|
f"**{role.name}** is already added, cannot add it again!"
|
|
)
|
|
return
|
|
|
|
log.info(f"Adding {role.name} ({role.id}) as a Whitelist role")
|
|
await interaction.response.send_message(
|
|
f"Adding **{role.name}** as a Whitelist role"
|
|
)
|
|
self.client.whitelistGrps[role.id] = config.WhitelistGroup(
|
|
name=role.name, roleID=role.id, permissions=perms
|
|
)
|
|
membsup = []
|
|
for memb in role.members:
|
|
membsup.append(memb.id)
|
|
memupcur = self.client.squadjs.cursor(buffered=True)
|
|
in_params = ','.join(['%s'] * len(membsup))
|
|
sqlstate = "SELECT * FROM DBLog_SteamUsers WHERE discordID IN (%s)" % in_params
|
|
memupcur.execute(sqlstate, membsup)
|
|
|
|
udata = memupcur.fetchall()
|
|
for data in udata:
|
|
self.client.whitelistGrps[role.id].addMember(config.WhitelistMember(data[2], "unknown", data[0]))
|
|
self.client.squadjs.commit()
|
|
|
|
@app_commands.command()
|
|
async def add(
|
|
self,
|
|
interaction: discord.Interaction,
|
|
role: discord.Role,
|
|
):
|
|
await self.baseperm(interaction, role, "reserve")
|
|
|
|
@app_commands.command()
|
|
async def addperm(
|
|
self,
|
|
interaction: discord.Interaction,
|
|
role: discord.Role,
|
|
perms: str
|
|
):
|
|
await self.baseperm(interaction, role, perms)
|
|
|
|
@app_commands.command()
|
|
async def remove(
|
|
self,
|
|
interaction: discord.Interaction,
|
|
role: discord.Role,
|
|
):
|
|
if not self.client.whitelistGrps.get(role.id):
|
|
await interaction.response.send_message(
|
|
f"**{role.name}** has not been added as a whitelisted group, unable to remove!"
|
|
)
|
|
return
|
|
|
|
log.info(f"Removing {role.name} ({role.id}) from Whitelisted role")
|
|
await interaction.response.send_message(
|
|
f"Removed **{role.name}** from Whitelisted roles"
|
|
)
|
|
self.client.whitelistGrps[role.id].delGroup()
|
|
self.client.whitelistGrps.pop(role.id)
|
|
|
|
@app_commands.command()
|
|
async def list_whitelisted_roles(self, interaction: discord.Interaction):
|
|
whitelisted_roles = []
|
|
if not interaction.guild:
|
|
await interaction.response.send_message(
|
|
"This command must be ran within a discord server"
|
|
)
|
|
return
|
|
|
|
for group in self.client.whitelistGrps:
|
|
role_id = self.client.whitelistGrps[group].discord_role_id
|
|
if not interaction.guild.get_role(role_id):
|
|
continue
|
|
whitelisted_roles.append(f"<@&{role_id}>")
|
|
|
|
embed_description = (
|
|
"\n".join(whitelisted_roles)
|
|
if whitelisted_roles
|
|
else "No whitelisted roles found, you can some using `/add`"
|
|
)
|
|
|
|
embed = discord.Embed(
|
|
title="Whitelisted Roles",
|
|
description=embed_description,
|
|
color=random.randint(0, 0xFFFFFF),
|
|
).set_footer(text="Users with these roles will be whitelisted on Squad")
|
|
|
|
await interaction.response.send_message(embed=embed)
|