1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76
<?php
namespace BEdita\Core\Model\Action;
use ArrayObject;
use BEdita\Core\Exception\InvalidDataException;
use Cake\Datasource\EntityInterface;
use Cake\ORM\Association\BelongsToMany;
use Cake\ORM\Association\HasMany;
class AddAssociatedAction extends UpdateAssociatedAction
{
use AssociatedTrait;
protected function update(EntityInterface $entity, $relatedEntities)
{
if ($this->Association instanceof BelongsToMany || $this->Association instanceof HasMany) {
if ($relatedEntities === null) {
$relatedEntities = [];
} elseif (!is_array($relatedEntities)) {
$relatedEntities = [$relatedEntities];
}
return $this->Association->getConnection()->transactional(function () use ($entity, $relatedEntities) {
$relatedEntities = new ArrayObject($relatedEntities);
$this->dispatchEvent('Associated.beforeSave', compact('entity', 'relatedEntities') + ['action' => 'add', 'association' => $this->Association]);
$relatedEntities = $this->diff($entity, $relatedEntities->getArrayCopy(), false);
if (!$this->Association->link($entity, $relatedEntities, ['atomic' => false])) {
return false;
}
foreach ($relatedEntities as $relatedEntity) {
if ($relatedEntity->has('_joinData') && $relatedEntity->get('_joinData')->getErrors()) {
throw new InvalidDataException(
__d('bedita', 'Error linking entities'),
(array)$relatedEntity->get('_joinData')->getErrors()
);
}
}
$this->dispatchEvent('Associated.afterSave', compact('entity', 'relatedEntities') + ['action' => 'add', 'association' => $this->Association]);
return count($relatedEntities);
});
}
throw new \RuntimeException(
__d('bedita', 'Unable to add additional links with association of type "{0}"', get_class($this->Association))
);
}
}