TranslatableListener::setTranslationFallback() - Code Metrics - Inspection of "bc fix" - ozo2003/HelperBundle - Measure and Improve Code Quality continuously with Scrutinizer

Passed
Branch — master (a0cc06)

by Dāvis
created 2017-10-13 07:16 UTC
TranslatableListener::setTranslationFallback() A

↳ Parent: TranslatableListener
Complexity

Conditions	1
Paths	1
Size

Total Lines	5
Code Lines	2
Duplication

Lines	0
Ratio	0 %
Importance

Changes
Metric	Value
cc	1
eloc	2
nc	1
nop	1
dl	0
loc	5
rs	9.4285
c	0
b	0
f	0
<?php

namespace Sludio\HelperBundle\Translatable\Listener;

use Doctrine\Common\EventArgs;
use Doctrine\ORM\ORMInvalidArgumentException;
filter:
    dependency_paths: ["lib/*"]
use Gedmo\Exception\InvalidArgumentException;
use Gedmo\Exception\RuntimeException;
use Gedmo\Tool\Wrapper\AbstractWrapper;
use Gedmo\Translatable\Mapping\Event\TranslatableAdapter;
use Gedmo\Translatable\TranslatableListener as BaseListener;

/**
 * The translation listener handles the generation and
 * loading of translations for entities which implements
 * the Translatable interface.
 *
 * This behavior can impact the performance of your application
 * since it does an additional query for each field to translate.
 *
 * Nevertheless the annotation metadata is properly cached and
 * it is not a big overhead to lookup all entity annotations since
 * the caching is activated for metadata
 *
 */
class TranslatableListener extends BaseListener
{
    /**
     * Query hint to override the fallback of translations
     * integer 1 for true, 0 false.
     */
    const HINT_FALLBACK = 'gedmo.translatable.fallback';

    /**
     * Query hint to override the fallback locale.
     */
    const HINT_TRANSLATABLE_LOCALE = 'gedmo.translatable.locale';

    /**
     * Query hint to use inner join strategy for translations.
     */
    const HINT_INNER_JOIN = 'gedmo.translatable.inner_join.translations';

    /**
     * Locale which is set on this listener.
     * If Entity being translated has locale defined it
     * will override this one.
     *
     * @var string
     */
    protected $locale = 'en_US';

    /**
     * Default locale, this changes behavior
     * to not update the original record field if locale
     * which is used for updating is not default. This
     * will load the default translation in other locales
     * if record is not translated yet.
     *
     * @var string
     */
    private $defaultLocale = 'en_US';

    /**
     * If this is set to false, when if entity does
     * not have a translation for requested locale
     * it will show a blank value.
     *
     * @var bool
     */
    private $translationFallback = false;

    /**
     * List of translations which do not have the foreign
     * key generated yet - MySQL case. These translations
     * will be updated with new keys on postPersist event.
     *
     * @var array
     */
    private $pendingTranslationInserts = [];

    /**
     * Currently in case if there is TranslationQueryWalker
     * in charge. We need to skip issuing additional queries
     * on load.
     *
     * @var bool
     */
    private $skipOnLoad = true;

    /**
     * Tracks locale the objects currently translated in.
     *
     * @var array
     */
    private $translatedInLocale = [];

    /**
     * Whether or not, to persist default locale
     * translation or keep it in original record.
     *
     * @var bool
     */
    private $persistDefaultLocaleTranslation = false;

    /**
     * Tracks translation object for default locale.
     *
     * @var array
     */
    private $translationInDefaultLocale = [];

    /**
     * Specifies the list of events to listen.
     *
     * @return array
     */
    public function getSubscribedEvents()
    {
        return [
            'postLoad',
            'postPersist',
            'preFlush',
            'onFlush',
            'loadClassMetadata',
        ];
    }

    /**
     * Set to skip or not onLoad event.
     *
     * @param bool $bool
     *
     * @return static
     */
    public function setSkipOnLoad($bool)
    {
        $this->skipOnLoad = (bool)$bool;

        return $this;
    }

    /**
     * Whether or not, to persist default locale
     * translation or keep it in original record.
     *
     * @param bool $bool
     *
     * @return static
     */
    public function setPersistDefaultLocaleTranslation($bool)
    {
        $this->persistDefaultLocaleTranslation = (bool)$bool;

        return $this;
    }

    /**
     * Check if should persist default locale
     * translation or keep it in original record.
     *
     * @return bool
     */
    public function getPersistDefaultLocaleTranslation()
    {
        return (bool)$this->persistDefaultLocaleTranslation;
    }

    /**
     * Add additional $translation for pending $oid object
     * which is being inserted.
     *
     * @param string $oid
     * @param object $translation
     */
    public function addPendingTranslationInsert($oid, $translation)
    {
        $this->pendingTranslationInserts[$oid][] = $translation;
    }

    /**
     * Maps additional metadata.
     *
     * @param EventArgs $eventArgs
     */
    public function loadClassMetadata(EventArgs $eventArgs)
    {
        $adapter = $this->getEventAdapter($eventArgs);
        $this->loadMetadataForObjectClass($adapter->getObjectManager(), $eventArgs->getClassMetadata());
        $this->loadMetadataForObjectClass($adapter->getObjectManager(), $eventArgs->/** @scrutinizer ignore-call */ getClassMetadata());
    }

    /**
     * Get the translation class to be used
     * for the object $class.
     *
     * @param TranslatableAdapter $adapter
     * @param string              $class
     *
     * @return string
     */
    public function getTranslationClass(TranslatableAdapter $adapter, $class)
    {
        return isset(self::$configurations[$this->name][$class]['translationClass']) ? self::$configurations[$this->name][$class]['translationClass'] : $adapter->getDefaultTranslationClass();
    }

    /**
     * Enable or disable translation fallback
     * to original record value.
     *
     * @param bool $bool
     *
     * @return static
     */
    public function setTranslationFallback($bool)
    {
        $this->translationFallback = (bool)$bool;

        return $this;
    }

    /**
     * Weather or not is using the translation
     * fallback to original record.
     *
     * @return bool
     */
    public function getTranslationFallback()
    {
        return $this->translationFallback;
    }

    /**
     * Set the locale to use for translation listener.
     *
     * @param string $locale
     *
     * @return static
     */
    public function setTranslatableLocale($locale)
    {
        $this->validateLocale($locale);
        $this->locale = $locale;

        return $this;
    }

    /**
     * Sets the default locale, this changes behavior
     * to not update the original record field if locale
     * which is used for updating is not default.
     *
     * @param string $locale
     *
     * @return static
     */
    public function setDefaultLocale($locale)
    {
        $this->validateLocale($locale);
        $this->defaultLocale = $locale;

        return $this;
    }

    /**
     * Gets the default locale.
     *
     * @return string
     */
    public function getDefaultLocale()
    {
        return $this->defaultLocale;
    }

    /**
     * Get currently set global locale, used
     * extensively during query execution.
     *
     * @return string
     */
    public function getListenerLocale()
    {
        return $this->locale;
    }

    /**
     * Gets the locale to use for translation. Loads object
     * defined locale first..
     *
     * @param object $object
     * @param object $meta
     * @param object $manager
     *
     * @throws \Gedmo\Exception\RuntimeException - if language or locale property is not
     *                                           found in entity
     *
     * @return string
     */
    public function getTranslatableLocale($object, $meta, $manager = null)
    {
        $locale = $this->locale;
        if (isset(self::$configurations[$this->name][$meta->name]['locale'])) {
            /** @var \ReflectionClass $class */
            $class = $meta->getReflectionClass();
            $reflectionProperty = $class->getProperty(self::$configurations[$this->name][$meta->name]['locale']);
            if (!$reflectionProperty) {
                $column = self::$configurations[$this->name][$meta->name]['locale'];
                throw new RuntimeException("There is no locale or language property ({$column}) found on object: {$meta->name}");
            }
            $reflectionProperty->setAccessible(true);
            $value = $reflectionProperty->getValue($object);
            if (is_object($value) && method_exists($value, '__toString')) {
                $value = (string)$value;
            }
            if ($this->isValidLocale($value)) {
                $locale = $value;
            }
        }

        return $locale;
    }

    /**
     * Handle translation changes in default locale.
     *
     * This has to be done in the preFlush because, when an entity has been loaded
     * in a different locale, no changes will be detected.
     *
     * @param EventArgs $args
     */
    public function preFlush(EventArgs $args)
    {
        $adapter = $this->getEventAdapter($args);
        $manager = $adapter->getObjectManager();
        $uow = $manager->getUnitOfWork();
        $uow = $manager->/** @scrutinizer ignore-call */ getUnitOfWork();

        foreach ($this->translationInDefaultLocale as $oid => $fields) {
            $trans = reset($fields);
            if ($adapter->usesPersonalTranslation(get_class($trans))) {
                $entity = $trans->getObject();
            } else {
                $entity = $uow->tryGetById($trans->getForeignKey(), $trans->getObjectClass());
            }

            if (!$entity) {
                continue;
            }

            try {
                $uow->scheduleForUpdate($entity);
            } catch (ORMInvalidArgumentException $e) {
                foreach ($fields as $field => $trans) {
                    $this->removeTranslationInDefaultLocale($oid, $field);
                }
            }
        }
    }

    /**
     * Looks for translatable objects being inserted or updated
     * for further processing.
     *
     * @param EventArgs $args
     */
    public function onFlush(EventArgs $args)
    {
        $adapter = $this->getEventAdapter($args);
        $manager = $adapter->getObjectManager();
        $uow = $manager->getUnitOfWork();
        // check all scheduled inserts for Translatable objects
        foreach ($adapter->getScheduledObjectInsertions($uow) as $object) {

            $meta = $manager->getClassMetadata(get_class($object));
            $config = $this->getConfiguration($manager, $meta->name);

            if (isset($config['fields'])) {
                $this->handleTranslatableObjectUpdate($adapter, $object, true);
            }
        }
        // check all scheduled updates for Translatable entities
        foreach ($adapter->getScheduledObjectUpdates($uow) as $object) {

            $meta = $manager->getClassMetadata(get_class($object));
            $config = $this->getConfiguration($manager, $meta->name);
            if (isset($config['fields'])) {
                $this->handleTranslatableObjectUpdate($adapter, $object, false);
            }
        }
        // check scheduled deletions for Translatable entities
        foreach ($adapter->getScheduledObjectDeletions($uow) as $object) {
            $meta = $manager->getClassMetadata(get_class($object));
            $config = $this->getConfiguration($manager, $meta->name);
            if (isset($config['fields'])) {
                $wrapped = AbstractWrapper::wrap($object, $manager);
                $transClass = $this->getTranslationClass($adapter, $meta->name);
                $adapter->removeAssociatedTranslations($wrapped, $transClass, $config['useObjectClass']);
            }
        }
    }

    /**
     * Checks for inserted object to update their translation
     * foreign keys.
     *
     * @param EventArgs $args
     */
    public function postPersist(EventArgs $args)
    {
        $adapter = $this->getEventAdapter($args);
        $manager = $adapter->getObjectManager();
        $object = $adapter->getObject();
        $meta = $manager->getClassMetadata(get_class($object));
        // check if entity is tracked by translatable and without foreign key
        if ($this->getConfiguration($manager, $meta->name) && count($this->pendingTranslationInserts)) {

            $oid = spl_object_hash($object);
            if (array_key_exists($oid, $this->pendingTranslationInserts)) {
                // load the pending translations without key
                $wrapped = AbstractWrapper::wrap($object, $manager);
                $objectId = $wrapped->getIdentifier();
                $translationClass = $this->getTranslationClass($adapter, get_class($object));
                foreach ($this->pendingTranslationInserts[$oid] as $translation) {
                    if ($adapter->usesPersonalTranslation($translationClass)) {
                        $translation->setObject($objectId);
                    } else {
                        $translation->setForeignKey($objectId);
                    }
                    $adapter->insertTranslationRecord($translation);
                }
                unset($this->pendingTranslationInserts[$oid]);
            }
        }
    }

    /**
     * After object is loaded, listener updates the translations
     * by currently used locale.
     *
     * @param EventArgs $args
     */
    public function postLoad(EventArgs $args)
    {
        $adapter = $this->getEventAdapter($args);
        $manager = $adapter->getObjectManager();
        $object = $adapter->getObject();
        $meta = $manager->getClassMetadata(get_class($object));
        $config = $this->getConfiguration($manager, $meta->name);

        if (isset($config['fields'])) {
            $locale = $this->getTranslatableLocale($object, $meta, $manager);
            $oid = spl_object_hash($object);
            $this->translatedInLocale[$oid] = $locale;
        }

        if ($this->skipOnLoad) {
            return;
        }

        if (isset($config['fields']) && ($locale !== $this->defaultLocale || $this->persistDefaultLocaleTranslation)) {

            // fetch translations
            $translationClass = $this->getTranslationClass($adapter, $config['useObjectClass']);
            $result = $adapter->loadTranslations($object, $translationClass, $locale, $config['useObjectClass']);
            // translate object's translatable properties
            foreach ($config['fields'] as $field) {
                $translated = '';
                foreach ((array)$result as $entry) {
                    if ($entry['field'] == $field) {
                        $translated = $entry['content'];
                        break;
                    }
                }
                // update translation
                if ($translated || (!$this->translationFallback && (!isset($config['fallback'][$field]) || !$config['fallback'][$field])) || ($this->translationFallback && isset($config['fallback'][$field]) && !$config['fallback'][$field])) {
                    $adapter->setTranslationValue($object, $field, $translated);
                    // ensure clean changeset
                    $adapter->setOriginalObjectProperty($manager->getUnitOfWork(), $oid, $field, $meta->getReflectionProperty($field)
                    $adapter->setOriginalObjectProperty($manager->getUnitOfWork(), $oid, $field, $meta->/** @scrutinizer ignore-call */ getReflectionProperty($field)
                        ->getValue($object));
                }
            }
        }
    }

    /**
     * {@inheritdoc}
     */
    protected function getNamespace()
    {
        return 'Sludio\HelperBundle\Translatable';
    }

    /**
     * Validates the given locale.
     *
     * @param string $locale - locale to validate
     *
     * @throws \Gedmo\Exception\InvalidArgumentException if locale is not valid
     */
    protected function validateLocale($locale)
    {
        if (!$this->isValidLocale($locale)) {
            throw new InvalidArgumentException('Locale or language cannot be empty and must be set through Listener or Entity');
        }
    }

    /**
     * Check if the given locale is valid.
     *
     * @param string $locale - locale to check
     *
     * @return bool
     */
    private function isValidlocale($locale)
    {
        return is_string($locale) && strlen($locale);
    }

    /**
     * Creates the translation for object being flushed.
     *
     * @param TranslatableAdapter $adapter
     * @param object              $object
     * @param bool                $isInsert
     *
     * @throws \UnexpectedValueException - if locale is not valid, or
     *                                   primary key is composite, missing or invalid
     */
    private function handleTranslatableObjectUpdate(TranslatableAdapter $adapter, $object, $isInsert)
    {
        $manager = $adapter->getObjectManager();
        $wrapped = AbstractWrapper::wrap($object, $manager);
        $meta = $wrapped->getMetadata();
        $config = $this->getConfiguration($manager, $meta->name);
        // no need cache, metadata is loaded only once in MetadataFactoryClass
        $translationClass = $this->getTranslationClass($adapter, $config['useObjectClass']);
        $translationMetadata = $manager->getClassMetadata($translationClass);

        // check for the availability of the primary key
        $objectId = $wrapped->getIdentifier();
        // load the currently used locale
        $locale = $this->getTranslatableLocale($object, $meta, $manager);

        $uow = $manager->getUnitOfWork();
        $oid = spl_object_hash($object);
        $changeSet = $adapter->getObjectChangeSet($uow, $object);
        $translatableFields = $config['fields'];
        foreach ($translatableFields as $field) {
            $wasPersistedSeparetely = false;
            $skip = isset($this->translatedInLocale[$oid]) && $locale === $this->translatedInLocale[$oid];
            $skip = $skip && !isset($changeSet[$field]) && !$this->getTranslationInDefaultLocale($oid, $field);
            if ($skip) {
                continue; // locale is same and nothing changed
            }
            $translation = null;
            foreach ($adapter->getScheduledObjectInsertions($uow) as $trans) {
                if ($locale !== $this->defaultLocale && get_class($trans) === $translationClass && $trans->getLocale() === $this->defaultLocale && $trans->getField() === $field && $this->belongsToObject($adapter, $trans, $object)) {
                    $this->setTranslationInDefaultLocale($oid, $field, $trans);
                    break;
                }
            }

            // lookup persisted translations
            foreach ($adapter->getScheduledObjectInsertions($uow) as $trans) {
                if (get_class($trans) !== $translationClass || $trans->getLocale() !== $locale || $trans->getField() !== $field) {
                    continue;
                }

                if ($adapter->usesPersonalTranslation($translationClass)) {
                    $wasPersistedSeparetely = $trans->getObject() === $object;
                } else {
                    $wasPersistedSeparetely = ($trans->getObjectClass() === $config['useObjectClass'] && $trans->getForeignKey() === $objectId);
                }

                if ($wasPersistedSeparetely) {
                    $translation = $trans;
                    break;
                }
            }

            // check if translation already is created
            if (!$isInsert && !$translation) {
                $translation = $adapter->findTranslation($wrapped, $locale, $field, $translationClass, $config['useObjectClass']);
            }

            // create new translation if translation not already created and locale is different from default locale, otherwise, we have the date in the original record
            $persistNewTranslation = !$translation && ($locale !== $this->defaultLocale || $this->persistDefaultLocaleTranslation);
            if ($persistNewTranslation) {
                $translation = $translationMetadata->newInstance();
                $translation = $translationMetadata->/** @scrutinizer ignore-call */ newInstance();
                $translation->setLocale($locale);
                $translation->setField($field);
                if ($adapter->usesPersonalTranslation($translationClass)) {
                    $translation->setObject($object);
                } else {
                    $translation->setObjectClass($config['useObjectClass']);
                    $translation->setForeignKey($objectId);
                }
            }

            if ($translation) {
                // set the translated field, take value using reflection
                $content = $adapter->getTranslationValue($object, $field);
                $translation->setContent($content);
                // check if need to update in database
                $transWrapper = AbstractWrapper::wrap($translation, $manager);
                if (((is_null($content) && !$isInsert) || is_bool($content) || is_int($content) || (is_string($content) && strlen($content) > 0) || !empty($content)) && ($isInsert || !$transWrapper->getIdentifier() || isset($changeSet[$field]))) {
                    if ($isInsert && !$objectId && !$adapter->usesPersonalTranslation($translationClass)) {
                        // if we do not have the primary key yet available
                        // keep this translation in memory to insert it later with foreign key
                        $this->pendingTranslationInserts[spl_object_hash($object)][] = $translation;
                    } else {
                        // persist and compute change set for translation
                        if ($wasPersistedSeparetely) {
                            $adapter->recomputeSingleObjectChangeset($uow, $translationMetadata, $translation);
                        } else {
                            $manager->persist($translation);
                            $uow->computeChangeSet($translationMetadata, $translation);
                        }
                    }
                }
            }

            if ($isInsert && $this->getTranslationInDefaultLocale($oid, $field) !== null) {
                // We can't rely on object field value which is created in non-default locale.
                // If we provide translation for default locale as well, the latter is considered to be trusted
                // and object content should be overridden.
                $wrapped->setPropertyValue($field, $this->getTranslationInDefaultLocale($oid, $field)->getContent());
                $adapter->recomputeSingleObjectChangeset($uow, $meta, $object);
                $this->removeTranslationInDefaultLocale($oid, $field);
            }
        }
        $this->translatedInLocale[$oid] = $locale;
        // check if we have default translation and need to reset the translation
        if (!$isInsert && strlen($this->defaultLocale)) {
            $this->validateLocale($this->defaultLocale);
            $modifiedChangeSet = $changeSet;
            foreach ($changeSet as $field => $changes) {
                if (in_array($field, $translatableFields)) {
                    if ($locale !== $this->defaultLocale) {
                        $adapter->setOriginalObjectProperty($uow, $oid, $field, $changes[0]);
                        unset($modifiedChangeSet[$field]);
                    }
                }
            }
            $adapter->recomputeSingleObjectChangeset($uow, $meta, $object);
            // cleanup current changeset only if working in a another locale different than de default one, otherwise the changeset will always be reverted
            if ($locale !== $this->defaultLocale) {
                $adapter->clearObjectChangeSet($uow, $oid);
                // recompute changeset only if there are changes other than reverted translations
                if ($modifiedChangeSet || $this->hasTranslationsInDefaultLocale($oid)) {
                    foreach ($modifiedChangeSet as $field => $changes) {
                        $adapter->setOriginalObjectProperty($uow, $oid, $field, $changes[0]);
                    }
                    foreach ($translatableFields as $field) {
                        if ($this->getTranslationInDefaultLocale($oid, $field) !== null) {
                            $wrapped->setPropertyValue($field, $this->getTranslationInDefaultLocale($oid, $field)
                                ->getContent());
                            $this->removeTranslationInDefaultLocale($oid, $field);
                        }
                    }
                    $adapter->recomputeSingleObjectChangeset($uow, $meta, $object);
                }
            }
        }
    }

    /**
     * Sets translation object which represents translation in default language.
     *
     * @param string $oid   hash of basic entity
     * @param string $field field of basic entity
     * @param mixed  $trans Translation object
     */
    public function setTranslationInDefaultLocale($oid, $field, $trans)
    {
        if (!isset($this->translationInDefaultLocale[$oid])) {
            $this->translationInDefaultLocale[$oid] = [];
        }
        $this->translationInDefaultLocale[$oid][$field] = $trans;
    }

    /**
     * @return bool
     */
    public function isSkipOnLoad()
    {
        return $this->skipOnLoad;
    }

    /**
     * Removes translation object which represents translation in default language.
     * This is for internal use only.
     *
     * @param string $oid   hash of the basic entity
     * @param string $field field of basic entity
     */
    private function removeTranslationInDefaultLocale($oid, $field)
    {
        if (isset($this->translationInDefaultLocale[$oid])) {
            if (isset($this->translationInDefaultLocale[$oid][$field])) {
                unset($this->translationInDefaultLocale[$oid][$field]);
            }
            if (!$this->translationInDefaultLocale[$oid]) {
                // We removed the final remaining elements from the
                // translationInDefaultLocale[$oid] array, so we might as well
                // completely remove the entry at $oid.
                unset($this->translationInDefaultLocale[$oid]);
            }
        }
    }

    /**
     * Gets translation object which represents translation in default language.
     * This is for internal use only.
     *
     * @param string $oid   hash of the basic entity
     * @param string $field field of basic entity
     *
     * @return mixed Returns translation object if it exists or NULL otherwise
     */
    private function getTranslationInDefaultLocale($oid, $field)
    {
        if (array_key_exists($oid, $this->translationInDefaultLocale)) {
            if (array_key_exists($field, $this->translationInDefaultLocale[$oid])) {
                $ret = $this->translationInDefaultLocale[$oid][$field];
            } else {
                $ret = null;
            }
        } else {
            $ret = null;
        }

        return $ret;
    }

    /**
     * Check if object has any translation object which represents translation in default language.
     * This is for internal use only.
     *
     * @param string $oid hash of the basic entity
     *
     * @return bool
     */
    public function hasTranslationsInDefaultLocale($oid)
    {
        return array_key_exists($oid, $this->translationInDefaultLocale);
    }

    /**
     * Checks if the translation entity belongs to the object in question.
     *
     * @param TranslatableAdapter $adapter
     * @param object              $trans
     * @param object              $object
     *
     * @return bool
     */
    private function belongsToObject(TranslatableAdapter $adapter, $trans, $object)
    {
        if ($adapter->usesPersonalTranslation(get_class($trans))) {
            return $trans->getObject() === $object;
        }

        return $trans->getForeignKey() === $object->getId() && ($trans->getObjectClass() === get_class($object));
    }
}

ozo2003 / HelperBundle

Branch — master (a0cc06)

TranslatableListener::setTranslationFallback() A

Complexity

Size

Duplication

Importance

Duplication Side-by-Side

Filter issues like
