View source
<?php
namespace Drupal\simple_sitemap\Plugin\simple_sitemap\UrlGenerator;
use Drupal\simple_sitemap\EntityHelper;
use Drupal\simple_sitemap\Logger;
use Drupal\simple_sitemap\Simplesitemap;
use Drupal\simple_sitemap\SitemapGenerator;
use Drupal\Core\Language\LanguageManagerInterface;
use Drupal\Core\Entity\EntityTypeManagerInterface;
use Drupal\Core\Menu\MenuTreeParameters;
use Symfony\Component\DependencyInjection\ContainerInterface;
use Drupal\Core\Menu\MenuLinkTree;
class EntityMenuLinkContentUrlGenerator extends UrlGeneratorBase {
protected $menuLinkTree;
protected $elements;
public function __construct(array $configuration, $plugin_id, $plugin_definition, Simplesitemap $generator, SitemapGenerator $sitemap_generator, LanguageManagerInterface $language_manager, EntityTypeManagerInterface $entity_type_manager, Logger $logger, EntityHelper $entityHelper, MenuLinkTree $menu_link_tree) {
parent::__construct($configuration, $plugin_id, $plugin_definition, $generator, $sitemap_generator, $language_manager, $entity_type_manager, $logger, $entityHelper);
$this->menuLinkTree = $menu_link_tree;
}
public static function create(ContainerInterface $container, array $configuration, $plugin_id, $plugin_definition) {
return new static($configuration, $plugin_id, $plugin_definition, $container
->get('simple_sitemap.generator'), $container
->get('simple_sitemap.sitemap_generator'), $container
->get('language_manager'), $container
->get('entity_type.manager'), $container
->get('simple_sitemap.logger'), $container
->get('simple_sitemap.entity_helper'), $container
->get('menu.link_tree'));
}
public function getDataSets() {
$menu_names = [];
$bundle_settings = $this->generator
->getBundleSettings();
if (!empty($bundle_settings['menu_link_content'])) {
foreach ($bundle_settings['menu_link_content'] as $bundle_name => $settings) {
if ($settings['index']) {
$menu_names[] = $bundle_name;
}
}
}
return $menu_names;
}
protected function processDataSet($link) {
if (!$link
->isEnabled()) {
return FALSE;
}
$url_object = $link
->getUrlObject();
if ($url_object
->isExternal()) {
return FALSE;
}
$meta_data = $link
->getMetaData();
if (empty($meta_data['entity_id'])) {
$entity_settings = $this->generator
->getBundleSettings('menu_link_content', $link
->getMenuName());
}
else {
$entity_settings = $this->generator
->getEntityInstanceSettings('menu_link_content', $meta_data['entity_id']);
if (empty($entity_settings['index'])) {
return FALSE;
}
}
if ($url_object
->isRouted()) {
$path = $url_object
->getInternalPath();
}
else {
if (strpos($uri = $url_object
->toUriString(), 'base:/') === 0) {
$path = $uri[6] === '/' ? substr($uri, 7) : substr($uri, 6);
}
else {
$path = $uri;
}
}
if ($this->batchSettings['remove_duplicates'] && $this
->pathProcessed($path)) {
return FALSE;
}
$url_object
->setOption('absolute', TRUE);
$entity = $this->entityHelper
->getEntityFromUrlObject($url_object);
$path_data = [
'url' => $url_object,
'lastmod' => !empty($entity) && method_exists($entity, 'getChangedTime') ? date_iso8601($entity
->getChangedTime()) : NULL,
'priority' => isset($entity_settings['priority']) ? $entity_settings['priority'] : NULL,
'changefreq' => !empty($entity_settings['changefreq']) ? $entity_settings['changefreq'] : NULL,
'images' => !empty($entity_settings['include_images']) && !empty($entity) ? $this
->getImages($entity
->getEntityTypeId(), $entity
->id()) : [],
'meta' => [
'path' => $path,
],
];
if (!empty($entity)) {
$path_data['meta']['entity_info'] = [
'entity_type' => $entity
->getEntityTypeId(),
'id' => $entity
->id(),
];
}
return $path_data;
}
protected function getBatchIterationElements($menu_name) {
$tree = $this->menuLinkTree
->load($menu_name, new MenuTreeParameters());
$tree = $this->menuLinkTree
->transform($tree, [
[
'callable' => 'menu.default_tree_manipulators:generateIndexAndSort',
],
]);
foreach ($tree as $i => $item) {
$this->elements[] = $item->link;
if ($item->hasChildren) {
$this
->getMenuChildrenRecursively($item);
}
}
$this->elements = array_values($this->elements);
if ($this
->needsInitialization()) {
$this
->initializeBatch(count($this->elements));
}
return $this
->isBatch() ? array_slice($this->elements, $this->context['sandbox']['progress'], $this->batchSettings['batch_process_limit']) : $this->elements;
}
protected function getMenuChildrenRecursively($parent) {
foreach ($parent->subtree as $i => $item) {
$this->elements[] = $item->link;
if ($item->hasChildren) {
$this
->getMenuChildrenRecursively($item);
}
}
}
}