import sqlite3
from loguru import logger as log

def init_database():
    conn = sqlite3.connect('products.db')
    cursor = conn.cursor()

    # Создаем таблицу Product
    cursor.execute('''
    CREATE TABLE IF NOT EXISTS Product (
        id INTEGER PRIMARY KEY AUTOINCREMENT,
        name TEXT NOT NULL,
        image_path TEXT
    )
    ''')

    # Создаем таблицу Param
    cursor.execute('''
    CREATE TABLE IF NOT EXISTS Param (
        id INTEGER PRIMARY KEY AUTOINCREMENT,
        name TEXT NOT NULL,
        val TEXT NOT NULL
    )
    ''')

    # Создаем таблицу Val
    cursor.execute('''
    CREATE TABLE IF NOT EXISTS Val (
        id INTEGER PRIMARY KEY AUTOINCREMENT,
        val TEXT NOT NULL
    )
    ''')

    # Создаем таблицу CParams
    cursor.execute('''
    CREATE TABLE IF NOT EXISTS CParams (
        id INTEGER PRIMARY KEY AUTOINCREMENT,
        name TEXT NOT NULL,
        val_id INTEGER,
        FOREIGN KEY (val_id) REFERENCES Val (id)
    )
    ''')

    # Создаем таблицу Category
    cursor.execute('''
    CREATE TABLE IF NOT EXISTS Category (
        id INTEGER PRIMARY KEY AUTOINCREMENT,
        name TEXT NOT NULL,
        prod_id INTEGER,
        image_path TEXT,
        c_params_id INTEGER,
        FOREIGN KEY (prod_id) REFERENCES Product (id),
        FOREIGN KEY (c_params_id) REFERENCES CParams (id)
    )
    ''')

    # Создаем таблицу Product_Param (связь многие-ко-многим)
    cursor.execute('''
    CREATE TABLE IF NOT EXISTS Product_Param (
        product_id INTEGER,
        param_id INTEGER,
        PRIMARY KEY (product_id, param_id),
        FOREIGN KEY (product_id) REFERENCES Product (id),
        FOREIGN KEY (param_id) REFERENCES Param (id)
    )
    ''')

    conn.commit()
    conn.close()
    log.info("База данных успешно инициализирована!")