Skip to content

Commit

Permalink
No more "This application did not respond" bug.
Browse files Browse the repository at this point in the history
  • Loading branch information
@CallocGD
CallocGD authored Dec 14, 2024
1 parent dec5fe8 commit e9e10c4
Showing 1 changed file with 123 additions and 64 deletions.
187 changes: 123 additions & 64 deletions defender.py
View file
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
from __future__ import annotations

from discord.ext import tasks

# from discord.ext.commands import Context, Greedy
from discord.ext import commands
from discord import Member, Guild, Embed, app_commands
Expand All @@ -12,46 +13,51 @@

from models import Defender
from typing import Union, Optional
import io

import io


bot = Defender("$")

# TODO: Slash commands? (LOL I already did that :P)
# TODO: Slash commands?


async def safe_prune(member:Member):

async def safe_prune(member: Member):
# All roles should be removed from the member

# Member should not be banned if they are moderators
if (member.guild_permissions.ban_members or member.guild_permissions.kick_members):
if member.guild_permissions.ban_members or member.guild_permissions.kick_members:
return None

# Remove all roles so channels can't be accessed
for r in member.guild.roles:
if member.get_role(r.id):
await member.remove_roles(r, reason="Violated The Rules")

try:
return await bot.prune_member(member)
except (discord.HTTPException, discord.Forbidden):
return None
return None


async def safe_ban(user:Union[discord.User, Member], interaction:discord.Interaction, reason:Optional[str] = None):
async def safe_ban(
user: Union[discord.User, Member],
interaction: discord.Interaction,
reason: Optional[str] = None,
):
if isinstance(user, Member):
if (user.guild_permissions.ban_members or user.guild_permissions.kick_members):
if user.guild_permissions.ban_members or user.guild_permissions.kick_members:
return None

async with suppress(discord.NotFound, discord.Forbidden, discord.HTTPException):
await interaction.guild.ban(user,reason=reason)
await interaction.channel.send(embed=Embed(title="Banned")
.add_field(name="Member ID", value=user.id)
.add_field(name="Reason", value=reason or "None Given")
.add_field(name="Name", value=user.name))

await interaction.guild.ban(user, reason=reason)
await interaction.followup.send(
embed=Embed(title="Banned")
.add_field(name="Member ID", value=user.id)
.add_field(name="Reason", value=reason or "None Given")
.add_field(name="Name", value=user.name)
)

# TODO: Shorten the length of this loop down to 5 minutes depending on the number of guilds...

@tasks.loop(minutes=15)
async def prune_loop():
Expand All @@ -63,83 +69,102 @@ async def prune_loop():

@bot.event
async def on_ready():

await bot.init_db()
# Try a prune loop immediatley if all requirements
# Try a prune loop immediatley if all requirements
# were ment to perform one if we originally did a startup before...
for guild in bot.guilds:
gm = await bot.get_guild_model(guild.id)
if gm.moderator_channel and gm.prune_role_id:
await bot.ban_pruned_members(gm.guild_id)
prune_loop.start()


@bot.event
async def on_guild_join(guild:Guild):
async def on_guild_join(guild: Guild):
await bot.create_guild_model(guild)


@bot.event
async def on_guild_remove(guild:Guild):
async def on_guild_remove(guild: Guild):
await bot.remove_guild_model(guild)


@bot.event
async def on_member_join(member:Member):
async def on_member_join(member: Member):
gm = await bot.get_guild_model(member.guild.id)
if gm.moderator_channel and gm.prune_role_id:
await bot.check_member(member)
# Smart to run this when we can...
await bot.ban_pruned_members(gm.guild_id)


@bot.tree.command(name="add-prune-role")
@commands.has_permissions(administrator=True)
async def register_prune_role(interaction: discord.Interaction, role_id:int):
async def register_prune_role(interaction: discord.Interaction, role: discord.Role):
"""Registers a prune Role to add to a user when joining a little too early, (reqiures admin)"""
await interaction.response.defer()
try:
await bot.update_guild_prune_role(role_id, interaction.guild.id)
await interaction.channel.send("Role Updated")
await bot.update_guild_prune_role(role.id, interaction.guild.id)
await interaction.followup.send("Role Updated")
except Exception as e:
await interaction.channel.send(f"Error {e.__name__} {e}")
await interaction.followup.send(f"Error {e.__name__} {e}")


@bot.tree.command(name="add-mod-channel")
@commands.has_permissions(administrator=True)
async def register_moderator_channel(interaction: discord.Interaction, channel_id:int):
async def register_moderator_channel(interaction: discord.Interaction, channel: Optional[discord.TextChannel] = None):
"""Registers a prune Role to add to a user when joining a little too early, (reqiures admin)"""
await interaction.response.defer()

try:
await bot.update_guild_mod_channel(channel_id, interaction.guild.id)
await interaction.channel.send("Moderator Channel Registered")
if not channel:
channel = interaction.channel
await bot.update_guild_mod_channel(channel.id, interaction.guild.id)
await interaction.followup.send("Moderator Channel Registered")
except Exception as e:
await interaction.channel.send(f"Error {e.__name__} {e}")
await interaction.followup.send(f"Error {e.__name__} {e}")


@bot.tree.command(name="requirements")
@commands.bot_has_permissions(ban_members=True)
@commands.has_permissions(administrator=True)
async def check_requirements(interaction: discord.Interaction):
"""Checks Guild Requirements for pruning members"""
"""Checks Guild Requirements for pruning members"""
await interaction.response.defer()
embed = Embed(title="Requirements")

model = await bot.get_guild_model(interaction.guild.id)
if not (model.moderator_channel or model.prune_role_id):
embed.add_field(name="Set Prune Role", value="Incomplete" if not model.prune_role_id else "Complete")
embed.add_field(name="Set Moderation Channel", value="Incomplete" if not model.moderator_channel else "Complete")
if not (model.moderator_channel or model.prune_role_id):
embed.add_field(
name="Set Prune Role",
value="Incomplete" if not model.prune_role_id else "Complete",
)
embed.add_field(
name="Set Moderation Channel",
value="Incomplete" if not model.moderator_channel else "Complete",
)
else:
embed.add_field(name="Good news", value="You did all the requirements for pruning away scammers/spammers/raiders etc...")

await interaction.channel.send(embed=embed)
embed.add_field(
name="Good news",
value="You did all the requirements for pruning away scammers/spammers/raiders etc...",
)

await interaction.followup.send(embed=embed)


# TODO: This command needs fixing IIRC
# @bot.tree.command(name="prune")
# @commands.bot_has_permissions(ban_members=True, manage_roles=True)
# @commands.has_permissions(ban_members=True)
# async def prune_users(ctx:Context, members:Greedy[Member]):
# """Prunes Single or Multiple members at a time useful for unsure ban-wipes"""

# model = await bot.get_guild_model(ctx.guild.id)
# if not (model.moderator_channel or model.prune_role_id):
# if not (model.moderator_channel or model.prune_role_id):
# embed = Embed(color=0x1f1137, title="Error Unfinished Requirements")
# embed.add_field(name="Set Prune Role", value="Incomplete" if not model.prune_role_id else "Complete")
# embed.add_field(name="Set Moderation Channel", value="Incomplete" if not model.moderator_channel else "Complete")
# return await ctx.send(embed=embed)

# await ctx.send("Pruning members...")

# async for result in amap(safe_prune , members, concurrency=2):
Expand All @@ -151,73 +176,101 @@ async def check_requirements(interaction: discord.Interaction):
# Where I upload a file of discordIDs and bans them all...


@bot.tree.command(name="massban", description="""Performs a masisve ban and bans Many Users given from a text file""")
@bot.tree.command(
name="massban",
description="""Performs a masisve ban and bans Many Users given from a text file""",
)
@commands.bot_has_permissions(ban_members=True, manage_roles=True)
@commands.has_permissions(administrator=True)
@app_commands.describe(
file="A Text file/list of members to ban by Developer ID Line for line, The User does not have to be in the server to perform a massban.",
reason="Reason A Group of Users are being banned for."
reason="Reason A Group of Users are being banned for.",
)
async def massban(interaction: discord.Interaction, file:discord.Attachment, reason:Optional[str] = None):
async def massban(
interaction: discord.Interaction,
file: discord.Attachment,
reason: Optional[str] = None,
):
"""Performs a masisve ban and bans Many Users given from a text file"""
await interaction.response.defer()

if not file.filename.endswith(".txt"):
return await interaction.channel.send("ERROR: Invalid File Type, textfiles with IDs on newlines required")
data = await file.read()
await interaction.channel.send("loading previous ban history to filter bans...")
return await interaction.channel.send(
"ERROR: Invalid File Type, textfiles with IDs on newlines required"
)
data = await file.read()
await interaction.followup.send("loading previous ban history to filter bans...")
entries_filter = {entry.user.id async for entry in interaction.guild.bans()}
await interaction.channel.send("Performing Massban")

await interaction.followup.send("Performing Massban")

for i in data.splitlines(keepends=False):
if uid := i.strip():
if uid.isdigit():
user_id = int(i.strip())
if user_id in entries_filter:
continue

async with suppress(discord.NotFound):
user = await bot.fetch_user(
user_id
)
user = await bot.fetch_user(user_id)
await safe_ban(user=user, interaction=interaction, reason=reason)


@bot.tree.command(name="blacklist", description="""makes a blacklist of banned users in the discord server""")

@bot.tree.command(
name="blacklist",
description="""makes a blacklist of banned users in the discord server""",
)
@commands.bot_has_permissions(ban_members=True, manage_roles=True)
@commands.has_permissions(administrator=True)
async def get_blacklist(interaction: discord.Interaction):
"""Be very careful when using this command as other users can lookup bad people"""
await interaction.response.defer()

# TODO: Local Database for banned users?

return await interaction.channel.send(
return await interaction.followup.send(
file=discord.File(
io.StringIO("\n".join(map(str, {entry.user.id async for entry in interaction.guild.bans()}))),
filename="blacklist.txt"
io.StringIO(
"\n".join(
map(
str, {entry.user.id async for entry in interaction.guild.bans()}
)
)
),
filename="blacklist.txt",
)
)



# Inspired by EvilPauze https://github.com/Alex1304/evilpauze
@bot.tree.command(name="lock-channel", description="""Locks down a discord server channel same to how EvilPauze works""")
# Inspired by EvilPauze
@bot.tree.command(
name="lock-channel",
description="""Locks down a discord server channel same to how EvilPauze works""",
)
@commands.bot_has_permissions(manage_channels=True)
@commands.has_permissions(manage_channels=True)
async def lock_channel(interaction: discord.Interaction, channel:Optional[discord.TextChannel] = None, moderators:bool=True):
async def lock_channel(
interaction: discord.Interaction,
channel: Optional[discord.TextChannel] = None,
moderators: bool = True,
):
await interaction.response.defer()
channel = channel or interaction.channel

assert channel, "No Channel Exists"

ldm = await bot.create_lockdown(interaction.guild, channel)

overwrite = discord.PermissionOverwrite()
# TODO: Make read_message flag chanagable upon performing more dangerous damage controls...
overwrite.send_messages = False
overwrite.read_messages = True

for role in interaction.guild.roles:
if role.permissions.send_messages:
if moderators and (role.permissions.ban_members or role.permissions.kick_members or role.permissions.deafen_members):
if moderators and (
role.permissions.ban_members
or role.permissions.kick_members
or role.permissions.deafen_members
):
# Moderators can talk here to help with damage control
continue

Expand All @@ -230,6 +283,12 @@ async def lock_channel(interaction: discord.Interaction, channel:Optional[discor
# TODO Unlock command...




def banner():
pass


def banner():

print(Fore.LIGHTBLUE_EX + """
Expand Down

0 comments on commit e9e10c4

Please sign in to comment.