import sqlite3
import datetime
import random
from werkzeug.security import generate_password_hash

DB_NAME = "db/rendiciones.db"

def get_db_connection():
    return sqlite3.connect(DB_NAME)

def populateDefaults():
    random.seed(42)
    conn = get_db_connection()
    c = conn.cursor()
    
    c.execute("SELECT COUNT(*) FROM zonas")
    if c.fetchone()[0] == 0:
        zonas = ['Norte', 'Quinta', 'RM', 'Sur']
        for zona in zonas:
            c.execute("INSERT INTO zonas (name) VALUES (?)", (zona,))
        conn.commit()

    c.execute("SELECT COUNT(*) FROM modulos")
    if c.fetchone()[0] == 0:
        c.execute("SELECT id, name FROM zonas")
        zona_map = {name: id for id, name in c.fetchall()}
        modulos_data = [
            ('ANTOFAGASTA', 'Norte'), ('COQUIMBO 1', 'Norte'), ('COQUIMBO 2', 'Norte'),
            ('SERENA 2', 'Norte'), ('SERENA 3', 'Norte'), ('LOS ANDES', 'Quinta'),
            ('VIÑA 1', 'Quinta'), ('VIÑA 2', 'Quinta'), ('CENTRO 2', 'RM'),
            ('IMPERIO 1', 'RM'), ('IMPERIO 2', 'RM'), ('MELIPILLA', 'RM'),
            ('PUENTE ALTO', 'RM'), ('QUILICURA', 'RM'), ('RANCAGUA', 'RM'),
            ('LINARES', 'Sur'), ('SAN FERNANDO', 'Sur'), ('TALCA', 'Sur')
        ]
        for mod_name, zona_name in modulos_data:
            c.execute("INSERT INTO modulos (zona_id, name) VALUES (?, ?)", (zona_map[zona_name], mod_name))
        conn.commit()

    c.execute("SELECT COUNT(*) FROM productos")
    if c.fetchone()[0] == 0:
        productos_base = [
            ('PACK LENTES DE SOL 1 x', 12990, 200),
            ('PACK LENTES DE PANTALLA', 12990, 200),
            ('PACK LENTES DE SOL 2 x', 19990, 400),
            ('PACK LENTES + ESTUCHE BLANDO', 17990, 400),
            ('PACK LENTES + STRAP', 17990, 400),
            ('PACK LENTES 1 x POLARIZADO + ESTUCHE BLANDO+ KIT', 23990, 1000),
            ('PACK LENTES GRANDES ANTIPARRA CON LIGA', 19990, 1000),
            ('ANTIPARRA MEDIANO', 14990, 800),
            ('ANTIPARRA PEQUEÑO', 9990, 200),
            ('PACK LENTES DE GRADUACION', 12990, 200),
            ('PACK LENTES FILTRO AZUL', 14990, 1000),
            ('JOCKEY (2 X PROD. SELECCIONADO)', 9990, 600),
            ('ESTUCHES MODA', 6990, 200),
            ('ESTUCHES CIERRE', 6990, 200),
            ('ESTUCHE DE LECTURA', 6990, 200),
            ('STRAP TELA', 4990, 150),
            ('STRAP DISEÑO', 6990, 200),
            ('STRAP CUERO', 6990, 200),
            ('LIMPIA CRISTAL + PAÑO', 2990, 100),
            ('LENTE LED', 14990, 1000),
            ('PULSERAS', 9990, 200),
            ('SUJETADORES PARA LENTES', 1000, 100),
            ('CORREAS DE CARTERAS DELGADAS Y GRUESAS', 9990, 600),
            ('ESTUCHE COLGANTE DE LENTES', 6990, 200),
            ('COLGANTE DE CELULAR (PULSERA)', 4990, 200),
            ('COLGANTE DE CELULAR (COLLAR)', 6990, 200),
            ('PACK DUO DE COLGANTE DE CELULAR', 7990, 300),
            ('JOCKEY, GORRAS, SOMBREROS, CUELLOS Y OTROS.', 14990, 1000),
            ('CARTERAS ORDENADOR', 9990, 600)
        ]
        c.execute("SELECT id FROM zonas")
        zonas_ids = [row[0] for row in c.fetchall()]
        
        # Fecha de activación artificial en el pasado para que las ventas históricas funcionen
        fecha_base_precios = (datetime.date.today() - datetime.timedelta(days=365*3)).strftime('%Y-%m-%d 00:00:00')

        for name, price, commission in productos_base:
            c.execute("INSERT OR IGNORE INTO productos (name) VALUES (?)", (name,))
            c.execute("SELECT id FROM productos WHERE name = ?", (name,))
            p_id = c.fetchone()[0]
            
            for zona_id in zonas_ids:
                c.execute('''INSERT INTO precios_historicos 
                          (producto_id, zona_id, price, commission, fecha_activacion) 
                          VALUES (?, ?, ?, ?, ?)''', 
                          (p_id, zona_id, price, commission, fecha_base_precios))
        conn.commit()

    c.execute("SELECT COUNT(*) FROM workers WHERE is_admin = 0")
    if c.fetchone()[0] == 0:
        c.execute("SELECT id FROM modulos")
        mod_ids = [row[0] for row in c.fetchall()]
        default_pass = generate_password_hash("123456")
        
        workers_data = [
            ("11.111.111-1", "Juan Perez", "+56 9 1111 1111", default_pass, 0, mod_ids[0], "Full Time"),
            ("22.222.222-2", "Maria Gonzalez", "+56 9 2222 2222", default_pass, 0, mod_ids[0], "Part Time"),
            ("33.333.333-3", "Pedro Soto", "+56 9 3333 3333", default_pass, 0, mod_ids[1], "Full Time"),
            ("44.444.444-4", "Ana Silva", "+56 9 4444 4444", default_pass, 0, mod_ids[1], "Part Time"),
            ("55.555.555-5", "Diego Lopez", "+56 9 5555 5555", default_pass, 0, mod_ids[2], "Full Time"),
            ("66.666.666-6", "Claudia Jara", "+56 9 6666 6666", default_pass, 0, mod_ids[2], "Full Time"),
            ("77.777.777-7", "Roberto Munoz", "+56 9 7777 7777", default_pass, 0, mod_ids[3], "Part Time"),
            ("88.888.888-8", "Elena Espinoza", "+56 9 8888 8888", default_pass, 0, mod_ids[3], "Part Time"),
            ("99.999.999-9", "Mauricio Rivas", "+56 9 9999 9999", default_pass, 0, mod_ids[4], "Full Time"),
            ("10.101.101-0", "Sofia Castro", "+56 9 1010 1010", default_pass, 0, mod_ids[4], "Part Time")
        ]
        for w in workers_data:
            c.execute("INSERT OR IGNORE INTO workers (rut, name, phone, password_hash, is_admin, modulo_id, tipo) VALUES (?, ?, ?, ?, ?, ?, ?)", w)
        conn.commit()

    c.execute("SELECT COUNT(*) FROM rendiciones")
    if c.fetchone()[0] == 0:
        c.execute("SELECT id, modulo_id, tipo FROM workers WHERE is_admin = 0")
        workers_list = c.fetchall()

        hoy = datetime.date.today()
        dias_totales = 365 * 2 + hoy.timetuple().tm_yday
        
        for i in range(dias_totales):
            fecha_actual = (hoy - datetime.timedelta(days=i))
            fecha_str = fecha_actual.isoformat()
            
            num_modulos_hoy = random.randint(3, 6)
            mods_activos = random.sample(range(len(modulos_data)), k=num_modulos_hoy)
            
            for mod_idx in mods_activos:
                target_mod_id = mod_idx + 1
                workers_in_mod = [w for w in workers_list if w[1] == target_mod_id]
                
                if not workers_in_mod:
                    continue
                
                main_worker = random.choice(workers_in_mod)
                w_id, m_id, w_tipo = main_worker
                
                w_comision = 1
                if w_tipo == "Part Time":
                    w_comision = random.choice([0, 1])

                companion_id = None
                comp_hora_ent = None
                comp_hora_sal = None
                comp_comision = 0
                
                other_workers_in_mod = [w for w in workers_in_mod if w[0] != w_id]
                if other_workers_in_mod and random.random() < 0.3:
                    companion = random.choice(other_workers_in_mod)
                    companion_id = companion[0]
                    comp_hora_ent = "11:00"
                    comp_hora_sal = "19:00"
                    comp_comision = random.choice([0, 1])

                v_debito = random.randint(15000, 80000)
                v_credito = random.randint(10000, 120000)
                v_efectivo = random.randint(5000, 50000)
                
                b_debito = max(1, v_debito // 15000)
                b_credito = max(1, v_credito // 15000)
                b_efectivo = max(1, v_efectivo // 15000)
                
                gastos = random.choice([0, 0, 0, 2000, 5000])
                
                c.execute('''INSERT INTO rendiciones 
                    (worker_id, worker_comision, companion_id, companion_hora_entrada, companion_hora_salida, 
                    companion_comision, modulo_id, fecha, hora_entrada, hora_salida, 
                    venta_debito, venta_credito, venta_efectivo, 
                    boletas_debito, boletas_credito, boletas_efectivo,
                    gastos, observaciones) 
                    VALUES (?, ?, ?, ?, ?, ?, ?, ?, '09:00', '21:00', ?, ?, ?, ?, ?, ?, ?, ?)''',
                    (w_id, w_comision, companion_id, comp_hora_ent, comp_hora_sal, 
                    comp_comision, m_id, fecha_str, v_debito, v_credito, v_efectivo, 
                    b_debito, b_credito, b_efectivo, 
                    gastos, f"Carga histórica {fecha_str}"))
                
                rend_id = c.lastrowid

                # Buscar el precio activo en el momento exacto de la venta histórica
                c.execute('''
                    SELECT p.id, ph.price, ph.commission 
                    FROM productos p 
                    JOIN precios_historicos ph ON p.id = ph.producto_id 
                    WHERE ph.zona_id = (SELECT zona_id FROM modulos WHERE id = ?) 
                      AND ph.fecha_activacion = (
                          SELECT MAX(fecha_activacion) 
                          FROM precios_historicos 
                          WHERE producto_id = p.id AND zona_id = ph.zona_id 
                          AND fecha_activacion <= ?
                      )
                ''', (m_id, fecha_str + ' 23:59:59'))
                prods_zona = c.fetchall()
                
                if prods_zona:
                    items_hoy = random.sample(prods_zona, k=min(len(prods_zona), random.randint(2, 6)))
                    for p_id, p_price, p_comm in items_hoy:
                        c.execute('''INSERT INTO rendicion_items 
                            (rendicion_id, producto_id, cantidad, precio_historico, comision_historica)
                            VALUES (?, ?, ?, ?, ?)''',
                            (rend_id, p_id, random.randint(1, 4), p_price, p_comm))
            
            if i % 100 == 0:
                conn.commit()
                
        conn.commit()
    
    conn.close()
    
def init_db():
    conn = get_db_connection()
    c = conn.cursor()
    c.execute('''CREATE TABLE IF NOT EXISTS zonas
                 (id INTEGER PRIMARY KEY AUTOINCREMENT,
                  name TEXT UNIQUE NOT NULL)''')
    c.execute('''CREATE TABLE IF NOT EXISTS modulos
                 (id INTEGER PRIMARY KEY AUTOINCREMENT,
                  zona_id INTEGER NOT NULL,
                  name TEXT NOT NULL,
                  FOREIGN KEY (zona_id) REFERENCES zonas(id))''')
    # 1. Catálogo Maestro (Solo el nombre)
    c.execute('''CREATE TABLE IF NOT EXISTS productos
                 (id INTEGER PRIMARY KEY AUTOINCREMENT,
                  name TEXT UNIQUE NOT NULL)''')

    # 2. Historial de Precios por Zona
    c.execute('''CREATE TABLE IF NOT EXISTS precios_historicos
                 (id INTEGER PRIMARY KEY AUTOINCREMENT,
                  producto_id INTEGER NOT NULL,
                  zona_id INTEGER NOT NULL,
                  price INTEGER NOT NULL,
                  commission INTEGER NOT NULL,
                  fecha_activacion DATETIME NOT NULL,
                  FOREIGN KEY (producto_id) REFERENCES productos(id),
                  FOREIGN KEY (zona_id) REFERENCES zonas(id))''')
    c.execute('''CREATE TABLE IF NOT EXISTS workers
                 (id INTEGER PRIMARY KEY AUTOINCREMENT,
                  rut TEXT UNIQUE NOT NULL,
                  name TEXT NOT NULL,
                  phone TEXT NOT NULL,
                  password_hash TEXT NOT NULL,
                  is_admin BOOLEAN DEFAULT 0,
                  modulo_id INTEGER,
                  tipo TEXT DEFAULT 'Full Time',
                  FOREIGN KEY (modulo_id) REFERENCES modulos(id))''')
    c.execute('''CREATE TABLE IF NOT EXISTS rendiciones
                (id INTEGER PRIMARY KEY AUTOINCREMENT,
                worker_id INTEGER NOT NULL,
                worker_comision BOOLEAN DEFAULT 1,
                companion_id INTEGER,
                modulo_id INTEGER NOT NULL,
                fecha DATE NOT NULL,
                hora_entrada TEXT NOT NULL,
                hora_salida TEXT NOT NULL,
                companion_hora_entrada TEXT,
                companion_hora_salida TEXT,
                companion_comision BOOLEAN DEFAULT 0,
                venta_debito INTEGER DEFAULT 0,
                venta_credito INTEGER DEFAULT 0,
                venta_mp INTEGER DEFAULT 0,
                venta_efectivo INTEGER DEFAULT 0,
                boletas_debito INTEGER DEFAULT 0,
                boletas_credito INTEGER DEFAULT 0,
                boletas_mp INTEGER DEFAULT 0,
                boletas_efectivo INTEGER DEFAULT 0,
                gastos INTEGER DEFAULT 0,
                observaciones TEXT,
                FOREIGN KEY (worker_id) REFERENCES workers(id),
                FOREIGN KEY (companion_id) REFERENCES workers(id),
                FOREIGN KEY (modulo_id) REFERENCES modulos(id))''')
    c.execute('''CREATE TABLE IF NOT EXISTS rendicion_items
                 (id INTEGER PRIMARY KEY AUTOINCREMENT,
                  rendicion_id INTEGER NOT NULL,
                  producto_id INTEGER NOT NULL,
                  cantidad INTEGER NOT NULL,
                  precio_historico INTEGER NOT NULL,
                  comision_historica INTEGER NOT NULL,
                  FOREIGN KEY (rendicion_id) REFERENCES rendiciones(id),
                  FOREIGN KEY (producto_id) REFERENCES productos(id))''')
    
    c.execute("SELECT id FROM workers WHERE is_admin = 1")
    if not c.fetchone():
        admin_pass = generate_password_hash("admin123")
        c.execute("INSERT INTO workers (rut, name, phone, password_hash, is_admin) VALUES (?, ?, ?, ?, ?)", 
                  ("1-9", "System Admin", "+56 9 0000 0000", admin_pass, 1))
    
    conn.commit()
    conn.close()
    populateDefaults()