Migrated settings cog

cogs/settings/__init__.py

@@ -1,163 +1,17 @@
-from typing import Any
-
 from discord.ext import commands
-from discord.message import Message
 
 from bot import TuneBot
 from context import CustomContext
 from utils.classes import BaseCog
-from utils.decorators import source_manager_only
-from utils.EmbedGenerator import EmbedGenerator
+from cogs.settings.interactions import SettingCommands
+from cogs.settings.interactions import SourceCommands
 
 
 class SettingsCog(BaseCog, name="Settings"):
-    @commands.group(aliases=["aj"], invoke_without_command=True)
-    @commands.cooldown(rate=1, per=5, type=commands.BucketType.user)
-    async def autojoin(self, ctx: CustomContext):
-        """Enable/Disable the bot automatically joining"""
-        await EmbedGenerator.Message(
-            ctx,
-            "Autojoin",
-            f"Usage:\n\n`{ctx.prefix}autojoin enable`\n`{ctx.prefix}autojoin disable`",
-        )
-
-    @autojoin.command(name="enable", aliases=["set"])
-    @commands.has_permissions(manage_channels=True)
-    @commands.cooldown(rate=1, per=5, type=commands.BucketType.user)
-    async def autojoin_set(self, ctx: CustomContext):
-        """Enable the bot automatically joining"""
-        voice_state = ctx.author.voice
-        if not voice_state:
-            embed = ctx.create_embed()
-            embed.title = "Please join a voice channel before running this command."
-            await ctx.send(embed=embed)
-            return
-
-        await ctx.autojoin.update(voice_state.channel.id, ctx.message.channel.id)
-        embed = ctx.create_embed()
-        embed.title = f"AutoJoin enabled for #{voice_state.channel.name}"
-        await ctx.send(embed=embed)
-
-    @autojoin.command(name="disable", aliases=["unset"])
-    @commands.has_permissions(manage_channels=True)
-    @commands.cooldown(rate=1, per=5, type=commands.BucketType.user)
-    async def autojoin_del(self, ctx: CustomContext):
-        """Disable the bot automatically joining"""
-        await ctx.autojoin.disable()
-        embed = ctx.create_embed()
-        embed.title = f"AutoJoin disabled"
-        await ctx.send(embed=embed)
-
-    @source_manager_only()
-    @commands.group(
-        name="source",
-        aliases=["src"],
-        invoke_without_command=True,
-        hidden=True,
-    )
-    async def source(self, ctx: CustomContext):
-        """Displays all possible options for the `source` command"""
-        prefix = self.bot.config["prefixes"][0]
-        embed = ctx.create_embed()
-        embed.title = "All options:"
-        embed.description = f"```{prefix}source list\n{prefix}source add <url>\n{prefix}source remove <url>\n{prefix}source sync```"
-        await ctx.send(embed=embed)
-
-    @source_manager_only()
-    @source.command(name="remove")
-    async def source_remove(self, ctx: CustomContext, source_url: str):
-        """Removes a source from the bot"""
-        if await ctx.playlist_source.remove(source_url):
-            prefix = self.bot.config["prefixes"][0]
-            embed = ctx.create_embed()
-            embed.title = "Removed source succesfully"
-            embed.description = (
-                f"Please use `{prefix}source sync` to persist these changes."
-            )
-            await ctx.send(embed=embed)
-            return
-
-        embed = ctx.create_embed()
-        embed.title = "Could not remove source, the specified source might not exist"
-        await ctx.send(embed=embed)
-
-    @source_manager_only()
-    @source.command(name="add")
-    async def source_add(self, ctx: CustomContext, source_url: str):
-        """
-        Add's a source to the bot
-
-        Supported sources: YouTube, SoundCloud, Bandcamp, Vimeo, Twitch and HTTP(S) URL's
-        """
-        embed = ctx.create_embed()
-        embed.title = "Started processing source"
-        message = await ctx.send(embed=embed)
-
-        query_result: Any = await self.bot.lavalink.get_tracks(source_url)
-
-        if query_result["loadType"] == "LOAD_FAILED":
-            embed = ctx.create_embed()
-            embed.title = "The specified URL is not a valid source"
-            embed.description = f"Supported sources: YouTube, SoundCloud, Bandcamp, Vimeo, Twitch and HTTP(S) URL's"
-            if isinstance(message, Message):
-                await message.edit(embed=embed)
-            else:
-                await ctx.send(embed=embed)
-            return
-
-        await ctx.playlist_source.add(source_url)
-        track_urls = [str(track["info"]["uri"]) for track in query_result["tracks"]]
-        await self.bot.global_playlist.add_tracks(track_urls)
-
-        embed = ctx.create_embed()
-        embed.title = "Finished processing source"
-        embed.description = f"Added {len(track_urls)} tracks"
-        if isinstance(message, Message):
-            await message.edit(embed=embed)
-        else:
-            await ctx.send(embed=embed)
-
-    @source_manager_only()
-    @source.command(name="list", aliases=["ls"])
-    async def source_list(self, ctx: CustomContext):
-        """Display a list of sources"""
-        # TODO: Implement pagination for sources
-        sources = await self.bot.global_playlist_source.fetch_sources()
-        if len(sources) > 0:
-            description = "\n".join([f"[{source}]({source})" for source in sources])
-        else:
-            description = "This bot has no sources yet"
-
-        embed = ctx.create_embed()
-        embed.title = "All sources:"
-        embed.description = description
-        await ctx.send(embed=embed)
-
-    @source_manager_only()
-    @source.command(name="sync")
-    async def source_sync(self, ctx: CustomContext):
-        """Forcefully resyncs all sources"""
-        failed_sources: list[str] = []
-        await self.bot.global_playlist.clear()
-        sources = await self.bot.global_playlist_source.fetch_sources()
-        for source_url in sources:
-            query_result: Any = await self.bot.lavalink.get_tracks(source_url)
-            if query_result["loadType"] == "LOAD_FAILED":
-                failed_sources.append(source_url)
-                continue
-
-            track_urls = [str(track["info"]["uri"]) for track in query_result["tracks"]]
-            await self.bot.global_playlist.add_tracks(track_urls)
-
-        embed = ctx.create_embed()
-        embed.title = f"Finished sync ({len(failed_sources)} issues)"
-        if len(failed_sources) > 0:
-            embed.description = "\n".join(
-                [f"[{source_url}]({source_url})" for source_url in failed_sources]
-            )
-
-        await ctx.send(embed=embed)
+    pass
 
 
 async def setup(bot: TuneBot):
     await bot.add_cog(SettingsCog(bot))
+    bot.tree.add_command(SettingCommands(bot), override=True)
+    bot.tree.add_command(SourceCommands(bot), override=True)