PluralKit/src/pluralkit/bot/commands/api_commands.py

from pluralkit.bot.commands import CommandContext

disclaimer = "Please note that this grants access to modify (and delete!) all your system data, so keep it safe and secure. If it leaks or you need a new one, you can invalidate this one with `pk;token refresh`."


async def token_root(ctx: CommandContext):
    if ctx.match("refresh") or ctx.match("expire") or ctx.match("invalidate") or ctx.match("update"):
        await token_refresh(ctx)
    else:
        await token_get(ctx)


async def token_get(ctx: CommandContext):
    system = await ctx.ensure_system()

    if system.token:
        token = system.token
    else:
        token = await system.refresh_token(ctx.conn)

    token_message = "Here's your API token: \n**`{}`**\n{}".format(token, disclaimer)
    return await ctx.reply_ok_dm(token_message)


async def token_refresh(ctx: CommandContext):
    system = await ctx.ensure_system()

    token = await system.refresh_token(ctx.conn)
    token_message = "Your previous API token has been invalidated. You will need to change it anywhere it's currently used.\nHere's your new API token:\n**`{}`**\n{}".format(
        token, disclaimer)
    return await ctx.reply_ok_dm(token_message)