Fix sitemap include (#141)

* fix: sitemap include --------- Co-authored-by: Marijan Klaric <[email protected]> Co-authored-by: Florian ALEXANDRE <[email protected]>
Novactive · Aug 31, 2023 · ebcb7dd · ebcb7dd
1 parent 3c08a04
commit ebcb7dd
Showing 1 changed file with 50 additions and 2 deletions.
diff --git a/bundle/Core/Sitemap/QueryFactory.php b/bundle/Core/Sitemap/QueryFactory.php
@@ -69,11 +69,10 @@ public function __invoke(): Query
         $config = $this->configResolver->getParameter('sitemap_includes', 'nova_ezseo');
         $criterions = array_merge(
             $criterions,
-            $this->getCriterionsForConfig(
+            $this->getCriterionsForIncludeConfig(
                 $config['contentTypeIdentifiers'],
                 $config['locations'],
                 $config['subtrees'],
-                false
             )
         );
 
@@ -142,4 +141,53 @@ function ($criterion) {
 
         return $criterions;
     }
+
+    private function getCriterionsForIncludeConfig(
+        array $contentTypeIdentifiers,
+        array $locationIds,
+        array $subtreeLocationsId
+    ) {
+        $contentTypeService = $this->repository->getContentTypeService();
+        $criterions = [];
+
+        $validContentTypeIdentifiers = [];
+        foreach ($contentTypeIdentifiers as $contentTypeIdentifier) {
+            try {
+                $contentTypeService->loadContentTypeByIdentifier($contentTypeIdentifier);
+            } catch (NotFoundException $exception) {
+                continue;
+            }
+            $validContentTypeIdentifiers[] = $contentTypeIdentifier;
+        }
+        if (count($validContentTypeIdentifiers) > 0) {
+            $criterions[] = new Criterion\ContentTypeIdentifier($validContentTypeIdentifiers);
+        }
+
+        $subtreePaths = [];
+        foreach ($subtreeLocationsId as $locationId) {
+            $includedLocation = $this->getLocation($locationId);
+            if (null === $includedLocation) {
+                continue;
+            }
+            $subtreePaths[] = $includedLocation->pathString;
+        }
+        if (count($subtreePaths) > 0) {
+            $criterions[] = new Criterion\Subtree($subtreePaths);
+        }
+
+        $validLocationIds = [];
+        foreach ($locationIds as $locationId) {
+            $includedLocation = $this->getLocation($locationId);
+            if (null === $includedLocation) {
+                continue;
+            }
+            $validLocationIds[] = $locationId;
+        }
+
+        if (count($validLocationIds) > 0) {
+            $criterions[] = new Criterion\LocationId($validLocationIds);
+        }
+
+        return $criterions;
+    }
 }