Propagates styles from bottom up

Author: jtreminio

src/Builder/CliMenuBuilder.php

@@ -188,12 +188,6 @@ public function addSubMenu(string $text, \Closure $callback) : self
         $menu = $builder->build();
         $menu->setParent($this->menu);
 
-        //we apply the parent theme if nothing was changed
-        //if no styles were changed in this sub-menu
-        if (!$menu->getStyle()->hasChangedFromDefaults()) {
-            $menu->setStyle($this->menu->getStyle());
-        }
-
         $this->menu->addItem($item = new MenuMenuItem(
             $text,
             $menu,
@@ -210,12 +204,6 @@ public function addSubMenuFromBuilder(string $text, CliMenuBuilder $builder) : s
         $menu = $builder->build();
         $menu->setParent($this->menu);
 
-        //we apply the parent theme if nothing was changed
-        //if no styles were changed in this sub-menu
-        if (!$menu->getStyle()->hasChangedFromDefaults()) {
-            $menu->setStyle($this->menu->getStyle());
-        }
-
         $this->menu->addItem($item = new MenuMenuItem(
             $text,
             $menu,
@@ -556,6 +544,28 @@ public function build() : CliMenu
             $this->style->setDisplaysExtra($this->itemsHaveExtra($this->menu->getItems()));
         }
 
+        if (!$this->subMenu) {
+            $this->propagateStyles($this->menu);
+        }
+
         return $this->menu;
     }
+
+    /**
+     * Pass styles from current menu to sub-menu
+     * only if sub-menu style has not be customized
+     */
+    private function propagateStyles(CliMenu $menu)
+    {
+        foreach ($menu->getItems() as $item) {
+            if ($item instanceof MenuMenuItem) {
+                $subMenu = $item->getSubMenu();
+
+                !$subMenu->getStyle()->hasChangedFromDefaults()
+                    && $subMenu->setStyle(clone $menu->getStyle());
+
+                $this->propagateStyles($subMenu);
+            }
+        }
+    }
 }

test/Builder/CliMenuBuilderTest.php

@@ -593,7 +593,7 @@ public function testSubMenuInheritsParentsStyle() : void
             ->build();
 
         self::assertSame('green', $menu->getItems()[0]->getSubMenu()->getStyle()->getBg());
-        self::assertSame($menu->getStyle(), $menu->getItems()[0]->getSubMenu()->getStyle());
+        self::assertEquals($menu->getStyle(), $menu->getItems()[0]->getSubMenu()->getStyle());
     }
 
     public function testSubMenuDoesNotInheritsParentsStyleWhenSubMenuStyleHasAlterations() : void