Resolver menu item entity class in switch menu event subscriber.

DarvinStudio · Apr 8, 2021 · 935a2f7 · 935a2f7
1 parent 06449f9
commit 935a2f7
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 3 deletions.
diff --git a/EventListener/SwitchMenuSubscriber.php b/EventListener/SwitchMenuSubscriber.php
@@ -35,6 +35,11 @@ class SwitchMenuSubscriber implements EventSubscriber
      */
     private $translationsInitializer;
 
+    /**
+     * @var array
+     */
+    private $entityOverride;
+
     /**
      * @var string[]
      */
@@ -48,12 +53,18 @@ class SwitchMenuSubscriber implements EventSubscriber
     /**
      * @param \Darvin\MenuBundle\Switcher\MenuSwitcher                            $menuSwitcher            Menu switcher
      * @param \Darvin\ContentBundle\Translatable\TranslationsInitializerInterface $translationsInitializer Translations initializer
+     * @param array                                                               $entityOverride          Entity override
      * @param string[]                                                            $locales                 Locales
      */
-    public function __construct(MenuSwitcher $menuSwitcher, TranslationsInitializerInterface $translationsInitializer, array $locales)
-    {
+    public function __construct(
+        MenuSwitcher $menuSwitcher,
+        TranslationsInitializerInterface $translationsInitializer,
+        array $entityOverride,
+        array $locales
+    ) {
         $this->menuSwitcher = $menuSwitcher;
         $this->translationsInitializer = $translationsInitializer;
+        $this->entityOverride = $entityOverride;
         $this->locales = $locales;
 
         $this->em = null;
@@ -142,7 +153,16 @@ public function preFlush(PreFlushEventArgs $args)
      */
     private function createMenuItem($menuAlias, SlugMapItem $slugMapItem)
     {
-        $menuItem = (new Item())
+        $class = Item::class;
+
+        if (isset($this->entityOverride[$class])) {
+            $class = $this->entityOverride[$class];
+        }
+
+        /** @var \Darvin\MenuBundle\Entity\Menu\Item $menuItem */
+        $menuItem = new $class();
+
+        $menuItem
             ->setMenu($menuAlias)
             ->setSlugMapItem($slugMapItem);
 

diff --git a/Resources/config/switcher.yml b/Resources/config/switcher.yml
@@ -30,6 +30,7 @@ services:
         arguments:
             - "@darvin_menu.switcher"
             - "@darvin_content.translatable.translations_initializer"
+            - "%darvin_admin.entity_override%"
             - "%locales%"
         tags:
             - { name: doctrine.event_subscriber, priority: 100 }