Merge pull request #1 from bolt/feature/update-on-content-save

Update index on saving content

Author: I-Valchev

src/Event/PostSaveSubscriber.php

@@ -0,0 +1,33 @@
+<?php
+
+declare(strict_types=1);
+
+namespace Bolt\TntSearch\Event;
+
+use Bolt\Event\ContentEvent;
+use Bolt\TntSearch\IndexGenerator;
+use Symfony\Component\EventDispatcher\EventSubscriberInterface;
+
+class PostSaveSubscriber implements EventSubscriberInterface
+{
+    /** @var IndexGenerator */
+    private $generator;
+
+    public function __construct(IndexGenerator $generator)
+    {
+        $this->generator = $generator;
+    }
+
+    public function onPostSave(ContentEvent $event): void
+    {
+        $content = $event->getContent();
+        $this->generator->update($content);
+    }
+
+    public static function getSubscribedEvents()
+    {
+        return [
+            ContentEvent::POST_SAVE => [['onPostSave']],
+        ];
+    }
+}

src/IndexGenerator.php

@@ -4,6 +4,9 @@
 
 namespace Bolt\TntSearch;
 
+use Bolt\Entity\Content;
+use Doctrine\ORM\EntityManagerInterface;
+
 class IndexGenerator
 {
     /** @var TntEngine */
@@ -12,18 +15,56 @@ class IndexGenerator
     /** @var \Bolt\Configuration\Config */
     private $boltConfig;
 
-    public function __construct(TntEngine $engine, \Bolt\Configuration\Config $boltConfig)
+    /** @var EntityManagerInterface */
+    private $em;
+
+    public function __construct(TntEngine $engine, \Bolt\Configuration\Config $boltConfig, EntityManagerInterface $em)
     {
         $this->engine = $engine;
         $this->boltConfig = $boltConfig;
+        $this->em = $em;
     }
 
     public function generate(): void
     {
         $tnt = $this->engine->get();
+        $index = $tnt->createIndex('records.index');
+        $index->disableOutput = true;
+
+        $index->query($this->getQuery());
+        $index->disableOutput = true;
+
+        // todo: makes the process a lot slower, but goes through all of them. Otherwise, the last page
+        // with less items than the step size gets ommitted.
+        $index->steps = 1;
+
+        $index->run();
+    }
+
+    public function update(Content $content): void
+    {
+        $query = $this->getQuery();
+        $whereClause = sprintf(' bolt_content.id = %d AND', $content->getId());
+        $position = mb_strpos($query, 'WHERE') + 5;
 
-        $indexer = $tnt->createIndex('records.index');
+        $query = substr_replace($query, $whereClause, $position, 0);
+        $statement = $this->em->getConnection()->prepare($query);
+        $statement->executeStatement();
+        $results = $statement->fetchAll();
 
+        $tnt = $this->engine->get();
+        $tnt->selectIndex('records.index');
+        $index = $tnt->getIndex();
+
+        // Delete the current document, then add all new fields.
+        $index->delete($content->getId());
+        foreach ($results as $result) {
+            $index->insert($result);
+        }
+    }
+
+    protected function getQuery(): string
+    {
         $contentTypes = $this->boltConfig->get('contenttypes')
             ->where('searchable', true)
             ->keys()
@@ -39,13 +80,6 @@ public function generate(): void
                 WHERE bolt_content.content_type IN (%s);
 EOD;
 
-        $indexer->query(sprintf($query, $contentTypes));
-        $indexer->disableOutput = true;
-
-        // todo: makes the process a lot slower, but goes through all of them. Otherwise, the last page
-        // with less items than the step size gets ommitted.
-        $indexer->steps = 1;
-
-        $indexer->run();
+        return sprintf($query, $contentTypes);
     }
 }