diff --git a/inc/submenu-check.php b/inc/submenu-check.php index e212c92..9709d04 100644 --- a/inc/submenu-check.php +++ b/inc/submenu-check.php @@ -1,37 +1,39 @@ -term_id); - $level = 0; - $stack = array('0'); - foreach($menu_items as $key => $menu_item) { - if($menu_item->object_id == get_the_ID()) $current_id = $menu_item->ID; - } - foreach($menu_items as $key => $menu_item) { - while($menu_item->menu_item_parent != array_pop($stack)) { - $level--; - } - $level++; - $stack[] = $menu_item->menu_item_parent; - $stack[] = $menu_item->ID; - $menu_items[$key]->level = $level; - if($menu_item->menu_item_parent == $current_id && $level > 2 || $menu_item->ID == $current_id && $level > 2){ - $has_sub_menu = true; - } - } - //print_r($menu_items); -} -if($has_sub_menu){ -?> - -
-
- -
-
+term_id); + $level = 0; + $stack = array('0'); + if(is_array($menu_items)){ + foreach($menu_items as $key => $menu_item) { + if($menu_item->object_id == get_the_ID()) $current_id = $menu_item->ID; + } + foreach($menu_items as $key => $menu_item) { + while($menu_item->menu_item_parent != array_pop($stack)) { + $level--; + } + $level++; + $stack[] = $menu_item->menu_item_parent; + $stack[] = $menu_item->ID; + $menu_items[$key]->level = $level; + if($menu_item->menu_item_parent == $current_id && $level > 2 || $menu_item->ID == $current_id && $level > 2){ + $has_sub_menu = true; + } + } + } + //print_r($menu_items); +} +if($has_sub_menu){ +?> + +
+
+ +
+
\ No newline at end of file