import sqlite3
from loguru import logger as log


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

    cursor.execute('''
        CREATE TABLE IF NOT EXISTS categories (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            name TEXT NOT NULL UNIQUE,
            description TEXT,
            parent_id INTEGER,
            image_path TEXT,
            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
            FOREIGN KEY (parent_id) REFERENCES categories (id)
        )
    ''')

    cursor.execute('''
        CREATE TABLE IF NOT EXISTS products (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            name TEXT NOT NULL,
            description TEXT,
            category_id INTEGER NOT NULL,
            diameter INTEGER,
            wall_thickness REAL,
            length INTEGER,
            color TEXT,
            application_type TEXT, -- внутренняя/наружная
            stiffness_class TEXT, -- SN4, SN2 и т.д.
            image_path TEXT,
            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
            FOREIGN KEY (category_id) REFERENCES categories (id)
        )
    ''')

    cursor.execute('''
        CREATE TABLE IF NOT EXISTS filters (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            name TEXT NOT NULL,
            filter_key TEXT NOT NULL UNIQUE, -- ключ для фильтрации
            filter_type TEXT DEFAULT 'single' -- 'single' или 'multiple'
        )
    ''')

    cursor.execute('''
        CREATE TABLE IF NOT EXISTS product_filters (
            id INTEGER PRIMARY KEY AUTOINCREMENT,
            product_id INTEGER NOT NULL,
            filter_key TEXT NOT NULL,
            value TEXT NOT NULL,
            FOREIGN KEY (product_id) REFERENCES products (id),
            FOREIGN KEY (filter_key) REFERENCES filters (filter_key),
            UNIQUE(product_id, filter_key)
        )
    ''')
    
    cursor.execute('CREATE INDEX IF NOT EXISTS idx_products_category ON products(category_id)')
    cursor.execute('CREATE INDEX IF NOT EXISTS idx_products_diameter ON products(diameter)')
    cursor.execute('CREATE INDEX IF NOT EXISTS idx_products_length ON products(length)')
    cursor.execute('CREATE INDEX IF NOT EXISTS idx_product_filters_key_value ON product_filters(filter_key, value)')
    
    # Миграция: переименовываем full_name → name
    cursor.execute("PRAGMA table_info(products)")
    columns = {row[1] for row in cursor.fetchall()}
    if 'full_name' in columns and 'name' not in columns:
        cursor.executescript('''
            PRAGMA foreign_keys=OFF;
            CREATE TABLE products_new (
                id INTEGER PRIMARY KEY AUTOINCREMENT,
                name TEXT NOT NULL,
                description TEXT,
                category_id INTEGER NOT NULL,
                diameter INTEGER,
                wall_thickness REAL,
                length INTEGER,
                color TEXT,
                application_type TEXT,
                stiffness_class TEXT,
                image_path TEXT,
                created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
                FOREIGN KEY (category_id) REFERENCES categories (id)
            );
            INSERT INTO products_new (id, name, description, category_id, diameter, wall_thickness, length, color, application_type, stiffness_class, image_path, created_at)
                SELECT id, full_name, description, category_id, diameter, wall_thickness, length, color, application_type, stiffness_class, image_path, created_at FROM products;
            DROP TABLE products;
            ALTER TABLE products_new RENAME TO products;
            CREATE INDEX IF NOT EXISTS idx_products_category ON products(category_id);
            CREATE INDEX IF NOT EXISTS idx_products_diameter ON products(diameter);
            CREATE INDEX IF NOT EXISTS idx_products_length ON products(length);
        ''')
        log.info("Migrated products table: renamed 'full_name' → 'name'")
    
    conn.commit()
    conn.close()
    log.success("База данных успешно инициализирована!")