<?php
namespace Omeka\Db\Event\Listener;

use Doctrine\ORM\Event\LoadClassMetadataEventArgs;

/**
 * Load the resource discriminator map dynamically.
 */
class ResourceDiscriminatorMap
{
    /**
     * @var array
     */
    protected $discriminatorMap;

    /**
     * Set the resource discriminator map.
     */
    public function __construct(array $discriminatorMap)
    {
        $this->discriminatorMap = $discriminatorMap;
    }

    /**
     * Attach the discriminator map to the Resource entity.
     *
     * @param LoadClassMetadataEventArgs $event
     */
    public function loadClassMetadata(LoadClassMetadataEventArgs $event)
    {
        $classMetadata = $event->getClassMetadata();
        if ('Omeka\Entity\Resource' == $classMetadata->name) {
            $classMetadata->discriminatorMap = $this->discriminatorMap;
        }
    }
}
