3runoDesign/setRobot

View on GitHub
app/Lib/NavWalker.php

Summary

Maintainability
A
2 hrs
Test Coverage
<?php

namespace App\Lib;

use Utils;

class NavWalker extends \Walker_Nav_Menu
{
    private $cpt; // Boolean, is current post a custom post type
    private $archive; // Stores the archive page for current URL

    public function __construct()
    {
        add_filter('nav_menu_css_class', array($this, 'cssClasses'), 10, 2);
        add_filter('nav_menu_item_id', '__return_null');
        $cpt = get_post_type();
        $this->cpt = in_array($cpt, get_post_types(array('_builtin' => false)));
        $this->archive = get_post_type_archive_link($cpt);
    }

    public function checkCurrent($classes)
    {
        return preg_match('/(current[-_])|active/', $classes);
    }

    // @codingStandardsIgnoreStart
    public function display_element($element, &$children_elements, $max_depth, $depth = 0, $args, &$output)
    {
        $element->is_subitem = ((!empty($children_elements[$element->ID]) && (($depth + 1) < $max_depth || ($max_depth === 0))));
        if ($element->is_subitem) {
            foreach ($children_elements[$element->ID] as $child) {
                if ($child->current_item_parent || (new Utils())->url_compare($this->archive, $child->url)) {
                    $element->classes[] = 'active';
                }
            }
        }
        $element->is_active = (!empty($element->url) && strpos($this->archive, $element->url));
        if ($element->is_active) {
            $element->classes[] = 'active';
        }
        parent::display_element($element, $children_elements, $max_depth, $depth, $args, $output);
    }

    // @codingStandardsIgnoreEnd
    public function cssClasses($classes, $item)
    {
        $slug = sanitize_title($item->title);
        // Fix core `active` behavior for custom post types
        if ($this->cpt) {
            $classes = str_replace('current_page_parent', '', $classes);
            if ($this->archive) {
                if ((new Utils())->url_compare($this->archive, $item->url)) {
                    $classes[] = 'active';
                }
            }
        }
        // Remove most core classes
        $classes = preg_replace('/(current(-menu-|[-_]page[-_])(item|parent|ancestor))/', 'active', $classes);
        $classes = preg_replace('/^((menu|page)[-_\w+]+)+/', '', $classes);
        // Re-add core `menu-item` class
        $classes[] = 'menu-item';
        // Re-add core `menu-item-has-children` class on parent elements
        if ($item->is_subitem) {
            $classes[] = 'menu-item-has-children';
        }
        // Add `menu-<slug>` class
        $classes[] = 'menu-' . $slug;
        $classes = array_unique($classes);
        $classes = array_map('trim', $classes);
        return array_filter($classes);
    }
}