Habitat Management

public function save()
{
    $id = $_POST['id_habitat'] ?? null;
    
    if ($id) {
        // UPDATE
        if (!hasPermission('habitats-edit')) {
            header('Location: /habitats/gest/start?msg=error');
            exit;
        }
        
        $habitatModel->update($id, $name, $description);
        
        // Handle new images
        if ($urlMobile || $urlTablet || $urlDesktop) {
            // Unlink old media
            $mediaModel->unlink('habitats', $id);
            // Link new media
            $mediaModel->link($mediaId, 'habitats', $id);
        }
    }
}

public function delete()
{
    if (!hasPermission('habitats-delete')) {
        header('Location: /habitats/gest/start?msg=error');
        exit;
    }

    $id = $_GET['id'] ?? null;
    if ($id) {
        // Delete media relation first
        $mediaModel->unlink('habitats', $id);
        
        // Delete habitat
        $habitatModel->delete($id);
    }
    
    header('Location: /habitats/gest/start?msg=deleted');
    exit;
}

// In Habitat model
public function getAnimalsByHabitatId($habitatId)
{
    $sql = "SELECT af.*, 
                   ag.animal_name, 
                   ag.gender,
                   s.specie_name,
                   c.category_name,
                   n.nutrition_type
            FROM animal_full af
            JOIN animal_general ag ON af.animal_g_id = ag.id_animal_g
            LEFT JOIN specie s ON ag.specie_id = s.id_specie
            LEFT JOIN category c ON s.category_id = c.id_category
            LEFT JOIN nutrition n ON af.nutrition_id = n.id_nutrition
            WHERE af.habitat_id = :habitat_id
            ORDER BY ag.animal_name ASC";
    
    $stmt = $this->db->prepare($sql);
    $stmt->execute([':habitat_id' => $habitatId]);
    return $stmt->fetchAll(PDO::FETCH_OBJ);
}

public function habitat1() {
    $id = $_GET['id'] ?? null;
    
    // Get habitat details
    $habitat = $habitatModel->getById($id);
    
    // Get animals in this habitat
    $animals = $habitatModel->getAnimalsByHabitatId($id);
    
    // Get latest health state for each animal
    $healthReportModel = new HealthStateReport();
    foreach ($animals as $animal) {
        $latestReport = $healthReportModel->getLatestByAnimalId(
            $animal->id_full_animal
        );
        $animal->latest_health_state = $latestReport->hsr_state ?? null;
    }
    
    // Load view
    include_once __DIR__ . '/../views/pages/habitat1.php';
}

public function edit()
{
    $userRoleName = $_SESSION['user']['role_name'] ?? null;
    if ($userRoleName !== 'Veterinary') {
        header('Location: /habitats/suggestion/start?msg=error');
        exit;
    }

    $id = $_GET['id'] ?? null;
    $suggestion = $suggestionModel->getById($id);
    $userId = $_SESSION['user']['id_user'] ?? null;

    // Check ownership
    if ($suggestion->suggested_by != $userId) {
        header('Location: /habitats/suggestion/start?msg=error&error=You can only edit your own suggestions');
        exit;
    }

    // Check status
    if ($suggestion->status !== 'pending') {
        header('Location: /habitats/suggestion/start?msg=error&error=You can only edit pending suggestions');
        exit;
    }
    
    include_once __DIR__ . '/../views/suggestion/edit.php';
}

public function delete()
{
    $id = $_GET['id'] ?? null;
    $userRoleName = $_SESSION['user']['role_name'] ?? null;
    $userId = $_SESSION['user']['id_user'] ?? null;

    $suggestion = $suggestionModel->getById($id);

    // Veterinarians can only delete accepted/rejected (not pending)
    if ($userRoleName === 'Veterinary') {
        if ($suggestion->suggested_by != $userId) {
            header('Location: /habitats/suggestion/start?msg=error&error=You can only delete your own suggestions');
            exit;
        }
        if ($suggestion->status === 'pending') {
            header('Location: /habitats/suggestion/start?msg=error&error=You cannot delete pending suggestions. Edit or wait for review.');
            exit;
        }
    }
    
    // Admins can only delete accepted/rejected (not pending - must review first)
    if ($userRoleName === 'Admin') {
        if ($suggestion->status === 'pending') {
            header('Location: /habitats/suggestion/start?msg=error&error=You cannot delete pending suggestions. Please accept or reject them first.');
            exit;
        }
    }

    $result = $suggestionModel->delete($id, $userRoleName);
}

public function start()
{
    $userRoleName = $_SESSION['user']['role_name'] ?? null;
    $userId = $_SESSION['user']['id_user'] ?? null;
    
    // Veterinarians see only their suggestions
    if ($userRoleName === 'Veterinary') {
        $suggestions = $suggestionModel->getByVeterinarian($userId);
    } else {
        // Admins see all suggestions
        $suggestions = $suggestionModel->getAll();
    }
    
    include_once __DIR__ . '/../views/suggestion/start.php';
}

// Handle Hero creation/update (optional)
$heroTitle = trim($_POST['hero_title'] ?? '');
$heroSubtitle = trim($_POST['hero_subtitle'] ?? '');

if (!empty($heroTitle)) {
    $heroModel = new Hero();
    $existingHero = $heroModel->getByHabitatId($habitatId);
    
    if ($existingHero) {
        // Update existing hero
        $heroModel->update(
            $existingHero->id_hero,
            $heroTitle,
            $heroSubtitle,
            'habitats',
            false,
            $habitatId
        );
    } else {
        // Create new hero
        $heroModel->create(
            $heroTitle,
            $heroSubtitle,
            'habitats',
            false,
            $habitatId
        );
    }
}