vendor/dachcom-digital/members/src/MembersBundle/Restriction/Dao.php line 53

Open in your IDE?
  1. <?php
  3. namespace MembersBundle\Restriction;
  5. use Pimcore\Model;
  7. class Dao extends Model\Dao\AbstractDao
  8. {
  9.     /**
  10.      * @var string
  11.      */
  12.     protected $tableName = 'members_restrictions';
  14.     /**
  15.      * @var string
  16.      */
  17.     protected $tableRelationName = 'members_group_relations';
  19.     /**
  20.      * @var Restriction
  21.      */
  22.     protected $model;
  24.     /**
  25.      * @param int $id
  26.      *
  27.      * @throws \Exception
  28.      */
  29.     public function getById($id)
  30.     {
  31.         $data = $this->db->fetchRow('SELECT * FROM ' . $this->tableName . ' WHERE id  = ?', [$id]);
  33.         if ($data === false) {
  34.             throw new \Exception('Object with the ID ' . $id . ' doesn\'t exists');
  35.         } else {
  36.             $data = $this->addRelationData($data);
  37.         }
  39.         $this->assignVariablesToModel($data);
  40.     }
  42.     /**
  43.      * @param string $field
  44.      * @param null   $value
  45.      * @param string $cType
  46.      *
  47.      * @throws \Exception
  48.      */
  49.     public function getByField($field, $value = null, $cType = 'page')
  50.     {
  51.         $data = $this->db->fetchRow('SELECT * FROM ' . $this->tableName . ' WHERE ' . $field . ' = ? AND ctype = ?', [
  52.             $value,
  53.             $cType
  54.         ]);
  56.         if ($data === false) {
  57.             throw new \Exception('Object (Type: ' . $cType . ') with the ' . $field . ' ' . $value . ' doesn\'t exists');
  58.         } else {
  59.             $data = $this->addRelationData($data);
  60.         }
  62.         $this->assignVariablesToModel($data);
  63.     }
  65.     /**
  66.      * @return bool
  67.      *
  68.      * @throws \Doctrine\DBAL\DBALException
  69.      * @throws \Doctrine\DBAL\Exception\InvalidArgumentException
  70.      */
  71.     public function save()
  72.     {
  73.         $saveData = [
  74.             'targetId'    => $this->model->getTargetId(),
  75.             'ctype'       => $this->model->getCtype(),
  76.             'isInherited' => (int) $this->model->isInherited(),
  77.             'inherit'     => (int) $this->model->getInherit()
  78.         ];
  80.         if ($this->model->getId() !== null) {
  81.             $this->db->update($this->tableName, $saveData, ['id' => $this->model->getId()]);
  82.         } else {
  83.             $this->db->insert($this->tableName, $saveData);
  84.             $this->model->setId($this->db->lastInsertId());
  85.         }
  87.         $this->saveRelations();
  89.         return true;
  90.     }
  92.     /**
  93.      * @param array $data
  94.      *
  95.      * @return mixed
  96.      */
  97.     private function addRelationData($data)
  98.     {
  99.         $relations = $this->db->fetchAll('SELECT * FROM ' . $this->tableRelationName . ' WHERE restrictionId  = ?', [$data['id']]);
  101.         if ($relations !== false) {
  102.             foreach ($relations as $relation) {
  103.                 $data['relatedGroups'][] = $relation['groupId'];
  104.             }
  105.         }
  107.         return $data;
  108.     }
  110.     /**
  111.      * @return bool
  112.      *
  113.      * @throws \Doctrine\DBAL\DBALException
  114.      * @throws \Doctrine\DBAL\Exception\InvalidArgumentException
  115.      */
  116.     public function saveRelations()
  117.     {
  118.         $groups = $this->model->getRelatedGroups();
  120.         //first, delete all!
  121.         $this->db->delete($this->tableRelationName, ['restrictionId' => $this->model->getId()]);
  123.         //set related Groups
  124.         if (empty($groups)) {
  125.             return false;
  126.         }
  128.         foreach ($groups as $groupId) {
  129.             $saveData = [
  130.                 'restrictionId' => $this->model->getId(),
  131.                 'groupId'       => (int) $groupId,
  132.             ];
  134.             $this->db->insert($this->tableRelationName, $saveData);
  135.         }
  137.         return true;
  138.     }
  140.     /**
  141.      * @throws \Doctrine\DBAL\DBALException
  142.      * @throws \Doctrine\DBAL\Exception\InvalidArgumentException
  143.      */
  144.     public function delete()
  145.     {
  146.         $this->db->delete($this->tableName, ['id' => $this->model->getId()]);
  147.         $this->db->delete($this->tableRelationName, ['restrictionId' => $this->model->getId()]);
  148.     }
  149. }