mirror of
https://github.com/UiO-Gaming/ShiteMusicBot.git
synced 2026-11-05 13:56:51 +01:00
242 lines
11 KiB
Python
242 lines
11 KiB
Python
from typing import Optional
|
|
|
|
import discord
|
|
from discord.ext import commands
|
|
|
|
from bot import MusicBot
|
|
from musicbot.utils import checks
|
|
from musicbot.utils.settingsmanager import Settings as SettingsManager
|
|
from musicbot.utils.userinteraction import ClearMode, Scroller
|
|
|
|
from .helpformatter import commandhelper
|
|
|
|
|
|
class Settings(commands.Cog):
|
|
def __init__(self, bot: MusicBot):
|
|
self.bot: MusicBot = bot
|
|
self.settings: SettingsManager = self.bot.settings
|
|
|
|
def format_prefixes(self, prefixes):
|
|
if prefixes is None:
|
|
prefixes = [self.bot.settings.default_prefix]
|
|
formatted = ''
|
|
for prefix in prefixes:
|
|
formatted += f'`{prefix}`, '
|
|
return formatted[:-2]
|
|
|
|
@commands.group(name='settings', hidden=True)
|
|
async def _set(self, ctx):
|
|
if ctx.invoked_subcommand is None:
|
|
ctx.localizer.prefix = 'help' # Ensure the bot looks for locales in the context of help, not cogmanager.
|
|
paginator = commandhelper(ctx, ctx.command, ctx.invoker, include_subcmd=True)
|
|
scroller = Scroller(ctx, paginator)
|
|
await scroller.start_scrolling(ClearMode.AnyExit)
|
|
|
|
@checks.is_admin()
|
|
@commands.guild_only()
|
|
@_set.command(name='serverlocale')
|
|
async def _set_guild_locale(self, ctx, locale):
|
|
self.bot.localizer.index_localizations()
|
|
self.bot.aliaser.index_localizations()
|
|
self.bot.localizer.load_localizations()
|
|
self.bot.aliaser.load_localizations()
|
|
|
|
if locale in self.bot.aliaser.localization_table.keys():
|
|
self.settings.set(ctx.guild, 'locale', locale)
|
|
locale = self.settings.get(ctx.guild, 'locale', 'default_lang')
|
|
await ctx.send(f'Locale set to `{locale}`')
|
|
else:
|
|
await ctx.send(f'`{locale}` is not a valid locale.')
|
|
|
|
@checks.is_admin()
|
|
@commands.guild_only()
|
|
@_set.command(name='serverprefix')
|
|
async def _set_guild_prefix(self, ctx, *prefixes):
|
|
if prefixes := list(prefixes):
|
|
self.settings.set(ctx.guild, 'prefixes', prefixes)
|
|
prefixes = self.settings.get(ctx.guild, 'prefixes')
|
|
await ctx.send(f'Server prefixes: {self.format_prefixes(prefixes)}')
|
|
|
|
@checks.is_admin()
|
|
@commands.guild_only()
|
|
@_set.command(name='resetprefix')
|
|
async def _reset_prefix(self, ctx):
|
|
self.settings.set(ctx.guild, 'prefixes', None)
|
|
prefixes = self.settings.get(ctx.guild, 'prefixes', 'default_prefix')
|
|
await ctx.send(self.format_prefixes(prefixes))
|
|
|
|
@checks.is_admin()
|
|
@commands.guild_only()
|
|
@_set.command(name='modrole')
|
|
async def set_mod_role(self, ctx, modrole: discord.Role):
|
|
await ctx.send(f'{modrole.name} {modrole.id}')
|
|
|
|
@checks.is_admin()
|
|
@commands.guild_only()
|
|
@_set.command(name='threshold')
|
|
async def set_vote_threshold(self, ctx, threshold: int = 50):
|
|
if not 0 <= threshold <= 100:
|
|
return await ctx.send('Must be between 0 and 100')
|
|
|
|
self.bot.settings.set(ctx.guild, 'vote_threshold', threshold)
|
|
threshold = self.bot.settings.get(ctx.guild, 'vote_threshold', 50)
|
|
await ctx.send(f'Vote threshold set to {threshold}%')
|
|
|
|
@checks.is_admin()
|
|
@commands.guild_only()
|
|
@_set.command(name='textchannels')
|
|
async def set_music_text(self, ctx, *channels: discord.TextChannel):
|
|
if channels:
|
|
channel_ids = [channel.id for channel in channels]
|
|
self.bot.settings.set(ctx.guild, 'channels.text', channel_ids)
|
|
else:
|
|
self.bot.settings.set(ctx.guild, 'channels.text', None)
|
|
|
|
if textchannels := self.bot.settings.get(ctx.guild, 'channels.text'):
|
|
embed = discord.Embed(title='Music command channels set', color=ctx.me.color)
|
|
configured_channels = [ctx.guild.get_channel(channel) for channel in textchannels]
|
|
mentioned = [channel.mention for channel in configured_channels if channel is not None]
|
|
embed.description = ', '.join(mentioned)
|
|
await ctx.send(embed=embed)
|
|
else:
|
|
embed = discord.Embed(color=ctx.me.color, description='Music command channels cleared')
|
|
await ctx.send(embed=embed)
|
|
|
|
@checks.is_admin()
|
|
@commands.guild_only()
|
|
@_set.command(name='musicchannels')
|
|
async def set_music_voice(self, ctx, *channels: discord.VoiceChannel):
|
|
if channels:
|
|
channel_ids = [channel.id for channel in channels]
|
|
self.bot.settings.set(ctx.guild, 'channels.music', channel_ids)
|
|
else:
|
|
self.bot.settings.set(ctx.guild, 'channels.music', None)
|
|
|
|
if musicchannels := self.bot.settings.get(ctx.guild, 'channels.music'):
|
|
embed = discord.Embed(title='Music channels set', color=ctx.me.color)
|
|
configured_channels = [ctx.guild.get_channel(channel) for channel in musicchannels]
|
|
mentioned = [channel.name for channel in configured_channels if channel is not None]
|
|
embed.description = ', '.join(mentioned)
|
|
await ctx.send(embed=embed)
|
|
else:
|
|
embed = discord.Embed(color=ctx.me.color, description='Music channels cleared')
|
|
await ctx.send(embed=embed)
|
|
|
|
@checks.is_admin()
|
|
@commands.guild_only()
|
|
@_set.command(name='listenchannels', hidden=True)
|
|
async def set_listen_voice(self, ctx, *channels: discord.VoiceChannel):
|
|
if channels:
|
|
channel_ids = [channel.id for channel in channels]
|
|
self.bot.settings.set(ctx.guild, 'channels.listen_only', channel_ids)
|
|
else:
|
|
self.bot.settings.set(ctx.guild, 'channels.listen_only', None)
|
|
|
|
if listenchannels := self.bot.settings.get(ctx.guild, 'channels.listen_only'):
|
|
embed = discord.Embed(title='Listen only channels set', color=ctx.me.color)
|
|
configured_channels = [ctx.guild.get_channel(channel) for channel in listenchannels]
|
|
mentioned = [channel.name for channel in configured_channels if channel is not None]
|
|
embed.description = ', '.join(mentioned)
|
|
await ctx.send(embed=embed)
|
|
else:
|
|
embed = discord.Embed(color=ctx.me.color, description='Listen only channels cleared')
|
|
await ctx.send(embed=embed)
|
|
|
|
@checks.is_admin()
|
|
@commands.guild_only()
|
|
@_set.command(name='djroles')
|
|
async def set_dj_roles(self, ctx, *roles: discord.Role):
|
|
role_ids = [role.id for role in roles]
|
|
if roles:
|
|
self.bot.settings.set(ctx.guild, 'roles.dj', role_ids)
|
|
else:
|
|
self.bot.settings.set(ctx.guild, 'roles.dj', None)
|
|
|
|
if djroles := self.bot.settings.get(ctx.guild, 'roles.dj'):
|
|
embed = discord.Embed(title='DJ roles set to', color=ctx.me.color)
|
|
configured_roles = [ctx.guild.get_role(role) for role in djroles]
|
|
mentioned = [role.name for role in configured_roles if role is not None]
|
|
embed.description = ', '.join(mentioned)
|
|
await ctx.send(embed=embed)
|
|
else:
|
|
embed = discord.Embed(color=ctx.me.color, description='DJ roles cleared')
|
|
await ctx.send(embed=embed)
|
|
|
|
@checks.is_admin()
|
|
@commands.guild_only()
|
|
@_set.command(name='maxduration')
|
|
async def set_max_track_duration(self, ctx, duration: Optional[int] = None):
|
|
self.bot.settings.set(ctx.guild, 'duration.max', duration)
|
|
|
|
if duration := self.bot.settings.get(ctx.guild, 'duration.max'):
|
|
embed = discord.Embed(title='Max track duration set to', color=ctx.me.color)
|
|
embed.description = f'{duration} minutes'
|
|
await ctx.send(embed=embed)
|
|
else:
|
|
embed = discord.Embed(color=ctx.me.color, description='Max track length removed')
|
|
await ctx.send(embed=embed)
|
|
|
|
@checks.is_admin()
|
|
@commands.guild_only()
|
|
@_set.command(name='dynamicmax')
|
|
async def set_track_duration_type(self, ctx, dynamic: bool = False):
|
|
self.bot.settings.set(ctx.guild, 'duration.is_dynamic', dynamic)
|
|
|
|
embed = discord.Embed(title='Max duration type set to', color=ctx.me.color)
|
|
if self.bot.settings.get(ctx.guild, 'duration.is_dynamic'):
|
|
embed.description = 'Dynamic'
|
|
else:
|
|
embed.description = 'Static'
|
|
await ctx.send(embed=embed)
|
|
|
|
@commands.guild_only()
|
|
@_set.command(name='current')
|
|
async def current_settings(self, ctx):
|
|
embed = discord.Embed(title='{current.title}', color=ctx.me.color)
|
|
embed.description = '{current.description}'
|
|
embed.set_thumbnail(url=ctx.guild.icon.url)
|
|
|
|
prefixes = self.bot.settings.get(ctx.guild, 'prefixes', 'default_prefix')
|
|
embed.add_field(name='{current.prefix}', value=self.format_prefixes(prefixes))
|
|
|
|
locale = f"{self.bot.settings.get(ctx.guild, 'locale', 'default_locale')}"
|
|
embed.add_field(name='{current.locale}', value=locale)
|
|
|
|
threshold = self.bot.settings.get(ctx.guild, 'vote_threshold', 50)
|
|
embed.add_field(name='{current.threshold}', value=f'{threshold}%')
|
|
|
|
is_dynamic = self.bot.settings.get(ctx.guild, 'duration.is_dynamic', 'default_is_dynamic')
|
|
embed.add_field(name='{current.dynamicmax}', value=is_dynamic)
|
|
|
|
if duration := self.bot.settings.get(ctx.guild, 'maxduration'):
|
|
embed.add_field(name='{current.maxduration}', value=f'{duration} minutes')
|
|
|
|
if textchannels := self.bot.settings.get(ctx.guild, 'channels.text'):
|
|
channels = [ctx.guild.get_channel(channel) for channel in textchannels]
|
|
mentioned = [channel.mention for channel in channels if channel is not None]
|
|
embed.add_field(name='{current.textchannels}', value='\n'.join(mentioned))
|
|
|
|
if voicechannels := self.bot.settings.get(ctx.guild, 'channels.music'):
|
|
channels = [ctx.guild.get_channel(channel) for channel in voicechannels]
|
|
mentioned = [channel.name for channel in channels if channel is not None]
|
|
embed.add_field(name='{current.musicchannels}', value='\n'.join(mentioned))
|
|
|
|
if listenchannels := self.bot.settings.get(ctx.guild, 'channels.listen_only'):
|
|
channels = [ctx.guild.get_channel(channel) for channel in listenchannels]
|
|
mentioned = [channel.name for channel in channels if channel is not None]
|
|
embed.add_field(name='{current.listenchannels}', value='\n'.join(mentioned))
|
|
|
|
djroles = self.bot.settings.get(ctx.guild, 'roles.dj')
|
|
if djroles:
|
|
roles = [ctx.guild.get_role(role) for role in djroles]
|
|
mentioned = [role.name for role in roles if role is not None]
|
|
embed.add_field(name='{current.djroles}', value=', '.join(mentioned))
|
|
|
|
embed = ctx.localizer.format_embed(embed, _guild=ctx.guild.name)
|
|
msg = await ctx.send(embed=embed)
|
|
return msg
|
|
|
|
|
|
async def setup(bot):
|
|
await bot.add_cog(Settings(bot))
|