1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
|
import datetime
import json
import logging
import os
import discord
from discord.ext import commands
from gtts import gTTS
logger = logging.getLogger(__name__)
class VcLogs(commands.Cog):
def __init__(self, bot):
self.bot = bot
@staticmethod
async def handle_state_change(member, channel, state_type, is_enabled, current_time, log_channel):
"""Helper method to handle voice state changes"""
state_messages = {
'deaf': {
True: (
f"[{current_time}] {member.name} ogłuchnął się na {channel.name}",
f"{member.display_name} jest bity przez ojca"
),
False: (
f"[{current_time}] {member.name} odgłuchał się na {channel.name}",
f"{member.display_name} uciek od ojca"
)
},
'mute': {
True: (
f"[{current_time}] {member.name} wyciszył się na {channel.name}",
f"{member.display_name} idzie robić jajecznice"
),
False: (
f"[{current_time}] {member.name} odciszył się na {channel.name}",
f"{member.display_name} zrobił jajecznice"
)
}
}
message, tts_text = state_messages[state_type][is_enabled]
await log_channel.send(message)
logger.info(f"Voice state change: {message}")
return tts_text, channel
async def play_tts(self, channel, text):
# Generate TTS audio
tts = gTTS(text=text, lang='pl')
tts.save('temp_tts.mp3')
# Get the voice client for the guild
voice_client = discord.utils.get(self.bot.voice_clients, guild=channel.guild)
# Handle connection state
if not voice_client or not voice_client.is_connected():
# If no voice client or not connected, create a new connection
try:
voice_client = await channel.connect()
except discord.ClientException as e:
if 'already connected' in str(e).lower():
# If already connected, find the existing connection
voice_client = next((vc for vc in self.bot.voice_clients if vc.guild == channel.guild), None)
else:
raise
elif voice_client.channel != channel:
# If connected to a different channel, move to the new one
await voice_client.move_to(channel)
# Play the TTS audio
if not voice_client.is_playing():
if os.path.exists('settings.json'):
with open('settings.json') as f:
data = json.load(f)
volume = data['volume']
else:
with open('settings.json', 'w') as f:
json.dump({'volume': '5'}, f, indent=4)
logger.info("settings.json created")
audio_source = discord.FFmpegPCMAudio('temp_tts.mp3', options=f'-filter:a "volume={volume}"')
voice_client.play(audio_source)
while voice_client.is_playing():
await discord.utils.sleep_until(datetime.datetime.now() + datetime.timedelta(seconds=1))
os.remove('temp_tts.mp3')
@commands.slash_command(
name="log_volume"
)
async def volume(self, ctx: discord.ApplicationContext, volume: int):
try:
if not os.path.exists('settings.json'):
with open('settings.json', 'w') as f:
json.dump({'volume': '5'}, f, indent=4)
logger.info("settings.json created")
with open('settings.json', 'w') as f:
json.dump({"volume": volume}, f, indent=4)
await ctx.followup.send(f"Volume has been set to: {volume}")
logger.info(f"Volume changed to: {volume}")
except Exception as e:
await ctx.respond(e, ephemeral=True)
logger.error(f"Error in volume command: {e}", exc_info=True)
@commands.Cog.listener()
async def on_voice_state_update(self, member, before, after):
log_channel = self.bot.get_channel(1213843195457183885)
current_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
tts_text = None
tts_channel = None
# Ignore if the user is a bot
if member.bot:
return
if before.channel is None and after.channel is not None:
message = f"[{current_time}] {member.name} dołączył na {after.channel.name}"
tts_text = f"{member.display_name} sie zespawnił"
tts_channel = after.channel
await log_channel.send(message)
logger.info(f"User joined voice channel: {message}")
elif before.channel is not None and after.channel is None:
message = f"[{current_time}] {member.name} rozłączył się z {before.channel.name}"
tts_text = f"{member.display_name} jest chujem"
tts_channel = before.channel
await log_channel.send(message)
logger.info(f"User left voice channel: {message}")
if before.self_deaf != after.self_deaf:
tts_text, tts_channel = await self.handle_state_change(
member, after.channel, 'deaf', after.self_deaf, current_time, log_channel
)
if tts_text and tts_channel:
await self.play_tts(tts_channel, tts_text)
return
if before.self_mute != after.self_mute:
tts_text, tts_channel = await self.handle_state_change(
member, after.channel, 'mute', after.self_mute, current_time, log_channel
)
if tts_text and tts_channel:
await self.play_tts(tts_channel, tts_text)
def setup(bot):
bot.add_cog(VcLogs(bot))
|
