Replace ENTRANCE.SOUND menu with dropdowns
All checks were successful
Build Heidi Docker image / build-docker (push) Successful in 14s
All checks were successful
Build Heidi Docker image / build-docker (push) Successful in 14s
This commit is contained in:
159
bot.py
159
bot.py
@ -1,8 +1,10 @@
|
|||||||
# Example: https://github.com/Rapptz/discord.py/blob/master/examples/app_commands/basic.py
|
# Example: https://github.com/Rapptz/discord.py/blob/master/examples/app_commands/basic.py
|
||||||
|
|
||||||
|
from ast import Call
|
||||||
import random, logging
|
import random, logging
|
||||||
|
from discord import DMChannel
|
||||||
from discord.app_commands import Choice
|
from discord.app_commands import Choice
|
||||||
from typing import Dict, List, Optional, Union, Callable, Any
|
from typing import Awaitable, Dict, List, Optional, Union, Callable, Any
|
||||||
from rich.traceback import install
|
from rich.traceback import install
|
||||||
|
|
||||||
from heidi_client import *
|
from heidi_client import *
|
||||||
@ -11,9 +13,7 @@ from heidi_client import *
|
|||||||
install(show_locals=True)
|
install(show_locals=True)
|
||||||
|
|
||||||
|
|
||||||
# @todo Only post in heidi-spam channel
|
|
||||||
# @todo yt-dlp music support
|
# @todo yt-dlp music support
|
||||||
# @todo Somehow upload voicelines more easily (from discord voice message?)
|
|
||||||
|
|
||||||
|
|
||||||
# Log to file
|
# Log to file
|
||||||
@ -85,6 +85,59 @@ async def on_voice_state_update(
|
|||||||
# Config Commands --------------------------------------------------------------------------------
|
# Config Commands --------------------------------------------------------------------------------
|
||||||
|
|
||||||
|
|
||||||
|
class EntranceSoundSoundSelect(discord.ui.Select):
|
||||||
|
def __init__(self, board: str, on_sound_select_callback):
|
||||||
|
self.board = board
|
||||||
|
self.on_sound_select_callback = on_sound_select_callback
|
||||||
|
|
||||||
|
options: List[discord.SelectOption] = [
|
||||||
|
discord.SelectOption(label=sound.split(".")[0], value=sound)
|
||||||
|
for sound in os.listdir(f"{SOUNDDIR}/{board}")
|
||||||
|
]
|
||||||
|
|
||||||
|
super().__init__(
|
||||||
|
placeholder="Select Sound", min_values=1, max_values=1, options=options
|
||||||
|
)
|
||||||
|
|
||||||
|
async def callback(self, interaction: Interaction):
|
||||||
|
await self.on_sound_select_callback(interaction, self.board, self.values[0])
|
||||||
|
|
||||||
|
|
||||||
|
class EntranceSoundSoundView(discord.ui.View):
|
||||||
|
def __init__(self, board: str, on_sound_select_callback):
|
||||||
|
super().__init__(timeout=600)
|
||||||
|
|
||||||
|
self.add_item(EntranceSoundSoundSelect(board, on_sound_select_callback))
|
||||||
|
|
||||||
|
|
||||||
|
class EntranceSoundBoardSelect(discord.ui.Select):
|
||||||
|
def __init__(self, on_sound_select_callback):
|
||||||
|
self.on_sound_select_callback = on_sound_select_callback
|
||||||
|
|
||||||
|
options: List[discord.SelectOption] = [
|
||||||
|
discord.SelectOption(label=board, value=board)
|
||||||
|
for board in os.listdir(f"{SOUNDDIR}")
|
||||||
|
]
|
||||||
|
|
||||||
|
super().__init__(
|
||||||
|
placeholder="Select Board", min_values=1, max_values=1, options=options
|
||||||
|
)
|
||||||
|
|
||||||
|
async def callback(self, interaction: Interaction):
|
||||||
|
await interaction.response.send_message(
|
||||||
|
f"Welchen sound willst du?",
|
||||||
|
view=EntranceSoundSoundView(self.values[0], self.on_sound_select_callback),
|
||||||
|
ephemeral=True,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class EntranceSoundBoardView(discord.ui.View):
|
||||||
|
def __init__(self, on_sound_select_callback):
|
||||||
|
super().__init__(timeout=600)
|
||||||
|
|
||||||
|
self.add_item(EntranceSoundBoardSelect(on_sound_select_callback))
|
||||||
|
|
||||||
|
|
||||||
async def user_config_key_autocomplete(
|
async def user_config_key_autocomplete(
|
||||||
interaction: Interaction, current: str
|
interaction: Interaction, current: str
|
||||||
) -> List[Choice[str]]:
|
) -> List[Choice[str]]:
|
||||||
@ -98,45 +151,6 @@ async def user_config_key_autocomplete(
|
|||||||
]
|
]
|
||||||
|
|
||||||
|
|
||||||
async def user_config_value_autocomplete(
|
|
||||||
interaction: Interaction, current: str
|
|
||||||
) -> List[Choice[str]]:
|
|
||||||
"""
|
|
||||||
Calls an autocomplete function depending on the entered config_key.
|
|
||||||
"""
|
|
||||||
autocompleters = {"ENTRANCE.SOUND": user_entrance_sound_autocomplete}
|
|
||||||
autocompleter = autocompleters[interaction.namespace.option]
|
|
||||||
|
|
||||||
print(f"config_value_autocomplete: calling {autocompleter.__name__}")
|
|
||||||
|
|
||||||
return autocompleter(interaction, current)
|
|
||||||
|
|
||||||
|
|
||||||
def user_entrance_sound_autocomplete(
|
|
||||||
interaction: Interaction, current: str
|
|
||||||
) -> List[Choice[str]]:
|
|
||||||
"""
|
|
||||||
Generates autocomplete options for the ENTRANCE.SOUND config key.
|
|
||||||
"""
|
|
||||||
boards: List[str] = os.listdir(SOUNDDIR)
|
|
||||||
all_sounds: Dict[str, List[str]] = {
|
|
||||||
board: os.listdir(f"{SOUNDDIR}/{board}/") for board in boards
|
|
||||||
} # These are all sounds, organized per board
|
|
||||||
|
|
||||||
# @todo Initially only suggest boards, because there are too many sounds to show them all
|
|
||||||
completions: List[Choice[str]] = []
|
|
||||||
for (
|
|
||||||
board,
|
|
||||||
board_sounds,
|
|
||||||
) in all_sounds.items(): # Iterate over all sounds, organized per board
|
|
||||||
for sound in board_sounds: # Iterate over board specific sounds
|
|
||||||
soundpath = f"{board}/{sound}"
|
|
||||||
if soundpath.lower().startswith(current.lower()):
|
|
||||||
completions += [Choice(name=soundpath.split(".")[0], value=soundpath)]
|
|
||||||
|
|
||||||
return completions
|
|
||||||
|
|
||||||
|
|
||||||
@client.tree.command(
|
@client.tree.command(
|
||||||
name="userconfig",
|
name="userconfig",
|
||||||
description="User-spezifische Heidi-Einstellungen (Heidi merkt sie sich in ihrem riesigen Gehirn).",
|
description="User-spezifische Heidi-Einstellungen (Heidi merkt sie sich in ihrem riesigen Gehirn).",
|
||||||
@ -144,32 +158,42 @@ def user_entrance_sound_autocomplete(
|
|||||||
@app_commands.rename(config_key="option")
|
@app_commands.rename(config_key="option")
|
||||||
@app_commands.describe(config_key="Die Option, welche du ändern willst.")
|
@app_commands.describe(config_key="Die Option, welche du ändern willst.")
|
||||||
@app_commands.autocomplete(config_key=user_config_key_autocomplete)
|
@app_commands.autocomplete(config_key=user_config_key_autocomplete)
|
||||||
@app_commands.rename(config_value="wert")
|
|
||||||
@app_commands.describe(
|
|
||||||
config_value="Der Wert, auf welche die Option gesetzt werden soll."
|
|
||||||
)
|
|
||||||
@app_commands.autocomplete(config_value=user_config_value_autocomplete)
|
|
||||||
@enforce_channel(HEIDI_SPAM_ID)
|
@enforce_channel(HEIDI_SPAM_ID)
|
||||||
async def user_config(
|
async def user_config(interaction: Interaction, config_key: str) -> None:
|
||||||
interaction: Interaction, config_key: str, config_value: str
|
|
||||||
) -> None:
|
|
||||||
"""
|
"""
|
||||||
Set a user config value for the calling user.
|
Set a user config value for the calling user.
|
||||||
"""
|
"""
|
||||||
# Only Members can set settings
|
# Only Members can set settings
|
||||||
if not isinstance(interaction.user, Member):
|
if not isinstance(interaction.user, Member):
|
||||||
print("User not a member")
|
print("User not a member")
|
||||||
await interaction.response.send_message("Heidi sagt: Komm in die Gruppe!", ephemeral=True)
|
await interaction.response.send_message(
|
||||||
|
"Heidi sagt: Komm in die Gruppe!", ephemeral=True
|
||||||
|
)
|
||||||
return
|
return
|
||||||
|
|
||||||
member: Member = interaction.user
|
member: Member = interaction.user
|
||||||
|
|
||||||
client.user_config[config_key][member.name] = config_value
|
async def on_sound_select_callback(interaction, board: str, sound: str):
|
||||||
client.write_user_config()
|
"""
|
||||||
|
This function is called, when an EntrySoundSoundSelect option is selected.
|
||||||
|
"""
|
||||||
|
client.user_config[config_key][member.name] = f"{board}/{sound}"
|
||||||
|
client.write_user_config()
|
||||||
|
|
||||||
|
await interaction.response.send_message(
|
||||||
|
f"Ok, ich schreibe {member.name}={board}/{sound} in mein fettes Gehirn!",
|
||||||
|
ephemeral=True,
|
||||||
|
)
|
||||||
|
|
||||||
|
# Views for different user config options are defined here
|
||||||
|
views = {"ENTRANCE.SOUND": (EntranceSoundBoardView, on_sound_select_callback)}
|
||||||
|
|
||||||
|
view, select_callback = views[config_key]
|
||||||
|
|
||||||
await interaction.response.send_message(
|
await interaction.response.send_message(
|
||||||
f"Ok, ich schreibe {member.name}={config_value} in mein fettes Gehirn!",
|
f"Aus welchem Soundboard soll dein sound sein?",
|
||||||
ephemeral=True
|
view=view(select_callback),
|
||||||
|
ephemeral=True,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
@ -295,7 +319,9 @@ async def say_voiceline(interaction: Interaction, board: str, sound: str) -> Non
|
|||||||
# Only Members can access voice channels
|
# Only Members can access voice channels
|
||||||
if not isinstance(interaction.user, Member):
|
if not isinstance(interaction.user, Member):
|
||||||
print("User not a member")
|
print("User not a member")
|
||||||
await interaction.response.send_message("Heidi sagt: Komm in die Gruppe!", ephemeral=True)
|
await interaction.response.send_message(
|
||||||
|
"Heidi sagt: Komm in die Gruppe!", ephemeral=True
|
||||||
|
)
|
||||||
return
|
return
|
||||||
|
|
||||||
member: Member = interaction.user
|
member: Member = interaction.user
|
||||||
@ -315,13 +341,17 @@ class InstantButton(discord.ui.Button):
|
|||||||
Handle a press of the button.
|
Handle a press of the button.
|
||||||
"""
|
"""
|
||||||
if not isinstance(interaction.user, Member):
|
if not isinstance(interaction.user, Member):
|
||||||
await interaction.response.send_message("Heidi mag keine discord.User, nur discord.Member!", ephemeral=True)
|
await interaction.response.send_message(
|
||||||
|
"Heidi mag keine discord.User, nur discord.Member!", ephemeral=True
|
||||||
|
)
|
||||||
return
|
return
|
||||||
|
|
||||||
await play_voice_line_for_member(interaction, interaction.user, self.board, self.sound)
|
await play_voice_line_for_member(
|
||||||
|
interaction, interaction.user, self.board, self.sound
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
class InstantButtons(discord.ui.View):
|
class InstantButtonsView(discord.ui.View):
|
||||||
def __init__(self, board: str, timeout=None):
|
def __init__(self, board: str, timeout=None):
|
||||||
super().__init__(timeout=timeout)
|
super().__init__(timeout=timeout)
|
||||||
|
|
||||||
@ -337,7 +367,9 @@ class InstantButtons(discord.ui.View):
|
|||||||
@app_commands.autocomplete(board=board_autocomplete)
|
@app_commands.autocomplete(board=board_autocomplete)
|
||||||
@enforce_channel(HEIDI_SPAM_ID)
|
@enforce_channel(HEIDI_SPAM_ID)
|
||||||
async def soundboard_buttons(interaction: Interaction, board: str) -> None:
|
async def soundboard_buttons(interaction: Interaction, board: str) -> None:
|
||||||
await interaction.response.send_message(f"Soundboard: {board.capitalize()}", view=InstantButtons(board))
|
await interaction.response.send_message(
|
||||||
|
f"Soundboard: {board.capitalize()}", view=InstantButtonsView(board)
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
# Contextmenu ------------------------------------------------------------------------------------
|
# Contextmenu ------------------------------------------------------------------------------------
|
||||||
@ -356,7 +388,9 @@ async def insult(
|
|||||||
|
|
||||||
if not member.dm_channel:
|
if not member.dm_channel:
|
||||||
print("Error creating DMChannel!")
|
print("Error creating DMChannel!")
|
||||||
await interaction.response.send_message("Heidi sagt: Gib mal DM Nummer süße*r!", ephemeral=True)
|
await interaction.response.send_message(
|
||||||
|
"Heidi sagt: Gib mal DM Nummer süße*r!", ephemeral=True
|
||||||
|
)
|
||||||
return
|
return
|
||||||
|
|
||||||
insults = [
|
insults = [
|
||||||
@ -377,8 +411,7 @@ async def insult(
|
|||||||
|
|
||||||
await member.dm_channel.send(random.choice(insults))
|
await member.dm_channel.send(random.choice(insults))
|
||||||
await interaction.response.send_message(
|
await interaction.response.send_message(
|
||||||
"Anzeige ist raus!",
|
"Anzeige ist raus!", ephemeral=True
|
||||||
ephemeral=True
|
|
||||||
) # with ephemeral = True only the caller can see the answer
|
) # with ephemeral = True only the caller can see the answer
|
||||||
|
|
||||||
|
|
||||||
|
Reference in New Issue
Block a user