| Current File : /var/www/pediatribu/wp-content/plugins/mailpoet/lib/Doctrine/EventListeners/TimestampListener.php |
<?php // phpcs:ignore SlevomatCodingStandard.TypeHints.DeclareStrictTypes.DeclareStrictTypesMissing
namespace MailPoet\Doctrine\EventListeners;
if (!defined('ABSPATH')) exit;
use MailPoet\Doctrine\EntityTraits\CreatedAtTrait;
use MailPoet\Doctrine\EntityTraits\UpdatedAtTrait;
use MailPoetVendor\Carbon\Carbon;
use MailPoetVendor\Doctrine\ORM\Event\LifecycleEventArgs;
use ReflectionObject;
class TimestampListener {
public function prePersist(LifecycleEventArgs $eventArgs) {
$entity = $eventArgs->getEntity();
$entityTraits = $this->getEntityTraits($entity);
$now = $this->getNow();
if (
in_array(CreatedAtTrait::class, $entityTraits, true)
&& method_exists($entity, 'setCreatedAt')
&& method_exists($entity, 'getCreatedAt')
&& !$entity->getCreatedAt()
) {
$entity->setCreatedAt(clone $now);
}
if (in_array(UpdatedAtTrait::class, $entityTraits, true) && method_exists($entity, 'setUpdatedAt')) {
$entity->setUpdatedAt(clone $now);
}
}
public function preUpdate(LifecycleEventArgs $eventArgs) {
$entity = $eventArgs->getEntity();
$entityTraits = $this->getEntityTraits($entity);
if (in_array(UpdatedAtTrait::class, $entityTraits, true) && method_exists($entity, 'setUpdatedAt')) {
$entity->setUpdatedAt($this->getNow());
}
}
private function getEntityTraits($entity) {
$entityReflection = new ReflectionObject($entity);
return $entityReflection->getTraitNames();
}
public function getNow(): \DateTimeInterface {
return Carbon::now()->millisecond(0);
}
}