CopyBot/Red-DiscordBot
1
0
Fork 0
mirror of https://github.com/Cog-Creators/Red-DiscordBot.git synced 2025-11-06 03:08:55 -05:00
Code Issues Packages Projects Releases Wiki Activity

Add [p]diagnoseissues command (#5243)

Browse Source 
* Split `Requires._transition_state` into reusable methods for Diagnoser

* (style) Fix line length

* Add missing .format()

* Add handling for user permissions, privilege level, Permissions rules

* Add missing awaits, use correct method in user perms check, add 'the'

* Fix .format() fields

* Add comment

* Add new file to labeler's configuration

* Add the command to the documentation

* All the work from DiagnoserCog

You can find the commit history of it here:
https://github.com/jack1142/DiagnoserCog

* Fix circular import

* Make channel argument optional

* Add a tip about channels from different servers
This commit is contained in:
jack1142 2021-09-05 19:01:46 +02:00 committed by GitHub
parent d84c8efa34
commit 86649e897f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
5 changed files with 1069 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.github/labeler.yml vendored
View File

@ -27,6 +27,7 @@
# Source
- redbot/*
- redbot/core/__init__.py
- redbot/core/_diagnoser.py
- redbot/core/_sharedlibdeprecation.py
- redbot/core/bot.py
- redbot/core/checks.py

27
docs/cog_guides/core.rst
View File

@ -770,6 +770,33 @@ This is limited to one message every 60 seconds per person.
**Arguments:**
- ``[message]`` - The message to send to the owner.
.. _core-command-diagnoseissues:
^^^^^^^^^^^^^^
diagnoseissues
^^^^^^^^^^^^^^
**Syntax**
.. code-block:: none
[p]diagnoseissues [channel] <member> <command_name>
**Description**
Diagnose issues with the command checks with ease!
If you want to diagnose the command from a text channel in a different server,
you can do so by using the command in DMs.
**Example:**
- ``[p]diagnoseissues #general @Slime ban`` - Diagnose why @Slime can't use ``[p]ban`` in #general channel.
**Arguments:**
- ``[channel]`` - The text channel that the command should be tested for. Defaults to the current channel.
- ``<member>`` - The member that should be considered as the command caller.
- ``<command_name>`` - The name of the command to test.
.. _core-command-dm:
^^

958
redbot/core/_diagnoser.py Normal file
View File

@ -0,0 +1,958 @@
from __future__ import annotations
import itertools
import string
from copy import copy
from dataclasses import dataclass
from functools import partial
from typing import TYPE_CHECKING, Awaitable, Callable, Iterable, List, Optional, Union
import discord
from redbot.core import commands
from redbot.core.i18n import Translator
from redbot.core.utils.chat_formatting import (
bold,
escape,
format_perms_list,
humanize_list,
inline,
)
if TYPE_CHECKING:
from redbot.core.bot import Red
_ = Translator("IssueDiagnoser", __file__)
@dataclass
class CheckResult:
success: bool
label: str
details: Union[List[CheckResult], str] = ""
resolution: str = ""
class IssueDiagnoserBase:
def __init__(
self,
bot: Red,
original_ctx: commands.Context,
channel: discord.TextChannel,
author: discord.Member,
command: commands.Command,
) -> None:
self.bot = bot
self._original_ctx = original_ctx
self.guild = channel.guild
self.channel = channel
self.author = author
self.command = command
self._prepared = False
self.message: discord.Message
self.ctx: commands.Context
async def _prepare(self) -> None:
if self._prepared:
return
self.message = copy(self._original_ctx.message)
self.message.author = self.author
self.message.channel = self.channel
self.message.content = self._original_ctx.prefix + self.command.qualified_name
# clear the cached properties
for attr in self.message._CACHED_SLOTS: # type: ignore[attr-defined]
try:
delattr(self.message, attr)
except AttributeError:
pass
self.ctx = await self.bot.get_context(self.message)
# reusable methods
async def _check_until_fail(
self,
label: str,
checks: Iterable[Callable[[], Awaitable[CheckResult]]],
*,
final_check_result: Optional[CheckResult] = None,
) -> CheckResult:
details = []
for check in checks:
check_result = await check()
details.append(check_result)
if not check_result.success:
return CheckResult(False, label, details, check_result.resolution)
if final_check_result is not None:
details.append(final_check_result)
return CheckResult(
final_check_result.success,
label,
details,
final_check_result.resolution,
)
return CheckResult(True, label, details)
def _format_command_name(self, command: Union[commands.Command, str]) -> str:
if not isinstance(command, str):
command = command.qualified_name
return inline(f"{self._original_ctx.clean_prefix}{command}")
def _format_multiple_resolutions(self, resolutions: Iterable[str]) -> str:
parts = [_("To fix this issue, you need to do one of these:")]
for idx, resolution in enumerate(resolutions):
parts.append(f"{string.ascii_lowercase[idx]}) {resolution}")
return "\n".join(parts)
class DetailedGlobalCallOnceChecksMixin(IssueDiagnoserBase):
async def _check_is_author_bot(self) -> CheckResult:
label = _("Check if the command caller is not a bot")
if not self.author.bot:
return CheckResult(True, label)
return CheckResult(
False,
label,
_("The user is a bot which prevents them from running any command."),
_("This cannot be fixed - bots should not be listening to other bots."),
)
async def _check_can_bot_send_messages(self) -> CheckResult:
label = _("Check if the bot can send messages in the given channel")
if self.channel.permissions_for(self.guild.me).send_messages:
return CheckResult(True, label)
return CheckResult(
False,
label,
_("Bot doesn't have permission to send messages in the given channel."),
_(
"To fix this issue, ensure that the permissions setup allows the bot"
" to send messages per Discord's role hierarchy:\n"
"https://support.discord.com/hc/en-us/articles/206141927"
),
)
# While the following 2 checks could show even more precise error message,
# it would require a usage of private attribute rather than the public API
# which increases maintanance burden for not that big of benefit.
async def _check_ignored_issues(self) -> CheckResult:
label = _("Check if the channel and the server aren't set to be ignored")
if await self.bot.ignored_channel_or_guild(self.message):
return CheckResult(True, label)
if self.channel.category is None:
resolution = _(
"To fix this issue, check the list returned by the {command} command"
" and ensure that the {channel} channel and the server aren't a part of that list."
).format(
command=self._format_command_name("ignore list"),
channel=self.channel.mention,
)
else:
resolution = _(
"To fix this issue, check the list returned by the {command} command"
" and ensure that the {channel} channel,"
" the channel category it belongs to ({channel_category}),"
" and the server aren't a part of that list."
).format(
command=self._format_command_name("ignore list"),
channel=self.channel.mention,
channel_category=self.channel.category.mention,
)
return CheckResult(
False,
label,
_("The bot is set to ignore commands in the given channel or this server."),
resolution,
)
async def _get_detailed_global_whitelist_blacklist_result(self, label: str) -> CheckResult:
global_whitelist = await self.bot._whiteblacklist_cache.get_whitelist()
if global_whitelist:
return CheckResult(
False,
label,
_("Global allowlist prevents the user from running this command."),
_(
"To fix this issue, you can either add the user to the allowlist,"
" or clear the allowlist.\n"
"If you want to keep the allowlist, you can run {command_1} which will"
" add {user} to the allowlist.\n"
"If you instead want to clear the allowlist and let all users"
" run commands freely, you can run {command_2} to do that."
).format(
command_1=self._format_command_name(f"allowlist add {self.author.id}"),
user=escape(str(self.author), formatting=True),
command_2=self._format_command_name("allowlist clear"),
),
)
return CheckResult(
False,
label,
_("Global blocklist prevents the user from running this command."),
_(
"To fix this issue, you can either remove the user from the blocklist,"
" or clear the blocklist.\n"
"If you want to keep the blocklist, you can run {command_1} which will"
" remove {user} from the blocklist.\n"
"If you instead want to clear the blocklist and let all users"
" run commands freely, you can run {command_2} to do that."
).format(
command_1=self._format_command_name(f"blocklist remove {self.author.id}"),
user=escape(str(self.author), formatting=True),
command_2=self._format_command_name("blocklist clear"),
),
)
async def _get_detailed_local_whitelist_blacklist_result(self, label: str) -> CheckResult:
# this method skips guild owner check as the earlier checks wouldn't fail
# if the user were guild owner
guild_whitelist = await self.bot._whiteblacklist_cache.get_whitelist(self.guild)
if guild_whitelist:
return CheckResult(
False,
label,
_("Local allowlist prevents the user from running this command."),
_(
"To fix this issue, you can either add the user or one of their roles"
" to the local allowlist, or clear the local allowlist.\n"
"If you want to keep the local allowlist, you can run {command_1} which will"
" add {user} to the local allowlist.\n"
"If you instead want to clear the local allowlist and let all users"
" run commands freely, you can run {command_2} to do that."
).format(
command_1=self._format_command_name(f"localallowlist add {self.author.id}"),
user=escape(str(self.author), formatting=True),
command_2=self._format_command_name("localallowlist clear"),
),
)
details = _("Local blocklist prevents the user from running this command.")
guild_blacklist = await self.bot._whiteblacklist_cache.get_blacklist(self.guild)
ids = {role.id for role in self.author.roles if not role.is_default()}
ids.add(self.author.id)
intersection = ids & guild_blacklist
try:
intersection.remove(self.author.id)
except KeyError:
# author is not part of the blocklist
to_remove = list(intersection)
role_names = [self.guild.get_role(role_id).name for role_id in to_remove]
return CheckResult(
False,
label,
details,
_(
"To fix this issue, you can either remove the user's roles"
" from the local blocklist, or clear the local blocklist.\n"
"If you want to keep the local blocklist, you can run {command_1} which will"
" remove the user's roles ({roles}) from the local blocklist.\n"
"If you instead want to clear the local blocklist and let all users"
" run commands freely, you can run {command_2} to do that."
).format(
command_1=self._format_command_name(
f"localblocklist remove {' '.join(map(str, to_remove))}"
),
roles=humanize_list(role_names),
command_2=self._format_command_name("localblocklist clear"),
),
)
if intersection:
# both author and some of their roles are part of the blocklist
to_remove = list(intersection)
role_names = [self.guild.get_role(role_id).name for role_id in to_remove]
to_remove.append(self.author.id)
return CheckResult(
False,
label,
details,
_(
"To fix this issue, you can either remove the user and their roles"
" from the local blocklist, or clear the local blocklist.\n"
"If you want to keep the local blocklist, you can run {command_1} which will"
" remove {user} and their roles ({roles}) from the local blocklist.\n"
"If you instead want to clear the local blocklist and let all users"
" run commands freely, you can run {command_2} to do that."
).format(
command_1=self._format_command_name(
f"localblocklist remove {' '.join(map(str, to_remove))}"
),
user=escape(str(self.author), formatting=True),
roles=humanize_list(role_names),
command_2=self._format_command_name("localblocklist clear"),
),
)
# only the author is part of the blocklist
return CheckResult(
False,
label,
details,
_(
"To fix this issue, you can either remove the user"
" from the local blocklist, or clear the local blocklist.\n"
"If you want to keep the local blocklist, you can run {command_1} which will"
" remove {user} from the local blocklist.\n"
"If you instead want to clear the local blocklist and let all users"
" run commands freely, you can run {command_2} to do that."
).format(
command_1=self._format_command_name(f"localblocklist remove {self.author.id}"),
user=escape(str(self.author), formatting=True),
command_2=self._format_command_name("localblocklist clear"),
),
)
async def _check_whitelist_blacklist_issues(self) -> CheckResult:
label = _("Allowlist and blocklist checks")
if await self.bot.allowed_by_whitelist_blacklist(self.author):
return CheckResult(True, label)
is_global = not await self.bot.allowed_by_whitelist_blacklist(who_id=self.author.id)
if is_global:
return await self._get_detailed_global_whitelist_blacklist_result(label)
return await self._get_detailed_local_whitelist_blacklist_result(label)
class DetailedCommandChecksMixin(IssueDiagnoserBase):
def _command_error_handler(
self,
msg: str,
label: str,
failed_with_message: str,
failed_without_message: str,
) -> CheckResult:
command = self.ctx.command
details = (
failed_with_message.format(command=self._format_command_name(command), message=msg)
if msg
else failed_without_message.format(command=self._format_command_name(command))
)
return CheckResult(
False,
label,
details,
)
async def _check_dpy_can_run(self) -> CheckResult:
label = _("Global, cog and command checks")
command = self.ctx.command
try:
if await super(commands.Command, command).can_run(self.ctx):
return CheckResult(True, label)
except commands.DisabledCommand:
details = (
_("The given command is disabled in this guild.")
if command is self.command
else _("One of the parents of the given command is disabled globally.")
)
return CheckResult(
False,
label,
details,
_(
"To fix this issue, you can run {command}"
" which will enable the {affected_command} command in this guild."
).format(
command=self._format_command_name(f"command enable guild {command}"),
affected_command=self._format_command_name(command),
),
)
except commands.CommandError:
# we want to narrow this down to specific type of checks (bot/cog/command)
pass
return await self._check_until_fail(
label,
(
self._check_dpy_can_run_bot,
self._check_dpy_can_run_cog,
self._check_dpy_can_run_command,
),
final_check_result=CheckResult(
False,
_("Other issues related to the checks"),
_(
"There's an issue related to the checks for {command}"
" but we're not able to determine the exact cause."
).format(command=self._format_command_name(command)),
_(
"To fix this issue, a manual review of"
" the global, cog and command checks is required."
),
),
)
async def _check_dpy_can_run_bot(self) -> CheckResult:
label = _("Global checks")
msg = ""
try:
if await self.bot.can_run(self.ctx):
return CheckResult(True, label)
except commands.CommandError as e:
msg = str(e)
return self._command_error_handler(
msg,
label,
_(
"One of the global checks for the command {command} failed with a message:\n"
"{message}"
),
_("One of the global checks for the command {command} failed without a message."),
)
async def _check_dpy_can_run_cog(self) -> CheckResult:
label = _("Cog check")
cog = self.ctx.command.cog
if cog is None:
return CheckResult(True, label)
local_check = commands.Cog._get_overridden_method(cog.cog_check)
if local_check is None:
return CheckResult(True, label)
msg = ""
try:
if await discord.utils.maybe_coroutine(local_check, self.ctx):
return CheckResult(True, label)
except commands.CommandError as e:
msg = str(e)
return self._command_error_handler(
msg,
label,
_("The cog check for the command {command} failed with a message:\n{message}"),
_("The cog check for the command {command} failed without a message."),
)
async def _check_dpy_can_run_command(self) -> CheckResult:
label = _("Command checks")
predicates = self.ctx.command.checks
if not predicates:
return CheckResult(True, label)
msg = ""
try:
if await discord.utils.async_all(predicate(self.ctx) for predicate in predicates):
return CheckResult(True, label)
except commands.CommandError as e:
msg = str(e)
return self._command_error_handler(
msg,
label,
_(
"One of the command checks for the command {command} failed with a message:\n"
"{message}"
),
_("One of the command checks for the command {command} failed without a message."),
)
async def _check_requires_command(self) -> CheckResult:
return await self._check_requires(_("Permissions verification"), self.ctx.command)
async def _check_requires_cog(self) -> CheckResult:
label = _("Permissions verification for {cog} cog").format(
cog=inline(self.ctx.cog.qualified_name)
)
if self.ctx.cog is None:
return CheckResult(True, label)
return await self._check_requires(label, self.ctx.cog)
async def _check_requires(
self, label: str, cog_or_command: commands.CogCommandMixin
) -> CheckResult:
original_perm_state = self.ctx.permission_state
try:
allowed = await cog_or_command.requires.verify(self.ctx)
except commands.DisabledCommand:
return CheckResult(
False,
label,
_("The cog of the given command is disabled in this guild."),
_(
"To fix this issue, you can run {command}"
" which will enable the {affected_cog} cog in this guild."
).format(
command=self._format_command_name(
f"command enablecog {self.ctx.cog.qualified_name}"
),
affected_cog=inline(self.ctx.cog.qualified_name),
),
)
except commands.BotMissingPermissions as e:
# No, go away, "some" can refer to a single permission so plurals are just fine here!
# Seriously. They are. Don't even question it.
details = (
_(
"Bot is missing some of the channel permissions ({permissions})"
" required by the {cog} cog."
).format(
permissions=format_perms_list(e.missing),
cog=inline(cog_or_command.qualified_name),
)
if cog_or_command is self.ctx.cog
else _(
"Bot is missing some of the channel permissions ({permissions})"
" required by the {command} command."
).format(
permissions=format_perms_list(e.missing),
command=self._format_command_name(cog_or_command),
)
)
return CheckResult(
False,
label,
details,
_(
"To fix this issue, grant the required permissions to the bot"
" through role settings or channel overrides."
),
)
if allowed:
return CheckResult(True, label)
self.ctx.permission_state = original_perm_state
return await self._check_until_fail(
label,
(
partial(self._check_requires_bot_owner, cog_or_command),
partial(self._check_requires_permission_hooks, cog_or_command),
partial(self._check_requires_permission_rules, cog_or_command),
),
# unless there's some bug here, we should probably never run into this
final_check_result=CheckResult(
False,
_("Other issues related to the permissions."),
_(
"Fatal error: There's an issue related to the permissions for the"
" {cog} cog but we're not able to determine the exact cause."
)
if cog_or_command is self.ctx.cog
else _(
"Fatal error: There's an issue related to the permissions for the"
" {command} command but we're not able to determine the exact cause."
),
_("This is an unexpected error, please report it on Red's issue tracker."),
),
)
async def _check_requires_bot_owner(
self, cog_or_command: commands.CogCommandMixin
) -> CheckResult:
label = _("Ensure that the command is not bot owner only")
if cog_or_command.requires.privilege_level is not commands.PrivilegeLevel.BOT_OWNER:
return CheckResult(True, label)
# we don't need to check whether the user is bot owner
# as call to `verify()` would already succeed if that were the case
return CheckResult(
False,
label,
_("The command is bot owner only and the given user is not a bot owner."),
_("This cannot be fixed - regular users cannot run bot owner only commands."),
)
async def _check_requires_permission_hooks(
self, cog_or_command: commands.CogCommandMixin
) -> CheckResult:
label = _("Permission hooks")
result = await self.bot.verify_permissions_hooks(self.ctx)
if result is None:
return CheckResult(True, label)
if result is True:
# this situation is abnormal as in this situation,
# call to `verify()` would already succeed and we wouldn't get to this point
return CheckResult(
False,
label,
_("Fatal error: the result of permission hooks is inconsistent."),
_("To fix this issue, a manual review of the installed cogs is required."),
)
return CheckResult(
False,
label,
_("The access has been denied by one of the bot's permissions hooks."),
_("To fix this issue, a manual review of the installed cogs is required."),
)
# Pinpointing a specific rule that denied the access is possible but it was considered
# to require more effort than it is worth it for the little benefit it gives.
# If this becomes a significant pain point for the users, this might get reconsidered.
async def _check_requires_permission_rules(
self, cog_or_command: commands.CogCommandMixin
) -> CheckResult:
label = _("User's discord permissions, privilege level and rules from Permissions cog")
should_invoke, next_state = cog_or_command.requires._get_transitioned_state(self.ctx)
if should_invoke is None:
return await self._check_requires_verify_user(label, cog_or_command)
elif isinstance(next_state, dict):
would_invoke = self._get_would_invoke(self.ctx)
if would_invoke is None:
return await self._check_requires_verify_user(label, cog_or_command)
next_state = next_state[would_invoke]
self.ctx.permission_state = next_state
if should_invoke:
return CheckResult(True, label)
return CheckResult(
False,
label,
_(
"The access has been denied due to the rules set for the {cog} cog"
" with Permissions cog."
).format(cog=inline(cog_or_command.qualified_name))
if cog_or_command is self.ctx.cog
else _(
"The access has been denied due to the rules set for the {command} command"
" with Permissions cog."
).format(command=self._format_command_name(cog_or_command)),
_("To fix the issue, a manual review of the rules is required."),
)
async def _check_requires_verify_user(
self, label: str, cog_or_command: commands.CogCommandMixin
) -> CheckResult:
return await self._check_until_fail(
label,
(
partial(self._check_requires_permission_checks, cog_or_command),
partial(self._check_requires_user_perms_and_privilege_level, cog_or_command),
),
final_check_result=CheckResult(
False,
_("Other issues related to the permissions."),
_(
"There's an issue related to the permissions of {cog} cog"
" but we're not able to determine the exact cause."
).format(cog=inline(cog_or_command.qualified_name))
if cog_or_command is self.ctx.cog
else _(
"There's an issue related to the permissions of {command} command"
" but we're not able to determine the exact cause."
).format(command=self._format_command_name(cog_or_command)),
_("To fix this issue, a manual review of the command is required."),
),
)
async def _check_requires_permission_checks(
self, cog_or_command: commands.CogCommandMixin
) -> CheckResult:
label = _("Permission checks")
if await cog_or_command.requires._verify_checks(self.ctx):
return CheckResult(True, label)
details = (
_("The access has been denied by one of the permissions checks of {cog} cog.").format(
cog=inline(cog_or_command.qualified_name)
)
if cog_or_command is self.ctx.cog
else _(
"The access has been denied by one of the permission checks of {command} command."
).format(command=self._format_command_name(cog_or_command))
)
return CheckResult(
False,
label,
details,
_("To fix this issue, a manual review of the permission checks is required."),
)
async def _check_requires_user_perms_and_privilege_level(
self, cog_or_command: commands.CogCommandMixin
) -> CheckResult:
label = _("User's discord permissions and privilege level")
requires = cog_or_command.requires
if await requires._verify_user(self.ctx):
print("HI!")
return CheckResult(True, label)
resolutions = []
if requires.user_perms is not None:
permissions = format_perms_list(requires.user_perms)
resolutions.append(
_(
"grant the required permissions to the user through role settings"
" or channel overrides"
)
)
details = (
_(
"The user is missing some of the channel permissions ({permissions})"
" required by the {cog} cog."
).format(permissions=permissions, cog=inline(cog_or_command.qualified_name))
if cog_or_command is self.ctx.cog
else _(
"The user is missing some of the channel permissions ({permissions})"
" required by the {command} command."
).format(
permissions=permissions, command=self._format_command_name(cog_or_command)
)
)
if requires.privilege_level is not None:
if requires.privilege_level is commands.PrivilegeLevel.GUILD_OWNER:
privilege_level = _("the guild owner")
else:
if requires.privilege_level is commands.PrivilegeLevel.MOD:
privilege_level = _("the mod role")
elif requires.privilege_level is commands.PrivilegeLevel.ADMIN:
privilege_level = _("the admin role")
else:
raise RuntimeError("Ran into unexpected privilege level.")
resolutions.append(_("assign appropriate role to the user"))
details = (
_(
"The user is missing the privilege level ({privilege_level})"
" required by the {cog} cog."
).format(
privilege_level=privilege_level, cog=inline(cog_or_command.qualified_name)
)
if cog_or_command is self.ctx.cog
else _(
"The user is missing the privilege level ({privilege_level})"
" required by the {command} command."
).format(
privilege_level=privilege_level,
command=self._format_command_name(cog_or_command),
)
)
if not resolutions:
# Neither `user_perms` nor `privilege_level` are responsible for the issue.
return CheckResult(True, label)
resolutions.append(_("add appropriate rule in the Permissions cog"))
if requires.user_perms is not None and requires.privilege_level is not None:
details = (
_(
"The user has neither the channel permissions ({permissions}) nor"
" the privilege level ({privilege_level}) required by the {cog} cog."
).format(
permissions=permissions,
privilege_level=privilege_level,
cog=inline(cog_or_command.qualified_name),
)
if cog_or_command is self.ctx.cog
else _(
"The user has neither the channel permissions ({permissions}) nor"
" the privilege level ({privilege_level}) required by the {command} command."
).format(
permissions=permissions,
privilege_level=privilege_level,
command=self._format_command_name(cog_or_command),
)
)
return CheckResult(
False,
label,
details,
self._format_multiple_resolutions(resolutions),
)
async def _check_dpy_checks_and_requires(self, command: commands.Command) -> CheckResult:
label = _("Checks and permissions verification for the command {command}").format(
command=self._format_command_name(command)
)
self.ctx.command = command
original_perm_state = self.ctx.permission_state
try:
can_run = await command.can_run(self.ctx, change_permission_state=True)
except commands.CommandError:
can_run = False
if can_run:
return CheckResult(True, label)
self.ctx.permission_state = original_perm_state
return await self._check_until_fail(
label,
(
self._check_dpy_can_run,
self._check_requires_command,
),
final_check_result=CheckResult(
False,
_("Other command checks"),
_("The given command is failing one of the required checks."),
_("To fix this issue, a manual review of the command's checks is required."),
),
)
class RootDiagnosersMixin(
DetailedGlobalCallOnceChecksMixin,
DetailedCommandChecksMixin,
IssueDiagnoserBase,
):
async def _check_global_call_once_checks_issues(self) -> CheckResult:
label = _("Global 'call once' checks")
# To avoid running core's global checks twice, we just run them all regularly
# and if it turns out that invokation would end here, we go back and check each of
# core's global check individually to give more precise error message.
try:
can_run = await self.bot.can_run(self.ctx, call_once=True)
except commands.CommandError:
pass
else:
if can_run:
return CheckResult(True, label)
return await self._check_until_fail(
label,
(
self._check_is_author_bot,
self._check_can_bot_send_messages,
self._check_ignored_issues,
self._check_whitelist_blacklist_issues,
),
final_check_result=CheckResult(
False,
_("Other global 'call once' checks"),
_(
"One of the global 'call once' checks implemented by a 3rd-party cog"
" prevents this command from being ran."
),
_("To fix this issue, a manual review of the installed cogs is required."),
),
)
async def _check_disabled_command_issues(self) -> CheckResult:
label = _("Check if the command is disabled")
command = self.command
for parent in reversed(command.parents):
if parent.enabled:
continue
return CheckResult(
False,
label,
_("One of the parents of the given command is disabled globally."),
_(
"To fix this issue, you can run {command}"
" which will enable the {affected_command} command globally."
).format(
command=self._format_command_name(f"command enable global {parent}"),
affected_command=self._format_command_name(parent),
),
)
if not command.enabled:
return CheckResult(
False,
label,
_("The given command is disabled globally."),
_(
"To fix this issue, you can run {command}"
" which will enable the {affected_command} command globally."
).format(
command=self._format_command_name(f"command enable global {command}"),
affected_command=self._format_command_name(command),
),
)
return CheckResult(True, label)
async def _check_can_run_issues(self) -> CheckResult:
label = _("Checks and permissions verification")
ctx = self.ctx
try:
can_run = await self.command.can_run(ctx, check_all_parents=True)
except commands.CommandError:
# we want to get more specific error by narrowing down the scope,
# so we just ignore handling this here
#
# NOTE: it might be worth storing this information in case we get to
# `final_check_result`, although that's not very likely
# If something like this gets implemented here in the future,
# similar exception handlers further down the line could do that as well.
pass
else:
if can_run:
return CheckResult(True, label)
ctx.permission_state = commands.PermState.NORMAL
ctx.command = self.command.root_parent or self.command
# slight discrepancy here - we're doing cog-level verify before top-level can_run
return await self._check_until_fail(
label,
itertools.chain(
(self._check_requires_cog,),
(
partial(self._check_dpy_checks_and_requires, command)
for command in itertools.chain(reversed(self.command.parents), (self.command,))
),
),
final_check_result=CheckResult(
False,
_("Other command checks"),
_("The given command is failing one of the required checks."),
_("To fix this issue, a manual review of the command's checks is required."),
),
)
class IssueDiagnoser(RootDiagnosersMixin, IssueDiagnoserBase):
def _get_message_from_check_result(
self, result: CheckResult, *, prefix: str = ""
) -> List[str]:
lines = []
if not result.details:
return []
if isinstance(result.details, str):
return [result.details]
for idx, subresult in enumerate(result.details, start=1):
status = (
_("Passed") + " \N{WHITE HEAVY CHECK MARK}"
if subresult.success
else _("Failed") + " \N{NO ENTRY}\N{VARIATION SELECTOR-16}"
)
lines.append(f"{prefix}{idx}. {subresult.label}: {status}")
lines.extend(
self._get_message_from_check_result(subresult, prefix=f" {prefix}{idx}.")
)
return lines
def _get_details_from_check_result(self, result: CheckResult) -> str:
if not result.details:
return ""
if isinstance(result.details, str):
return result.details
return self._get_details_from_check_result(result.details[-1])
async def diagnose(self) -> str:
await self._prepare()
lines = []
result = await self._check_until_fail(
"",
(
self._check_global_call_once_checks_issues,
self._check_disabled_command_issues,
self._check_can_run_issues,
),
)
if result.success:
lines.append(
_(
"All checks passed and no issues were detected."
" Make sure that the given parameters correspond to"
" the channel, user, and command name that have been problematic.\n\n"
"If you still can't find the issue, it is likely that one of the 3rd-party"
" cogs you're using adds a global or cog local before invoke hook that"
" prevents the command from getting invoked as this can't be diagnosed"
" with this tool."
)
)
else:
lines.append(_("The bot has been able to identify the issue."))
details = self._get_details_from_check_result(result)
if details:
lines.append(bold(_("Detected issue: ")) + details)
if result.resolution:
lines.append(bold(_("Solution: ")) + result.resolution)
lines.append(_("\nHere's a detailed report in case you need it:"))
lines.append(">>> " + bold(_("Channel: ")) + self.channel.mention)
lines.append(bold(_("Command caller: ")) + escape(str(self.author), formatting=True))
lines.append(bold(_("Command: ")) + self._format_command_name(self.command))
lines.append(bold(_("Tests that have been ran:")))
lines.extend(self._get_message_from_check_result(result))
return "\n".join(lines)

36
redbot/core/commands/requires.py
View File

@ -521,26 +521,15 @@ class Requires:
raise BotMissingPermissions(missing=self._missing_perms(self.bot_perms, bot_perms))
async def _transition_state(self, ctx: "Context") -> bool:
prev_state = ctx.permission_state
cur_state = self._get_rule_from_ctx(ctx)
should_invoke, next_state = transition_permstate_to(prev_state, cur_state)
should_invoke, next_state = self._get_transitioned_state(ctx)
if should_invoke is None:
# NORMAL invocation, we simply follow standard procedure
should_invoke = await self._verify_user(ctx)
elif isinstance(next_state, dict):
# NORMAL to PASSIVE_ALLOW; should we proceed as normal or transition?
# We must check what would happen normally, if no explicit rules were set.
default_rule = PermState.NORMAL
if ctx.guild is not None:
default_rule = self.get_rule(self.DEFAULT, guild_id=ctx.guild.id)
if default_rule is PermState.NORMAL:
default_rule = self.get_rule(self.DEFAULT, self.GLOBAL)
if default_rule == PermState.ACTIVE_DENY:
would_invoke = False
elif default_rule == PermState.ACTIVE_ALLOW:
would_invoke = True
else:
would_invoke = self._get_would_invoke(ctx)
if would_invoke is None:
would_invoke = await self._verify_user(ctx)
next_state = next_state[would_invoke]
@ -548,6 +537,25 @@ class Requires:
ctx.permission_state = next_state
return should_invoke
def _get_transitioned_state(self, ctx: "Context") -> TransitionResult:
prev_state = ctx.permission_state
cur_state = self._get_rule_from_ctx(ctx)
return transition_permstate_to(prev_state, cur_state)
def _get_would_invoke(self, ctx: "Context") -> Optional[bool]:
default_rule = PermState.NORMAL
if ctx.guild is not None:
default_rule = self.get_rule(self.DEFAULT, guild_id=ctx.guild.id)
if default_rule is PermState.NORMAL:
default_rule = self.get_rule(self.DEFAULT, self.GLOBAL)
if default_rule == PermState.ACTIVE_DENY:
return False
elif default_rule == PermState.ACTIVE_ALLOW:
return True
else:
return None
async def _verify_user(self, ctx: "Context") -> bool:
checks_pass = await self._verify_checks(ctx)
if checks_pass is False:

61
redbot/core/core_commands.py
View File

@ -36,6 +36,7 @@ from . import (
errors,
i18n,
)
from ._diagnoser import IssueDiagnoser
from .utils import AsyncIter
from .utils._internal_utils import fetch_latest_red_version_info
from .utils.predicates import MessagePredicate
@ -3680,6 +3681,66 @@ class Core(commands.commands._RuleDropper, commands.Cog, CoreLogic):
await ctx.send("".join(response))
# You may ask why this command is owner-only,
# cause after all it could be quite useful to guild owners!
# Truth to be told, that would require us to make some part of this
# more end-user friendly rather than just bot owner friendly - terms like
# 'global call once checks' are not of any use to someone who isn't bot owner.
@commands.is_owner()
@commands.command()
async def diagnoseissues(
self,
ctx: commands.Context,
channel: Optional[discord.TextChannel],
member: Union[discord.Member, discord.User],
*,
command_name: str,
) -> None:
"""
Diagnose issues with the command checks with ease!
If you want to diagnose the command from a text channel in a different server,
you can do so by using the command in DMs.
**Example:**
- `[p]diagnoseissues #general @Slime ban` - Diagnose why @Slime can't use `[p]ban` in #general channel.
**Arguments:**
- `[channel]` - The text channel that the command should be tested for. Defaults to the current channel.
- `<member>` - The member that should be considered as the command caller.
- `<command_name>` - The name of the command to test.
"""
if channel is None:
channel = ctx.channel
if not isinstance(channel, discord.TextChannel):
await ctx.send(_("The channel needs to be passed when using this command in DMs."))
return
command = self.bot.get_command(command_name)
if command is None:
await ctx.send("Command not found!")
return
# This is done to allow the bot owner to diagnose a command
# while not being a part of the server.
if isinstance(member, discord.User):
maybe_member = channel.guild.get_member(member.id)
if maybe_member is None:
await ctx.send(_("The given user is not a member of the diagnosed server."))
return
member = maybe_member
if not channel.permissions_for(member).send_messages:
# Let's make Flame happy here
await ctx.send(
_(
"Don't try to fool me, the given member can't access the {channel} channel!"
).format(channel=channel.mention)
)
return
issue_diagnoser = IssueDiagnoser(self.bot, ctx, channel, member, command)
await ctx.send(await issue_diagnoser.diagnose())
@commands.group(aliases=["whitelist"])
@checks.is_owner()
async def allowlist(self, ctx: commands.Context):