mirror of
https://github.com/pacnpal/Pac-cogs.git
synced 2025-12-20 02:41:06 -05:00
Improve command handling with better error handling, type hints, and organization
- Add proper error handling with CommandError - Add proper error context support - Add better command organization - Add proper type hints - Add proper docstrings - Add better error recovery - Add proper error context creation - Add better response handling - Add proper validation - Add better logging - Add proper response types - Add better error messages - Add proper status reporting - Add new status commands
This commit is contained in:
pacnpal
@@ -4,253 +4,700 @@ import discord
|
||||
from redbot.core.commands import Context, hybrid_group, guild_only, admin_or_permissions
|
||||
from discord import app_commands
|
||||
import logging
|
||||
from ..response_handler import handle_response
|
||||
from typing import Optional, Any, Dict, TypedDict, Union, List
|
||||
from enum import Enum, auto
|
||||
|
||||
from ..response_handler import handle_response, ResponseType
|
||||
from ...utils.exceptions import (
|
||||
CommandError,
|
||||
ErrorContext,
|
||||
ErrorSeverity
|
||||
)
|
||||
from ...core.settings import VideoFormat, VideoQuality
|
||||
|
||||
logger = logging.getLogger("VideoArchiver")
|
||||
|
||||
def setup_settings_commands(cog):
|
||||
"""Set up settings commands for the cog"""
|
||||
class SettingCategory(Enum):
|
||||
"""Setting categories"""
|
||||
CHANNELS = auto()
|
||||
VIDEO = auto()
|
||||
MESSAGES = auto()
|
||||
PERFORMANCE = auto()
|
||||
|
||||
@cog.hybrid_group(name="settings", fallback="show")
|
||||
class SettingValidation(TypedDict):
|
||||
"""Type definition for setting validation"""
|
||||
valid: bool
|
||||
error: Optional[str]
|
||||
details: Dict[str, Any]
|
||||
|
||||
class SettingUpdate(TypedDict):
|
||||
"""Type definition for setting update"""
|
||||
setting: str
|
||||
old_value: Any
|
||||
new_value: Any
|
||||
category: SettingCategory
|
||||
|
||||
async def validate_setting(
|
||||
category: SettingCategory,
|
||||
setting: str,
|
||||
value: Any
|
||||
) -> SettingValidation:
|
||||
"""
|
||||
Validate a setting value.
|
||||
|
||||
Args:
|
||||
category: Setting category
|
||||
setting: Setting name
|
||||
value: Value to validate
|
||||
|
||||
Returns:
|
||||
Validation result
|
||||
"""
|
||||
validation = SettingValidation(
|
||||
valid=True,
|
||||
error=None,
|
||||
details={"category": category.name, "setting": setting, "value": value}
|
||||
)
|
||||
|
||||
try:
|
||||
if category == SettingCategory.VIDEO:
|
||||
if setting == "format":
|
||||
if value not in [f.value for f in VideoFormat]:
|
||||
validation.update({
|
||||
"valid": False,
|
||||
"error": f"Invalid format. Must be one of: {', '.join(f.value for f in VideoFormat)}"
|
||||
})
|
||||
elif setting == "quality":
|
||||
if not 144 <= value <= 4320:
|
||||
validation.update({
|
||||
"valid": False,
|
||||
"error": "Quality must be between 144 and 4320"
|
||||
})
|
||||
elif setting == "max_file_size":
|
||||
if not 1 <= value <= 100:
|
||||
validation.update({
|
||||
"valid": False,
|
||||
"error": "Size must be between 1 and 100 MB"
|
||||
})
|
||||
|
||||
elif category == SettingCategory.MESSAGES:
|
||||
if setting == "duration":
|
||||
if not 0 <= value <= 168:
|
||||
validation.update({
|
||||
"valid": False,
|
||||
"error": "Duration must be between 0 and 168 hours (1 week)"
|
||||
})
|
||||
elif setting == "template":
|
||||
placeholders = ["{author}", "{channel}", "{original_message}"]
|
||||
if not any(ph in value for ph in placeholders):
|
||||
validation.update({
|
||||
"valid": False,
|
||||
"error": f"Template must include at least one placeholder: {', '.join(placeholders)}"
|
||||
})
|
||||
|
||||
elif category == SettingCategory.PERFORMANCE:
|
||||
if setting == "concurrent_downloads":
|
||||
if not 1 <= value <= 5:
|
||||
validation.update({
|
||||
"valid": False,
|
||||
"error": "Concurrent downloads must be between 1 and 5"
|
||||
})
|
||||
|
||||
except Exception as e:
|
||||
validation.update({
|
||||
"valid": False,
|
||||
"error": f"Validation error: {str(e)}"
|
||||
})
|
||||
|
||||
return validation
|
||||
|
||||
def setup_settings_commands(cog: Any) -> Any:
|
||||
"""
|
||||
Set up settings commands for the cog.
|
||||
|
||||
Args:
|
||||
cog: VideoArchiver cog instance
|
||||
|
||||
Returns:
|
||||
Main settings command group
|
||||
"""
|
||||
|
||||
@hybrid_group(name="settings", fallback="show")
|
||||
@guild_only()
|
||||
async def settings(ctx: Context):
|
||||
async def settings(ctx: Context) -> None:
|
||||
"""Show current archiver settings."""
|
||||
try:
|
||||
# Check if config manager is ready
|
||||
if not cog.config_manager:
|
||||
raise CommandError(
|
||||
"Configuration system is not ready",
|
||||
context=ErrorContext(
|
||||
"SettingsCommands",
|
||||
"show_settings",
|
||||
{"guild_id": ctx.guild.id},
|
||||
ErrorSeverity.HIGH
|
||||
)
|
||||
)
|
||||
|
||||
# Defer the response immediately for slash commands
|
||||
if hasattr(ctx, "interaction") and ctx.interaction:
|
||||
await ctx.defer()
|
||||
|
||||
embed = await cog.config_manager.format_settings_embed(ctx.guild)
|
||||
await handle_response(ctx, embed=embed)
|
||||
except Exception as e:
|
||||
logger.error(f"Error showing settings: {e}")
|
||||
await handle_response(
|
||||
ctx, "An error occurred while showing settings."
|
||||
ctx,
|
||||
embed=embed,
|
||||
response_type=ResponseType.INFO
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
error = f"Failed to show settings: {str(e)}"
|
||||
logger.error(error, exc_info=True)
|
||||
raise CommandError(
|
||||
error,
|
||||
context=ErrorContext(
|
||||
"SettingsCommands",
|
||||
"show_settings",
|
||||
{"guild_id": ctx.guild.id},
|
||||
ErrorSeverity.MEDIUM
|
||||
)
|
||||
)
|
||||
|
||||
@settings.command(name="setchannel")
|
||||
@guild_only()
|
||||
@admin_or_permissions(administrator=True)
|
||||
@app_commands.describe(channel="The channel where archived videos will be stored")
|
||||
async def set_archive_channel(ctx: Context, channel: discord.TextChannel):
|
||||
async def set_archive_channel(ctx: Context, channel: discord.TextChannel) -> None:
|
||||
"""Set the channel where archived videos will be stored."""
|
||||
try:
|
||||
# Check if config manager is ready
|
||||
if not cog.config_manager:
|
||||
raise CommandError(
|
||||
"Configuration system is not ready",
|
||||
context=ErrorContext(
|
||||
"SettingsCommands",
|
||||
"set_archive_channel",
|
||||
{"guild_id": ctx.guild.id},
|
||||
ErrorSeverity.HIGH
|
||||
)
|
||||
)
|
||||
|
||||
# Defer the response immediately for slash commands
|
||||
if hasattr(ctx, "interaction") and ctx.interaction:
|
||||
await ctx.defer()
|
||||
|
||||
# Check channel permissions
|
||||
bot_member = ctx.guild.me
|
||||
required_perms = discord.Permissions(
|
||||
send_messages=True,
|
||||
embed_links=True,
|
||||
attach_files=True,
|
||||
read_message_history=True
|
||||
)
|
||||
channel_perms = channel.permissions_for(bot_member)
|
||||
if not all(getattr(channel_perms, perm) for perm in required_perms):
|
||||
raise CommandError(
|
||||
"Missing required permissions in target channel",
|
||||
context=ErrorContext(
|
||||
"SettingsCommands",
|
||||
"set_archive_channel",
|
||||
{
|
||||
"guild_id": ctx.guild.id,
|
||||
"channel_id": channel.id,
|
||||
"missing_perms": [
|
||||
perm for perm in required_perms
|
||||
if not getattr(channel_perms, perm)
|
||||
]
|
||||
},
|
||||
ErrorSeverity.MEDIUM
|
||||
)
|
||||
)
|
||||
|
||||
await cog.config_manager.update_setting(
|
||||
ctx.guild.id, "archive_channel", channel.id
|
||||
ctx.guild.id,
|
||||
"archive_channel",
|
||||
channel.id
|
||||
)
|
||||
await handle_response(
|
||||
ctx, f"Archive channel has been set to {channel.mention}."
|
||||
ctx,
|
||||
f"Archive channel has been set to {channel.mention}.",
|
||||
response_type=ResponseType.SUCCESS
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Error setting archive channel: {e}")
|
||||
await handle_response(
|
||||
ctx, "An error occurred while setting the archive channel."
|
||||
error = f"Failed to set archive channel: {str(e)}"
|
||||
logger.error(error, exc_info=True)
|
||||
raise CommandError(
|
||||
error,
|
||||
context=ErrorContext(
|
||||
"SettingsCommands",
|
||||
"set_archive_channel",
|
||||
{"guild_id": ctx.guild.id, "channel_id": channel.id},
|
||||
ErrorSeverity.HIGH
|
||||
)
|
||||
)
|
||||
|
||||
@settings.command(name="setlog")
|
||||
@guild_only()
|
||||
@admin_or_permissions(administrator=True)
|
||||
@app_commands.describe(channel="The channel where log messages will be sent")
|
||||
async def set_log_channel(ctx: Context, channel: discord.TextChannel):
|
||||
async def set_log_channel(ctx: Context, channel: discord.TextChannel) -> None:
|
||||
"""Set the channel where log messages will be sent."""
|
||||
try:
|
||||
# Check if config manager is ready
|
||||
if not cog.config_manager:
|
||||
raise CommandError(
|
||||
"Configuration system is not ready",
|
||||
context=ErrorContext(
|
||||
"SettingsCommands",
|
||||
"set_log_channel",
|
||||
{"guild_id": ctx.guild.id},
|
||||
ErrorSeverity.HIGH
|
||||
)
|
||||
)
|
||||
|
||||
# Defer the response immediately for slash commands
|
||||
if hasattr(ctx, "interaction") and ctx.interaction:
|
||||
await ctx.defer()
|
||||
|
||||
# Check channel permissions
|
||||
bot_member = ctx.guild.me
|
||||
required_perms = discord.Permissions(
|
||||
send_messages=True,
|
||||
embed_links=True,
|
||||
read_message_history=True
|
||||
)
|
||||
channel_perms = channel.permissions_for(bot_member)
|
||||
if not all(getattr(channel_perms, perm) for perm in required_perms):
|
||||
raise CommandError(
|
||||
"Missing required permissions in target channel",
|
||||
context=ErrorContext(
|
||||
"SettingsCommands",
|
||||
"set_log_channel",
|
||||
{
|
||||
"guild_id": ctx.guild.id,
|
||||
"channel_id": channel.id,
|
||||
"missing_perms": [
|
||||
perm for perm in required_perms
|
||||
if not getattr(channel_perms, perm)
|
||||
]
|
||||
},
|
||||
ErrorSeverity.MEDIUM
|
||||
)
|
||||
)
|
||||
|
||||
await cog.config_manager.update_setting(
|
||||
ctx.guild.id, "log_channel", channel.id
|
||||
ctx.guild.id,
|
||||
"log_channel",
|
||||
channel.id
|
||||
)
|
||||
await handle_response(
|
||||
ctx, f"Log channel has been set to {channel.mention}."
|
||||
ctx,
|
||||
f"Log channel has been set to {channel.mention}.",
|
||||
response_type=ResponseType.SUCCESS
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Error setting log channel: {e}")
|
||||
await handle_response(
|
||||
ctx, "An error occurred while setting the log channel."
|
||||
error = f"Failed to set log channel: {str(e)}"
|
||||
logger.error(error, exc_info=True)
|
||||
raise CommandError(
|
||||
error,
|
||||
context=ErrorContext(
|
||||
"SettingsCommands",
|
||||
"set_log_channel",
|
||||
{"guild_id": ctx.guild.id, "channel_id": channel.id},
|
||||
ErrorSeverity.HIGH
|
||||
)
|
||||
)
|
||||
|
||||
@settings.command(name="addchannel")
|
||||
@guild_only()
|
||||
@admin_or_permissions(administrator=True)
|
||||
@app_commands.describe(channel="The channel to monitor for videos")
|
||||
async def add_enabled_channel(ctx: Context, channel: discord.TextChannel):
|
||||
async def add_enabled_channel(ctx: Context, channel: discord.TextChannel) -> None:
|
||||
"""Add a channel to monitor for videos."""
|
||||
try:
|
||||
# Check if config manager is ready
|
||||
if not cog.config_manager:
|
||||
raise CommandError(
|
||||
"Configuration system is not ready",
|
||||
context=ErrorContext(
|
||||
"SettingsCommands",
|
||||
"add_enabled_channel",
|
||||
{"guild_id": ctx.guild.id},
|
||||
ErrorSeverity.HIGH
|
||||
)
|
||||
)
|
||||
|
||||
# Defer the response immediately for slash commands
|
||||
if hasattr(ctx, "interaction") and ctx.interaction:
|
||||
await ctx.defer()
|
||||
|
||||
# Check channel permissions
|
||||
bot_member = ctx.guild.me
|
||||
required_perms = discord.Permissions(
|
||||
read_messages=True,
|
||||
read_message_history=True
|
||||
)
|
||||
channel_perms = channel.permissions_for(bot_member)
|
||||
if not all(getattr(channel_perms, perm) for perm in required_perms):
|
||||
raise CommandError(
|
||||
"Missing required permissions in target channel",
|
||||
context=ErrorContext(
|
||||
"SettingsCommands",
|
||||
"add_enabled_channel",
|
||||
{
|
||||
"guild_id": ctx.guild.id,
|
||||
"channel_id": channel.id,
|
||||
"missing_perms": [
|
||||
perm for perm in required_perms
|
||||
if not getattr(channel_perms, perm)
|
||||
]
|
||||
},
|
||||
ErrorSeverity.MEDIUM
|
||||
)
|
||||
)
|
||||
|
||||
enabled_channels = await cog.config_manager.get_setting(
|
||||
ctx.guild.id, "enabled_channels"
|
||||
ctx.guild.id,
|
||||
"enabled_channels"
|
||||
)
|
||||
if channel.id in enabled_channels:
|
||||
await handle_response(
|
||||
ctx, f"{channel.mention} is already being monitored."
|
||||
ctx,
|
||||
f"{channel.mention} is already being monitored.",
|
||||
response_type=ResponseType.WARNING
|
||||
)
|
||||
return
|
||||
|
||||
enabled_channels.append(channel.id)
|
||||
await cog.config_manager.update_setting(
|
||||
ctx.guild.id, "enabled_channels", enabled_channels
|
||||
ctx.guild.id,
|
||||
"enabled_channels",
|
||||
enabled_channels
|
||||
)
|
||||
await handle_response(
|
||||
ctx, f"Now monitoring {channel.mention} for videos."
|
||||
ctx,
|
||||
f"Now monitoring {channel.mention} for videos.",
|
||||
response_type=ResponseType.SUCCESS
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Error adding enabled channel: {e}")
|
||||
await handle_response(
|
||||
ctx, "An error occurred while adding the channel."
|
||||
error = f"Failed to add enabled channel: {str(e)}"
|
||||
logger.error(error, exc_info=True)
|
||||
raise CommandError(
|
||||
error,
|
||||
context=ErrorContext(
|
||||
"SettingsCommands",
|
||||
"add_enabled_channel",
|
||||
{"guild_id": ctx.guild.id, "channel_id": channel.id},
|
||||
ErrorSeverity.HIGH
|
||||
)
|
||||
)
|
||||
|
||||
@settings.command(name="removechannel")
|
||||
@guild_only()
|
||||
@admin_or_permissions(administrator=True)
|
||||
@app_commands.describe(channel="The channel to stop monitoring")
|
||||
async def remove_enabled_channel(ctx: Context, channel: discord.TextChannel):
|
||||
async def remove_enabled_channel(ctx: Context, channel: discord.TextChannel) -> None:
|
||||
"""Remove a channel from video monitoring."""
|
||||
try:
|
||||
# Check if config manager is ready
|
||||
if not cog.config_manager:
|
||||
raise CommandError(
|
||||
"Configuration system is not ready",
|
||||
context=ErrorContext(
|
||||
"SettingsCommands",
|
||||
"remove_enabled_channel",
|
||||
{"guild_id": ctx.guild.id},
|
||||
ErrorSeverity.HIGH
|
||||
)
|
||||
)
|
||||
|
||||
# Defer the response immediately for slash commands
|
||||
if hasattr(ctx, "interaction") and ctx.interaction:
|
||||
await ctx.defer()
|
||||
|
||||
enabled_channels = await cog.config_manager.get_setting(
|
||||
ctx.guild.id, "enabled_channels"
|
||||
ctx.guild.id,
|
||||
"enabled_channels"
|
||||
)
|
||||
if channel.id not in enabled_channels:
|
||||
await handle_response(
|
||||
ctx, f"{channel.mention} is not being monitored."
|
||||
ctx,
|
||||
f"{channel.mention} is not being monitored.",
|
||||
response_type=ResponseType.WARNING
|
||||
)
|
||||
return
|
||||
|
||||
enabled_channels.remove(channel.id)
|
||||
await cog.config_manager.update_setting(
|
||||
ctx.guild.id, "enabled_channels", enabled_channels
|
||||
ctx.guild.id,
|
||||
"enabled_channels",
|
||||
enabled_channels
|
||||
)
|
||||
await handle_response(
|
||||
ctx, f"Stopped monitoring {channel.mention} for videos."
|
||||
ctx,
|
||||
f"Stopped monitoring {channel.mention} for videos.",
|
||||
response_type=ResponseType.SUCCESS
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Error removing enabled channel: {e}")
|
||||
await handle_response(
|
||||
ctx, "An error occurred while removing the channel."
|
||||
error = f"Failed to remove enabled channel: {str(e)}"
|
||||
logger.error(error, exc_info=True)
|
||||
raise CommandError(
|
||||
error,
|
||||
context=ErrorContext(
|
||||
"SettingsCommands",
|
||||
"remove_enabled_channel",
|
||||
{"guild_id": ctx.guild.id, "channel_id": channel.id},
|
||||
ErrorSeverity.HIGH
|
||||
)
|
||||
)
|
||||
|
||||
@settings.command(name="setformat")
|
||||
@guild_only()
|
||||
@admin_or_permissions(administrator=True)
|
||||
@app_commands.describe(format="The video format to use (mp4, webm, or mkv)")
|
||||
async def set_video_format(ctx: Context, format: str):
|
||||
async def set_video_format(ctx: Context, format: str) -> None:
|
||||
"""Set the video format for archived videos."""
|
||||
try:
|
||||
# Check if config manager is ready
|
||||
if not cog.config_manager:
|
||||
raise CommandError(
|
||||
"Configuration system is not ready",
|
||||
context=ErrorContext(
|
||||
"SettingsCommands",
|
||||
"set_video_format",
|
||||
{"guild_id": ctx.guild.id},
|
||||
ErrorSeverity.HIGH
|
||||
)
|
||||
)
|
||||
|
||||
# Defer the response immediately for slash commands
|
||||
if hasattr(ctx, "interaction") and ctx.interaction:
|
||||
await ctx.defer()
|
||||
|
||||
# Validate format
|
||||
format = format.lower()
|
||||
if format not in ["mp4", "webm", "mkv"]:
|
||||
validation = await validate_setting(
|
||||
SettingCategory.VIDEO,
|
||||
"format",
|
||||
format
|
||||
)
|
||||
if not validation["valid"]:
|
||||
await handle_response(
|
||||
ctx, "Invalid format. Please use mp4, webm, or mkv."
|
||||
ctx,
|
||||
validation["error"],
|
||||
response_type=ResponseType.ERROR
|
||||
)
|
||||
return
|
||||
|
||||
await cog.config_manager.update_setting(
|
||||
ctx.guild.id, "video_format", format
|
||||
ctx.guild.id,
|
||||
"video_format",
|
||||
format
|
||||
)
|
||||
await handle_response(ctx, f"Video format has been set to {format}.")
|
||||
except Exception as e:
|
||||
logger.error(f"Error setting video format: {e}")
|
||||
await handle_response(
|
||||
ctx, "An error occurred while setting the video format."
|
||||
ctx,
|
||||
f"Video format has been set to {format}.",
|
||||
response_type=ResponseType.SUCCESS
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
error = f"Failed to set video format: {str(e)}"
|
||||
logger.error(error, exc_info=True)
|
||||
raise CommandError(
|
||||
error,
|
||||
context=ErrorContext(
|
||||
"SettingsCommands",
|
||||
"set_video_format",
|
||||
{"guild_id": ctx.guild.id, "format": format},
|
||||
ErrorSeverity.HIGH
|
||||
)
|
||||
)
|
||||
|
||||
@settings.command(name="setquality")
|
||||
@guild_only()
|
||||
@admin_or_permissions(administrator=True)
|
||||
@app_commands.describe(quality="The video quality (144-4320)")
|
||||
async def set_video_quality(ctx: Context, quality: int):
|
||||
async def set_video_quality(ctx: Context, quality: int) -> None:
|
||||
"""Set the video quality for archived videos."""
|
||||
try:
|
||||
# Check if config manager is ready
|
||||
if not cog.config_manager:
|
||||
raise CommandError(
|
||||
"Configuration system is not ready",
|
||||
context=ErrorContext(
|
||||
"SettingsCommands",
|
||||
"set_video_quality",
|
||||
{"guild_id": ctx.guild.id},
|
||||
ErrorSeverity.HIGH
|
||||
)
|
||||
)
|
||||
|
||||
# Defer the response immediately for slash commands
|
||||
if hasattr(ctx, "interaction") and ctx.interaction:
|
||||
await ctx.defer()
|
||||
|
||||
if not 144 <= quality <= 4320:
|
||||
# Validate quality
|
||||
validation = await validate_setting(
|
||||
SettingCategory.VIDEO,
|
||||
"quality",
|
||||
quality
|
||||
)
|
||||
if not validation["valid"]:
|
||||
await handle_response(
|
||||
ctx, "Quality must be between 144 and 4320."
|
||||
ctx,
|
||||
validation["error"],
|
||||
response_type=ResponseType.ERROR
|
||||
)
|
||||
return
|
||||
|
||||
await cog.config_manager.update_setting(
|
||||
ctx.guild.id, "video_quality", quality
|
||||
ctx.guild.id,
|
||||
"video_quality",
|
||||
quality
|
||||
)
|
||||
await handle_response(
|
||||
ctx, f"Video quality has been set to {quality}p."
|
||||
ctx,
|
||||
f"Video quality has been set to {quality}p.",
|
||||
response_type=ResponseType.SUCCESS
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Error setting video quality: {e}")
|
||||
await handle_response(
|
||||
ctx, "An error occurred while setting the video quality."
|
||||
error = f"Failed to set video quality: {str(e)}"
|
||||
logger.error(error, exc_info=True)
|
||||
raise CommandError(
|
||||
error,
|
||||
context=ErrorContext(
|
||||
"SettingsCommands",
|
||||
"set_video_quality",
|
||||
{"guild_id": ctx.guild.id, "quality": quality},
|
||||
ErrorSeverity.HIGH
|
||||
)
|
||||
)
|
||||
|
||||
@settings.command(name="setmaxsize")
|
||||
@guild_only()
|
||||
@admin_or_permissions(administrator=True)
|
||||
@app_commands.describe(size="The maximum file size in MB (1-100)")
|
||||
async def set_max_file_size(ctx: Context, size: int):
|
||||
async def set_max_file_size(ctx: Context, size: int) -> None:
|
||||
"""Set the maximum file size for archived videos."""
|
||||
try:
|
||||
# Check if config manager is ready
|
||||
if not cog.config_manager:
|
||||
raise CommandError(
|
||||
"Configuration system is not ready",
|
||||
context=ErrorContext(
|
||||
"SettingsCommands",
|
||||
"set_max_file_size",
|
||||
{"guild_id": ctx.guild.id},
|
||||
ErrorSeverity.HIGH
|
||||
)
|
||||
)
|
||||
|
||||
# Defer the response immediately for slash commands
|
||||
if hasattr(ctx, "interaction") and ctx.interaction:
|
||||
await ctx.defer()
|
||||
|
||||
if not 1 <= size <= 100:
|
||||
await handle_response(ctx, "Size must be between 1 and 100 MB.")
|
||||
# Validate size
|
||||
validation = await validate_setting(
|
||||
SettingCategory.VIDEO,
|
||||
"max_file_size",
|
||||
size
|
||||
)
|
||||
if not validation["valid"]:
|
||||
await handle_response(
|
||||
ctx,
|
||||
validation["error"],
|
||||
response_type=ResponseType.ERROR
|
||||
)
|
||||
return
|
||||
|
||||
await cog.config_manager.update_setting(
|
||||
ctx.guild.id, "max_file_size", size
|
||||
ctx.guild.id,
|
||||
"max_file_size",
|
||||
size
|
||||
)
|
||||
await handle_response(
|
||||
ctx, f"Maximum file size has been set to {size}MB."
|
||||
ctx,
|
||||
f"Maximum file size has been set to {size}MB.",
|
||||
response_type=ResponseType.SUCCESS
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Error setting max file size: {e}")
|
||||
await handle_response(
|
||||
ctx, "An error occurred while setting the maximum file size."
|
||||
error = f"Failed to set max file size: {str(e)}"
|
||||
logger.error(error, exc_info=True)
|
||||
raise CommandError(
|
||||
error,
|
||||
context=ErrorContext(
|
||||
"SettingsCommands",
|
||||
"set_max_file_size",
|
||||
{"guild_id": ctx.guild.id, "size": size},
|
||||
ErrorSeverity.HIGH
|
||||
)
|
||||
)
|
||||
|
||||
@settings.command(name="setmessageduration")
|
||||
@guild_only()
|
||||
@admin_or_permissions(administrator=True)
|
||||
@app_commands.describe(hours="How long to keep messages in hours (0-168)")
|
||||
async def set_message_duration(ctx: Context, hours: int):
|
||||
async def set_message_duration(ctx: Context, hours: int) -> None:
|
||||
"""Set how long to keep archived messages."""
|
||||
try:
|
||||
# Check if config manager is ready
|
||||
if not cog.config_manager:
|
||||
raise CommandError(
|
||||
"Configuration system is not ready",
|
||||
context=ErrorContext(
|
||||
"SettingsCommands",
|
||||
"set_message_duration",
|
||||
{"guild_id": ctx.guild.id},
|
||||
ErrorSeverity.HIGH
|
||||
)
|
||||
)
|
||||
|
||||
# Defer the response immediately for slash commands
|
||||
if hasattr(ctx, "interaction") and ctx.interaction:
|
||||
await ctx.defer()
|
||||
|
||||
if not 0 <= hours <= 168:
|
||||
# Validate duration
|
||||
validation = await validate_setting(
|
||||
SettingCategory.MESSAGES,
|
||||
"duration",
|
||||
hours
|
||||
)
|
||||
if not validation["valid"]:
|
||||
await handle_response(
|
||||
ctx, "Duration must be between 0 and 168 hours (1 week)."
|
||||
ctx,
|
||||
validation["error"],
|
||||
response_type=ResponseType.ERROR
|
||||
)
|
||||
return
|
||||
|
||||
await cog.config_manager.update_setting(
|
||||
ctx.guild.id, "message_duration", hours
|
||||
ctx.guild.id,
|
||||
"message_duration",
|
||||
hours
|
||||
)
|
||||
await handle_response(
|
||||
ctx, f"Message duration has been set to {hours} hours."
|
||||
ctx,
|
||||
f"Message duration has been set to {hours} hours.",
|
||||
response_type=ResponseType.SUCCESS
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Error setting message duration: {e}")
|
||||
await handle_response(
|
||||
ctx, "An error occurred while setting the message duration."
|
||||
error = f"Failed to set message duration: {str(e)}"
|
||||
logger.error(error, exc_info=True)
|
||||
raise CommandError(
|
||||
error,
|
||||
context=ErrorContext(
|
||||
"SettingsCommands",
|
||||
"set_message_duration",
|
||||
{"guild_id": ctx.guild.id, "hours": hours},
|
||||
ErrorSeverity.HIGH
|
||||
)
|
||||
)
|
||||
|
||||
@settings.command(name="settemplate")
|
||||
@@ -259,61 +706,122 @@ def setup_settings_commands(cog):
|
||||
@app_commands.describe(
|
||||
template="The message template to use. Available placeholders: {author}, {channel}, {original_message}"
|
||||
)
|
||||
async def set_message_template(ctx: Context, template: str):
|
||||
async def set_message_template(ctx: Context, template: str) -> None:
|
||||
"""Set the template for archived messages."""
|
||||
try:
|
||||
# Check if config manager is ready
|
||||
if not cog.config_manager:
|
||||
raise CommandError(
|
||||
"Configuration system is not ready",
|
||||
context=ErrorContext(
|
||||
"SettingsCommands",
|
||||
"set_message_template",
|
||||
{"guild_id": ctx.guild.id},
|
||||
ErrorSeverity.HIGH
|
||||
)
|
||||
)
|
||||
|
||||
# Defer the response immediately for slash commands
|
||||
if hasattr(ctx, "interaction") and ctx.interaction:
|
||||
await ctx.defer()
|
||||
|
||||
if not any(
|
||||
ph in template for ph in ["{author}", "{channel}", "{original_message}"]
|
||||
):
|
||||
# Validate template
|
||||
validation = await validate_setting(
|
||||
SettingCategory.MESSAGES,
|
||||
"template",
|
||||
template
|
||||
)
|
||||
if not validation["valid"]:
|
||||
await handle_response(
|
||||
ctx,
|
||||
"Template must include at least one placeholder: {author}, {channel}, or {original_message}",
|
||||
validation["error"],
|
||||
response_type=ResponseType.ERROR
|
||||
)
|
||||
return
|
||||
|
||||
await cog.config_manager.update_setting(
|
||||
ctx.guild.id, "message_template", template
|
||||
ctx.guild.id,
|
||||
"message_template",
|
||||
template
|
||||
)
|
||||
await handle_response(
|
||||
ctx, f"Message template has been set to: {template}"
|
||||
ctx,
|
||||
f"Message template has been set to: {template}",
|
||||
response_type=ResponseType.SUCCESS
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Error setting message template: {e}")
|
||||
await handle_response(
|
||||
ctx, "An error occurred while setting the message template."
|
||||
error = f"Failed to set message template: {str(e)}"
|
||||
logger.error(error, exc_info=True)
|
||||
raise CommandError(
|
||||
error,
|
||||
context=ErrorContext(
|
||||
"SettingsCommands",
|
||||
"set_message_template",
|
||||
{"guild_id": ctx.guild.id},
|
||||
ErrorSeverity.HIGH
|
||||
)
|
||||
)
|
||||
|
||||
@settings.command(name="setconcurrent")
|
||||
@guild_only()
|
||||
@admin_or_permissions(administrator=True)
|
||||
@app_commands.describe(count="Number of concurrent downloads (1-5)")
|
||||
async def set_concurrent_downloads(ctx: Context, count: int):
|
||||
async def set_concurrent_downloads(ctx: Context, count: int) -> None:
|
||||
"""Set the number of concurrent downloads allowed."""
|
||||
try:
|
||||
# Check if config manager is ready
|
||||
if not cog.config_manager:
|
||||
raise CommandError(
|
||||
"Configuration system is not ready",
|
||||
context=ErrorContext(
|
||||
"SettingsCommands",
|
||||
"set_concurrent_downloads",
|
||||
{"guild_id": ctx.guild.id},
|
||||
ErrorSeverity.HIGH
|
||||
)
|
||||
)
|
||||
|
||||
# Defer the response immediately for slash commands
|
||||
if hasattr(ctx, "interaction") and ctx.interaction:
|
||||
await ctx.defer()
|
||||
|
||||
if not 1 <= count <= 5:
|
||||
# Validate count
|
||||
validation = await validate_setting(
|
||||
SettingCategory.PERFORMANCE,
|
||||
"concurrent_downloads",
|
||||
count
|
||||
)
|
||||
if not validation["valid"]:
|
||||
await handle_response(
|
||||
ctx, "Concurrent downloads must be between 1 and 5."
|
||||
ctx,
|
||||
validation["error"],
|
||||
response_type=ResponseType.ERROR
|
||||
)
|
||||
return
|
||||
|
||||
await cog.config_manager.update_setting(
|
||||
ctx.guild.id, "concurrent_downloads", count
|
||||
ctx.guild.id,
|
||||
"concurrent_downloads",
|
||||
count
|
||||
)
|
||||
await handle_response(
|
||||
ctx, f"Concurrent downloads has been set to {count}."
|
||||
ctx,
|
||||
f"Concurrent downloads has been set to {count}.",
|
||||
response_type=ResponseType.SUCCESS
|
||||
)
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"Error setting concurrent downloads: {e}")
|
||||
await handle_response(
|
||||
ctx, "An error occurred while setting concurrent downloads."
|
||||
error = f"Failed to set concurrent downloads: {str(e)}"
|
||||
logger.error(error, exc_info=True)
|
||||
raise CommandError(
|
||||
error,
|
||||
context=ErrorContext(
|
||||
"SettingsCommands",
|
||||
"set_concurrent_downloads",
|
||||
{"guild_id": ctx.guild.id, "count": count},
|
||||
ErrorSeverity.HIGH
|
||||
)
|
||||
)
|
||||
|
||||
# Store commands in cog for access
|
||||
|
||||
Reference in New Issue
Block a user