const { outputProgress } = require('./utils/progress');
const { getConnection } = require('./utils/db');

async function calculateSalesForecasts(startTime, totalProducts, processedCount) {
    const connection = await getConnection();
    try {
        outputProgress({
            status: 'running',
            operation: 'Calculating sales forecasts',
            current: Math.floor(totalProducts * 0.98),
            total: totalProducts,
            elapsed: formatElapsedTime(startTime),
            remaining: estimateRemaining(startTime, Math.floor(totalProducts * 0.98), totalProducts),
            rate: calculateRate(startTime, Math.floor(totalProducts * 0.98)),
            percentage: '98'
        });

        // First, create a temporary table for forecast dates
        await connection.query(`
            CREATE TEMPORARY TABLE IF NOT EXISTS temp_forecast_dates (
                forecast_date DATE,
                day_of_week INT,
                month INT,
                PRIMARY KEY (forecast_date)
            )
        `);

        await connection.query(`
            INSERT INTO temp_forecast_dates
            SELECT 
                DATE_ADD(CURRENT_DATE, INTERVAL n DAY) as forecast_date,
                DAYOFWEEK(DATE_ADD(CURRENT_DATE, INTERVAL n DAY)) as day_of_week,
                MONTH(DATE_ADD(CURRENT_DATE, INTERVAL n DAY)) as month
            FROM (
                SELECT a.N + b.N * 10 as n 
                FROM 
                    (SELECT 0 as N UNION SELECT 1 UNION SELECT 2 UNION SELECT 3 UNION SELECT 4 UNION
                     SELECT 5 UNION SELECT 6 UNION SELECT 7 UNION SELECT 8 UNION SELECT 9) a,
                    (SELECT 0 as N UNION SELECT 1 UNION SELECT 2) b
                ORDER BY n
                LIMIT 31
            ) numbers
        `);

        // Create temporary table for daily sales stats
        await connection.query(`
            CREATE TEMPORARY TABLE IF NOT EXISTS temp_daily_sales AS
            SELECT 
                o.pid,
                DAYOFWEEK(o.date) as day_of_week,
                SUM(o.quantity) as daily_quantity,
                SUM(o.price * o.quantity) as daily_revenue,
                COUNT(DISTINCT DATE(o.date)) as day_count
            FROM orders o
            WHERE o.canceled = false
                AND o.date >= DATE_SUB(CURRENT_DATE, INTERVAL 90 DAY)
            GROUP BY o.pid, DAYOFWEEK(o.date)
        `);

        // Create temporary table for product stats
        await connection.query(`
            CREATE TEMPORARY TABLE IF NOT EXISTS temp_product_stats AS
            SELECT 
                pid,
                AVG(daily_revenue) as overall_avg_revenue,
                SUM(day_count) as total_days
            FROM temp_daily_sales
            GROUP BY pid
        `);

        // Calculate product-level forecasts
        await connection.query(`
            INSERT INTO sales_forecasts (
                pid,
                forecast_date,
                forecast_units,
                forecast_revenue,
                confidence_level,
                last_calculated_at
            )
            SELECT 
                ds.pid,
                fd.forecast_date,
                GREATEST(0, 
                    AVG(ds.daily_quantity) * 
                    (1 + COALESCE(sf.seasonality_factor, 0))
                ) as forecast_units,
                GREATEST(0, 
                    COALESCE(
                        CASE 
                            WHEN SUM(ds.day_count) >= 4 THEN AVG(ds.daily_revenue)
                            ELSE ps.overall_avg_revenue
                        END * 
                        (1 + COALESCE(sf.seasonality_factor, 0)) *
                        (0.95 + (RAND() * 0.1)),
                        0
                    )
                ) as forecast_revenue,
                CASE
                    WHEN ps.total_days >= 60 THEN 90
                    WHEN ps.total_days >= 30 THEN 80
                    WHEN ps.total_days >= 14 THEN 70
                    ELSE 60
                END as confidence_level,
                NOW() as last_calculated_at
            FROM temp_daily_sales ds
            JOIN temp_product_stats ps ON ds.pid = ps.pid
            CROSS JOIN temp_forecast_dates fd
            LEFT JOIN sales_seasonality sf ON fd.month = sf.month
            GROUP BY ds.pid, fd.forecast_date, ps.overall_avg_revenue, ps.total_days, sf.seasonality_factor
            HAVING AVG(ds.daily_quantity) > 0
            ON DUPLICATE KEY UPDATE
                forecast_units = VALUES(forecast_units),
                forecast_revenue = VALUES(forecast_revenue),
                confidence_level = VALUES(confidence_level),
                last_calculated_at = NOW()
        `);

        // Create temporary table for category stats
        await connection.query(`
            CREATE TEMPORARY TABLE IF NOT EXISTS temp_category_sales AS
            SELECT 
                pc.cat_id,
                DAYOFWEEK(o.date) as day_of_week,
                SUM(o.quantity) as daily_quantity,
                SUM(o.price * o.quantity) as daily_revenue,
                COUNT(DISTINCT DATE(o.date)) as day_count
            FROM orders o
            JOIN product_categories pc ON o.pid = pc.pid
            WHERE o.canceled = false
                AND o.date >= DATE_SUB(CURRENT_DATE, INTERVAL 90 DAY)
            GROUP BY pc.cat_id, DAYOFWEEK(o.date)
        `);

        await connection.query(`
            CREATE TEMPORARY TABLE IF NOT EXISTS temp_category_stats AS
            SELECT 
                cat_id,
                AVG(daily_revenue) as overall_avg_revenue,
                SUM(day_count) as total_days
            FROM temp_category_sales
            GROUP BY cat_id
        `);

        // Calculate category-level forecasts
        await connection.query(`
            INSERT INTO category_forecasts (
                category_id,
                forecast_date,
                forecast_units,
                forecast_revenue,
                confidence_level,
                last_calculated_at
            )
            SELECT 
                cs.cat_id as category_id,
                fd.forecast_date,
                GREATEST(0, 
                    AVG(cs.daily_quantity) * 
                    (1 + COALESCE(sf.seasonality_factor, 0))
                ) as forecast_units,
                GREATEST(0, 
                    COALESCE(
                        CASE 
                            WHEN SUM(cs.day_count) >= 4 THEN AVG(cs.daily_revenue)
                            ELSE ct.overall_avg_revenue
                        END * 
                        (1 + COALESCE(sf.seasonality_factor, 0)) *
                        (0.95 + (RAND() * 0.1)),
                        0
                    )
                ) as forecast_revenue,
                CASE
                    WHEN ct.total_days >= 60 THEN 90
                    WHEN ct.total_days >= 30 THEN 80
                    WHEN ct.total_days >= 14 THEN 70
                    ELSE 60
                END as confidence_level,
                NOW() as last_calculated_at
            FROM temp_category_sales cs
            JOIN temp_category_stats ct ON cs.cat_id = ct.cat_id
            CROSS JOIN temp_forecast_dates fd
            LEFT JOIN sales_seasonality sf ON fd.month = sf.month
            GROUP BY cs.cat_id, fd.forecast_date, ct.overall_avg_revenue, ct.total_days, sf.seasonality_factor
            HAVING AVG(cs.daily_quantity) > 0
            ON DUPLICATE KEY UPDATE
                forecast_units = VALUES(forecast_units),
                forecast_revenue = VALUES(forecast_revenue),
                confidence_level = VALUES(confidence_level),
                last_calculated_at = NOW()
        `);

        // Clean up temporary tables
        await connection.query(`
            DROP TEMPORARY TABLE IF EXISTS temp_forecast_dates;
            DROP TEMPORARY TABLE IF EXISTS temp_daily_sales;
            DROP TEMPORARY TABLE IF EXISTS temp_product_stats;
            DROP TEMPORARY TABLE IF EXISTS temp_category_sales;
            DROP TEMPORARY TABLE IF EXISTS temp_category_stats;
        `);

        return Math.floor(totalProducts * 1.0);
    } finally {
        connection.release();
    }
}

module.exports = calculateSalesForecasts;