vendor/sulu/sulu/src/Sulu/Bundle/WebsiteBundle/Sitemap/SitemapGenerator.php line 61

Open in your IDE?
  1. <?php
  3. /*
  4.  * This file is part of Sulu.
  5.  *
  6.  * (c) Sulu GmbH
  7.  *
  8.  * This source file is subject to the MIT license that is bundled
  9.  * with this source code in the file LICENSE.
  10.  */
  12. namespace Sulu\Bundle\WebsiteBundle\Sitemap;
  14. use Sulu\Component\Content\Query\ContentQueryBuilderInterface;
  15. use Sulu\Component\Content\Query\ContentQueryExecutorInterface;
  16. use Sulu\Component\Webspace\Manager\WebspaceManagerInterface;
  18. /**
  19.  * Generates a sitemap structure for xml or html.
  20.  */
  21. class SitemapGenerator implements SitemapGeneratorInterface
  22. {
  23.     /**
  24.      * @var ContentQueryExecutorInterface
  25.      */
  26.     private $contentQuery;
  28.     /**
  29.      * @var WebspaceManagerInterface
  30.      */
  31.     private $webspaceManager;
  33.     /**
  34.      * @var ContentQueryBuilderInterface
  35.      */
  36.     private $contentQueryBuilder;
  38.     public function __construct(
  39.         ContentQueryExecutorInterface $contentQuery,
  40.         WebspaceManagerInterface $webspaceManager,
  41.         ContentQueryBuilderInterface $contentQueryBuilder
  42.     ) {
  43.         $this->contentQuery = $contentQuery;
  44.         $this->webspaceManager = $webspaceManager;
  45.         $this->contentQueryBuilder = $contentQueryBuilder;
  46.     }
  48.     public function generateAllLocals($webspaceKey, $flat = false)
  49.     {
  50.         $webSpaceSitemap = $this->getWebspaceSitemap($webspaceKey);
  51.         $webSpaceSitemap->setSitemap(
  52.             $this->generateByLocals($webspaceKey, $webSpaceSitemap->getLocalizations(), $flat)
  53.         );
  55.         return $webSpaceSitemap;
  56.     }
  58.     public function generate($webspaceKey, $locale, $flat = false)
  59.     {
  60.         $webspaceSitemapInformation = $this->getWebspaceSitemap($webspaceKey);
  61.         $sitemap = $this->generateByLocals($webspaceKey, [$locale], $flat);
  62.         if (1 === \count($sitemap) && !$flat) {
  63.             $sitemap = $sitemap[0];
  64.         }
  65.         $webspaceSitemapInformation->setSitemap(
  66.             $sitemap
  67.         );
  69.         return $webspaceSitemapInformation;
  70.     }
  72.     /**
  73.      * @param string $webspaceKey
  74.      *
  75.      * @return WebspaceSitemap
  76.      */
  77.     private function getWebspaceSitemap($webspaceKey)
  78.     {
  79.         $webspace = $this->webspaceManager->findWebspaceByKey($webspaceKey);
  81.         $webspaceSitemap = new WebspaceSitemap();
  82.         $webspaceSitemap->setWebspaceKey($webspace->getKey());
  84.         $defaultLocalization = $webspace->getDefaultLocalization();
  85.         if ($defaultLocalization) {
  86.             $webspaceSitemap->setDefaultLocalization($defaultLocalization->getLocale());
  87.         }
  88.         foreach ($webspace->getAllLocalizations() as $localization) {
  89.             $webspaceSitemap->addLocalization($localization->getLocale());
  90.         }
  92.         return $webspaceSitemap;
  93.     }
  95.     /**
  96.      * @param string $webspaceKey
  97.      * @param array $locales
  98.      * @param bool $flat
  99.      *
  100.      * @return array
  101.      */
  102.     private function generateByLocals($webspaceKey, $locales, $flat = false)
  103.     {
  104.         return $this->contentQuery->execute(
  105.             $webspaceKey,
  106.             $locales,
  107.             $this->contentQueryBuilder,
  108.             $flat,
  109.             -1,
  110.             null,
  111.             null,
  112.             true
  113.         );
  114.     }
  115. }