diff --git a/birthday/birthday.py b/birthday/birthday.py index 6e5496a..059e91a 100644 --- a/birthday/birthday.py +++ b/birthday/birthday.py @@ -22,80 +22,80 @@ class Birthday(commands.Cog): self.config.register_guild(**default_guild) self.birthday_tasks = {} - @app_commands.command(name="setrole") + @commands.hybrid_command(name="setrole") @app_commands.guild_only() @app_commands.describe(role="The role to set as the birthday role") @checks.is_owner() - async def set_role(self, interaction: discord.Interaction, role: discord.Role): + async def set_role(self, ctx: commands.Context, role: discord.Role): """Set the birthday role.""" - await self.config.guild(interaction.guild).birthday_role.set(role.id) - await interaction.response.send_message(f"Birthday role set to {role.name}") + await self.config.guild(ctx.guild).birthday_role.set(role.id) + await ctx.send(f"Birthday role set to {role.name}") - @app_commands.command(name="settimezone") + @commands.hybrid_command(name="settimezone") @app_commands.guild_only() @app_commands.describe(tz="The timezone for role expiration (e.g., UTC, America/New_York)") @checks.is_owner() - async def set_timezone(self, interaction: discord.Interaction, tz: str): + async def set_timezone(self, ctx: commands.Context, tz: str): """Set the timezone for the birthday role expiration.""" try: ZoneInfo(tz) - await self.config.guild(interaction.guild).timezone.set(tz) - await interaction.response.send_message(f"Timezone set to {tz}") + await self.config.guild(ctx.guild).timezone.set(tz) + await ctx.send(f"Timezone set to {tz}") except ZoneInfoNotFoundError: - await interaction.response.send_message(f"Invalid timezone: {tz}. Please use a valid IANA time zone identifier.") + await ctx.send(f"Invalid timezone: {tz}. Please use a valid IANA time zone identifier.") - @app_commands.command(name="setchannel") + @commands.hybrid_command(name="setchannel") @app_commands.guild_only() @app_commands.describe(channel="The channel for birthday announcements") @checks.is_owner() - async def set_channel(self, interaction: discord.Interaction, channel: discord.TextChannel): + async def set_channel(self, ctx: commands.Context, channel: discord.TextChannel): """Set the channel for birthday announcements.""" - await self.config.guild(interaction.guild).birthday_channel.set(channel.id) - await interaction.response.send_message(f"Birthday announcement channel set to {channel.mention}") + await self.config.guild(ctx.guild).birthday_channel.set(channel.id) + await ctx.send(f"Birthday announcement channel set to {channel.mention}") - @app_commands.command(name="addrole") + @commands.hybrid_command(name="addrole") @app_commands.guild_only() @app_commands.describe(role="The role to allow using the birthday command") - async def add_allowed_role(self, interaction: discord.Interaction, role: discord.Role): + async def add_allowed_role(self, ctx: commands.Context, role: discord.Role): """Add a role that can use the birthday command.""" - async with self.config.guild(interaction.guild).allowed_roles() as allowed_roles: + async with self.config.guild(ctx.guild).allowed_roles() as allowed_roles: if role.id not in allowed_roles: allowed_roles.append(role.id) - await interaction.response.send_message(f"Added {role.name} to the list of roles that can use the birthday command.") + await ctx.send(f"Added {role.name} to the list of roles that can use the birthday command.") - @app_commands.command(name="removerole") + @commands.hybrid_command(name="removerole") @app_commands.guild_only() @app_commands.describe(role="The role to remove from using the birthday command") - async def remove_allowed_role(self, interaction: discord.Interaction, role: discord.Role): + async def remove_allowed_role(self, ctx: commands.Context, role: discord.Role): """Remove a role from using the birthday command.""" - async with self.config.guild(interaction.guild).allowed_roles() as allowed_roles: + async with self.config.guild(ctx.guild).allowed_roles() as allowed_roles: if role.id in allowed_roles: allowed_roles.remove(role.id) - await interaction.response.send_message(f"Removed {role.name} from the list of roles that can use the birthday command.") + await ctx.send(f"Removed {role.name} from the list of roles that can use the birthday command.") - @app_commands.command(name="birthday") + @commands.hybrid_command(name="birthday") @app_commands.guild_only() @app_commands.describe(member="The member to give the birthday role to") - async def birthday(self, interaction: discord.Interaction, member: discord.Member): + async def birthday(self, ctx: commands.Context, member: discord.Member): """Assign the birthday role to a user until midnight in the set timezone.""" # Check if the user has permission to use this command - allowed_roles = await self.config.guild(interaction.guild).allowed_roles() - if not any(role.id in allowed_roles for role in interaction.user.roles): - return await interaction.response.send_message("You don't have permission to use this command.", ephemeral=True) + allowed_roles = await self.config.guild(ctx.guild).allowed_roles() + if not any(role.id in allowed_roles for role in ctx.author.roles): + return await ctx.send("You don't have permission to use this command.", ephemeral=True) - birthday_role_id = await self.config.guild(interaction.guild).birthday_role() + birthday_role_id = await self.config.guild(ctx.guild).birthday_role() if not birthday_role_id: - return await interaction.response.send_message("The birthday role hasn't been set. An admin needs to set it using `/setrole`.", ephemeral=True) + return await ctx.send("The birthday role hasn't been set. An admin needs to set it using `/setrole`.", ephemeral=True) - birthday_role = interaction.guild.get_role(birthday_role_id) + birthday_role = ctx.guild.get_role(birthday_role_id) if not birthday_role: - return await interaction.response.send_message("The birthday role doesn't exist anymore. Please ask an admin to set it again.", ephemeral=True) + return await ctx.send("The birthday role doesn't exist anymore. Please ask an admin to set it again.", ephemeral=True) # Assign the role, ignoring hierarchy try: await member.add_roles(birthday_role, reason="Birthday role") except discord.Forbidden: - return await interaction.response.send_message("I don't have permission to assign that role.", ephemeral=True) + return await ctx.send("I don't have permission to assign that role.", ephemeral=True) # Generate birthday message with random cakes (or pie) cakes = random.randint(0, 5) @@ -105,55 +105,55 @@ class Birthday(commands.Cog): message = f"🎉 Happy Birthday, {member.mention}! Here's your cake{'s' if cakes > 1 else ''}: " + "🎂" * cakes # Get the birthday announcement channel - birthday_channel_id = await self.config.guild(interaction.guild).birthday_channel() + birthday_channel_id = await self.config.guild(ctx.guild).birthday_channel() if birthday_channel_id: channel = self.bot.get_channel(birthday_channel_id) if not channel: # If the set channel doesn't exist anymore - channel = interaction.channel + channel = ctx.channel else: - channel = interaction.channel + channel = ctx.channel await channel.send(message) - await interaction.response.send_message("Birthday role assigned!", ephemeral=True) + await ctx.send("Birthday role assigned!", ephemeral=True) # Schedule role removal - timezone = await self.config.guild(interaction.guild).timezone() + timezone = await self.config.guild(ctx.guild).timezone() try: tz = ZoneInfo(timezone) except ZoneInfoNotFoundError: - await interaction.followup.send("Warning: Invalid timezone set. Defaulting to UTC.", ephemeral=True) + await ctx.send("Warning: Invalid timezone set. Defaulting to UTC.", ephemeral=True) tz = ZoneInfo("UTC") now = datetime.now(tz) midnight = datetime.combine(now.date() + timedelta(days=1), time.min).replace(tzinfo=tz) - await self.schedule_birthday_role_removal(interaction.guild, member, birthday_role, midnight) + await self.schedule_birthday_role_removal(ctx.guild, member, birthday_role, midnight) - @app_commands.command(name="bdaycheck") + @commands.hybrid_command(name="bdaycheck") @app_commands.guild_only() - async def bdaycheck(self, interaction: discord.Interaction): + async def bdaycheck(self, ctx: commands.Context): """Check the upcoming birthday role removal tasks.""" # Check if the user has permission to use this command - allowed_roles = await self.config.guild(interaction.guild).allowed_roles() - if not any(role.id in allowed_roles for role in interaction.user.roles): - return await interaction.response.send_message("You don't have permission to use this command.", ephemeral=True) + allowed_roles = await self.config.guild(ctx.guild).allowed_roles() + if not any(role.id in allowed_roles for role in ctx.author.roles): + return await ctx.send("You don't have permission to use this command.", ephemeral=True) - scheduled_tasks = await self.config.guild(interaction.guild).scheduled_tasks() + scheduled_tasks = await self.config.guild(ctx.guild).scheduled_tasks() if not scheduled_tasks: - return await interaction.response.send_message("There are no scheduled tasks.", ephemeral=True) + return await ctx.send("There are no scheduled tasks.", ephemeral=True) message = "Upcoming birthday role removal tasks:\n" for member_id, task_info in scheduled_tasks.items(): - member = interaction.guild.get_member(int(member_id)) + member = ctx.guild.get_member(int(member_id)) if not member: continue - role = interaction.guild.get_role(task_info["role_id"]) + role = ctx.guild.get_role(task_info["role_id"]) if not role: continue - remove_at = datetime.fromisoformat(task_info["remove_at"]).replace(tzinfo=ZoneInfo(await self.config.guild(interaction.guild).timezone())) + remove_at = datetime.fromisoformat(task_info["remove_at"]).replace(tzinfo=ZoneInfo(await self.config.guild(ctx.guild).timezone())) message += f"- {member.display_name} ({member.id}): {role.name} will be removed at {remove_at}\n" - await interaction.response.send_message(message, ephemeral=True) + await ctx.send(message, ephemeral=True) async def schedule_birthday_role_removal(self, guild, member, role, when): """Schedule the removal of the birthday role.""" diff --git a/overseerr/overseerr.py b/overseerr/overseerr.py index e75a4b9..3621af6 100644 --- a/overseerr/overseerr.py +++ b/overseerr/overseerr.py @@ -17,44 +17,44 @@ class Overseerr(commands.Cog): } self.config.register_global(**default_global) - @app_commands.command(name="seturl") + @commands.hybrid_command(name="seturl") @app_commands.guild_only() @app_commands.describe(url="The URL of your Overseerr instance") @commands.admin() - async def set_url(self, interaction: discord.Interaction, url: str): + async def set_url(self, ctx: commands.Context, url: str): """Set the Overseerr URL.""" url = url.rstrip('/') await self.config.overseerr_url.set(url) - await interaction.response.send_message(f"Overseerr URL set to: {url}") + await ctx.send(f"Overseerr URL set to: {url}") - @app_commands.command(name="setapikey") + @commands.hybrid_command(name="setapikey") @app_commands.guild_only() @app_commands.describe(api_key="Your Overseerr API key") @commands.admin() - async def set_apikey(self, interaction: discord.Interaction, api_key: str): + async def set_apikey(self, ctx: commands.Context, api_key: str): """Set the Overseerr API key.""" await self.config.overseerr_api_key.set(api_key) - await interaction.response.send_message("Overseerr API key has been set.") + await ctx.send("Overseerr API key has been set.") - @app_commands.command(name="setadminrole") + @commands.hybrid_command(name="setadminrole") @app_commands.guild_only() @app_commands.describe(role_name="The name of the admin role") @commands.admin() - async def set_adminrole(self, interaction: discord.Interaction, role_name: str): + async def set_adminrole(self, ctx: commands.Context, role_name: str): """Set the admin role name for Overseerr approvals.""" await self.config.admin_role_name.set(role_name) - await interaction.response.send_message(f"Admin role for Overseerr approvals set to: {role_name}") + await ctx.send(f"Admin role for Overseerr approvals set to: {role_name}") - @app_commands.command(name="request") + @commands.hybrid_command(name="request") @app_commands.guild_only() @app_commands.describe(query="The name of the movie or TV show to search for") - async def request(self, interaction: discord.Interaction, query: str): + async def request(self, ctx: commands.Context, query: str): """Search and request a movie or TV show on Overseerr.""" overseerr_url = await self.config.overseerr_url() overseerr_api_key = await self.config.overseerr_api_key() if not overseerr_url or not overseerr_api_key: - await interaction.response.send_message("Overseerr is not configured. Please ask an admin to set it up using `/seturl` and `/setapikey`.", ephemeral=True) + await ctx.send("Overseerr is not configured. Please ask an admin to set it up using `/seturl` and `/setapikey`.", ephemeral=True) return search_url = f"{overseerr_url}/api/v1/search" @@ -63,27 +63,27 @@ class Overseerr(commands.Cog): "Content-Type": "application/json" } - # Defer the response since this might take a while - await interaction.response.defer() + # Send initial response + initial_message = await ctx.send("Searching...") async with aiohttp.ClientSession() as session: async with session.get(search_url, headers=headers, params={"query": query}) as resp: if resp.status != 200: - await interaction.followup.send(f"Error from Overseerr API: {resp.status}", ephemeral=True) + await initial_message.edit(content=f"Error from Overseerr API: {resp.status}") return try: search_results = await resp.json() except Exception as e: - await interaction.followup.send(f"Failed to parse JSON: {e}", ephemeral=True) + await initial_message.edit(content=f"Failed to parse JSON: {e}") return if 'results' not in search_results: - await interaction.followup.send(f"No results found for '{query}'. API Response: {search_results}", ephemeral=True) + await initial_message.edit(content=f"No results found for '{query}'. API Response: {search_results}") return if not search_results['results']: - await interaction.followup.send(f"No results found for '{query}'.", ephemeral=True) + await initial_message.edit(content=f"No results found for '{query}'.") return # Create select menu for results @@ -114,6 +114,10 @@ class Overseerr(commands.Cog): ) async def select_callback(select_interaction: discord.Interaction): + if select_interaction.user != ctx.author: + await select_interaction.response.send_message("This menu is not for you!", ephemeral=True) + return + selected_index = int(select_interaction.data['values'][0]) selected_result = search_results['results'][selected_index] media_type = selected_result['mediaType'] @@ -157,23 +161,23 @@ class Overseerr(commands.Cog): select.callback = select_callback view = discord.ui.View() view.add_item(select) - await interaction.followup.send("Search results:", view=view) + await initial_message.edit(content="Search results:", view=view) - @app_commands.command(name="approve") + @commands.hybrid_command(name="approve") @app_commands.guild_only() @app_commands.describe(request_id="The ID of the request to approve") - async def approve(self, interaction: discord.Interaction, request_id: int): + async def approve(self, ctx: commands.Context, request_id: int): """Approve a request on Overseerr.""" admin_role_name = await self.config.admin_role_name() - if not any(role.name == admin_role_name for role in interaction.user.roles): - await interaction.response.send_message(f"You need the '{admin_role_name}' role to approve requests.", ephemeral=True) + if not any(role.name == admin_role_name for role in ctx.author.roles): + await ctx.send(f"You need the '{admin_role_name}' role to approve requests.", ephemeral=True) return overseerr_url = await self.config.overseerr_url() overseerr_api_key = await self.config.overseerr_api_key() if not overseerr_url or not overseerr_api_key: - await interaction.response.send_message("Overseerr is not configured. Please ask an admin to set it up using `/seturl` and `/setapikey`.", ephemeral=True) + await ctx.send("Overseerr is not configured. Please ask an admin to set it up using `/seturl` and `/setapikey`.", ephemeral=True) return approve_url = f"{overseerr_url}/api/v1/request/{request_id}/approve" @@ -185,9 +189,9 @@ class Overseerr(commands.Cog): async with aiohttp.ClientSession() as session: async with session.post(approve_url, headers=headers) as resp: if resp.status == 200: - await interaction.response.send_message(f"Request {request_id} has been approved!") + await ctx.send(f"Request {request_id} has been approved!") else: - await interaction.response.send_message(f"Failed to approve request {request_id}. Please check the request ID and try again.", ephemeral=True) + await ctx.send(f"Failed to approve request {request_id}. Please check the request ID and try again.", ephemeral=True) async def get_media_status(self, media_id, media_type): overseerr_url = await self.config.overseerr_url() diff --git a/videoarchiver/commands.py b/videoarchiver/commands.py index 182fc0c..a82f1e7 100644 --- a/videoarchiver/commands.py +++ b/videoarchiver/commands.py @@ -15,182 +15,183 @@ class VideoArchiverCommands(commands.Cog): self.processor = processor super().__init__() - @app_commands.command(name="settings") + @commands.hybrid_group(name="videoarchiver", aliases=["va"], fallback="settings") @app_commands.guild_only() @commands.admin_or_permissions(administrator=True) - async def settings(self, interaction: discord.Interaction): - """Show VideoArchiver settings""" - embed = await self.config.format_settings_embed(interaction.guild) - await interaction.response.send_message(embed=embed) + async def videoarchiver(self, ctx: commands.Context): + """Video Archiver configuration commands""" + if ctx.invoked_subcommand is None: + embed = await self.config.format_settings_embed(ctx.guild) + await ctx.send(embed=embed) - @app_commands.command(name="updateytdlp") + @videoarchiver.command(name="updateytdlp") @app_commands.guild_only() @checks.is_owner() - async def update_ytdlp(self, interaction: discord.Interaction): + async def update_ytdlp(self, ctx: commands.Context): """Update yt-dlp to the latest version""" success, message = await self.update_checker.update_yt_dlp() - await interaction.response.send_message("✅ " + message if success else "❌ " + message) + await ctx.send("✅ " + message if success else "❌ " + message) - @app_commands.command(name="toggleupdates") + @videoarchiver.command(name="toggleupdates") @app_commands.guild_only() @commands.admin_or_permissions(administrator=True) - async def toggle_update_check(self, interaction: discord.Interaction): + async def toggle_update_check(self, ctx: commands.Context): """Toggle yt-dlp update notifications""" - state = await self.config.toggle_setting(interaction.guild.id, "disable_update_check") + state = await self.config.toggle_setting(ctx.guild.id, "disable_update_check") status = "disabled" if state else "enabled" - await interaction.response.send_message(f"Update notifications {status}") + await ctx.send(f"Update notifications {status}") - @app_commands.command(name="addrole") + @videoarchiver.command(name="addrole") @app_commands.guild_only() @app_commands.describe(role="The role to allow (leave empty for @everyone)") - async def add_allowed_role(self, interaction: discord.Interaction, role: Optional[discord.Role] = None): + async def add_allowed_role(self, ctx: commands.Context, role: Optional[discord.Role] = None): """Add a role that's allowed to trigger archiving""" if not role: # If no role is specified, clear the list to allow everyone - await self.config.update_setting(interaction.guild.id, "allowed_roles", []) - await interaction.response.send_message("Allowed role set to @everyone (all users can trigger archiving)") + await self.config.update_setting(ctx.guild.id, "allowed_roles", []) + await ctx.send("Allowed role set to @everyone (all users can trigger archiving)") return - await self.config.add_to_list(interaction.guild.id, "allowed_roles", role.id) - await interaction.response.send_message(f"Added {role.name} to allowed roles") + await self.config.add_to_list(ctx.guild.id, "allowed_roles", role.id) + await ctx.send(f"Added {role.name} to allowed roles") - @app_commands.command(name="removerole") + @videoarchiver.command(name="removerole") @app_commands.guild_only() @app_commands.describe(role="The role to remove") - async def remove_allowed_role(self, interaction: discord.Interaction, role: discord.Role): + async def remove_allowed_role(self, ctx: commands.Context, role: discord.Role): """Remove a role from allowed roles""" - await self.config.remove_from_list(interaction.guild.id, "allowed_roles", role.id) - await interaction.response.send_message(f"Removed {role.name} from allowed roles") + await self.config.remove_from_list(ctx.guild.id, "allowed_roles", role.id) + await ctx.send(f"Removed {role.name} from allowed roles") - @app_commands.command(name="listroles") + @videoarchiver.command(name="listroles") @app_commands.guild_only() - async def list_allowed_roles(self, interaction: discord.Interaction): + async def list_allowed_roles(self, ctx: commands.Context): """List all roles allowed to trigger archiving""" - roles = await self.config.get_setting(interaction.guild.id, "allowed_roles") + roles = await self.config.get_setting(ctx.guild.id, "allowed_roles") if not roles: - await interaction.response.send_message( + await ctx.send( "No roles are currently set (all users can trigger archiving)" ) return role_names = [ r.name if r else "@everyone" - for r in [interaction.guild.get_role(role_id) for role_id in roles] + for r in [ctx.guild.get_role(role_id) for role_id in roles] ] - await interaction.response.send_message(f"Allowed roles: {', '.join(role_names)}") + await ctx.send(f"Allowed roles: {', '.join(role_names)}") - @app_commands.command(name="setconcurrent") + @videoarchiver.command(name="setconcurrent") @app_commands.guild_only() @app_commands.describe(count="Number of concurrent downloads (1-5)") - async def set_concurrent_downloads(self, interaction: discord.Interaction, count: app_commands.Range[int, 1, 5]): + async def set_concurrent_downloads(self, ctx: commands.Context, count: app_commands.Range[int, 1, 5]): """Set the number of concurrent downloads""" - await self.config.update_setting(interaction.guild.id, "concurrent_downloads", count) - await interaction.response.send_message(f"Concurrent downloads set to {count}") + await self.config.update_setting(ctx.guild.id, "concurrent_downloads", count) + await ctx.send(f"Concurrent downloads set to {count}") - @app_commands.command(name="setchannel") + @videoarchiver.command(name="setchannel") @app_commands.guild_only() @app_commands.describe(channel="The archive channel") - async def set_archive_channel(self, interaction: discord.Interaction, channel: discord.TextChannel): + async def set_archive_channel(self, ctx: commands.Context, channel: discord.TextChannel): """Set the archive channel""" - await self.config.update_setting(interaction.guild.id, "archive_channel", channel.id) - await interaction.response.send_message(f"Archive channel set to {channel.mention}") + await self.config.update_setting(ctx.guild.id, "archive_channel", channel.id) + await ctx.send(f"Archive channel set to {channel.mention}") - @app_commands.command(name="setnotification") + @videoarchiver.command(name="setnotification") @app_commands.guild_only() @app_commands.describe(channel="The notification channel") - async def set_notification_channel(self, interaction: discord.Interaction, channel: discord.TextChannel): + async def set_notification_channel(self, ctx: commands.Context, channel: discord.TextChannel): """Set the notification channel (where archive messages appear)""" await self.config.update_setting( - interaction.guild.id, "notification_channel", channel.id + ctx.guild.id, "notification_channel", channel.id ) - await interaction.response.send_message(f"Notification channel set to {channel.mention}") + await ctx.send(f"Notification channel set to {channel.mention}") - @app_commands.command(name="setlogchannel") + @videoarchiver.command(name="setlogchannel") @app_commands.guild_only() @app_commands.describe(channel="The log channel") - async def set_log_channel(self, interaction: discord.Interaction, channel: discord.TextChannel): + async def set_log_channel(self, ctx: commands.Context, channel: discord.TextChannel): """Set the log channel for error messages and notifications""" - await self.config.update_setting(interaction.guild.id, "log_channel", channel.id) - await interaction.response.send_message(f"Log channel set to {channel.mention}") + await self.config.update_setting(ctx.guild.id, "log_channel", channel.id) + await ctx.send(f"Log channel set to {channel.mention}") - @app_commands.command(name="addmonitor") + @videoarchiver.command(name="addmonitor") @app_commands.guild_only() @app_commands.describe(channel="The channel to monitor (leave empty to monitor all channels)") - async def add_monitored_channel(self, interaction: discord.Interaction, channel: Optional[discord.TextChannel] = None): + async def add_monitored_channel(self, ctx: commands.Context, channel: Optional[discord.TextChannel] = None): """Add a channel to monitor for videos""" if not channel: # If no channel is specified, clear the list to monitor all channels - await self.config.update_setting(interaction.guild.id, "monitored_channels", []) - await interaction.response.send_message("Now monitoring all channels for videos") + await self.config.update_setting(ctx.guild.id, "monitored_channels", []) + await ctx.send("Now monitoring all channels for videos") return - await self.config.add_to_list(interaction.guild.id, "monitored_channels", channel.id) - await interaction.response.send_message(f"Now monitoring {channel.mention} for videos") + await self.config.add_to_list(ctx.guild.id, "monitored_channels", channel.id) + await ctx.send(f"Now monitoring {channel.mention} for videos") - @app_commands.command(name="removemonitor") + @videoarchiver.command(name="removemonitor") @app_commands.guild_only() @app_commands.describe(channel="The channel to stop monitoring") - async def remove_monitored_channel(self, interaction: discord.Interaction, channel: discord.TextChannel): + async def remove_monitored_channel(self, ctx: commands.Context, channel: discord.TextChannel): """Remove a channel from monitoring""" await self.config.remove_from_list( - interaction.guild.id, "monitored_channels", channel.id + ctx.guild.id, "monitored_channels", channel.id ) - await interaction.response.send_message(f"Stopped monitoring {channel.mention}") + await ctx.send(f"Stopped monitoring {channel.mention}") - @app_commands.command(name="setformat") + @videoarchiver.command(name="setformat") @app_commands.guild_only() @app_commands.describe(format="The video format (e.g., mp4, webm)") - async def set_video_format(self, interaction: discord.Interaction, format: Literal["mp4", "webm"]): + async def set_video_format(self, ctx: commands.Context, format: Literal["mp4", "webm"]): """Set the video format""" - await self.config.update_setting(interaction.guild.id, "video_format", format.lower()) - await interaction.response.send_message(f"Video format set to {format.lower()}") + await self.config.update_setting(ctx.guild.id, "video_format", format.lower()) + await ctx.send(f"Video format set to {format.lower()}") - @app_commands.command(name="setquality") + @videoarchiver.command(name="setquality") @app_commands.guild_only() @app_commands.describe(quality="Maximum video quality in pixels (e.g., 1080)") - async def set_video_quality(self, interaction: discord.Interaction, quality: app_commands.Range[int, 144, 4320]): + async def set_video_quality(self, ctx: commands.Context, quality: app_commands.Range[int, 144, 4320]): """Set the maximum video quality""" - await self.config.update_setting(interaction.guild.id, "video_quality", quality) - await interaction.response.send_message(f"Maximum video quality set to {quality}p") + await self.config.update_setting(ctx.guild.id, "video_quality", quality) + await ctx.send(f"Maximum video quality set to {quality}p") - @app_commands.command(name="setmaxsize") + @videoarchiver.command(name="setmaxsize") @app_commands.guild_only() @app_commands.describe(size="Maximum file size in MB") - async def set_max_file_size(self, interaction: discord.Interaction, size: app_commands.Range[int, 1, 100]): + async def set_max_file_size(self, ctx: commands.Context, size: app_commands.Range[int, 1, 100]): """Set the maximum file size""" - await self.config.update_setting(interaction.guild.id, "max_file_size", size) - await interaction.response.send_message(f"Maximum file size set to {size}MB") + await self.config.update_setting(ctx.guild.id, "max_file_size", size) + await ctx.send(f"Maximum file size set to {size}MB") - @app_commands.command(name="toggledelete") + @videoarchiver.command(name="toggledelete") @app_commands.guild_only() - async def toggle_delete_after_repost(self, interaction: discord.Interaction): + async def toggle_delete_after_repost(self, ctx: commands.Context): """Toggle whether to delete local files after reposting""" - state = await self.config.toggle_setting(interaction.guild.id, "delete_after_repost") - await interaction.response.send_message(f"Delete after repost: {state}") + state = await self.config.toggle_setting(ctx.guild.id, "delete_after_repost") + await ctx.send(f"Delete after repost: {state}") - @app_commands.command(name="setduration") + @videoarchiver.command(name="setduration") @app_commands.guild_only() @app_commands.describe(hours="Duration in hours (0 for permanent)") - async def set_message_duration(self, interaction: discord.Interaction, hours: app_commands.Range[int, 0, 720]): + async def set_message_duration(self, ctx: commands.Context, hours: app_commands.Range[int, 0, 720]): """Set how long to keep archive messages""" - await self.config.update_setting(interaction.guild.id, "message_duration", hours) - await interaction.response.send_message(f"Archive message duration set to {hours} hours") + await self.config.update_setting(ctx.guild.id, "message_duration", hours) + await ctx.send(f"Archive message duration set to {hours} hours") - @app_commands.command(name="settemplate") + @videoarchiver.command(name="settemplate") @app_commands.guild_only() @app_commands.describe(template="Message template using {author}, {url}, and {original_message}") - async def set_message_template(self, interaction: discord.Interaction, template: str): + async def set_message_template(self, ctx: commands.Context, template: str): """Set the archive message template""" - await self.config.update_setting(interaction.guild.id, "message_template", template) - await interaction.response.send_message(f"Archive message template set to:\n{template}") + await self.config.update_setting(ctx.guild.id, "message_template", template) + await ctx.send(f"Archive message template set to:\n{template}") - @app_commands.command(name="enablesites") + @videoarchiver.command(name="enablesites") @app_commands.guild_only() @app_commands.describe(sites="Sites to enable (leave empty for all sites)") - async def enable_sites(self, interaction: discord.Interaction, sites: Optional[str] = None): + async def enable_sites(self, ctx: commands.Context, *, sites: Optional[str] = None): """Enable specific sites""" if sites is None: - await self.config.update_setting(interaction.guild.id, "enabled_sites", []) - await interaction.response.send_message("All sites enabled") + await self.config.update_setting(ctx.guild.id, "enabled_sites", []) + await ctx.send("All sites enabled") return site_list = [s.strip().lower() for s in sites.split()] @@ -200,19 +201,19 @@ class VideoArchiverCommands(commands.Cog): valid_sites = set(ie.IE_NAME.lower() for ie in ydl._ies) invalid_sites = [s for s in site_list if s not in valid_sites] if invalid_sites: - await interaction.response.send_message( + await ctx.send( f"Invalid sites: {', '.join(invalid_sites)}\nValid sites: {', '.join(valid_sites)}" ) return - await self.config.update_setting(interaction.guild.id, "enabled_sites", site_list) - await interaction.response.send_message(f"Enabled sites: {', '.join(site_list)}") + await self.config.update_setting(ctx.guild.id, "enabled_sites", site_list) + await ctx.send(f"Enabled sites: {', '.join(site_list)}") - @app_commands.command(name="listsites") + @videoarchiver.command(name="listsites") @app_commands.guild_only() - async def list_sites(self, interaction: discord.Interaction): + async def list_sites(self, ctx: commands.Context): """List all available sites and currently enabled sites""" - enabled_sites = await self.config.get_setting(interaction.guild.id, "enabled_sites") + enabled_sites = await self.config.get_setting(ctx.guild.id, "enabled_sites") embed = discord.Embed( title="Video Sites Configuration", color=discord.Color.blue() @@ -240,14 +241,14 @@ class VideoArchiverCommands(commands.Cog): inline=False, ) - await interaction.response.send_message(embed=embed) + await ctx.send(embed=embed) - @app_commands.command(name="queue") + @videoarchiver.command(name="queue") @app_commands.guild_only() @commands.admin_or_permissions(administrator=True) - async def show_queue(self, interaction: discord.Interaction): + async def show_queue(self, ctx: commands.Context): """Show current queue status with basic metrics""" - status = self.processor.queue_manager.get_queue_status(interaction.guild.id) + status = self.processor.queue_manager.get_queue_status(ctx.guild.id) embed = discord.Embed( title="Video Processing Queue Status", @@ -278,15 +279,15 @@ class VideoArchiverCommands(commands.Cog): inline=False ) - embed.set_footer(text="Use /queuemetrics for detailed performance metrics") - await interaction.response.send_message(embed=embed) + embed.set_footer(text="Use /videoarchiver queuemetrics for detailed performance metrics") + await ctx.send(embed=embed) - @app_commands.command(name="queuemetrics") + @videoarchiver.command(name="queuemetrics") @app_commands.guild_only() @commands.admin_or_permissions(administrator=True) - async def show_queue_metrics(self, interaction: discord.Interaction): + async def show_queue_metrics(self, ctx: commands.Context): """Show detailed queue performance metrics""" - status = self.processor.queue_manager.get_queue_status(interaction.guild.id) + status = self.processor.queue_manager.get_queue_status(ctx.guild.id) metrics = status['metrics'] embed = discord.Embed( @@ -330,12 +331,12 @@ class VideoArchiverCommands(commands.Cog): ) embed.set_footer(text="Metrics are updated in real-time as videos are processed") - await interaction.response.send_message(embed=embed) + await ctx.send(embed=embed) - @app_commands.command(name="clearqueue") + @videoarchiver.command(name="clearqueue") @app_commands.guild_only() @commands.admin_or_permissions(administrator=True) - async def clear_queue(self, interaction: discord.Interaction): + async def clear_queue(self, ctx: commands.Context): """Clear the video processing queue for this guild""" - cleared = await self.processor.queue_manager.clear_guild_queue(interaction.guild.id) - await interaction.response.send_message(f"Cleared {cleared} items from the queue") + cleared = await self.processor.queue_manager.clear_guild_queue(ctx.guild.id) + await ctx.send(f"Cleared {cleared} items from the queue")