136 lines
4.3 KiB
Python
136 lines
4.3 KiB
Python
import discord
|
|
from discord.ext import commands
|
|
import asyncio
|
|
import os
|
|
from dotenv import load_dotenv
|
|
import yt_dlp
|
|
from ytmusicapi import YTMusic
|
|
|
|
load_dotenv()
|
|
songs_dir = os.getenv("SONGS_DIR")
|
|
intents = discord.Intents.default()
|
|
intents.message_content = True
|
|
intents.voice_states = True
|
|
bot = commands.Bot(command_prefix='!', intents=intents)
|
|
queued_songs = []
|
|
voice_client = None
|
|
|
|
def search_song(query):
|
|
ytmusic = YTMusic()
|
|
search_results = ytmusic.search(query, filter='songs', limit=1)
|
|
if search_results:
|
|
song_id = search_results[0]['videoId']
|
|
return song_id
|
|
else:
|
|
return None
|
|
|
|
def search_video(query):
|
|
ydl_opts = {
|
|
'default_search': 'auto',
|
|
'format': 'best',
|
|
'quiet': True
|
|
}
|
|
|
|
with yt_dlp.YoutubeDL(ydl_opts) as ydl:
|
|
try:
|
|
result = ydl.extract_info(query, download=False)
|
|
if 'entries' in result:
|
|
# Take the first result
|
|
video_id = result['entries'][0]['id']
|
|
return video_id
|
|
else:
|
|
return None
|
|
except yt_dlp.DownloadError:
|
|
return None
|
|
|
|
def download_song(query):
|
|
song_id = search_video(query)
|
|
ydl_opts = {
|
|
'format': 'bestaudio/best',
|
|
'outtmpl': f'{songs_dir}/%(title)s.%(ext)s',
|
|
'postprocessors': [{
|
|
'key': 'FFmpegExtractAudio',
|
|
'preferredcodec': 'mp3',
|
|
'preferredquality': '192',
|
|
}],
|
|
}
|
|
if not os.path.exists(songs_dir):
|
|
os.makedirs(songs_dir)
|
|
with yt_dlp.YoutubeDL(ydl_opts) as ydl:
|
|
info = ydl.extract_info(f'https://music.youtube.com/watch?v={song_id}', download=True)
|
|
filename = ydl.prepare_filename(info)
|
|
filename = os.path.splitext(filename)[0] + '.mp3'
|
|
return filename
|
|
|
|
@bot.command()
|
|
async def play(ctx, *, query):
|
|
global voice_client
|
|
if voice_client and voice_client.is_connected():
|
|
await add_to_queue(ctx, query)
|
|
else:
|
|
await connect_and_play(ctx, query)
|
|
|
|
@bot.command()
|
|
async def skip(ctx):
|
|
global voice_client
|
|
if voice_client and voice_client.is_playing():
|
|
voice_client.stop()
|
|
|
|
@bot.command()
|
|
async def stop(ctx):
|
|
global voice_client
|
|
for file in os.listdir(songs_dir):
|
|
file_path = os.path.join(songs_dir, file)
|
|
if os.path.isfile(file_path):
|
|
os.unlink(file_path)
|
|
if voice_client and voice_client.is_connected():
|
|
queued_songs.clear()
|
|
if voice_client.is_playing():
|
|
voice_client.stop()
|
|
await voice_client.disconnect()
|
|
voice_client = None
|
|
await ctx.send("Stopped playback and cleared queue.")
|
|
|
|
async def add_to_queue(ctx, query):
|
|
global voice_client
|
|
song_path = download_song(query)
|
|
print(song_path)
|
|
|
|
if song_path:
|
|
song_name = os.path.relpath(song_path, start=songs_dir) # Get the song name relative to songs_dir
|
|
queued_songs.append((song_path, song_name))
|
|
await ctx.send(f"**Added to queue:** `{song_name}`")
|
|
else:
|
|
await ctx.send("Can't find a match for that song.")
|
|
|
|
async def connect_and_play(ctx, query):
|
|
global voice_client
|
|
voice_channel = ctx.author.voice.channel
|
|
if voice_channel:
|
|
song_path = download_song(query)
|
|
print(song_path)
|
|
|
|
if song_path:
|
|
song_name = os.path.relpath(song_path, start=songs_dir) # Get the song name relative to songs_dir
|
|
queued_songs.append((song_path, song_name))
|
|
voice_client = await play_queued_songs(ctx, voice_channel)
|
|
else:
|
|
await ctx.send("Can't find a match for that song.")
|
|
|
|
async def play_queued_songs(ctx, voice_channel):
|
|
global voice_client
|
|
voice_client = await voice_channel.connect()
|
|
while queued_songs:
|
|
song_path, song_name = queued_songs.pop(0) # Get the first song in the queue
|
|
await ctx.send(f"**Now playing:** `{song_name}`")
|
|
voice_client.play(discord.FFmpegPCMAudio(song_path))
|
|
while voice_client and voice_client.is_playing(): # Check if voice_client is not None
|
|
await asyncio.sleep(1)
|
|
if voice_client:
|
|
for file in os.listdir(songs_dir):
|
|
file_path = os.path.join(songs_dir, file)
|
|
if os.path.isfile(file_path):
|
|
os.unlink(file_path)
|
|
await voice_client.disconnect()
|
|
|
|
bot.run(os.getenv("DISCORD_TOKEN"))
|