DoctrineExtensions/lib/Gedmo/Translator/TranslationProxy.php

<?php

namespace Gedmo\Translator;

use Doctrine\Common\Collections\Collection;

/**
 * Proxy class for Entity/Document translations.
 *
 * @author  Konstantin Kudryashov <ever.zet@gmail.com>
 * @link    http://www.gediminasm.org
 * @license MIT License (http://www.opensource.org/licenses/mit-license.php)
 */
class TranslationProxy
{
    protected $locale;
    private $translatable;
    private $properties = array();
    private $class;
    private $coll;

    /**
     * Initializes translations collection
     *
     * @param   Object      $translatable   object to translate
     * @param   string      $locale         translation name
     * @param   array       $properties     object properties to translate
     * @param   string      $class          translation entity|document class
     * @param   Collection  $coll           translations collection
     */
    public function __construct($translatable, $locale, array $properties, $class, Collection $coll)
    {
        $this->translatable = $translatable;
        $this->locale       = $locale;
        $this->properties   = $properties;
        $this->class        = $class;
        $this->coll         = $coll;

        $translationClass = new \ReflectionClass($class);
        if (!$translationClass->implementsInterface('Gedmo\Translator\TranslationInterface')) {
            throw new \InvalidArgumentException(sprintf(
                'Translation class should implement Gedmo\Translator\TranslationInterface, "%s" given',
                $class
            ));
        }
    }

    public function __call($method, $arguments)
    {
        $matches = array();
        if (preg_match('/^(set|get)(.*)$/', $method, $matches)) {
            $property = strtolower($matches[2]);

            if (in_array($property, $this->properties)) {
                switch ($matches[1]) {
                    case 'get':
                        return $this->getTranslatedValue($property);
                    case 'set':
                        if (isset($arguments[0])) {
                            $this->setTranslatedValue($property, $arguments[0]);
                            return $this;
                        }
                }
            }
        }

        $return = call_user_func_array(array($this->translatable, $method), $arguments);

        if ($this->translatable === $return) {
            return $this;
        }

        return $return;
    }

    public function __get($property)
    {
        if (in_array($property, $this->properties)) {
            if (method_exists($this, $getter = 'get'.ucfirst($property))) {
                return $this->$getter;
            }

            return $this->getTranslatedValue($property);
        }

        return $this->translatable->$property;
    }

    public function __set($property, $value)
    {
        if (in_array($property, $this->properties)) {
            if (method_exists($this, $setter = 'set'.ucfirst($property))) {
                return $this->$setter($value);
            }

            return $this->setTranslatedValue($property, $value);
        }

        $this->translatable->property = $value;
    }

    /**
     * Returns translated value for specific property.
     *
     * @param   string  $property   property name
     *
     * @return  mixed
     */
    public function getTranslatedValue($property)
    {
        return $this->findOrCreateTranslationForProperty($property, $this->locale)->getValue();
    }

    /**
     * Sets translated value for specific property.
     *
     * @param   string  $property   property name
     * @param   string  $value      value
     */
    public function setTranslatedValue($property, $value)
    {
        $this->findOrCreateTranslationForProperty($property, $this->locale)->setValue($value);
    }

    /**
     * Finds existing or creates new translation for specified property
     *
     * @param   string  $property   object property name
     * @param   string  $locale     locale name
     *
     * @return  Translation
     */
    private function findOrCreateTranslationForProperty($property, $locale)
    {
        foreach ($this->coll as $translation) {
            if ($locale === $translation->getLocale() && $property === $translation->getProperty()) {
                return $translation;
            }
        }

        $translation = new $this->class;
        $translation->setTranslatable($this->translatable);
        $translation->setProperty($property);
        $translation->setLocale($locale);
        $this->coll->add($translation);

        return $translation;
    }
}