ModelClassResolver.php

<?php

namespace AppBundle\Pomm\Resolver;

use ArrayObject;
use PommProject\ModelManager\Model\FlexibleEntity;

class ModelClassResolver extends ArrayObject
{
    /**
     * @param Array $modelClasses
     */
    public function __construct(Array $modelClasses)
    {
        foreach ($modelClasses as $modelClass) {
            $this[$modelClass::FLEXIBLE_ENTITY_CLASS] = $modelClass;
        }
    }

    /**
     * Resolve the model class for a flexible entity.
     *
     * @param  FlexibleEntity $entity
     *
     * @return string
     */
    public function resolve(FlexibleEntity $entity)
    {
        $flexibleEntityCLass = get_class($entity);

        if (!isset($this[$flexibleEntityCLass])) {
            throw new \Exception('Could not resolve model class');
        }

        return $this[$flexibleEntityCLass];
    }
}