Compare commits
10 Commits
v3.0
...
3335009692
| Author | SHA1 | Date | |
|---|---|---|---|
| 3335009692 | |||
| d7604b6604 | |||
| 2e493e404b | |||
| 16822e0212 | |||
| 9d78352ea5 | |||
| 13b3e9910a | |||
| 1b89d2ef3b | |||
| 6debffbd77 | |||
| e08c1c0204 | |||
| 82f0387675 |
2
Heidi_User.conf
Normal file
2
Heidi_User.conf
Normal file
@ -0,0 +1,2 @@
|
||||
[ENTRANCE.SOUND]
|
||||
|
||||
411
bot.py
411
bot.py
@ -1,220 +1,20 @@
|
||||
# Example: https://github.com/Rapptz/discord.py/blob/master/examples/app_commands/basic.py
|
||||
|
||||
import os, re, random, logging, asyncio, discord, configparser
|
||||
from discord import app_commands
|
||||
import random, logging
|
||||
from discord.app_commands import Choice
|
||||
from functools import reduce
|
||||
from dotenv import load_dotenv
|
||||
from typing import Dict, List, Optional, Union
|
||||
|
||||
# TODO: Reenable + extend textgen
|
||||
# from textgen import textgen
|
||||
# from textgen_markov import MarkovTextGenerator
|
||||
# from textgen_lstm import LSTMTextGenerator
|
||||
|
||||
# TODO: Reenable + extend scraper
|
||||
# from models import Models
|
||||
|
||||
# We're fancy today
|
||||
from typing import Dict, List, Optional, Union, Callable, Any
|
||||
from rich.traceback import install
|
||||
|
||||
from heidi_client import *
|
||||
|
||||
# Install rich traceback
|
||||
install(show_locals=True)
|
||||
load_dotenv()
|
||||
|
||||
|
||||
# ================================================================================================ #
|
||||
# ================================================================================================ #
|
||||
# NOTE: Always set this correctly:
|
||||
DOCKER = os.getenv("DOCKER") == "True"
|
||||
# ================================================================================================ #
|
||||
# ================================================================================================ #
|
||||
|
||||
|
||||
# TODO: Only post in heidi-spam channel
|
||||
# TODO: yt-dlp music support
|
||||
# TODO: Somehow upload voicelines more easily (from discord voice message?)
|
||||
|
||||
# IDs of the servers Heidi is used on
|
||||
LINUS_GUILD = discord.Object(id=431154792308408340)
|
||||
TEST_GUILD = discord.Object(id=821511861178204161)
|
||||
|
||||
CONFIGPATH = "/config" if DOCKER else "."
|
||||
USERCONFIGNAME = "Heidi_User.conf"
|
||||
|
||||
|
||||
class HeidiClient(discord.Client):
|
||||
def __init__(self, *, intents: discord.Intents):
|
||||
super().__init__(status="Nur eine kann GNTM werden!", intents=intents)
|
||||
|
||||
# Separate object that keeps all application command state
|
||||
self.tree = app_commands.CommandTree(self)
|
||||
|
||||
# Handle persistent user configuration
|
||||
self.user_config = configparser.ConfigParser()
|
||||
if not os.path.exists(f"{CONFIGPATH}/{USERCONFIGNAME}"):
|
||||
os.mknod(f"{CONFIGPATH}/{USERCONFIGNAME}")
|
||||
self.user_config.read(f"{CONFIGPATH}/{USERCONFIGNAME}")
|
||||
self.update_to_default_user_config()
|
||||
self.print_user_config()
|
||||
|
||||
# self.models = Models() # scraped model list
|
||||
|
||||
# automatic actions on all messages
|
||||
# on_message_triggers is a map with tuples of two functions: (predicate, action)
|
||||
# the predicate receives the message as argument
|
||||
# if the predicate is true the action is performed
|
||||
self.on_message_triggers = {
|
||||
# lambda m: m.author.nick.lower() in self.models.get_in_names(): self.autoreact_to_girls,
|
||||
lambda m: "jeremy" in m.author.nick.lower(): self._autoreact_to_jeremy,
|
||||
lambda m: "kardashian" in m.author.nick.lower()
|
||||
or "jenner" in m.author.nick.lower(): self._autoreact_to_kardashian,
|
||||
}
|
||||
|
||||
# automatic actions on voice state changes
|
||||
# on_voice_state_triggers is a map with tuples of two functions: (predicate, action)
|
||||
# the predicate receives the member, before- and after-state as arguments
|
||||
# if the predicate is true, the action is performed
|
||||
self.on_voice_state_triggers = {
|
||||
lambda m, b, a: b.channel != a.channel
|
||||
and a.channel != None
|
||||
and isinstance(a.channel, discord.VoiceChannel): self._play_entrance_sound,
|
||||
}
|
||||
|
||||
# Textgen
|
||||
# self.textgen_models: dict[str, textgen] = {
|
||||
# # The name must correspond to the name of the training text file
|
||||
# "kommunistisches_manifest": LSTMTextGenerator(10),
|
||||
# "musk": LSTMTextGenerator(10),
|
||||
# "bibel": LSTMTextGenerator(10)
|
||||
# "bibel": MarkovTextGenerator(3), # Prefix length of 3
|
||||
# "kommunistisches_manifest": MarkovTextGenerator(3),
|
||||
# "musk": MarkovTextGenerator(3)
|
||||
# }
|
||||
|
||||
# for name, model in self.textgen_models.items():
|
||||
# model.init(name) # Loads the textfile
|
||||
|
||||
# if os.path.exists(f"weights/{name}_lstm_model.pt"):
|
||||
# model.load()
|
||||
# elif not DOCKER:
|
||||
# model.train()
|
||||
# else:
|
||||
# print("Error: Can't load model", name)
|
||||
|
||||
# print("Generating test sentence for", name)
|
||||
# self.textgen_models[name].generate_sentence()
|
||||
|
||||
# Synchronize commands to guilds
|
||||
async def setup_hook(self):
|
||||
self.tree.copy_global_to(guild=LINUS_GUILD)
|
||||
await self.tree.sync(guild=LINUS_GUILD)
|
||||
|
||||
self.tree.copy_global_to(guild=TEST_GUILD)
|
||||
await self.tree.sync(guild=TEST_GUILD)
|
||||
|
||||
def update_to_default_user_config(self):
|
||||
"""
|
||||
Adds config keys to the config, if they don't exist yet.
|
||||
"""
|
||||
user_config_sections = ["ENTRANCE.SOUND"]
|
||||
|
||||
for section in user_config_sections:
|
||||
if section not in self.user_config:
|
||||
print(f"Adding section {section} to {CONFIGPATH}/{USERCONFIGNAME}")
|
||||
self.user_config[section] = dict()
|
||||
|
||||
self.write_user_config()
|
||||
|
||||
def print_user_config(self):
|
||||
print("Read persistent configuration:\n")
|
||||
|
||||
for section in self.user_config.sections():
|
||||
print(f"[{section}]")
|
||||
for key in self.user_config[section]:
|
||||
print(f"{key}={self.user_config[section][key]}")
|
||||
|
||||
print("")
|
||||
|
||||
def write_user_config(self):
|
||||
if not os.path.exists(f"{CONFIGPATH}/{USERCONFIGNAME}"):
|
||||
print(f"Error: {CONFIGPATH}/{USERCONFIGNAME} doesn't exist!")
|
||||
return
|
||||
|
||||
print(f"Writing {CONFIGPATH}/{USERCONFIGNAME}")
|
||||
|
||||
with open(f"{CONFIGPATH}/{USERCONFIGNAME}", "w") as file:
|
||||
self.user_config.write(file)
|
||||
|
||||
# Commands -----------------------------------------------------------------------------------
|
||||
|
||||
# async def list_models_in(self, message):
|
||||
# """
|
||||
# wer ist dabei?
|
||||
# """
|
||||
# await message.channel.send("\n".join(self.models.get_in_names()))
|
||||
|
||||
# async def list_models_out(self, message):
|
||||
# """
|
||||
# wer ist raus? (Liste der Keks welche ge*ickt wurden)
|
||||
# """
|
||||
# await message.channel.send("\n".join(self.models.get_out_names()))
|
||||
|
||||
# async def show_model_picture(self, message):
|
||||
# """
|
||||
# gib Bild von <Name>
|
||||
# """
|
||||
# name = message.content.split()[-1]
|
||||
# picture = discord.Embed()
|
||||
# picture.set_image(url=self.models.get_image(name))
|
||||
# picture.set_footer(text=name)
|
||||
# await message.channel.send(embed=picture)
|
||||
|
||||
# Automatic Actions --------------------------------------------------------------------------
|
||||
|
||||
# @staticmethod
|
||||
# async def autoreact_to_girls(message):
|
||||
# """
|
||||
# ❤ aktives Model
|
||||
# """
|
||||
# await message.add_reaction("❤")
|
||||
|
||||
@staticmethod
|
||||
async def _autoreact_to_jeremy(message: discord.Message):
|
||||
"""
|
||||
🧀 Jeremy
|
||||
"""
|
||||
await message.add_reaction("🧀")
|
||||
|
||||
@staticmethod
|
||||
async def _autoreact_to_kardashian(message: discord.Message):
|
||||
"""
|
||||
💄 Kardashian
|
||||
"""
|
||||
await message.add_reaction("💄")
|
||||
|
||||
async def _play_entrance_sound(
|
||||
self,
|
||||
member: discord.Member,
|
||||
before: discord.VoiceState,
|
||||
after: discord.VoiceState,
|
||||
):
|
||||
soundpath: Union[str, None] = self.user_config["ENTRANCE.SOUND"].get(
|
||||
member.name, None
|
||||
)
|
||||
|
||||
if soundpath == None:
|
||||
print(f"User {member.name} has not set an entrance sound")
|
||||
return
|
||||
|
||||
board, sound = soundpath.split("/")
|
||||
|
||||
# Wait a bit to not have simultaneous joins
|
||||
await asyncio.sleep(1)
|
||||
|
||||
await play_voice_line_for_member(None, member, board, sound)
|
||||
|
||||
|
||||
# ------------------------------------------------------------------------------------------------
|
||||
|
||||
# Log to file
|
||||
handler = logging.FileHandler(filename="discord.log", encoding="utf-8", mode="w")
|
||||
@ -228,20 +28,27 @@ intents.voice_states = True # Allow to process on_voice_state_update
|
||||
# Set up our client
|
||||
client = HeidiClient(intents=intents)
|
||||
|
||||
|
||||
# Events -----------------------------------------------------------------------------------------
|
||||
# NOTE: I defined the events outside of the Client class, don't know if I like it or not...
|
||||
# NOTE: I defined the events outside the Client class, don't know if I like it or not...
|
||||
|
||||
|
||||
@client.event
|
||||
async def on_ready():
|
||||
if client.user != None:
|
||||
async def on_ready() -> None:
|
||||
"""
|
||||
This event triggers when the Heidi client has finished connecting.
|
||||
"""
|
||||
if client.user is not None:
|
||||
print(f"{client.user} (id: {client.user.id}) has connected to Discord!")
|
||||
else:
|
||||
print("client.user is None!")
|
||||
|
||||
|
||||
@client.event
|
||||
async def on_message(message: discord.Message):
|
||||
async def on_message(message: Message) -> None:
|
||||
"""
|
||||
This event triggers when a message is sent in any text channel.
|
||||
"""
|
||||
# Skip Heidis own messages
|
||||
if message.author == client.user:
|
||||
return
|
||||
@ -257,8 +64,11 @@ async def on_message(message: discord.Message):
|
||||
|
||||
@client.event
|
||||
async def on_voice_state_update(
|
||||
member: discord.Member, before: discord.VoiceState, after: discord.VoiceState
|
||||
):
|
||||
member: Member, before: VoiceState, after: VoiceState
|
||||
) -> None:
|
||||
"""
|
||||
This event triggers when a member joins/changes/leaves a voice channel or mutes/unmutes.
|
||||
"""
|
||||
# Skip Heidis own voice state updates (e.g. on /say)
|
||||
if member._user == client.user:
|
||||
return
|
||||
@ -267,7 +77,7 @@ async def on_voice_state_update(
|
||||
# python iterates over the keys of a map
|
||||
for predicate in client.on_voice_state_triggers:
|
||||
if predicate(member, before, after):
|
||||
action = client.on_voice_state_triggers[predicate]
|
||||
action: Callable = client.on_voice_state_triggers[predicate]
|
||||
print(f"on_voice_state_update: calling {action.__name__}")
|
||||
await action(member, before, after)
|
||||
|
||||
@ -276,8 +86,11 @@ async def on_voice_state_update(
|
||||
|
||||
|
||||
async def user_config_key_autocomplete(
|
||||
interaction: discord.Interaction, current: str
|
||||
interaction: Interaction, current: str
|
||||
) -> List[Choice[str]]:
|
||||
"""
|
||||
Suggest a value from the user config keys (each .conf section is a key).
|
||||
"""
|
||||
return [
|
||||
Choice(name=key, value=key)
|
||||
for key in client.user_config.sections()
|
||||
@ -286,7 +99,7 @@ async def user_config_key_autocomplete(
|
||||
|
||||
|
||||
async def user_config_value_autocomplete(
|
||||
interaction: discord.Interaction, current: str
|
||||
interaction: Interaction, current: str
|
||||
) -> List[Choice[str]]:
|
||||
"""
|
||||
Calls an autocomplete function depending on the entered config_key.
|
||||
@ -300,7 +113,7 @@ async def user_config_value_autocomplete(
|
||||
|
||||
|
||||
def user_entrance_sound_autocomplete(
|
||||
interaction: discord.Interaction, current: str
|
||||
interaction: Interaction, current: str
|
||||
) -> List[Choice[str]]:
|
||||
"""
|
||||
Generates autocomplete options for the ENTRANCE.SOUND config key.
|
||||
@ -311,6 +124,7 @@ def user_entrance_sound_autocomplete(
|
||||
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,
|
||||
@ -337,15 +151,18 @@ def user_entrance_sound_autocomplete(
|
||||
)
|
||||
@app_commands.autocomplete(config_value=user_config_value_autocomplete)
|
||||
async def user_config(
|
||||
interaction: discord.Interaction, config_key: str, config_value: str
|
||||
):
|
||||
interaction: Interaction, config_key: str, config_value: str
|
||||
) -> None:
|
||||
"""
|
||||
Set a user config value for the calling user.
|
||||
"""
|
||||
# Only Members can set settings
|
||||
if not isinstance(interaction.user, discord.Member):
|
||||
if not isinstance(interaction.user, Member):
|
||||
print("User not a member")
|
||||
await interaction.response.send_message("Heidi sagt: Komm in die Gruppe!")
|
||||
return
|
||||
|
||||
member: discord.Member = interaction.user
|
||||
member: Member = interaction.user
|
||||
|
||||
client.user_config[config_key][member.name] = config_value
|
||||
client.write_user_config()
|
||||
@ -358,21 +175,11 @@ async def user_config(
|
||||
# Commands ---------------------------------------------------------------------------------------
|
||||
|
||||
|
||||
@client.tree.command(
|
||||
name="giblinkbruder",
|
||||
description="Heidi hilft mit dem Link zu deiner Lieblingsshow im Qualitätsfernsehen.",
|
||||
)
|
||||
async def show_link(interaction: discord.Interaction):
|
||||
link_pro7 = "https://www.prosieben.de/tv/germanys-next-topmodel/livestream"
|
||||
link_joyn = "https://www.joyn.de/serien/germanys-next-topmodel"
|
||||
|
||||
await interaction.response.send_message(
|
||||
f"ProSieben: {link_pro7}\nJoyn: {link_joyn}"
|
||||
)
|
||||
|
||||
|
||||
@client.tree.command(name="heidi", description="Heidi!")
|
||||
async def heidi_exclaim(interaction: discord.Interaction):
|
||||
async def heidi_exclaim(interaction: Interaction) -> None:
|
||||
"""
|
||||
Print a random Heidi quote.
|
||||
"""
|
||||
messages = [
|
||||
"Die sind doch fast 18!",
|
||||
"Heidi!",
|
||||
@ -388,16 +195,27 @@ async def heidi_exclaim(interaction: discord.Interaction):
|
||||
@client.tree.command(name="miesmuschel", description="Was denkt Heidi?")
|
||||
@app_commands.rename(question="frage")
|
||||
@app_commands.describe(question="Heidi wird es beantworten!")
|
||||
async def magic_shell(interaction: discord.Interaction, question: str):
|
||||
async def magic_shell(interaction: Interaction, question: str) -> None:
|
||||
"""
|
||||
Answer a yes/no question.
|
||||
"""
|
||||
# Should be equal amounts of yes/no answers, to have a 50/50 chance.
|
||||
choices = [
|
||||
"Ja!",
|
||||
"Jo.",
|
||||
"Jo",
|
||||
"Total!",
|
||||
"Natürlich.",
|
||||
"Natürlich",
|
||||
"Klaro Karo",
|
||||
"Offensichtlich Sherlock",
|
||||
"Tom sagt Ja",
|
||||
|
||||
"Nein!",
|
||||
"Nö.",
|
||||
"Nä.",
|
||||
"Niemals!",
|
||||
"Nur über meine Leiche du Hurensohn!",
|
||||
"In deinen Träumen.",
|
||||
"Tom sagt Nein"
|
||||
]
|
||||
question = question.strip()
|
||||
question_mark = "" if question[-1] == "?" else "?"
|
||||
@ -412,52 +230,25 @@ async def magic_shell(interaction: discord.Interaction, question: str):
|
||||
@app_commands.describe(option_a="Ist es vielleicht dies?")
|
||||
@app_commands.rename(option_b="oder")
|
||||
@app_commands.describe(option_b="Oder doch eher das?")
|
||||
async def choose(interaction: discord.Interaction, option_a: str, option_b: str):
|
||||
async def choose(interaction: Interaction, option_a: str, option_b: str) -> None:
|
||||
"""
|
||||
Select an answer from two options.
|
||||
"""
|
||||
options = [option_a.strip(), option_b.strip()]
|
||||
await interaction.response.send_message(
|
||||
f"{options[0]} oder {options[1]}?\nHeidi sagt: {random.choice(options)}"
|
||||
)
|
||||
|
||||
|
||||
# TextGen ----------------------------------------------------------------------------------------
|
||||
|
||||
|
||||
# async def quote_model_autocomplete(interaction: discord.Interaction, current: str) -> list[Choice[str]]:
|
||||
# models = client.textgen_models.keys()
|
||||
# return [Choice(name=model, value=model) for model in models]
|
||||
|
||||
# @client.tree.command(name="zitat", description="Heidi zitiert!")
|
||||
# @app_commands.rename(quote_model = "style")
|
||||
# @app_commands.describe(quote_model = "Woraus soll Heidi zitieren?")
|
||||
# @app_commands.autocomplete(quote_model = quote_model_autocomplete)
|
||||
# async def quote(interaction: discord.Interaction, quote_model: str):
|
||||
# generated_quote = client.textgen_models[quote_model].generate_sentence()
|
||||
# joined_quote = " ".join(generated_quote)
|
||||
# await interaction.response.send_message(f"Heidi zitiert: \"{joined_quote}\"")
|
||||
|
||||
# @client.tree.command(name="vervollständige", description="Heidi beendet den Satz!")
|
||||
# @app_commands.rename(prompt = "satzanfang")
|
||||
# @app_commands.describe(prompt = "Der Satzanfang wird vervollständigt.")
|
||||
# @app_commands.rename(quote_model = "style")
|
||||
# @app_commands.describe(quote_model = "Woraus soll Heidi vervollständigen?")
|
||||
# @app_commands.autocomplete(quote_model = quote_model_autocomplete)
|
||||
# async def complete(interaction: discord.Interaction, prompt: str, quote_model: str):
|
||||
# prompt = re.sub(r"[^a-zäöüß'.,]+", " ", prompt.lower()) # only keep valid chars
|
||||
# generated_quote = client.textgen_models[quote_model].complete_sentence(prompt.split())
|
||||
# joined_quote = " ".join(generated_quote)
|
||||
# await interaction.response.send_message(f"Heidi sagt: \"{joined_quote}\"")
|
||||
|
||||
|
||||
# Sounds -----------------------------------------------------------------------------------------
|
||||
|
||||
|
||||
SOUNDDIR: str = "/sounds" if DOCKER else "./heidi-sounds"
|
||||
|
||||
|
||||
# Example: https://discordpy.readthedocs.io/en/latest/interactions/api.html?highlight=autocomplete#discord.app_commands.autocomplete
|
||||
async def board_autocomplete(
|
||||
interaction: discord.Interaction, current: str
|
||||
interaction: Interaction, current: str
|
||||
) -> List[Choice[str]]:
|
||||
"""
|
||||
Suggest a sound board.
|
||||
"""
|
||||
boards: List[str] = os.listdir(SOUNDDIR)
|
||||
|
||||
return [
|
||||
@ -468,8 +259,11 @@ async def board_autocomplete(
|
||||
|
||||
|
||||
async def sound_autocomplete(
|
||||
interaction: discord.Interaction, current: str
|
||||
interaction: Interaction, current: str
|
||||
) -> List[Choice[str]]:
|
||||
"""
|
||||
Suggest a sound from an already selected board.
|
||||
"""
|
||||
board: str = interaction.namespace.board
|
||||
sounds: List[str] = list(
|
||||
map(lambda x: x.split(".")[0], os.listdir(f"{SOUNDDIR}/{board}/"))
|
||||
@ -488,14 +282,17 @@ async def sound_autocomplete(
|
||||
@app_commands.describe(sound="Was soll Heidi sagen?")
|
||||
@app_commands.autocomplete(board=board_autocomplete)
|
||||
@app_commands.autocomplete(sound=sound_autocomplete)
|
||||
async def say_voiceline(interaction: discord.Interaction, board: str, sound: str):
|
||||
async def say_voiceline(interaction: Interaction, board: str, sound: str) -> None:
|
||||
"""
|
||||
Play a voiceline in the calling member's current voice channel.
|
||||
"""
|
||||
# Only Members can access voice channels
|
||||
if not isinstance(interaction.user, discord.Member):
|
||||
if not isinstance(interaction.user, Member):
|
||||
print("User not a member")
|
||||
await interaction.response.send_message("Heidi sagt: Komm in die Gruppe!")
|
||||
return
|
||||
|
||||
member: discord.Member = interaction.user
|
||||
member: Member = interaction.user
|
||||
|
||||
await play_voice_line_for_member(interaction, member, board, sound)
|
||||
|
||||
@ -506,8 +303,11 @@ async def say_voiceline(interaction: discord.Interaction, board: str, sound: str
|
||||
# Callable on members
|
||||
@client.tree.context_menu(name="beleidigen")
|
||||
async def insult(
|
||||
interaction: discord.Interaction, member: discord.Member
|
||||
): # with message: discord.Message this can be called on a message
|
||||
interaction: Interaction, member: Member
|
||||
) -> None: # with message: discord.Message this can be called on a message
|
||||
"""
|
||||
Send an insult to a member via direct message.
|
||||
"""
|
||||
if not member.dm_channel:
|
||||
await member.create_dm()
|
||||
|
||||
@ -538,63 +338,6 @@ async def insult(
|
||||
) # with ephemeral = True only the caller can see the answer
|
||||
|
||||
|
||||
# Helpers ----------------------------------------------------------------------------------------
|
||||
|
||||
|
||||
async def play_voice_line(
|
||||
interaction: Union[discord.Interaction, None],
|
||||
voice_channel: discord.VoiceChannel,
|
||||
board: str,
|
||||
sound: str,
|
||||
):
|
||||
try:
|
||||
open(f"{SOUNDDIR}/{board}/{sound}.mkv")
|
||||
except IOError:
|
||||
print("Error: Invalid soundfile!")
|
||||
if interaction != None:
|
||||
await interaction.response.send_message(
|
||||
f'Heidi sagt: "{board}/{sound}" kanninich finden bruder'
|
||||
)
|
||||
return
|
||||
|
||||
if interaction != None:
|
||||
await interaction.response.send_message(f'Heidi sagt: "{board}/{sound}"')
|
||||
|
||||
audio_source = discord.FFmpegPCMAudio(
|
||||
f"{SOUNDDIR}/{board}/{sound}.mkv"
|
||||
) # only works from docker
|
||||
voice_client = await voice_channel.connect()
|
||||
voice_client.play(audio_source)
|
||||
|
||||
while voice_client.is_playing():
|
||||
await asyncio.sleep(1)
|
||||
|
||||
await voice_client.disconnect()
|
||||
|
||||
|
||||
async def play_voice_line_for_member(
|
||||
interaction: Union[discord.Interaction, None],
|
||||
member: discord.Member,
|
||||
board: str,
|
||||
sound: str,
|
||||
):
|
||||
# Member needs to be in voice channel to hear audio (Heidi needs to know the channel to join)
|
||||
if (
|
||||
member == None
|
||||
or member.voice == None
|
||||
or member.voice.channel == None
|
||||
or not isinstance(member.voice.channel, discord.VoiceChannel)
|
||||
):
|
||||
print("User not in (valid) voice channel!")
|
||||
if interaction != None:
|
||||
await interaction.response.send_message("Heidi sagt: Komm in den Channel!")
|
||||
return
|
||||
|
||||
voice_channel: discord.VoiceChannel = member.voice.channel
|
||||
|
||||
await play_voice_line(interaction, voice_channel, board, sound)
|
||||
|
||||
|
||||
# ------------------------------------------------------------------------------------------------
|
||||
|
||||
|
||||
|
||||
18
flake.lock
generated
18
flake.lock
generated
@ -6,11 +6,11 @@
|
||||
"systems": "systems"
|
||||
},
|
||||
"locked": {
|
||||
"lastModified": 1700815693,
|
||||
"narHash": "sha256-JtKZEQUzosrCwDsLgm+g6aqbP1aseUl1334OShEAS3s=",
|
||||
"lastModified": 1701787589,
|
||||
"narHash": "sha256-ce+oQR4Zq9VOsLoh9bZT8Ip9PaMLcjjBUHVPzW5d7Cw=",
|
||||
"owner": "numtide",
|
||||
"repo": "devshell",
|
||||
"rev": "7ad1c417c87e98e56dcef7ecd0e0a2f2e5669d51",
|
||||
"rev": "44ddedcbcfc2d52a76b64fb6122f209881bd3e1e",
|
||||
"type": "github"
|
||||
},
|
||||
"original": {
|
||||
@ -24,11 +24,11 @@
|
||||
"systems": "systems_2"
|
||||
},
|
||||
"locked": {
|
||||
"lastModified": 1694529238,
|
||||
"narHash": "sha256-zsNZZGTGnMOf9YpHKJqMSsa0dXbfmxeoJ7xHlrt+xmY=",
|
||||
"lastModified": 1701680307,
|
||||
"narHash": "sha256-kAuep2h5ajznlPMD9rnQyffWG8EM/C73lejGofXvdM8=",
|
||||
"owner": "numtide",
|
||||
"repo": "flake-utils",
|
||||
"rev": "ff7b65b44d01cf9ba6a71320833626af21126384",
|
||||
"rev": "4022d587cbbfd70fe950c1e2083a02621806a725",
|
||||
"type": "github"
|
||||
},
|
||||
"original": {
|
||||
@ -55,11 +55,11 @@
|
||||
},
|
||||
"nixpkgs_2": {
|
||||
"locked": {
|
||||
"lastModified": 1700856099,
|
||||
"narHash": "sha256-RnEA7iJ36Ay9jI0WwP+/y4zjEhmeN6Cjs9VOFBH7eVQ=",
|
||||
"lastModified": 1701693815,
|
||||
"narHash": "sha256-7BkrXykVWfkn6+c1EhFA3ko4MLi3gVG0p9G96PNnKTM=",
|
||||
"owner": "NixOS",
|
||||
"repo": "nixpkgs",
|
||||
"rev": "0bd59c54ef06bc34eca01e37d689f5e46b3fe2f1",
|
||||
"rev": "09ec6a0881e1a36c29d67497693a67a16f4da573",
|
||||
"type": "github"
|
||||
},
|
||||
"original": {
|
||||
|
||||
BIN
heidi-sounds/basic/Suiii.mkv
(Stored with Git LFS)
Normal file
BIN
heidi-sounds/basic/Suiii.mkv
(Stored with Git LFS)
Normal file
Binary file not shown.
BIN
heidi-sounds/drache/Hagebuddne.mkv
(Stored with Git LFS)
Normal file
BIN
heidi-sounds/drache/Hagebuddne.mkv
(Stored with Git LFS)
Normal file
Binary file not shown.
BIN
heidi-sounds/henri/Ich bin der Pablo.mkv
(Stored with Git LFS)
Normal file
BIN
heidi-sounds/henri/Ich bin der Pablo.mkv
(Stored with Git LFS)
Normal file
Binary file not shown.
BIN
heidi-sounds/henri/Kann ich behilflich sein.mkv
(Stored with Git LFS)
Normal file
BIN
heidi-sounds/henri/Kann ich behilflich sein.mkv
(Stored with Git LFS)
Normal file
Binary file not shown.
BIN
heidi-sounds/henri/Yakari.mkv
(Stored with Git LFS)
Normal file
BIN
heidi-sounds/henri/Yakari.mkv
(Stored with Git LFS)
Normal file
Binary file not shown.
BIN
heidi-sounds/tit/Ab in Knast.mkv
(Stored with Git LFS)
Normal file
BIN
heidi-sounds/tit/Ab in Knast.mkv
(Stored with Git LFS)
Normal file
Binary file not shown.
BIN
heidi-sounds/tit/Bitconnect.mkv
(Stored with Git LFS)
Normal file
BIN
heidi-sounds/tit/Bitconnect.mkv
(Stored with Git LFS)
Normal file
Binary file not shown.
BIN
heidi-sounds/vinz/VINZENT.mkv
(Stored with Git LFS)
Normal file
BIN
heidi-sounds/vinz/VINZENT.mkv
(Stored with Git LFS)
Normal file
Binary file not shown.
134
heidi_client.py
Normal file
134
heidi_client.py
Normal file
@ -0,0 +1,134 @@
|
||||
import configparser
|
||||
from discord import app_commands, Message, VoiceState
|
||||
|
||||
from heidi_constants import *
|
||||
from heidi_helpers import *
|
||||
|
||||
|
||||
class HeidiClient(discord.Client):
|
||||
def __init__(self, *, intents: discord.Intents):
|
||||
super().__init__(status="Nur eine kann GNTM werden!", intents=intents)
|
||||
|
||||
# Separate object that keeps all application command state
|
||||
self.tree = app_commands.CommandTree(self)
|
||||
|
||||
# Handle persistent user configuration
|
||||
self.user_config = configparser.ConfigParser()
|
||||
if not os.path.exists(f"{CONFIGPATH}/{USERCONFIGNAME}"):
|
||||
open(f"{CONFIGPATH}/{USERCONFIGNAME}", "x")
|
||||
self.user_config.read(f"{CONFIGPATH}/{USERCONFIGNAME}")
|
||||
self.update_to_default_user_config()
|
||||
self.print_user_config()
|
||||
|
||||
# automatic actions on all messages
|
||||
# on_message_triggers is a map with tuples of two functions: (predicate, action)
|
||||
# the predicate receives the message as argument
|
||||
# if the predicate is true the action is performed
|
||||
self.on_message_triggers = {
|
||||
# lambda m: m.author.nick.lower() in self.models.get_in_names(): self.autoreact_to_girls,
|
||||
lambda m: "jeremy" in m.author.nick.lower(): self._autoreact_to_jeremy,
|
||||
lambda m: "kardashian" in m.author.nick.lower()
|
||||
or "jenner" in m.author.nick.lower(): self._autoreact_to_kardashian,
|
||||
}
|
||||
|
||||
# automatic actions on voice state changes
|
||||
# on_voice_state_triggers is a map with tuples of two functions: (predicate, action)
|
||||
# the predicate receives the member, before- and after-state as arguments
|
||||
# if the predicate is true, the action is performed
|
||||
self.on_voice_state_triggers = {
|
||||
lambda m, b, a: b.channel != a.channel
|
||||
and a.channel is not None
|
||||
and isinstance(a.channel, VoiceChannel): self._play_entrance_sound,
|
||||
}
|
||||
|
||||
# Synchronize commands to guilds
|
||||
async def setup_hook(self):
|
||||
self.tree.copy_global_to(guild=LINUS_GUILD)
|
||||
await self.tree.sync(guild=LINUS_GUILD)
|
||||
|
||||
self.tree.copy_global_to(guild=TEST_GUILD)
|
||||
await self.tree.sync(guild=TEST_GUILD)
|
||||
|
||||
def update_to_default_user_config(self) -> None:
|
||||
"""
|
||||
Adds config keys to the config, if they don't exist yet.
|
||||
This writes the user config file.
|
||||
"""
|
||||
user_config_sections = ["ENTRANCE.SOUND"]
|
||||
|
||||
for section in user_config_sections:
|
||||
if section not in self.user_config:
|
||||
print(f"Adding section {section} to {CONFIGPATH}/{USERCONFIGNAME}")
|
||||
self.user_config[section] = dict()
|
||||
|
||||
self.write_user_config()
|
||||
|
||||
def print_user_config(self) -> None:
|
||||
"""
|
||||
Print the current user config from memory.
|
||||
This does not read the user config file.
|
||||
"""
|
||||
print("Heidi User Config:\n")
|
||||
|
||||
for section in self.user_config.sections():
|
||||
print(f"[{section}]")
|
||||
for key in self.user_config[section]:
|
||||
print(f"{key}={self.user_config[section][key]}")
|
||||
|
||||
print("")
|
||||
|
||||
def write_user_config(self) -> None:
|
||||
"""
|
||||
Write the current configuration to disk.
|
||||
"""
|
||||
if not os.path.exists(f"{CONFIGPATH}/{USERCONFIGNAME}"):
|
||||
print(f"Error: {CONFIGPATH}/{USERCONFIGNAME} doesn't exist!")
|
||||
return
|
||||
|
||||
print(f"Writing {CONFIGPATH}/{USERCONFIGNAME}")
|
||||
|
||||
with open(f"{CONFIGPATH}/{USERCONFIGNAME}", "w") as file:
|
||||
self.user_config.write(file)
|
||||
|
||||
# Automatic Actions ------------------------------------------------------------------------------
|
||||
|
||||
@staticmethod
|
||||
async def _autoreact_to_jeremy(message: Message) -> None:
|
||||
"""
|
||||
🧀 Jeremy.
|
||||
This function is set in on_message_triggers and triggered by the on_message event.
|
||||
"""
|
||||
await message.add_reaction("🧀")
|
||||
|
||||
@staticmethod
|
||||
async def _autoreact_to_kardashian(message: Message) -> None:
|
||||
"""
|
||||
💄 Kardashian.
|
||||
This function is set in on_message_triggers and triggered by the on_message event.
|
||||
"""
|
||||
await message.add_reaction("💄")
|
||||
|
||||
async def _play_entrance_sound(
|
||||
self,
|
||||
member: Member,
|
||||
before: VoiceState,
|
||||
after: VoiceState,
|
||||
) -> None:
|
||||
"""
|
||||
Play a sound when a member joins a voice channel.
|
||||
This function is set in on_voice_state_triggers and triggered by the on_voice_state_update event.
|
||||
"""
|
||||
soundpath: Union[str, None] = self.user_config["ENTRANCE.SOUND"].get(
|
||||
member.name, None
|
||||
)
|
||||
|
||||
if soundpath is None:
|
||||
print(f"User {member.name} has not set an entrance sound")
|
||||
return
|
||||
|
||||
board, sound = soundpath.split("/")
|
||||
|
||||
# Wait a bit to not have simultaneous joins
|
||||
await asyncio.sleep(1)
|
||||
|
||||
await play_voice_line_for_member(None, member, board, sound)
|
||||
26
heidi_constants.py
Normal file
26
heidi_constants.py
Normal file
@ -0,0 +1,26 @@
|
||||
import os
|
||||
import discord
|
||||
from dotenv import load_dotenv
|
||||
|
||||
# This is run when this file is imported
|
||||
load_dotenv()
|
||||
|
||||
print("Debug: Importing heidi_constants.py")
|
||||
|
||||
|
||||
# ================================================================================================ #
|
||||
# ================================================================================================ #
|
||||
# NOTE: Always set this correctly:
|
||||
DOCKER = os.getenv("DOCKER") == "True"
|
||||
# ================================================================================================ #
|
||||
# ================================================================================================ #
|
||||
|
||||
# Constants
|
||||
CONFIGPATH = "/config" if DOCKER else "."
|
||||
USERCONFIGNAME = "Heidi_User.conf"
|
||||
|
||||
SOUNDDIR: str = "/sounds" if DOCKER else "./heidi-sounds"
|
||||
|
||||
# IDs of the servers Heidi is used on
|
||||
LINUS_GUILD = discord.Object(id=431154792308408340)
|
||||
TEST_GUILD = discord.Object(id=821511861178204161)
|
||||
68
heidi_helpers.py
Normal file
68
heidi_helpers.py
Normal file
@ -0,0 +1,68 @@
|
||||
import asyncio
|
||||
from typing import Union
|
||||
|
||||
import discord
|
||||
from discord import Interaction, VoiceChannel, Member
|
||||
|
||||
from heidi_constants import *
|
||||
|
||||
print("Debug: Importing heidi_helpers.py")
|
||||
|
||||
async def play_voice_line(
|
||||
interaction: Union[Interaction, None],
|
||||
voice_channel: VoiceChannel,
|
||||
board: str,
|
||||
sound: str,
|
||||
) -> None:
|
||||
"""
|
||||
Play a voice line in the specified channel.
|
||||
"""
|
||||
try:
|
||||
open(f"{SOUNDDIR}/{board}/{sound}.mkv")
|
||||
except IOError:
|
||||
print("Error: Invalid soundfile!")
|
||||
if interaction is not None:
|
||||
await interaction.response.send_message(
|
||||
f'Heidi sagt: "{board}/{sound}" kanninich finden bruder'
|
||||
)
|
||||
return
|
||||
|
||||
if interaction is not None:
|
||||
await interaction.response.send_message(f'Heidi sagt: "{board}/{sound}"')
|
||||
|
||||
audio_source = discord.FFmpegPCMAudio(
|
||||
f"{SOUNDDIR}/{board}/{sound}.mkv"
|
||||
) # only works from docker
|
||||
voice_client = await voice_channel.connect()
|
||||
voice_client.play(audio_source)
|
||||
|
||||
while voice_client.is_playing():
|
||||
await asyncio.sleep(1)
|
||||
|
||||
await voice_client.disconnect()
|
||||
|
||||
|
||||
async def play_voice_line_for_member(
|
||||
interaction: Union[Interaction, None],
|
||||
member: Member,
|
||||
board: str,
|
||||
sound: str,
|
||||
) -> None:
|
||||
"""
|
||||
Play a voice line in the member's current channel.
|
||||
"""
|
||||
# Member needs to be in voice channel to hear audio (Heidi needs to know the channel to join)
|
||||
if (
|
||||
member is None
|
||||
or member.voice is None
|
||||
or member.voice.channel is None
|
||||
or not isinstance(member.voice.channel, VoiceChannel)
|
||||
):
|
||||
print("User not in (valid) voice channel!")
|
||||
if interaction is not None:
|
||||
await interaction.response.send_message("Heidi sagt: Komm in den Channel!")
|
||||
return
|
||||
|
||||
voice_channel: VoiceChannel = member.voice.channel
|
||||
|
||||
await play_voice_line(interaction, voice_channel, board, sound)
|
||||
Reference in New Issue
Block a user