Hrochobot/hrochobot/cogs/roles.py

import discord
import hrochobot.utils.data as data
from discord.ext import commands
from discord.utils import get

ROLES_JSON = "roles"

class Roles(commands.Cog):
    def __init__(self, bot):
        self.bot = bot

    secret_roles = discord.SlashCommandGroup(
        "secretroles",
        "Commands for management of secret roles.",
        guild_only=True,
        checks=[commands.has_permissions(manage_roles=True)]
    )

    @secret_roles.command(description="Adds a new secret role.")
    @discord.option("role", discord.role.Role, description="Role locked behind a password.")
    @discord.option("password", str, description="Password for given role.")
    async def add(self, ctx, role, password):
        roles = data.load_guild_data(ctx.author.guild.id, ROLES_JSON)
        if password in roles["secret_roles"]:
            return await ctx.respond(f"Password ``{password}`` is already used.", ephemeral=True)
 
        roles["secret_roles"][password] = role.id
        data.dump_guild_data(ctx.author.guild.id, ROLES_JSON, roles)
        return await ctx.respond(f"Secret role {role.mention} added with password {password}.", ephemeral=True)
    
    @secret_roles.command(description="Lists all passwords and their secret roles")
    async def list(self, ctx):
        roles = data.load_guild_data(ctx.author.guild.id, ROLES_JSON)
        if len(roles["secret_roles"]) == 0:
            return await ctx.respond(f"No current secret roles.", ephemeral=True)
        msg = ""
        for passwd, role in roles["secret_roles"].items():
            role = get(ctx.author.guild.roles, id=role)
            msg += f"``{passwd}``: {role.mention}\n"
        return await ctx.respond(msg, ephemeral=True)
    
    @secret_roles.command(description="Deletes given password and its secret role.")
    @discord.option("password", str, description="Password to be deleted.")
    async def delete(self, ctx, password):
        roles = data.load_guild_data(ctx.author.guild.id, ROLES_JSON)
        if password not in roles["secret_roles"]:
            return await ctx.respond(f"Role with passowrd {password} does not exist.", ephemeral=True)
 
        role = get(ctx.author.guild.roles, id=roles["secret_roles"][password])
        del roles["secret_roles"][password]
        data.dump_guild_data(ctx.author.guild.id, ROLES_JSON, roles)
        return await ctx.respond(
            f"Secret role {role.mention} no longer obtainable with password {password}.",
            ephemeral=True
        )

    @discord.slash_command(description="Gives a secret role locked by a password.", guild_only=True)
    @discord.option("password", str, description="Password for secret role.")
    async def secretrole(self, ctx, password):
        roles = data.load_guild_data(ctx.author.guild.id, ROLES_JSON)
        if password in roles["secret_roles"]:
            author = ctx.author
            role = get(author.guild.roles, id=roles["secret_roles"][password])
            await author.add_roles(role, reason="Roles assigned for password knowledge.")
            return await ctx.respond(f"You now have role {role.mention}.", ephemeral=True)
        else:
            return await ctx.respond("Incorrect password.", ephemeral=True)

def setup(bot):
    bot.add_cog(Roles(bot))