Start Theme Repo

src/Theme/Extenders/Nav_Menu.php

@@ -0,0 +1,39 @@
+<?php
+
+namespace RomhackPlaza\Theme\Extenders;
+defined( '\ABSPATH' ) || exit;
+
+use RomhackPlaza\Theme\Snippets;
+class Nav_Menu extends Abstract_Extender {
+
+    public function can_extend(): bool {
+        return true;
+    }
+
+    public function extend(): void {
+
+        $this->add_filter( 'nav_menu_item_title', [ $this, 'add_expand' ], 10, 4 );
+
+    }
+
+    /**
+     * @param string $title The menu item's title.
+     * @param object $item The current menu item.
+     * @param array $args An array of wp_nav_menu() arguments.
+     * @param int $depth Depth of menu item. Used for padding.
+     * @return string $title The menu item's title with dropdown icon.
+     */
+    public function add_expand( $title, $item, $args, $depth ): string {
+
+        if( $args->theme_location === 'primary' || $args->theme_location === 'secondary' ) {
+            foreach ( $item->classes as $value ) {
+                if( $value === 'menu-item-has-children' || $value === 'page_item_has_children' )
+                    $title .= Snippets::svg_icon( 'expand' );
+            }
+        }
+
+        return $title;
+
+    }
+
+}