363 lines
9.8 KiB
Python
363 lines
9.8 KiB
Python
from http import server
|
|
import sqlite3
|
|
|
|
import discord
|
|
import asyncio
|
|
|
|
from .translate import search_song
|
|
|
|
db_path = "./data/music.db"
|
|
|
|
FFMPEG_OPTS = {
|
|
'before_options':
|
|
'-reconnect 1 -reconnect_streamed 1 -reconnect_delay_max 5',
|
|
|
|
'options':
|
|
'-vn'
|
|
}
|
|
|
|
# Creates the tables if they don't exist
|
|
def initialize_tables():
|
|
# Connect to the database
|
|
conn = sqlite3.connect(db_path)
|
|
cursor = conn.cursor()
|
|
|
|
# Create servers table if it doesn't exist
|
|
cursor.execute('''CREATE TABLE IF NOT EXISTS servers (
|
|
server_id TEXT PRIMARY KEY,
|
|
is_playing INTEGER DEFAULT 0,
|
|
song_name TEXT
|
|
);''')
|
|
|
|
# Set all to not playing
|
|
cursor.execute("UPDATE servers SET is_playing = 0;")
|
|
|
|
# Create queue table if it doesn't exist
|
|
cursor.execute('''CREATE TABLE IF NOT EXISTS songs (
|
|
server_id TEXT NOT NULL,
|
|
song_link TEXT,
|
|
queued_by TEXT,
|
|
position INTEGER NOT NULL,
|
|
|
|
title TEXT,
|
|
thumbnail TEXT,
|
|
duration INTEGER,
|
|
|
|
PRIMARY KEY (position),
|
|
FOREIGN KEY (server_id) REFERENCES servers(server_id)
|
|
);''')
|
|
# Clear all entries
|
|
cursor.execute("DELETE FROM songs;")
|
|
|
|
# Commit the changes and close the connection
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
|
|
# Queue a song in the db
|
|
async def add_song(server_id, details, queued_by):
|
|
# Connect to db
|
|
conn = sqlite3.connect(db_path)
|
|
cursor = conn.cursor()
|
|
|
|
await add_server(server_id, cursor, conn)
|
|
|
|
max_order_num = await get_max(server_id, cursor) + 1
|
|
|
|
if isinstance(details, str):
|
|
cursor.execute("""
|
|
INSERT INTO songs (server_id,
|
|
song_link,
|
|
queued_by,
|
|
position,
|
|
title,
|
|
thumbnail,
|
|
duration)
|
|
VALUES (?, ?, ?, ?, ?, ?, ?)
|
|
""", (server_id,
|
|
"Not grabbed",
|
|
queued_by,
|
|
max_order_num,
|
|
details,
|
|
"Unkown",
|
|
"Unkown"))
|
|
else:
|
|
cursor.execute("""
|
|
INSERT INTO songs (server_id,
|
|
song_link,
|
|
queued_by,
|
|
position,
|
|
title,
|
|
thumbnail,
|
|
duration)
|
|
VALUES (?, ?, ?, ?, ?, ?, ?)
|
|
""", (server_id,
|
|
details['url'],
|
|
queued_by,
|
|
max_order_num,
|
|
details['title'],
|
|
details['thumbnail'],
|
|
details['duration']))
|
|
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
return max_order_num
|
|
|
|
|
|
# Pop song from server
|
|
async def pop(server_id, ignore=False):
|
|
# Connect to db
|
|
conn = sqlite3.connect(db_path)
|
|
cursor = conn.cursor()
|
|
|
|
# JUST INCASE!
|
|
await add_server(server_id, cursor, conn)
|
|
|
|
cursor.execute('''SELECT *
|
|
FROM songs
|
|
WHERE server_id = ?
|
|
ORDER BY position
|
|
LIMIT 1;''', (server_id,))
|
|
result = cursor.fetchone()
|
|
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
if result == None:
|
|
return None
|
|
elif ignore:
|
|
await mark_song_as_finished(server_id, result[3])
|
|
return None
|
|
elif result[1] == "Not grabbed":
|
|
# Fetch song info
|
|
song = await search_song(result[4])
|
|
if song == []:
|
|
return None
|
|
else:
|
|
song = song[0]
|
|
|
|
await set_current_song(server_id, song['title'])
|
|
await mark_song_as_finished(server_id, result[3])
|
|
return song['url']
|
|
|
|
await set_current_song(server_id, result[4])
|
|
await mark_song_as_finished(server_id, result[3])
|
|
|
|
return result[1]
|
|
|
|
|
|
# Add server to db if first time queuing
|
|
async def add_server(server_id, cursor, conn):
|
|
# Check if the server exists
|
|
cursor.execute('''SELECT COUNT(*)
|
|
FROM servers
|
|
WHERE server_id = ?''', (server_id,))
|
|
|
|
result = cursor.fetchone()
|
|
server_exists = result[0] > 0
|
|
|
|
# If the server doesn't exist, add it
|
|
if not server_exists:
|
|
cursor.execute('''INSERT INTO servers (server_id)
|
|
VALUES (?)''', (server_id,))
|
|
conn.commit()
|
|
|
|
|
|
# set song as played and update indexes
|
|
async def mark_song_as_finished(server_id, order_num):
|
|
# Connect to the database
|
|
conn = sqlite3.connect(db_path)
|
|
cursor = conn.cursor()
|
|
|
|
# Update the song as finished
|
|
cursor.execute('''DELETE FROM songs
|
|
WHERE server_id = ? AND position = ?''',
|
|
(server_id, order_num))
|
|
|
|
# Close connection
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
|
|
# set the current playing song of the server
|
|
async def set_current_song(server_id, title):
|
|
# Connect to the database
|
|
conn = sqlite3.connect(db_path)
|
|
cursor = conn.cursor()
|
|
|
|
cursor.execute(''' UPDATE servers
|
|
SET song_name = ?
|
|
WHERE server_id = ?''',
|
|
(title, server_id))
|
|
|
|
# Close connection
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
|
|
async def get_current_song(server_id):
|
|
# Connect to the database
|
|
conn = sqlite3.connect(db_path)
|
|
cursor = conn.cursor()
|
|
|
|
cursor.execute(''' SELECT song_name
|
|
FROM servers
|
|
WHERE server_id = ?
|
|
LIMIT 1;''',
|
|
(server_id,))
|
|
|
|
result = cursor.fetchone()
|
|
|
|
# Close connection
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
return result[0] if result else "Nothing"
|
|
|
|
|
|
# Grab max order from server
|
|
async def get_max(server_id, cursor):
|
|
cursor.execute(f"""
|
|
SELECT MAX(position)
|
|
FROM songs
|
|
WHERE server_id = ?
|
|
""", (server_id,))
|
|
result = cursor.fetchone()
|
|
|
|
# Highnest number or 0
|
|
max_order_num = result[0] if result[0] is not None else -1
|
|
|
|
return max_order_num
|
|
|
|
|
|
# Sets the playing variable in a server to true or false
|
|
async def update_server(server_id, playing: bool):
|
|
# Connect to database
|
|
conn = sqlite3.connect(db_path)
|
|
cursor = conn.cursor()
|
|
|
|
# add server to db if not present
|
|
await add_server(server_id, cursor, conn)
|
|
|
|
value = 1 if playing else 0
|
|
|
|
# Update field
|
|
cursor.execute("""UPDATE servers
|
|
SET is_playing = ?
|
|
WHERE server_id = ?
|
|
""", (value, server_id))
|
|
|
|
# Close connection
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
|
|
async def is_server_playing(server_id):
|
|
# Connect to db
|
|
conn = sqlite3.connect(db_path)
|
|
cursor = conn.cursor()
|
|
|
|
# add server to db if not present
|
|
await add_server(server_id, cursor, conn)
|
|
|
|
cursor.execute("""SELECT is_playing
|
|
FROM servers
|
|
WHERE server_id = ?""",
|
|
(server_id,))
|
|
|
|
result = cursor.fetchone()
|
|
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
return True if result[0] == 1 else False
|
|
|
|
|
|
# Delete all songs from a server
|
|
async def clear(server_id):
|
|
# Connect to db
|
|
conn = sqlite3.connect(db_path)
|
|
cursor = conn.cursor()
|
|
|
|
await add_server(server_id, cursor, conn)
|
|
await update_server(server_id, False)
|
|
|
|
# Delete all songs from the server
|
|
cursor.execute('''DELETE FROM songs WHERE server_id = ?''', (server_id,))
|
|
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
|
|
# Grabs all songs from a server for display purposes
|
|
async def grab_songs(server_id):
|
|
# Connect to db
|
|
conn = sqlite3.connect(db_path)
|
|
cursor = conn.cursor()
|
|
|
|
await add_server(server_id, cursor, conn)
|
|
|
|
# Grabs all songs from the server
|
|
cursor.execute('''SELECT title, duration, queued_by
|
|
FROM songs
|
|
WHERE server_id = ?
|
|
ORDER BY position
|
|
LIMIT 10''', (server_id,))
|
|
songs = cursor.fetchall()
|
|
max = await get_max(server_id, cursor)
|
|
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
return max, songs
|
|
|
|
|
|
# Play and loop songs in server
|
|
async def play(ctx):
|
|
"""Main playback loop - plays songs from queue sequentially"""
|
|
server_id = ctx.guild.id
|
|
voice_client = ctx.voice_client
|
|
|
|
# Safety check
|
|
if voice_client is None:
|
|
await update_server(server_id, False)
|
|
return
|
|
|
|
# Wait until current song finishes
|
|
while voice_client.is_playing():
|
|
await asyncio.sleep(0.5)
|
|
|
|
# Get next song
|
|
url = await pop(server_id)
|
|
|
|
# If no songs left, update status and return
|
|
if url is None:
|
|
await update_server(server_id, False)
|
|
return
|
|
|
|
try:
|
|
# Create audio source
|
|
audio_source = discord.FFmpegPCMAudio(url, **FFMPEG_OPTS)
|
|
|
|
# Play with callback to continue queue
|
|
def after_playing(error):
|
|
if error:
|
|
print(f"Player error: {error}")
|
|
# Schedule the next song in the event loop
|
|
if voice_client and not voice_client.is_connected():
|
|
return
|
|
coro = play(ctx)
|
|
fut = asyncio.run_coroutine_threadsafe(coro, ctx.bot.loop)
|
|
try:
|
|
fut.result()
|
|
except Exception as e:
|
|
print(f"Error playing next song: {e}")
|
|
|
|
voice_client.play(audio_source, after=after_playing)
|
|
|
|
except Exception as e:
|
|
print(f"Error starting playback: {e}")
|
|
# Try to continue with next song
|
|
await play(ctx)
|