Implement Doctrine discriminator maps for dynamic entity management

Author: james2001

src/Services/MeilisearchService.php

@@ -57,9 +57,7 @@ public function __construct(NormalizerInterface $normalizer, Engine $engine, arr
 
     public function isSearchable($className): bool
     {
-        if (is_object($className)) {
-            $className = ClassUtils::getClass($className);
-        }
+        $className = $this->getBaseClassName($className);
 
         return in_array($className, $this->searchableEntities, true);
     }
@@ -76,6 +74,8 @@ public function getConfiguration(): Collection
 
     public function searchableAs(string $className): string
     {
+        $className = $this->getBaseClassName($className);
+
         $indexes = new Collection($this->getConfiguration()->get('indices'));
         $index = $indexes->firstWhere('class', $className);
 
@@ -207,7 +207,8 @@ public function count(string $className, string $query = '', array $searchParams
 
     public function shouldBeIndexed(object $entity): bool
     {
-        $className = ClassUtils::getClass($entity);
+        $className = $this->getBaseClassName($entity);
+
         $propertyPath = $this->indexIfMapping[$className];
 
         if (null !== $propertyPath) {
@@ -221,6 +222,21 @@ public function shouldBeIndexed(object $entity): bool
         return true;
     }
 
+    private function getBaseClassName(mixed $object_or_class): string
+    {
+        foreach ($this->searchableEntities as $class) {
+            if (is_a($object_or_class, $class, true)) {
+                return $class;
+            }
+        }
+
+        if (is_object($object_or_class)) {
+            return ClassUtils::getClass($object_or_class);
+        }
+
+        return $object_or_class;
+    }
+
     private function setSearchableEntities(): void
     {
         $searchable = [];
@@ -313,7 +329,7 @@ private function makeSearchServiceResponseFrom(
         foreach (array_chunk($entities, $this->configuration->get('batchSize')) as $chunk) {
             $searchableEntitiesChunk = [];
             foreach ($chunk as $entity) {
-                $entityClassName = ClassUtils::getClass($entity);
+                $entityClassName = $this->getBaseClassName($entity);
 
                 $searchableEntitiesChunk[] = new SearchableEntity(
                     $this->searchableAs($entityClassName),