jarvis/jarvis-bot
jarvis/jarvis-bot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Migrate admin/roleping, ref #91

Browse source
This commit is contained in:
Zeva Rose 2022-02-02 21:18:07 -07:00
parent 9aa7075266
commit a9c75207da
1 changed files with 95 additions and 153 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

246
jarvis/cogs/admin/roleping.py
View file

@ -1,41 +1,37 @@
"""J.A.R.V.I.S. RolepingCog.""" """J.A.R.V.I.S. RolepingCog."""
from datetime import datetime, timedelta from datetime import datetime, timedelta
from ButtonPaginator import Paginator from dis_snek import InteractionContext, Permissions, Snek
from discord import Member, Role from dis_snek.ext.paginators import Paginator
from discord.ext.commands import Bot from dis_snek.models.discord.embed import EmbedField
from discord_slash import SlashContext, cog_ext from dis_snek.models.discord.role import Role
from discord_slash.model import ButtonStyle from dis_snek.models.discord.user import Member
from discord_slash.utils.manage_commands import create_option from dis_snek.models.snek.application_commands import (
OptionTypes,
slash_command,
slash_option,
)
from jarvis.db.models import Roleping from jarvis.db.models import Roleping
from jarvis.utils import build_embed from jarvis.utils import build_embed
from jarvis.utils.cachecog import CacheCog from jarvis.utils.cachecog import CacheCog
from jarvis.utils.field import Field
from jarvis.utils.permissions import admin_or_permissions from jarvis.utils.permissions import admin_or_permissions
class RolepingCog(CacheCog): class RolepingCog(CacheCog):
"""J.A.R.V.I.S. RolepingCog.""" """J.A.R.V.I.S. RolepingCog."""
def __init__(self, bot: Bot): def __init__(self, bot: Snek):
super().__init__(bot) super().__init__(bot)
@cog_ext.cog_subcommand( @slash_command(
base="roleping", name="roleping", sub_cmd_name="add", sub_cmd_description="Add a role to roleping"
name="add",
description="Add a role to roleping",
options=[
create_option(
name="role",
description="Role to add to roleping",
option_type=8,
required=True,
) )
], @slash_option(
name="role", description="Role to add", option_type=OptionTypes.ROLE, required=True
) )
@admin_or_permissions(manage_guild=True) @admin_or_permissions(Permissions.MANAGE_GUILD)
async def _roleping_add(self, ctx: SlashContext, role: Role) -> None: async def _roleping_add(self, ctx: InteractionContext, role: Role) -> None:
roleping = Roleping.objects(guild=ctx.guild.id, role=role.id).first() roleping = Roleping.objects(guild=ctx.guild.id, role=role.id).first()
if roleping: if roleping:
await ctx.send(f"Role `{role.name}` already in roleping.", hidden=True) await ctx.send(f"Role `{role.name}` already in roleping.", hidden=True)
@ -49,21 +45,12 @@ class RolepingCog(CacheCog):
).save() ).save()
await ctx.send(f"Role `{role.name}` added to roleping.") await ctx.send(f"Role `{role.name}` added to roleping.")
@cog_ext.cog_subcommand( @slash_command(name="roleping", sub_cmd_name="remove", sub_cmd_description="Remove a role")
base="roleping", @slash_option(
name="remove", name="role", description="Role to remove", option_type=OptionTypes.ROLE, required=True
description="Remove a role from the roleping",
options=[
create_option(
name="role",
description="Role to remove from roleping",
option_type=8,
required=True,
) )
], @admin_or_permissions(Permissions.MANAGE_GUILD)
) async def _roleping_remove(self, ctx: InteractionContext, role: Role) -> None:
@admin_or_permissions(manage_guild=True)
async def _roleping_remove(self, ctx: SlashContext, role: Role) -> None:
roleping = Roleping.objects(guild=ctx.guild.id, role=role.id) roleping = Roleping.objects(guild=ctx.guild.id, role=role.id)
if not roleping: if not roleping:
await ctx.send("Roleping does not exist", hidden=True) await ctx.send("Roleping does not exist", hidden=True)
@ -72,12 +59,8 @@ class RolepingCog(CacheCog):
roleping.delete() roleping.delete()
await ctx.send(f"Role `{role.name}` removed from roleping.") await ctx.send(f"Role `{role.name}` removed from roleping.")
@cog_ext.cog_subcommand( @slash_command(name="roleping", sub_cmd_name="list", description="Lick all blocklisted roles")
base="roleping", async def _roleping_list(self, ctx: InteractionContext) -> None:
name="list",
description="List all blocklisted roles",
)
async def _roleping_list(self, ctx: SlashContext) -> None:
exists = self.check_cache(ctx) exists = self.check_cache(ctx)
if exists: if exists:
await ctx.defer(hidden=True) await ctx.defer(hidden=True)
@ -108,17 +91,17 @@ class RolepingCog(CacheCog):
description=role.mention, description=role.mention,
color=str(role.color), color=str(role.color),
fields=[ fields=[
Field( EmbedField(
name="Created At", name="Created At",
value=roleping.created_at.strftime("%a, %b %d, %Y %I:%M %p"), value=roleping.created_at.strftime("%a, %b %d, %Y %I:%M %p"),
inline=False, inline=False,
), ),
Field(name="Active", value=str(roleping.active)), EmbedField(name="Active", value=str(roleping.active)),
Field( EmbedField(
name="Bypass Users", name="Bypass Users",
value="\n".join(bypass_users), value="\n".join(bypass_users),
), ),
Field( EmbedField(
name="Bypass Roles", name="Bypass Roles",
value="\n".join(bypass_roles), value="\n".join(bypass_roles),
), ),
@ -129,23 +112,12 @@ class RolepingCog(CacheCog):
if not admin: if not admin:
admin = self.bot.user admin = self.bot.user
embed.set_author(name=admin.nick or admin.name, icon_url=admin.avatar_url) embed.set_author(name=admin.display_name, icon_url=admin.avatar_url)
embed.set_footer(text=f"{admin.name}#{admin.discriminator} | {admin.id}") embed.set_footer(text=f"{admin.name}#{admin.discriminator} | {admin.id}")
embeds.append(embed) embeds.append(embed)
paginator = Paginator( paginator = Paginator.create_from_embeds(self.bot, embeds, timeout=300)
bot=self.bot,
ctx=ctx,
embeds=embeds,
only=ctx.author,
timeout=60 * 5, # 5 minute timeout
disable_after_timeout=True,
use_extend=len(embeds) > 2,
left_button_style=ButtonStyle.grey,
right_button_style=ButtonStyle.grey,
basic_buttons=["◀", "▶"],
)
self.cache[hash(paginator)] = { self.cache[hash(paginator)] = {
"user": ctx.author.id, "user": ctx.author.id,
@ -155,32 +127,26 @@ class RolepingCog(CacheCog):
"paginator": paginator, "paginator": paginator,
} }
await paginator.start() await paginator.send(ctx)
@cog_ext.cog_subcommand( @slash_command(
base="roleping", name="roleping",
subcommand_group="bypass", description="Block roles from being pinged",
name="user", group_name="bypass",
description="Add a user as a bypass to a roleping", group_description="Allow specific users/roles to ping rolepings",
base_desc="Block roles from being pinged", sub_cmd_name="user",
sub_group_desc="Allow specific users/roles to ping rolepings", sub_cmd_description="Add a user as a bypass to a roleping",
options=[
create_option(
name="user",
description="User to add",
option_type=6,
required=True,
),
create_option(
name="rping",
description="Rolepinged role",
option_type=8,
required=True,
),
],
) )
@admin_or_permissions(manage_guild=True) @slash_option(
async def _roleping_bypass_user(self, ctx: SlashContext, user: Member, rping: Role) -> None: name="user", description="User to add", option_type=OptionTypes.USER, required=True
)
@slash_option(
name="rping", description="Rolepinged role", option_type=OptionTypes.ROLE, required=True
)
@admin_or_permissions(Permissions.MANAGE_GUILD)
async def _roleping_bypass_user(
self, ctx: InteractionContext, user: Member, rping: Role
) -> None:
roleping = Roleping.objects(guild=ctx.guild.id, role=rping.id).first() roleping = Roleping.objects(guild=ctx.guild.id, role=rping.id).first()
if not roleping: if not roleping:
await ctx.send(f"Roleping not configured for {rping.mention}", hidden=True) await ctx.send(f"Roleping not configured for {rping.mention}", hidden=True)
@ -208,32 +174,22 @@ class RolepingCog(CacheCog):
roleping.bypass["users"].append(user.id) roleping.bypass["users"].append(user.id)
roleping.save() roleping.save()
await ctx.send(f"{user.nick or user.name} user bypass added for `{rping.name}`") await ctx.send(f"{user.display_name} user bypass added for `{rping.name}`")
@cog_ext.cog_subcommand( @slash_command(
base="roleping", name="roleping",
subcommand_group="bypass", group_name="bypass",
name="role", sub_cmd_name="role",
description="Add a role as a bypass to a roleping", description="Add a role as a bypass to roleping",
base_desc="Block roles from being pinged",
sub_group_desc="Allow specific users/roles to ping rolepings",
options=[
create_option(
name="role",
description="Role to add",
option_type=8,
required=True,
),
create_option(
name="rping",
description="Rolepinged role",
option_type=8,
required=True,
),
],
) )
@admin_or_permissions(manage_guild=True) @slash_option(
async def _roleping_bypass_role(self, ctx: SlashContext, role: Role, rping: Role) -> None: name="role", description="Role to add", option_type=OptionTypes.ROLE, required=True
)
@slash_option(
name="rping", description="Rolepinged role", option_type=OptionTypes.ROLE, required=True
)
@admin_or_permissions(Permissions.MANAGE_GUILD)
async def _roleping_bypass_role(self, ctx: InteractionContext, role: Role, rping: Role) -> None:
roleping = Roleping.objects(guild=ctx.guild.id, role=rping.id).first() roleping = Roleping.objects(guild=ctx.guild.id, role=rping.id).first()
if not roleping: if not roleping:
await ctx.send(f"Roleping not configured for {rping.mention}", hidden=True) await ctx.send(f"Roleping not configured for {rping.mention}", hidden=True)
@ -255,30 +211,24 @@ class RolepingCog(CacheCog):
roleping.save() roleping.save()
await ctx.send(f"{role.name} role bypass added for `{rping.name}`") await ctx.send(f"{role.name} role bypass added for `{rping.name}`")
@cog_ext.cog_subcommand( @slash_command(
base="roleping", name="roleping",
subcommand_group="restore", description="Block roles from being pinged",
name="user", group_name="restore",
description="Remove a role bypass", group_description="Remove a roleping bypass",
base_desc="Block roles from being pinged", sub_cmd_name="user",
sub_group_desc="Remove a bypass from a roleping (restoring it)", sub_cmd_description="Remove a bypass from a roleping (restoring it)",
options=[
create_option(
name="user",
description="User to add",
option_type=6,
required=True,
),
create_option(
name="rping",
description="Rolepinged role",
option_type=8,
required=True,
),
],
) )
@admin_or_permissions(manage_guild=True) @slash_option(
async def _roleping_restore_user(self, ctx: SlashContext, user: Member, rping: Role) -> None: name="user", description="User to remove", option_type=OptionTypes.USER, required=True
)
@slash_option(
name="rping", description="Rolepinged role", option_type=OptionTypes.ROLE, required=True
)
@admin_or_permissions(Permissions.MANAGE_GUILD)
async def _roleping_restore_user(
self, ctx: InteractionContext, user: Member, rping: Role
) -> None:
roleping = Roleping.objects(guild=ctx.guild.id, role=rping.id).first() roleping = Roleping.objects(guild=ctx.guild.id, role=rping.id).first()
if not roleping: if not roleping:
await ctx.send(f"Roleping not configured for {rping.mention}", hidden=True) await ctx.send(f"Roleping not configured for {rping.mention}", hidden=True)
@ -290,32 +240,24 @@ class RolepingCog(CacheCog):
roleping.bypass["users"].delete(user.id) roleping.bypass["users"].delete(user.id)
roleping.save() roleping.save()
await ctx.send(f"{user.nick or user.name} user bypass removed for `{rping.name}`") await ctx.send(f"{user.display_name} user bypass removed for `{rping.name}`")
@cog_ext.cog_subcommand( @slash_command(
base="roleping", name="roleping",
subcommand_group="restore", group_name="restore",
name="role", sub_cmd_name="role",
description="Remove a role bypass", description="Remove a bypass from a roleping (restoring it)",
base_desc="Block roles from being pinged", )
sub_group_desc="Remove a bypass from a roleping (restoring it)", @slash_option(
options=[ name="role", description="Role to remove", option_type=OptionTypes.ROLE, required=True
create_option( )
name="role", @slash_option(
description="Role to add", name="rping", description="Rolepinged role", option_type=OptionTypes.ROLE, required=True
option_type=8,
required=True,
),
create_option(
name="rping",
description="Rolepinged role",
option_type=8,
required=True,
),
],
) )
@admin_or_permissions(manage_guild=True) @admin_or_permissions(manage_guild=True)
async def _roleping_restore_role(self, ctx: SlashContext, role: Role, rping: Role) -> None: async def _roleping_restore_role(
self, ctx: InteractionContext, role: Role, rping: Role
) -> None:
roleping = Roleping.objects(guild=ctx.guild.id, role=rping.id).first() roleping = Roleping.objects(guild=ctx.guild.id, role=rping.id).first()
if not roleping: if not roleping:
await ctx.send(f"Roleping not configured for {rping.mention}", hidden=True) await ctx.send(f"Roleping not configured for {rping.mention}", hidden=True)