angelxmoreno/cakephp-linked-entities

<?php

namespace LinkedEntities\Model\Behavior;

use Cake\ORM\Entity;
use Cake\Utility\Inflector;

/**
 * Class LinkableEntityUserBehavior
 * @package LinkedEntities\Model\Behavior
 */
class LinkableEntityUserBehavior extends BehaviorBase
{
    const METHOD_PREFIXES = ['add', 'remove'];

    public function initialize(array $config)
    {
        parent::initialize($config);
        $this->associateUser();
    }

    protected function associateUser()
    {
        $this->getTable()->hasMany(self::MODEL_NAME, ['className' => self::FULL_MODEL_NAME]);

        $links = $this->getConfig('links');

        foreach ($links as $modelAlias => $linkOptions) {
            $this->associateModelAliasToUser($modelAlias, $linkOptions);
            $this->injectModelAliasMethods($modelAlias);
        }
    }

    /**
     * @param string $modelAlias
     * @param array $linkOptions
     */
    protected function associateModelAliasToUser($modelAlias, array $linkOptions)
    {
        $association_options = $this->buildUserBelongToManyAssociationOptions($linkOptions);
        $this->getTable()->belongsToMany($modelAlias, $association_options);
    }

    /**
     * @param string $modelAlias
     */
    protected function injectModelAliasMethods($modelAlias)
    {
        foreach (self::METHOD_PREFIXES as $prefix) {
            $method = $prefix . $modelAlias;
            $this->setConfig("implementedMethods.{$method}", $method);
        }
    }

    /**
     * @param string $name
     * @param array $arguments
     */
    public function __call($name, array $arguments)
    {
        $prefix = $this->getMethodPrefix($name);
        $modelAlias = str_replace($prefix, '', $name);

        if (in_array($prefix, self::METHOD_PREFIXES) &&
            $this->isValidAlias($modelAlias)
        ) {
            $method = $prefix . 'EntityLink';
            array_unshift($arguments, $modelAlias);
            call_user_func_array([$this, $method], $arguments);
        }
    }

    /**
     * @param string $modelAlias
     * @param Entity $user
     * @param Entity $entity
     */
    public function addEntityLink($modelAlias, Entity $user, Entity $entity)
    {
        if (!$this->isValidAlias($modelAlias)) {
            throw new \InvalidArgumentException(sprintf(
                '"%s" is not a valid LinkableEntity alias for %s',
                $modelAlias,
                $this->getTable()->getAlias()
            ));
        }

        $data = $this->buildLinkedEntity($modelAlias, $user, $entity);

        $link = $this->getTable()->{self::MODEL_NAME}->newEntity($data);
        $user->{self::MODEL_PROPERTY} = $user->{self::MODEL_PROPERTY} ?: [];
        $user->{self::MODEL_PROPERTY}[] = $link;
        $user->setAccess(self::MODEL_PROPERTY, true);
        $user->setDirty(self::MODEL_PROPERTY);
    }

    /**
     * @param string $modelAlias
     * @param Entity $user
     * @param Entity $entity
     */
    public function removeEntityLink($modelAlias, Entity $user, Entity $entity)
    {
        if (!$this->isValidAlias($modelAlias)) {
            throw new \InvalidArgumentException(sprintf(
                '"%s" is not a valid LinkableEntity alias for %s',
                $modelAlias,
                $this->getTable()->getAlias()
            ));
        }

        $table = $this->getTable()->{self::MODEL_NAME};
        $conditions = $this->buildLinkedEntity($modelAlias, $user, $entity);
        $link = $table->find()->where($conditions)->first();
        $table->delete($link);
        $user->isDirty(self::MODEL_PROPERTY);
    }

    /**
     * @param string $modelAlias
     * @return bool
     */
    protected function isValidAlias($modelAlias)
    {
        return in_array($modelAlias, array_keys($this->getConfig('links')));
    }

    /**
     * @param string $modelAlias
     * @param Entity $user
     * @param Entity $entity
     * @return array
     */
    protected function buildLinkedEntity($modelAlias, Entity $user, Entity $entity)
    {
        return [
            'user_id' => $user->get($this->getTable()->getPrimaryKey()),
            'type' => $this->getConfig("links.{$modelAlias}.linkType"),
            'foreign_model' => $this->getConfig("links.{$modelAlias}.className"),
            'foreign_key' => $entity->get($this->getTable()->{$modelAlias}->getPrimaryKey()),
        ];
    }

    /**
     * @param string $method
     * @return string
     */
    protected function getMethodPrefix($method)
    {
        list($prefix) = explode('_', Inflector::tableize($method));

        return $prefix;
    }
}