trait.php

use TYPO3\Flow\Annotations as Flow;

/**
 * Trait which can be used to add automatic storage of
 * unknown class properties into a related CouchDB document.
 */
trait EntityWithDocumentTrait {

  /**
	 * @var string
	 */
	protected $document = '';

	/**
	 * @var \...\Domain\Model\Document
	 * @Flow\Transient
	 */
	protected $documentObject;

	/**
	 * @var \Doctrine\ODM\CouchDB\DocumentManager
	 * @Flow\Transient
	 */
	protected $documentManager;

	/**
	 * @var \Radmiraal\CouchDB\Persistence\DocumentManagerFactory
	 * @Flow\Transient
	 */
	protected $documentManagementFactory;

	/**
	 * @param \Radmiraal\CouchDB\Persistence\DocumentManagerFactory $documentManagerFactory
	 * @return void
	 */
	public function injectDocumentManagerFactory(\Radmiraal\CouchDB\Persistence\DocumentManagerFactory $documentManagerFactory) {
		$this->documentManagementFactory = $documentManagerFactory;
		$this->documentManager = $this->documentManagementFactory->create();
	}

	/**
	 * @param mixed $document
	 * @return void
	 */
	public function setDocument($document) {
		if ($document instanceof \...\Domain\Model\Document) {
			$this->documentObject = $document;
			$this->document = $document->getId();
		} elseif (is_string($document)) {
			$this->document = $document;
			$this->documentObject = $this->documentManager->find('...\Domain\Model\Document', $document);
		}
	}

	/**
	 * @return \...\Domain\Model\Document
	 */
	public function getDocument() {
		if (!empty($this->document)) {
			$document = $this->documentManager->find('...\Domain\Model\Document', $this->document);
			if (!empty($document)) {
				$this->documentObject = $document;
				$this->document = $document->getId();
				return $this->documentObject;
			}
		}

		$document = new \...\Domain\Model\Document();
		$this->documentManager->persist($document);
		$this->documentManager->flush();
		$this->document = $document->getId();
		$this->documentObject = $document;
		return $this->documentObject;
	}

	/**
	 * @param string $property
	 * @return mixed
	 */
	public function __get($property) {
		if (property_exists($this, $property)) {
			return $this->$property;
		}

		return $this->getDocument()->__get($property);
	}

	/**
	 * Magic setter
	 *
	 * @param string $property
	 * @param mixed $value
	 * @return void
	 */
	public function __set($property, $value) {
		if (substr($property, 0, 5) === 'Flow_') {
			return;
		}
		if (property_exists($this, $property)) {
			$this->$property = $value;
		} else {
			$this->getDocument()->__set($property, $value);
		}
	}

	/**
	 * Magic get* / set* method
	 *
	 * @param string $method
	 * @param array $arguments
	 * @return mixed
	 */
	function __call($method, array $arguments) {
		if (strlen($method) <= 3) {
			return NULL;
		}

		$methodName = '__' . substr($method, 0, 3);

		if ($methodName !== '__set' && $methodName !== '__get') {
			return NULL;
		}

		$var = lcfirst(substr($method, 3));
		return call_user_func_array(array($this, $methodName), array_merge(array($var), $arguments));
	}

}