TCollectionEmbeddedObjects.class.php

<?php

namespace wcf\data;

use wcf\system\message\embedded\object\MessageEmbeddedObjectManager;

/**
 * Trait for dbo collections with message embedded objects.
 *
 * @author      Marcel Werk
 * @copyright   2001-2025 WoltLab GmbH
 * @license     GNU Lesser General Public License <http://opensource.org/licenses/lgpl-license.php>
 * @since       6.2
 */
trait TCollectionEmbeddedObjects
{
    private bool $embeddedObjectsLoaded = false;

    public function loadEmbeddedObjects(string $messageObjectType): void
    {
        if ($this->embeddedObjectsLoaded) {
            return;
        }

        $this->embeddedObjectsLoaded = true;

        $objectIDs = $this->getEmbeddedObjectIDs();
        if ($objectIDs === []) {
            return;
        }

        MessageEmbeddedObjectManager::getInstance()->loadObjects(
            $messageObjectType,
            $objectIDs
        );
    }

    /**
     * @return int[]
     */
    private function getEmbeddedObjectIDs(): array
    {
        \assert($this instanceof DatabaseObjectCollection);

        return \array_map(
            static fn($object) => $object->getObjectID(),
            \array_filter($this->getObjects(), fn($object) => $object->hasEmbeddedObjects === 1)
        );
    }
}