#!/usr/bin/env python3
"""
Script per prelevare dati da MySQL e inviare comandi SITE FTP
"""

import mysql.connector
from utils.database.connection import connetti_db
from utils.config import users_loader as setting
from ftplib import FTP
import logging
import sys
from typing import List, Tuple

# Configurazione logging
logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(levelname)s - %(message)s'
)
logger = logging.getLogger(__name__)

# Configurazione server FTP
FTP_CONFIG = {
    'host': 'localhost',
    'user': 'admin',
    'password': 'batt1l0',
    'port': 2121
}

def connect_ftp() -> FTP:
    """Connessione al server FTP"""
    try:
        ftp = FTP()
        ftp.connect(FTP_CONFIG['host'], FTP_CONFIG['port'])
        ftp.login(FTP_CONFIG['user'], FTP_CONFIG['password'])
        logger.info("Connessione FTP stabilita")
        return ftp
    except Exception as e:
        logger.error(f"Errore connessione FTP: {e}")
        sys.exit(1)

def fetch_data_from_db(connection: mysql.connector.MySQLConnection) -> List[Tuple]:
    """Preleva i dati dal database"""
    try:
        cursor = connection.cursor()

        # Modifica questa query secondo le tue esigenze
        query = """
        SELECT username, password
        FROM ase_lar.ftp_accounts
        """

        cursor.execute(query)
        results = cursor.fetchall()

        logger.info(f"Prelevate {len(results)} righe dal database")
        return results

    except mysql.connector.Error as e:
        logger.error(f"Errore query database: {e}")
        return []
    finally:
        cursor.close()

def send_site_command(ftp: FTP, command: str) -> bool:
    """Invia un comando SITE al server FTP"""
    try:
        # Il comando SITE viene inviato usando sendcmd
        response = ftp.sendcmd(f"SITE {command}")
        logger.info(f"Comando SITE '{command}' inviato. Risposta: {response}")
        return True
    except Exception as e:
        logger.error(f"Errore invio comando SITE '{command}': {e}")
        return False

def main():
    """Funzione principale"""
    logger.info("Avvio script caricamento utenti FTP")
    cfg = setting.Config()

    # Connessioni
    db_connection = connetti_db(cfg)
    ftp_connection = connect_ftp()

    try:
        # Preleva dati dal database
        data = fetch_data_from_db(db_connection)

        if not data:
            logger.warning("Nessun dato trovato nel database")
            return

        success_count = 0
        error_count = 0

        # Processa ogni riga
        for row in data:
            username, password = row

            # Costruisci il comando SITE completo
            ftp_site_command = f'addu {username} {password}'

            logger.info(f"Sending ftp command: {ftp_site_command}")

            # Invia comando SITE
            if send_site_command(ftp_connection, ftp_site_command):
                success_count += 1
            else:
                error_count += 1

        logger.info(f"Elaborazione completata. Successi: {success_count}, Errori: {error_count}")

    except Exception as e:
        logger.error(f"Errore generale: {e}")

    finally:
        # Chiudi connessioni
        try:
            ftp_connection.quit()
            logger.info("Connessione FTP chiusa")
        except Exception as e:
            logger.error(f"Errore chiusura connessione FTP: {e}")

        try:
            db_connection.close()
            logger.info("Connessione MySQL chiusa")
        except Exception as e:
            logger.error(f"Errore chiusura connessione MySQL: {e}")

if __name__ == "__main__":
    main()