为 WordPress 菜单添加自定义选项打造超级菜单

// 设置添加字段
function zm_menu_custom_fields( $item_id, $item ) {
	$menu_code = get_post_meta( $item_id, 'zm_menu_code_checkbox', true );
	$menu_shortcode = get_post_meta( $item_id, 'zm_menu_shortcode', true );
	?>

	<p class="zm-menu-code-label description" style="margin: 10px 0">
		<label for="zm-menu-code-<?php echo $item_id; ?>" >
			<input type="checkbox" id="zm-menu-code-<?php echo $item_id; ?>" name="zm_menu_code[<?php echo $item_id; ?>]" <?php checked( $menu_code, true ); ?> />
			为顶级菜单添加CSS类（仅在顶级菜单勾选）
		</label>
	</p>

	<p class="zm-menu-shortcode-label description">
		<label for="zm-menu-shortcode-<?php echo $item_id; ?>">
			输入短代码（仅在菜单子项目中使用）
		</label>
		<input type="hidden" class="nav-menu-id" value="<?php echo $item_id; ?>" />
		<div class="logged-input-holder">
			<input type="text" name="zm_menu_shortcode[<?php echo $item_id; ?>]" id="zm-menu-shortcode-<?php echo $item_id ;?>" value="<?php echo esc_attr( $menu_shortcode ); ?>" style="width: 100%;">
		</div>
	</p>
	<?php
}

add_action( 'wp_nav_menu_item_custom_fields', 'zm_menu_custom_fields', 10, 2 );

// 保存字段
function zm_save_menu_custom_item_meta( $menu_id, $menu_item_db_id ) {
	$button_value = ( isset($_POST['zm_menu_code'][$menu_item_db_id] ) && $_POST['zm_menu_code'][$menu_item_db_id] == 'on' ) ? true : false;
	update_post_meta( $menu_item_db_id, 'zm_menu_code_checkbox', $button_value );

	if ( isset( $_POST['zm_menu_shortcode'][$menu_item_db_id] ) ) {
		$sanitized_data = sanitize_text_field( $_POST['zm_menu_shortcode'][$menu_item_db_id] );
		update_post_meta( $menu_item_db_id, 'zm_menu_shortcode', $sanitized_data );
	} else {
		delete_post_meta( $menu_item_db_id, 'zm_menu_shortcode' );
	}
}

add_action( 'wp_update_nav_menu_item', 'zm_save_menu_custom_item_meta', 10, 2 );

// 前端输出
function zm_menu_custom_output( $title, $item ) {
	if( is_object( $item ) && isset( $item->ID ) ) {
		$menu_shortcode = get_post_meta( $item->ID, 'zm_menu_shortcode', true );

		// 输出菜单标题，可以删除
		if ( ! empty( $menu_shortcode ) ) {
			$title = '<span class="zm-menu-custom-title">'.$title.'</span>';
		}

		// 输出短代码
		if ( ! empty( $menu_shortcode ) ) {
			$title .= '<span class="show-menu-shortcode">' . do_shortcode( $menu_shortcode ) . '</span>';
		}
	}

	return $title;
}

add_filter( 'nav_menu_item_title', 'zm_menu_custom_output', 10, 2 );

// 添加CSS类
function zm_menu_custom_add_class( $classes, $menu_item ) {
	$menu_code = get_post_meta( $menu_item->ID, 'zm_menu_code_checkbox', true );
	$menu_shortcode = get_post_meta( $menu_item->ID, 'zm_menu_shortcode', true );

	// 为顶级菜单添加CSS类
	if ( ! empty( $menu_code ) ) {
		$classes[] = 'zm-menu-code';
	}
	// 为子菜单添加CSS类
	if ( ! empty( $menu_shortcode ) ) {
		$classes[] = 'zm-menu-shortcode';
	}

	return $classes;
}
add_filter( 'nav_menu_css_class', 'zm_menu_custom_add_class', 10, 2 );