<?php
require 'includes/db.php';

try {
    // Check if company_id exists in deals table
    $stmt = $pdo->query("SHOW COLUMNS FROM deals LIKE 'company_id'");
    if ($stmt->rowCount() == 0) {
        // Add company_id column
        $pdo->exec("ALTER TABLE deals ADD COLUMN company_id INT DEFAULT NULL AFTER id");
        echo "Added company_id column to deals table.\n";
        
        // Populate company_id based on property_id -> company_id (via properties table)
        // OR agent_id -> company_id (via users table)
        // Let's try to infer from properties first as deals are usually property-centric
        
        // Assuming properties table has company_id (which it should in multi-tenant setup)
        $pdo->exec("UPDATE deals d 
                    JOIN properties p ON d.property_id = p.id 
                    SET d.company_id = p.company_id 
                    WHERE d.company_id IS NULL");
        echo "Populated company_id from properties table.\n";
        
        // Fallback: update from agent's company_id
        $pdo->exec("UPDATE deals d 
                    JOIN users u ON d.agent_id = u.id 
                    SET d.company_id = u.company_id 
                    WHERE d.company_id IS NULL");
        echo "Populated remaining company_id from agents.\n";
        
    } else {
        echo "company_id column already exists in deals table.\n";
    }

} catch (PDOException $e) {
    echo "Error: " . $e->getMessage() . "\n";
}
?>