SearchableSubscriber re-index entity on change/deletion

Author: patrickbrouwers

README.md

@@ -6,6 +6,22 @@
 ]
 ```
 
+```
+'events'     => [
+    'listeners'   => [],
+    'subscribers' => [
+        LaravelDoctrine\Scout\Subscribers\SearachableSubscriber::class
+    ]
+],
+```
+
+```php
+class Post implements LaravelDoctrine\Scout\Searchable
+{
+    use LaravelDoctrine\Scout\Indexable;
+}
+```
+
 To index all records of a certain entity:
 
 ```

src/Searchable.php

@@ -0,0 +1,29 @@
+<?php
+
+namespace LaravelDoctrine\Scout;
+
+use Doctrine\ORM\Mapping\ClassMetadata;
+
+interface Searchable
+{
+    /**
+     * Get primary key value
+     * @return int
+     */
+    public function getKey();
+
+    /**
+     * @return string
+     */
+    public function searchableAs();
+
+    /**
+     * @return string
+     */
+    public function toSearchableArray();
+
+    /**
+     * @param ClassMetadata $classMetadata
+     */
+    public function setClassMetaData(ClassMetadata $classMetadata);
+}

src/SearchableRepository.php

@@ -83,6 +83,24 @@ public function makeAllSearchable()
         });
     }
 
+    /**
+     * Make specific entities searchable
+     * @param Collection $entities
+     */
+    public function makeEntitiesSearchable(Collection $entities)
+    {
+        $this->searchableUsing()->update($entities);
+    }
+
+    /**
+     * Remove specific searchable entities
+     * @param Collection $entities
+     */
+    public function removeSearchableEntities(Collection $entities)
+    {
+        $this->searchableUsing()->delete($entities);
+    }
+
     /**
      * @param  int      $count
      * @param  callable $callback

src/Subscribers/SearachableSubscriber.php

@@ -0,0 +1,139 @@
+<?php
+
+namespace LaravelDoctrine\Scout\Subscribers;
+
+use Doctrine\Common\EventSubscriber;
+use Doctrine\ORM\EntityManagerInterface;
+use Doctrine\ORM\Event\LifecycleEventArgs;
+use Doctrine\ORM\Event\PostFlushEventArgs;
+use Doctrine\ORM\Events;
+use Illuminate\Support\Collection;
+use Laravel\Scout\EngineManager;
+use LaravelDoctrine\Scout\Searchable;
+use LaravelDoctrine\Scout\SearchableRepository;
+
+class SearachableSubscriber implements EventSubscriber
+{
+    /**
+     * @var array
+     */
+    private $indexable = [];
+
+    /**
+     * @var array
+     */
+    private $deleteable = [];
+
+    /**
+     * @return array
+     */
+    public function getSubscribedEvents()
+    {
+        return [
+            Events::postPersist,
+            Events::postUpdate,
+            Events::preRemove,
+            Events::postFlush
+        ];
+    }
+
+    /**
+     * @param PostFlushEventArgs $args
+     */
+    public function postFlush(PostFlushEventArgs $args)
+    {
+        foreach ($this->indexable as $event) {
+            $this->indexEntity($args->getEntityManager(), $event);
+        }
+
+        foreach ($this->deleteable as $event) {
+            $this->removeEntity($args->getEntityManager(), $event);
+        }
+    }
+
+    /**
+     * @param LifecycleEventArgs $event
+     */
+    public function postPersist(LifecycleEventArgs $event)
+    {
+        $this->scheduleIndexing($event);
+    }
+
+    /**
+     * @param LifecycleEventArgs $event
+     */
+    public function postUpdate(LifecycleEventArgs $event)
+    {
+        $this->scheduleIndexing($event);
+    }
+
+    /**
+     * @param LifecycleEventArgs $event
+     */
+    public function preRemove(LifecycleEventArgs $event)
+    {
+        $object = $event->getObject();
+
+        if ($object instanceof Searchable) {
+            $this->deleteable[] = clone $object;
+        }
+    }
+
+    /**
+     * @param LifecycleEventArgs $event
+     */
+    public function scheduleIndexing(LifecycleEventArgs $event)
+    {
+        $object = $event->getObject();
+
+        if ($object instanceof Searchable) {
+            $this->indexable[] = $object;
+        }
+    }
+
+    /**
+     * @param EntityManagerInterface $em
+     * @param Searchable             $object
+     */
+    private function indexEntity(EntityManagerInterface $em, Searchable $object)
+    {
+        $repository = $this->getRepository($em, $object);
+
+        $repository->makeEntitiesSearchable(new Collection([$object]));
+    }
+
+    /**
+     * @param EntityManagerInterface $em
+     * @param Searchable             $object
+     */
+    private function removeEntity(EntityManagerInterface $em, Searchable $object)
+    {
+        $repository = $this->getRepository($em, $object);
+
+        $repository->removeSearchableEntities(new Collection([$object]));
+    }
+
+    /**
+     * @param  EntityManagerInterface $em
+     * @param  Searchable             $object
+     * @return SearchableRepository
+     */
+    private function getRepository(EntityManagerInterface $em, Searchable $object)
+    {
+        $class      = get_class($object);
+        $cmd        = $em->getClassMetadata($class);
+        $repository = $em->getRepository($class);
+
+        if (!$repository instanceof SearchableRepository) {
+            $repository = new SearchableRepository(
+                $em,
+                $cmd,
+                app(EngineManager::class)
+            );
+        }
+
+        $object->setClassMetaData($cmd);
+
+        return $repository;
+    }
+}