function dpuk_add_sub_menu_toggle2( $output, $item, $depth=0, $args=[], $id=0 ) {
$arrow = '<i class="dropdown dashicons dashicons-arrow-down"></i>';
if ( '#' === $item->url ) {
$item_output = $args->before;
$item_output .= '<span>'. $item->title;
$item_output .= '</span>';
$item_output .= $args->after;
if ( 0 === $depth && in_array( 'menu-item-has-children', $item->classes, true ) ) {
// Add toggle button.
$item_output .= '<button class="sub-menu-toggle" aria-expanded="false" onClick="expandSubMenu(this)">';
$item_output .= $arrow;
$item_output .= '<span class="screen-reader-text">' . esc_html__( 'Open menu', 'nachala' ) . '</span>';
$item_output .= '</button>';
}
return $item_output;
} else {
if ( 0 === $depth && in_array( 'menu-item-has-children', $item->classes, true ) ) {
// Add toggle button.
$output .= '<button class="sub-menu-toggle" aria-expanded="false" onClick="expandSubMenu(this)">';
$output .= $arrow;
$output .= '<span class="screen-reader-text">' . esc_html__( 'Open menu', 'nachala' ) . '</span>';
$output .= '</button>';
}
return $output;
}
}
add_filter( 'walker_nav_menu_start_el', '
dpuk_sub_menu_toggle2', 10, 4 );