import sqlite3

DB_PATH = 'data.db'

def init_db():
    """
    Initialize the database and create necessary tables.
    """
    conn = sqlite3.connect(DB_PATH)
    cursor = conn.cursor()

    # Create the react_roles table
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS react_roles (
            message_id INTEGER,
            role_id INTEGER,
            emoji TEXT,
            description TEXT
        )
    ''')

    # Create the react_role_categories table
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS react_role_categories (
            message_id INTEGER,
            category_name TEXT
        )
    ''')

    conn.commit()
    conn.close()

def add_react_role_category_to_db(message_id: int, category_name: str):
    """
    Add a new react role category to the database.
    :param message_id: The ID of the message to which the category is associated.
    :param category_name: The name of the category (e.g., "Category1").
    """
    conn = sqlite3.connect(DB_PATH)
    cursor = conn.cursor()

    # Create the table if it doesn't exist
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS react_role_categories (
            message_id INTEGER,
            category_name TEXT
        )
    ''')

    # Insert the new react-role category into the database
    cursor.execute('''
        INSERT INTO react_role_categories (message_id, category_name)
        VALUES (?, ?)
    ''', (message_id, category_name.lower()))

    conn.commit()
    conn.close()

def add_react_role_to_db(message_id: int, role_id: int, emoji: str, description: str):
    """
    Add a new react role to the database.
    :param message_id: The message ID of the react-role message.
    :param role_id: The ID of the role to assign.
    :param emoji: The emoji to react with.
    :param description: The description of the role.
    """
    conn = sqlite3.connect(DB_PATH)
    cursor = conn.cursor()

    # Create the table if it doesn't exist
    cursor.execute('''
        CREATE TABLE IF NOT EXISTS react_roles (
            message_id INTEGER,
            role_id INTEGER,
            emoji TEXT,
            description TEXT
        )
    ''')

    # Insert the new react-role into the database
    cursor.execute('''
        INSERT INTO react_roles (message_id, role_id, emoji, description)
        VALUES (?, ?, ?, ?)
    ''', (message_id, role_id, emoji, description))

    conn.commit()
    conn.close()

def get_react_roles_categories():
    """
    Get all react role categories from the database.
    :return: A list of tuples containing message_id and category_name.
    """
    conn = sqlite3.connect(DB_PATH)
    cursor = conn.cursor()

    # Fetch all react-role categories from the database
    cursor.execute('''
        SELECT * FROM react_role_categories
    ''')
    categories = cursor.fetchall()

    conn.close()
    return categories

def get_react_roles_by_category(category_name: str):
    """
    Get all react roles for a specific category from the database.
    :param category_name: The name of the category (e.g., "Category1").
    :return: A list of tuples containing message_id, role_id, emoji, and description.
    """
    # Get the message id of the category
    categories = get_react_roles_categories()
    message_id = next((cat[0] for cat in categories if cat[1] == category_name.lower()), None)

    conn = sqlite3.connect(DB_PATH)
    cursor = conn.cursor()

    # Fetch all react-roles for the specified category from the database
    cursor.execute('''
        SELECT * FROM react_roles WHERE message_id = ?
    ''', (message_id,))
    roles = cursor.fetchall()

    conn.close()
    return roles

def get_react_roles_by_message_id(message_id: int):
    """
    Get all react roles for a specific message ID from the database.
    :param message_id: The ID of the message to which the roles are associated.
    :return: A list of tuples containing message_id, role_id, emoji, and description.
    """
    conn = sqlite3.connect(DB_PATH)
    cursor = conn.cursor()

    # Fetch all react-roles for the specified message ID from the database
    cursor.execute('''
        SELECT * FROM react_roles WHERE message_id = ?
    ''', (message_id,))
    roles = cursor.fetchall()

    conn.close()
    return roles