<?php
require_once 'db/config.php';

echo "Recalculating funding totals...\n";

// 1. Reset all funding_raised to 0
db()->exec("UPDATE startups SET funding_raised = 0");
db()->exec("UPDATE funding_rounds SET funding_raised = 0");

// 2. Update funding_rounds totals
$stmt = db()->query("SELECT funding_round_id, SUM(amount) as total FROM investments WHERE status != 'rejected' GROUP BY funding_round_id");
$roundTotals = $stmt->fetchAll();

foreach ($roundTotals as $t) {
    if (!$t['funding_round_id']) continue;
    echo "Updating Round ID: {$t['funding_round_id']} with £{$t['total']}\n";
    $stmt2 = db()->prepare("UPDATE funding_rounds SET funding_raised = ? WHERE id = ?");
    $stmt2->execute([$t['total'], $t['funding_round_id']]);
}

// 3. Update startups totals (sum of all their rounds)
$stmt = db()->query("SELECT startup_id, SUM(amount) as total FROM investments WHERE status != 'rejected' GROUP BY startup_id");
$startupTotals = $stmt->fetchAll();

foreach ($startupTotals as $t) {
    if (!$t['startup_id']) continue;
    echo "Updating Startup ID: {$t['startup_id']} with £{$t['total']}\n";
    $stmt3 = db()->prepare("UPDATE startups SET funding_raised = ? WHERE id = ?");
    $stmt3->execute([$t['total'], $t['startup_id']]);
}

echo "Done!\n";