load_hooks();
}
/**
* Loads hooks.
*
* @since 3.3.0
*/
public function load_hooks() {
global $wp_version;
if ( version_compare( $wp_version, '5.4', '<' ) ) {
add_filter( 'wp_edit_nav_menu_walker', array( $this, 'edit_nav_menu_walker' ) );
}
add_action( 'wp_update_nav_menu_item', array( $this, 'update_nav_menu_item' ), 10, 2 );
add_filter( 'wp_setup_nav_menu_item', array( $this, 'setup_nav_menu_item' ) );
add_action( 'admin_enqueue_scripts' , array( $this, 'enqueue_scripts' ) );
add_action( 'wp_nav_menu_item_custom_fields', array( $this, 'nav_menu_item_fields' ), 5, 4 );
// add_action( 'wp_nav_menu_item_custom_fields_customize_template', array( $this, 'nav_menu_item_fields' ), 5 ); // @todo Work this out.
add_action( 'wpmem_nav_menu_logged_in_criteria', array( $this, 'add_product_criteria' ) );
// Handles removing front end menu items.
if ( ! is_admin() ) {
add_filter( 'wp_get_nav_menu_items', array( $this, 'exclude_menu_items' ), 20 );
}
}
/**
* Override the Admin Menu Walker
*
* @since 3.3.0
*
* @global object $wpmem
*
* @return object WP_Members_Walker_Nav_Menu
*/
public function edit_nav_menu_walker( $walker ) {
global $wpmem;
require_once( $wpmem->path . 'includes/walkers/class-wp-members-walker-nav-menu.php' );
return 'WP_Members_Walker_Nav_Menu';
}
/**
* Add fields to hook added in Walker
*
* @since 3.3.0
*
* @global object $wpmem
*
* @param string $item_id
* @param object $item
* @param $depth
* @param array $args
*/
public function nav_menu_item_fields( $item_id, $item, $depth, $args ) {
$restrictions = get_post_meta( $item->ID, $this->post_meta, true );
$logged_in_out = '';
if ( is_array( $restrictions ) || $restrictions == 'in' ) {
$logged_in_out = 'in';
} else if ( $restrictions == 'out' ){
$logged_in_out = 'out';
}
$hidden = $logged_in_out == 'in' ? '' : 'display: none;';
?>
url . 'assets/js/wpmem-nav-menu' . wpmem_get_suffix() . '.js', array( 'jquery' ), $wpmem->version, true );
}
}
/**
* Save custom menu fields.
*
* @since 3.3.0
*
* @global object $wpmem
*/
public function update_nav_menu_item( $menu_id, $menu_item_db_id ) {
$product_names = array();
foreach ( wpmem_get_memberships() as $key => $value ) {
$product_names[ $key ] = $value['title'];
}
// Verify this came from our screen and with proper authorization.
if ( ! isset( $_POST[ $this->nonce_name ] ) || ! wp_verify_nonce( $_POST[ $this->nonce_name ], $this->nonce_field ) ){
return;
}
$saved_data = false;
if ( isset( $_POST['wpmem_logged_in_out'][ $menu_item_db_id ] ) && 'in' == $_POST['wpmem_logged_in_out'][ $menu_item_db_id ] && isset( $_POST['wpmem_product'][ $menu_item_db_id ] ) ) {
$custom_fields = array();
foreach( (array) $_POST['wpmem_product'][ $menu_item_db_id ] as $product ) {
if ( array_key_exists ( $product, $product_names ) ) {
$custom_fields['products'][] = sanitize_text_field( $product );
}
}
if ( ! empty ( $custom_fields ) ) {
$saved_data = $custom_fields;
}
} elseif ( isset( $_POST['wpmem_logged_in_out'][ $menu_item_db_id ] ) && in_array( $_POST['wpmem_logged_in_out'][ $menu_item_db_id ], array( 'in', 'out' ) ) ) {
$saved_data = sanitize_text_field( $_POST['wpmem_logged_in_out'][ $menu_item_db_id ] );
}
if ( $saved_data ) {
update_post_meta( $menu_item_db_id, $this->post_meta, $saved_data );
} else {
delete_post_meta( $menu_item_db_id, $this->post_meta );
}
}
/**
* Add custom field to $item object.
*
* @since 3.3.0
*
* @global object $wpmem
*
* @param object $menu_item
* @return object $menu_item
*/
public function setup_nav_menu_item( $menu_item ) {
global $wpmem;
if ( is_object( $menu_item ) && isset( $menu_item->ID ) ) {
$restrictions = get_post_meta( $menu_item->ID, $this->post_meta, true );
// If there are any restrictions, set them in the menu item.
if ( ! empty( $restrictions ) ) {
$menu_item->restrictions = $restrictions;
}
}
return $menu_item;
}
/**
* Exclude menu items based on custom criteria.
*
* @since 3.3.0
*
* @global object $wpmem
*
* @param array $items
* @return array $items
*/
public function exclude_menu_items( $items ) {
global $wpmem;
$hide_children_of = array();
if ( ! empty( $items ) ) {
// Iterate and remove set items.
foreach ( $items as $key => $item ) {
$visible = true;
// Hide any item that is the child of a hidden item.
if ( isset( $item->menu_item_parent ) && in_array( $item->menu_item_parent, $hide_children_of ) ) {
$visible = false;
}
// Check items that have products.
if ( $visible && isset( $item->restrictions ) ) {
// Check all logged in, all logged out, or role.
switch( $item->restrictions ) {
case 'in' :
$visible = ( is_user_logged_in() ) ? true : false;
break;
case 'out' :
$visible = ( ! is_user_logged_in() ) ? true : false;
break;
default:
if ( 1 == $wpmem->enable_products ) {
$visible = false;
if ( is_array( $item->restrictions ) && ! empty( $item->restrictions ) ) {
foreach ( $item->restrictions['products'] as $product ) {
if ( wpmem_user_has_access( $product ) ) {
$visible = true;
}
}
}
} else {
$visible = true;
}
break;
}
}
/**
* Filters item visibility value.
*
* @since 3.3.0
*
* @param bool $visible
* @param object $item
*/
$visible = apply_filters( 'wpmem_menu_item_visibility', $visible, $item );
// Unset non-visible item.
if ( ! $visible ) {
if ( isset( $item->ID ) ) {
$hide_children_of[] = $item->ID;
}
unset( $items[ $key ] ) ;
}
}
}
return $items;
}
}