import sqlite3

db_path = "./data/music.db"


# 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,
                    )''')

    # Create queue table if it doesn't exist
    cursor.execute('''CREATE TABLE IF NOT EXISTS queue (
                        server_id TEXT,
                        song_link TEXT,
                        queued_by TEXT,
                        index INTEGER,
                        has_played INTEGER DEFAULT 0,
                        PRIMARY KEY (server_id, order_num)
                    )''')

    # Commit the changes and close the connection
    conn.commit()
    conn.close()


# Queue a song in the db
def add_song(server_id, song_link, queued_by):
    # Connect to db
    conn = sqlite3.connect(db_path)
    cursor = conn.cursor()

    # Grab current index
    cursor.execute(f"""
                   SELECT MAX(index)
                   FROM queue
                   WHERE server_id = ?
                   """, (server_id,))
    result = cursor.fetchone()

    # Highnest number or 0
    max_order_num = result[0] + 1 if result[0] is not None else 0

    cursor.execute("""
                   INSERT INTO queue (server_id, song_link, queued_by, index)
                   VALUES (?, ?, ?, ?)
                   """, (server_id, song_link, queued_by, max_order_num))

    conn.commit()
    conn.close()


# Add server to db if first time queuing
def add_server(server_id, cursor):
    # 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,))


# set song as played and update indexes
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 queue
                   WHERE server_id = ? AND order_num = ?''',
                   (server_id, order_num))
    #cursor.execute('''UPDATE queue
    #               SET is_finished = 1
    #               WHERE server_id = ? AND index = ?''',
    #               (server_id, order_num))

    # Get the order numbers of unplayed songs
    cursor.execute('''SELECT index
                   FROM queue
                   WHERE server_id = ? AND is_finished = 0''', (server_id,))
    unplayed_order_nums = [row[0] for row in cursor.fetchall()]

    # Update the order numbers of unplayed songs
    for new_order, old_order in enumerate(unplayed_order_nums, start=1):
        cursor.execute('''UPDATE queue
                       SET order_num = ?
                       WHERE server_id = ? AND order_num = ?''',
                       (new_order, server_id, old_order))

    # Close connection
    conn.commit()
    conn.close()