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. }