V.basic update to excluded nav classes, allowing custom classes

Log1x · Apr 21, 2021 · f9baca6 · f9baca6
1 parent 51f6710
commit f9baca6
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/src/MenuBuilder.php b/src/MenuBuilder.php
@@ -35,15 +35,19 @@ class MenuBuilder
     ];
 
     /**
-     * Blacklisted Classes
+     * Excluded Classes
      *
      * @var array
      */
-    protected $classes = [
+    protected $excluded_classes = [
         'current-menu',
         'current_page',
         'sub-menu',
-        'menu-item',
+        // 'menu-item',
+        'menu-item-type-post_type',
+        'menu-item-object-page',
+        'menu-item-type-custom',
+        'menu-item-object-custom',
         'menu_item',
         'page-item',
         'page_item',
@@ -88,10 +92,7 @@ protected function filter($menu = [])
 
         return array_map(function ($item) {
             $classes = array_filter($item->classes, function ($class) {
-                return array_key_exists(
-                    $class,
-                    array_flip($this->classes)
-                );
+                return !in_array($class, $this->excluded_classes);
             });
 
             $item->classes = is_array($classes) ? implode(' ', $classes) : $classes;