paper-orm/src/Persistence/EntityPersistence.php

<?php

namespace PhpDevCommunity\PaperORM\Persistence;

use PhpDevCommunity\PaperORM\Entity\EntityInterface;
use PhpDevCommunity\PaperORM\Event\PreCreateEvent;
use PhpDevCommunity\PaperORM\Event\PreUpdateEvent;
use PhpDevCommunity\PaperORM\Hydrator\ReadOnlyEntityHydrator;
use PhpDevCommunity\PaperORM\Mapper\ColumnMapper;
use PhpDevCommunity\PaperORM\Mapper\EntityMapper;
use PhpDevCommunity\PaperORM\Platform\PlatformInterface;
use PhpDevCommunity\PaperORM\Proxy\ProxyInterface;
use PhpDevCommunity\PaperORM\Serializer\SerializerToDb;
use PhpDevCommunity\Sql\QueryBuilder;
use Psr\EventDispatcher\EventDispatcherInterface;

class EntityPersistence
{
    private PlatformInterface $platform;
    private \SplObjectStorage $managed;

    private ?EventDispatcherInterface $dispatcher;
    public function __construct(PlatformInterface $platform, EventDispatcherInterface $dispatcher = null)
    {
        $this->platform = $platform;
        $this->dispatcher = $dispatcher;
        $this->managed = new \SplObjectStorage();
    }

    public function insert(object $entity): int
    {
        /**
         * @var EntityInterface $entity
         */
        $this->checkEntity($entity);
        if ($entity->getPrimaryKeyValue() !== null || $this->managed->contains($entity)) {
            throw new \LogicException(sprintf(
                '%s cannot insert entity of type "%s": entity already managed (has primary key, is a proxy, or is attached).',
                static::class,
                get_class($entity)
            ));
        }

        if ($this->dispatcher) {
            $this->dispatcher->dispatch(new PreCreateEvent($entity));
        }
        $schema = $this->platform->getSchema();
        $tableName = EntityMapper::getTable($entity);
        $qb = QueryBuilder::insert($schema->quote($tableName));

        $values = [];
        foreach ((new SerializerToDb($entity))->serialize() as $key => $value) {
            $qb->setValue($schema->quote($key), ":$key");
            $values[$key] = $value;
        }
        $conn = $this->platform->getConnection();
        $rows = $conn->executeStatement($qb, $values);
        $lastInsertId = $conn->getPdo()->lastInsertId();
        if ($rows > 0) {
            $primaryKeyColumn = ColumnMapper::getPrimaryKeyColumnName($entity);
            (new ReadOnlyEntityHydrator())->hydrate($entity, [$primaryKeyColumn => $lastInsertId]);
            $this->managed->attach($entity);
        }
        return $rows;
    }

    public function update(object $entity): int
    {
        /**
         * @var ProxyInterface|EntityInterface $entity
         */
        $this->checkEntity($entity, true);
        if ($entity->getPrimaryKeyValue() === null) {
            throw new \LogicException(static::class . sprintf(' Cannot update an entity %s without a primary key ', get_class($entity)));
        }

        if ($entity instanceof ProxyInterface) {
            if (!$entity->__wasModified()) {
                return 0;
            }
        }

        if ($this->dispatcher) {
            $this->dispatcher->dispatch(new PreUpdateEvent($entity));
        }

        if ($entity instanceof ProxyInterface) {
            $propertiesModified = $entity->__getPropertiesModified();
        } else {
            $propertiesModified = [];
        }

        $tableName = EntityMapper::getTable($entity);
        $schema = $this->platform->getSchema();
        $qb = QueryBuilder::update($schema->quote($tableName))
            ->where(
                sprintf('%s = %s',
                    $schema->quote(ColumnMapper::getPrimaryKeyColumnName($entity)),
                    $entity->getPrimaryKeyValue()
                )
            );

        $values = [];
        foreach ((new SerializerToDb($entity))->serialize($propertiesModified) as $key => $value) {
            $qb->set($schema->quote($key), ":$key");
            $values[$key] = $value;
        }
        $conn = $this->platform->getConnection();
        $rows = $conn->executeStatement($qb, $values);
        if ($rows > 0 && $entity instanceof ProxyInterface) {
            $entity->__reset();
        }
        return $rows;
    }

    public function delete(object $entity): int
    {
        /**
         * @var ProxyInterface|EntityInterface $entity
         */
        $this->checkEntity($entity, true);
        if ($entity->getPrimaryKeyValue() === null) {
            throw new \LogicException(static::class . sprintf(' Cannot delete an entity %s without a primary key ', get_class($entity)));
        }

        $tableName = EntityMapper::getTable($entity);
        $schema = $this->platform->getSchema();
        $qb = QueryBuilder::delete($schema->quote($tableName))
            ->where(sprintf('%s = :id', $schema->quote(ColumnMapper::getPrimaryKeyColumnName($entity))));

        $conn = $this->platform->getConnection();
        $rows = $conn->executeStatement($qb,  [
            'id' => $entity->getPrimaryKeyValue(),
        ]);
        if ($rows > 0) {
            if ($entity instanceof ProxyInterface) {
                $entity->__destroy();
            }
            if ($this->managed->contains($entity)) {
                $this->managed->detach($entity);
            }
        }
        return $rows;
    }

    private function checkEntity(object $entity,  bool $requireManaged = false): void
    {
        if (!$entity instanceof EntityInterface) {
            throw new \LogicException(sprintf(
                'Invalid entity of type "%s". Expected an instance of "%s".',
                get_class($entity),
                EntityInterface::class
            ));
        }

        if ($requireManaged) {
            $isManaged = $this->managed->contains($entity);
            $isProxy = $entity instanceof ProxyInterface && $entity->__isInitialized();
            if (!$isManaged && !$isProxy) {
                throw new \LogicException(sprintf(
                    'Entity of type "%s" is not managed by ORM',
                    get_class($entity)
                ));
            }
        }
    }

}