Files
zyrarbot/bot.py
T

656 lines
26 KiB
Python

import discord
from discord import app_commands
from discord.ext import commands
import asyncio
import os
from typing import Optional
from dotenv import load_dotenv
from ytdl import YTDLSource
# Load environment variables
load_dotenv()
TOKEN = os.getenv('DISCORD_TOKEN')
GUILD_ID = os.getenv('GUILD_ID')
class MusicBot(commands.Cog):
def __init__(self, bot):
self.bot = bot
self.queues = {} # guild_id -> list
self.currents = {} # guild_id -> player
self.volumes = {} # guild_id -> volume float
self.loop_currents = {} # guild_id -> bool
self.loop_playlists = {} # guild_id -> bool
self.last_channels = {} # guild_id -> text channel
self.history = {} # guild_id -> list of dicts
self.on_state_change_callback = None
def load_history(self, guild_id: int):
if guild_id in self.history:
return self.history[guild_id]
config_path = "config/history.json"
if os.path.exists(config_path):
try:
import json
with open(config_path, "r") as f:
data = json.load(f)
raw_history = data.get(str(guild_id), [])
unique_history = []
seen = set()
for item in raw_history:
key = (item.get("title", ""), item.get("url", ""))
if key not in seen:
seen.add(key)
unique_history.append(item)
self.history[guild_id] = unique_history
except Exception as e:
print(f"Error loading history: {e}")
self.history[guild_id] = []
else:
self.history[guild_id] = []
return self.history[guild_id]
def save_history(self, guild_id: int):
config_path = "config/history.json"
try:
import json
os.makedirs(os.path.dirname(config_path), exist_ok=True)
data = {}
if os.path.exists(config_path):
with open(config_path, "r") as f:
data = json.load(f)
data[str(guild_id)] = self.history.get(guild_id, [])
with open(config_path, "w") as f:
json.dump(data, f, indent=4)
except Exception as e:
print(f"Error saving history: {e}")
def add_to_history(self, guild_id: int, player):
if not player or not player.title:
return
history = self.load_history(guild_id)
entry = {
"id": player.id,
"title": player.title,
"url": player.data.get("webpage_url") or player.url,
"duration": player.duration,
"is_local": player.data.get("is_local", False),
"filepath": player.data.get("filepath")
}
history = [h for h in history if h.get("title") != entry["title"] and (not entry.get("url") or h.get("url") != entry["url"])]
history.insert(0, entry)
self.history[guild_id] = history[:50]
self.save_history(guild_id)
def get_queue(self, guild_id: int):
if guild_id not in self.queues:
self.queues[guild_id] = []
return self.queues[guild_id]
def get_current(self, guild_id: int):
return self.currents.get(guild_id)
def set_current(self, guild_id: int, player):
self.currents[guild_id] = player
def get_volume(self, guild_id: int):
return self.volumes.get(guild_id, 1.0)
def set_volume(self, guild_id: int, volume: float):
self.volumes[guild_id] = volume
def get_loop_current(self, guild_id: int):
return self.loop_currents.get(guild_id, False)
def set_loop_current(self, guild_id: int, loop: bool):
self.loop_currents[guild_id] = loop
def get_loop_playlist(self, guild_id: int):
return self.loop_playlists.get(guild_id, False)
def set_loop_playlist(self, guild_id: int, loop: bool):
self.loop_playlists[guild_id] = loop
async def trigger_state_change(self):
if self.on_state_change_callback:
try:
if asyncio.iscoroutinefunction(self.on_state_change_callback):
await self.on_state_change_callback()
else:
self.on_state_change_callback()
except Exception as e:
print(f"Error executing on_state_change_callback: {e}")
def load_saved_channel_id(self, guild_id: int):
config_path = "config/config.json"
if os.path.exists(config_path):
try:
import json
with open(config_path, "r") as f:
data = json.load(f)
val = data.get(str(guild_id))
return int(val) if val is not None else None
except Exception as e:
print(f"Error loading config: {e}")
return None
def save_channel_id(self, guild_id: int, channel_id: Optional[int]):
config_path = "config/config.json"
try:
import json
os.makedirs(os.path.dirname(config_path), exist_ok=True)
data = {}
if os.path.exists(config_path):
with open(config_path, "r") as f:
data = json.load(f)
if channel_id is None:
if str(guild_id) in data:
del data[str(guild_id)]
else:
data[str(guild_id)] = channel_id
with open(config_path, "w") as f:
json.dump(data, f, indent=4)
except Exception as e:
print(f"Error saving config: {e}")
def get_notification_channel(self, guild_id: int):
if self.last_channels.get(guild_id):
return self.last_channels[guild_id]
saved_id = self.load_saved_channel_id(guild_id)
if saved_id:
channel = self.bot.get_channel(int(saved_id))
if channel:
self.last_channels[guild_id] = channel
return channel
# Fallback to the first accessible channel in the guild
guild = self.bot.get_guild(guild_id)
if guild:
for tc in guild.text_channels:
if tc.permissions_for(guild.me).send_messages:
self.last_channels[guild_id] = tc
return tc
return None
async def ensure_voice(self, interaction):
if not interaction.user.voice:
await interaction.response.send_message(
"Nie jesteś połączony z kanałem głosowym.",
ephemeral=True
)
return False
if interaction.guild.voice_client is None:
await interaction.user.voice.channel.connect()
elif interaction.guild.voice_client.channel != interaction.user.voice.channel:
await interaction.response.send_message(
"Jestem już połączony z innym kanałem głosowym.",
ephemeral=True
)
return False
return True
@app_commands.command(name="play", description="Odtwórz utwór z YouTube")
@app_commands.describe(query="Link YouTube lub zapytanie wyszukiwania")
async def play(self, interaction: discord.Interaction, query: str):
"""Odtwórz utwór z YouTube"""
if not await self.ensure_voice(interaction):
return
await interaction.response.defer()
guild_id = interaction.guild.id
self.last_channels[guild_id] = interaction.channel
try:
filepath = os.path.join("uploads", os.path.basename(query))
if os.path.exists(filepath) and not query.startswith("http"):
player = await YTDLSource.from_file(filepath, loop=self.bot.loop)
else:
player = await YTDLSource.from_url(query, loop=self.bot.loop, stream=True)
if interaction.guild.voice_client.is_playing():
self.get_queue(guild_id).append(player)
await self.trigger_state_change()
await interaction.followup.send(
f"Dodano do kolejki: {discord.utils.escape_markdown(player.title)}",
ephemeral=False
)
else:
self.set_current(guild_id, player)
self.add_to_history(guild_id, player)
await self.trigger_state_change()
interaction.guild.voice_client.play(
player,
after=lambda e: asyncio.run_coroutine_threadsafe(
self.play_next(guild_id),
self.bot.loop
)
)
await interaction.followup.send(
f"Teraz odtwarzane: {discord.utils.escape_markdown(player.title)}",
ephemeral=False
)
except Exception as e:
await interaction.followup.send(
f"Wystąpił błąd: {e}",
ephemeral=True
)
async def uploads_autocomplete(
self,
interaction: discord.Interaction,
current: str,
) -> list[app_commands.Choice[str]]:
uploads_dir = "uploads"
if not os.path.exists(uploads_dir):
return []
files = [f for f in os.listdir(uploads_dir) if os.path.isfile(os.path.join(uploads_dir, f)) and not f.startswith(".")]
filtered = [f for f in files if current.lower() in f.lower()][:25]
return [app_commands.Choice(name=f, value=f) for f in filtered]
@app_commands.command(name="playfile", description="Odtwórz plik przesłany na serwer")
@app_commands.describe(filename="Wybierz plik z listy przesłanych")
@app_commands.autocomplete(filename=uploads_autocomplete)
async def playfile(self, interaction: discord.Interaction, filename: str):
"""Odtwórz plik przesłany na serwer"""
if not await self.ensure_voice(interaction):
return
await interaction.response.defer()
guild_id = interaction.guild.id
self.last_channels[guild_id] = interaction.channel
filepath = os.path.join("uploads", os.path.basename(filename))
if not os.path.exists(filepath):
await interaction.followup.send("Nie znaleziono pliku na serwerze.", ephemeral=True)
return
try:
player = await YTDLSource.from_file(filepath, loop=self.bot.loop)
if interaction.guild.voice_client.is_playing():
self.get_queue(guild_id).append(player)
await self.trigger_state_change()
await interaction.followup.send(
f"Dodano do kolejki: {discord.utils.escape_markdown(player.title)}",
ephemeral=False
)
else:
self.set_current(guild_id, player)
self.add_to_history(guild_id, player)
await self.trigger_state_change()
interaction.guild.voice_client.play(
player,
after=lambda e: asyncio.run_coroutine_threadsafe(
self.play_next(guild_id),
self.bot.loop
)
)
await interaction.followup.send(
f"Teraz odtwarzane: {discord.utils.escape_markdown(player.title)}",
ephemeral=False
)
except Exception as e:
await interaction.followup.send(
f"Wystąpił błąd: {e}",
ephemeral=True
)
async def play_next(self, guild_id: int):
voice_client = discord.utils.get(self.bot.voice_clients, guild__id=guild_id)
if not voice_client:
self.set_current(guild_id, None)
return
loop_current = self.get_loop_current(guild_id)
current = self.get_current(guild_id)
if loop_current and current:
try:
if current.data.get('is_local'):
player = await YTDLSource.from_file(current.data['filepath'], loop=self.bot.loop, title=current.title)
else:
url = current.data.get('webpage_url') or current.url
player = await YTDLSource.from_url(url, loop=self.bot.loop, stream=True)
self.set_current(guild_id, player)
await self.trigger_state_change()
voice_client.play(
player,
after=lambda e: asyncio.run_coroutine_threadsafe(
self.play_next(guild_id),
self.bot.loop
)
)
await self.send_to_last_channel(guild_id, f"Teraz odtwarzane (zapętlone): {discord.utils.escape_markdown(player.title)}")
return
except Exception as e:
print(f"Error repeating song: {e}")
elif self.get_loop_playlist(guild_id) and current:
try:
if current.data.get('is_local'):
new_player = await YTDLSource.from_file(current.data['filepath'], loop=self.bot.loop, title=current.title)
else:
url = current.data.get('webpage_url') or current.url
new_player = await YTDLSource.from_url(url, loop=self.bot.loop, stream=True)
self.get_queue(guild_id).append(new_player)
await self.trigger_state_change()
except Exception as e:
print(f"Error re-adding track to playlist loop: {e}")
queue = self.get_queue(guild_id)
if len(queue) > 0:
next_player = queue.pop(0)
self.set_current(guild_id, next_player)
self.add_to_history(guild_id, next_player)
await self.trigger_state_change()
voice_client.play(
next_player,
after=lambda e: asyncio.run_coroutine_threadsafe(
self.play_next(guild_id),
self.bot.loop
)
)
await self.send_to_last_channel(guild_id, f"Teraz odtwarzane: {discord.utils.escape_markdown(next_player.title)}")
else:
self.set_current(guild_id, None)
await self.trigger_state_change()
await self.send_to_last_channel(guild_id, "Kolejka jest pusta.")
async def send_to_last_channel(self, guild_id: int, text: str = None, embed: discord.Embed = None):
channel = self.get_notification_channel(guild_id)
if channel:
try:
await channel.send(content=text if text else None, embed=embed)
except Exception as e:
print(f"Failed to send message: {e}")
@app_commands.command(name="queue", description="Pokaż aktualną kolejkę")
async def show_queue(self, interaction: discord.Interaction):
"""Pokaż aktualną kolejkę"""
guild_id = interaction.guild.id
queue = self.get_queue(guild_id)
if not queue:
await interaction.response.send_message(
"Kolejka jest pusta.",
ephemeral=True
)
return
queue_list = "\n".join([f"{i + 1}. {discord.utils.escape_markdown(item.title)}" for i, item in enumerate(queue)])
await interaction.response.send_message(
f"Aktualna kolejka:\n{queue_list}",
ephemeral=False
)
async def history_autocomplete(
self,
interaction: discord.Interaction,
current: str,
) -> list[app_commands.Choice[str]]:
guild_id = interaction.guild.id
history = self.load_history(guild_id)
filtered = [h for h in history if current.lower() in h["title"].lower()][:25]
return [app_commands.Choice(name=h["title"][:100], value=str(h["id"])) for h in filtered]
@app_commands.command(name="history", description="Pokaż historię ostatnio odtwarzanych utworów")
async def show_history(self, interaction: discord.Interaction):
"""Pokaż historię odtwarzania"""
guild_id = interaction.guild.id
history = self.load_history(guild_id)
if not history:
await interaction.response.send_message(
"Historia odtwarzania jest pusta.",
ephemeral=True
)
return
history_list = "\n".join([f"{i + 1}. {discord.utils.escape_markdown(item['title'])}" for i, item in enumerate(history[:15])])
await interaction.response.send_message(
f"📜 Historia odtwarzania (ostatnie 15):\n{history_list}",
ephemeral=False
)
@app_commands.command(name="replay", description="Odtwórz ponownie utwór z historii")
@app_commands.describe(track_id="Wybierz utwór z historii odtwarzania")
@app_commands.autocomplete(track_id=history_autocomplete)
async def replay(self, interaction: discord.Interaction, track_id: str):
"""Odtwórz z historii"""
if not await self.ensure_voice(interaction):
return
await interaction.response.defer()
guild_id = interaction.guild.id
self.last_channels[guild_id] = interaction.channel
history = self.load_history(guild_id)
entry = next((item for item in history if str(item["id"]) == track_id), None)
# If not found by ID, maybe user typed an index number 1..15
if not entry and track_id.isdigit():
idx = int(track_id) - 1
if 0 <= idx < len(history):
entry = history[idx]
if not entry:
await interaction.followup.send("Nie znaleziono utworu w historii.", ephemeral=True)
return
try:
if entry.get("is_local"):
filepath = entry.get("filepath") or entry.get("url")
if not os.path.exists(filepath):
await interaction.followup.send("Plik lokalny z historii już nie istnieje na serwerze.", ephemeral=True)
return
player = await YTDLSource.from_file(filepath, loop=self.bot.loop, title=entry["title"])
else:
player = await YTDLSource.from_url(entry["url"], loop=self.bot.loop, stream=True)
if interaction.guild.voice_client.is_playing():
self.get_queue(guild_id).append(player)
await self.trigger_state_change()
await interaction.followup.send(
f"Dodano do kolejki z historii: {discord.utils.escape_markdown(player.title)}",
ephemeral=False
)
else:
self.set_current(guild_id, player)
self.add_to_history(guild_id, player)
await self.trigger_state_change()
interaction.guild.voice_client.play(
player,
after=lambda e: asyncio.run_coroutine_threadsafe(
self.play_next(guild_id),
self.bot.loop
)
)
await interaction.followup.send(
f"Teraz odtwarzane z historii: {discord.utils.escape_markdown(player.title)}",
ephemeral=False
)
except Exception as e:
await interaction.followup.send(f"Wystąpił błąd: {e}", ephemeral=True)
@app_commands.command(name="skip", description="Pomiń aktualny utwór")
async def skip(self, interaction: discord.Interaction):
"""Pomiń aktualny utwór"""
guild_id = interaction.guild.id
if self.get_loop_current(guild_id):
self.set_loop_current(guild_id, False)
if interaction.guild.voice_client and (interaction.guild.voice_client.is_playing() or interaction.guild.voice_client.is_paused()):
interaction.guild.voice_client.stop()
await self.trigger_state_change()
await interaction.response.send_message(
"Pominięto aktualny utwór.",
ephemeral=False
)
else:
await interaction.response.send_message(
"Nic nie jest odtwarzane.",
ephemeral=True
)
@app_commands.command(name="pause", description="Wstrzymaj odtwarzanie")
async def pause(self, interaction: discord.Interaction):
"""Wstrzymaj odtwarzanie"""
if interaction.guild.voice_client.is_playing():
interaction.guild.voice_client.pause()
await self.trigger_state_change()
await interaction.response.send_message(
"Odtwarzanie wstrzymane.",
ephemeral=False
)
else:
await interaction.response.send_message(
"Nic nie jest odtwarzane.",
ephemeral=True
)
@app_commands.command(name="resume", description="Wznów odtwarzanie")
async def resume(self, interaction: discord.Interaction):
"""Wznów odtwarzanie"""
if interaction.guild.voice_client.is_paused():
interaction.guild.voice_client.resume()
await self.trigger_state_change()
await interaction.response.send_message(
"Odtwarzanie wznowione.",
ephemeral=False
)
else:
await interaction.response.send_message(
"Odtwarzanie nie jest wstrzymane.",
ephemeral=True
)
@app_commands.command(name="stop", description="Zatrzymaj odtwarzanie i wyczyść kolejkę")
async def stop(self, interaction: discord.Interaction):
"""Zatrzymaj odtwarzanie i wyczyść kolejkę"""
guild_id = interaction.guild.id
if interaction.guild.voice_client.is_playing():
interaction.guild.voice_client.stop()
self.get_queue(guild_id).clear()
self.set_current(guild_id, None)
await self.trigger_state_change()
await interaction.response.send_message(
"Zatrzymano odtwarzanie i wyczyszczono kolejkę.",
ephemeral=False
)
@app_commands.command(name="volume", description="Zmień głośność odtwarzacza")
@app_commands.describe(volume="Poziom głośności (1-100)")
async def volume(self, interaction: discord.Interaction, volume: int):
"""Zmień głośność odtwarzacza"""
guild_id = interaction.guild.id
if interaction.guild.voice_client is None:
return await interaction.response.send_message(
"Nie połączono z kanałem głosowym.",
ephemeral=True
)
if 0 < volume <= 100:
self.set_volume(guild_id, volume / 100)
interaction.guild.voice_client.source.volume = self.get_volume(guild_id)
await self.trigger_state_change()
await interaction.response.send_message(
f"Zmieniono głośność na {volume}%",
ephemeral=False
)
return None
else:
await interaction.response.send_message(
"Wprowadź wartość z przedziału od 1 do 100.",
ephemeral=True
)
return None
@app_commands.command(name="nowplaying", description="Pokaż aktualnie odtwarzany utwór")
async def now_playing(self, interaction: discord.Interaction):
"""Pokaż aktualnie odtwarzany utwór"""
guild_id = interaction.guild.id
current = self.get_current(guild_id)
if current:
await interaction.response.send_message(
f"Teraz odtwarzane: {discord.utils.escape_markdown(current.title)}",
ephemeral=False
)
else:
await interaction.response.send_message(
"Nic nie jest obecnie odtwarzane.",
ephemeral=True
)
@app_commands.command(name="leave", description="Odłącz bota od kanału głosowego")
async def leave(self, interaction: discord.Interaction):
"""Odłącz bota od kanału głosowego"""
guild_id = interaction.guild.id
if interaction.guild.voice_client is not None:
await interaction.guild.voice_client.disconnect()
self.get_queue(guild_id).clear()
self.set_current(guild_id, None)
await interaction.response.send_message(
"Rozłączono z kanału głosowego.",
ephemeral=False
)
else:
await interaction.response.send_message(
"Nie jestem na kanale głosowym.",
ephemeral=True
)
@app_commands.command(name="loop", description="Włącz/wyłącz zapętlanie aktualnego utworu")
async def loop_command(self, interaction: discord.Interaction):
"""Włącz/wyłącz zapętlanie aktualnego utworu"""
guild_id = interaction.guild.id
loop_current = not self.get_loop_current(guild_id)
self.set_loop_current(guild_id, loop_current)
if loop_current:
self.set_loop_playlist(guild_id, False)
state = "włączone" if loop_current else "wyłączone"
await interaction.response.send_message(
f"Zapętlanie aktualnego utworu zostało {state}.",
ephemeral=False
)
@app_commands.command(name="loopplaylist", description="Włącz/wyłącz zapętlanie całej kolejki (tryb playlisty)")
async def loopplaylist_command(self, interaction: discord.Interaction):
"""Włącz/wyłącz zapętlanie całej kolejki"""
guild_id = interaction.guild.id
loop_pl = not self.get_loop_playlist(guild_id)
self.set_loop_playlist(guild_id, loop_pl)
if loop_pl:
self.set_loop_current(guild_id, False)
state = "włączone (utwory po zakończeniu wracają na koniec kolejki)" if loop_pl else "wyłączone"
await interaction.response.send_message(
f"🔁 Zapętlanie całej kolejki: **{state}**.",
ephemeral=False
)
class MyBot(commands.Bot):
def __init__(self):
intents = discord.Intents.default()
intents.message_content = True
intents.voice_states = True
super().__init__(command_prefix="!", intents=intents)
async def setup_hook(self):
await self.add_cog(MusicBot(self))
if GUILD_ID:
guild = discord.Object(id=int(GUILD_ID))
self.tree.copy_global_to(guild=guild)
await self.tree.sync(guild=guild)
print(f"Synced commands to guild {GUILD_ID}")
else:
await self.tree.sync()
print("Synced commands globally")
async def on_ready(self):
print(f'Logged in as {self.user.name} ({self.user.id})')
print('------')
bot = MyBot()