155 lines
5.5 KiB
Python
155 lines
5.5 KiB
Python
from discord import (
|
|
ApplicationContext,
|
|
SlashCommandGroup,
|
|
VoiceChannel,
|
|
Member,
|
|
VoiceState,
|
|
Cog,
|
|
option,
|
|
utils,
|
|
)
|
|
from discord.ext import commands
|
|
from functions import (
|
|
appendLog,
|
|
changeNomicPerms,
|
|
createUserVoice,
|
|
getMsg,
|
|
guildConfGet,
|
|
guildConfReset,
|
|
guildConfSet,
|
|
isUserVoice,
|
|
isVoiceOfUser,
|
|
loadJson,
|
|
makeEmbed,
|
|
removeUserVoice,
|
|
strToColor,
|
|
)
|
|
|
|
|
|
# =========================================================================================================================
|
|
class CogChannel(commands.Cog):
|
|
def __init__(self, client):
|
|
self.client = client
|
|
|
|
@Cog.listener()
|
|
async def on_voice_state_update(
|
|
self, member: Member, before: VoiceState, after: VoiceState
|
|
):
|
|
config = loadJson("config.json")
|
|
|
|
vc_from = before.channel
|
|
vc_to = after.channel
|
|
|
|
# If user left vc
|
|
if vc_to is None:
|
|
if isUserVoice(vc_from):
|
|
if isVoiceOfUser(vc_from, member):
|
|
await removeUserVoice(vc_from)
|
|
return
|
|
else:
|
|
if loadJson("config.json")["enable_nomic"]:
|
|
await changeNomicPerms("deny", vc_from, member)
|
|
|
|
# If user joined vc
|
|
else:
|
|
if isUserVoice(vc_from):
|
|
if isVoiceOfUser(vc_from, member):
|
|
await removeUserVoice(vc_from)
|
|
else:
|
|
if loadJson("config.json")["enable_nomic"]:
|
|
await changeNomicPerms("deny", vc_from, member)
|
|
if isUserVoice(vc_to):
|
|
if loadJson("config.json")["enable_nomic"]:
|
|
await changeNomicPerms("allow", vc_to, member)
|
|
if vc_to.id == guildConfGet(vc_to.guild, "channel"):
|
|
if guildConfGet(vc_to.guild, "category") is not None:
|
|
voice_chan = await createUserVoice(
|
|
vc_to,
|
|
utils.get(
|
|
vc_to.guild.categories,
|
|
id=guildConfGet(vc_to.guild, "category"),
|
|
),
|
|
member,
|
|
)
|
|
try:
|
|
await member.move_to(voice_chan)
|
|
except:
|
|
await removeUserVoice(voice_chan)
|
|
else:
|
|
appendLog(
|
|
f"Category for guild {vc_to.guild} ({str(vc_to.guild.id)}) is not set",
|
|
guild=vc_to.guild,
|
|
)
|
|
|
|
channel = SlashCommandGroup("channel", "Commands related to parent voice channel")
|
|
|
|
@channel.command(
|
|
name="set",
|
|
description="Select the voice channel that will be parent to private ones",
|
|
)
|
|
@option("channel", description="Parent Voice Channel")
|
|
async def channel_set(self, ctx: ApplicationContext, channel: VoiceChannel):
|
|
config = loadJson("config.json")
|
|
if ctx.guild is not None:
|
|
guildConfSet(ctx.guild, "channel", channel.id)
|
|
await ctx.respond(
|
|
embed=makeEmbed(
|
|
title=getMsg("set_channel_title", ctx.guild),
|
|
description=getMsg("set_channel_description", ctx.guild).format(
|
|
channel.name
|
|
),
|
|
color=strToColor(config["color_ok"]),
|
|
)
|
|
)
|
|
if guildConfGet(ctx.guild, "category") is None:
|
|
await ctx.respond(
|
|
embed=makeEmbed(
|
|
title=getMsg("hint_none_category_title", ctx.guild),
|
|
description=getMsg("hint_none_category_description", ctx.guild),
|
|
color=strToColor(config["color_warn"]),
|
|
)
|
|
)
|
|
else:
|
|
await ctx.respond(
|
|
embed=makeEmbed(
|
|
title=getMsg("dm_title", ctx.guild),
|
|
description=getMsg("dm_description", ctx.guild),
|
|
color=strToColor(config["color_error"]),
|
|
)
|
|
)
|
|
|
|
@channel.command(
|
|
name="reset", description="Reset the currently selected parent voice channel"
|
|
)
|
|
async def channel_reset(self, ctx: ApplicationContext):
|
|
config = loadJson("config.json")
|
|
if ctx.guild is not None:
|
|
if guildConfGet(ctx.guild, "channel") is not None:
|
|
guildConfReset(ctx.guild, "channel")
|
|
await ctx.respond(
|
|
embed=makeEmbed(
|
|
title=getMsg("reset_channel_title", ctx.guild),
|
|
description=getMsg("reset_channel_description", ctx.guild),
|
|
color=strToColor(config["color_ok"]),
|
|
)
|
|
)
|
|
else:
|
|
await ctx.respond(
|
|
embed=makeEmbed(
|
|
title=getMsg("hint_none_channel_title", ctx.guild),
|
|
description=getMsg("hint_none_channel_description", ctx.guild),
|
|
color=strToColor(config["color_warn"]),
|
|
)
|
|
)
|
|
else:
|
|
await ctx.respond(
|
|
embed=makeEmbed(
|
|
title=getMsg("dm_title", ctx.guild),
|
|
description=getMsg("dm_description", ctx.guild),
|
|
color=strToColor(config["color_error"]),
|
|
)
|
|
)
|
|
|
|
|
|
# =========================================================================================================================
|