WordPress WooCommerce根据用户角色添加新的自定义菜单和限制

您的代码中仍然有一些小错误,一些遗漏的东西,并且由于WooCommerce 3,在第2步中对我的帐户端点进行了一些相关更改。有些事情也可以简化。

为了避免不允许的用户角色访问某些禁止的部分或端点,您可以使用挂接在 template_redirect 钩子,将用户重定向到允许的部分。

以下是完整的代码:

// Custom function that get My account menu item data based on user roles
function get_menu_item_by_user_role() {
    $user_roles = wp_get_current_user()->roles;

    if ( ! empty($user_roles) ) {
        $menu_item = [];

        // if ( in_array('mindesk_var_account', $user_roles) ) {
        if ( in_array( 'mindesk_var_account', $user_roles ) ) {
            $menu_item = [ 'clients' => __( "Clients", "woocommerce" ) ];
        }
        elseif( in_array( 'mindesk_owner_account', $user_roles ) ) {
            $menu_item = [ 'children' => __( "Children", "woocommerce" ) ];
        }
    }
    return $menu_item;
}

// Step 1 - Add Link (Tab) to My Account menu
add_filter ( 'woocommerce_account_menu_items', 'add_mindesk_custom_menu_items', 40 );
function add_mindesk_custom_menu_items( $menu_items ){
    $new_item = get_menu_item_by_user_role();

    if ( ! empty($new_item) ) {
        $menu_items = array_slice( $menu_items, 0, 5, true ) + $new_item + array_slice( $menu_items, 5, null, true );
    }
    return $menu_items;
}

// Step 2 - Enable endpoint (and endpoint permalink) - Since WooCommerce 3
add_filter( 'woocommerce_get_query_vars', 'add_mindesk_menu_item_endpoint' );
function add_mindesk_menu_item_endpoint( $query_vars ) {
    $query_vars['clients']  = 'clients';
    $query_vars['children'] = 'children';

    return $query_vars;
}

// Step 3. Content for the new page in My Account, woocommerce_account_{ENDPOINT NAME}_endpoint
add_action( 'woocommerce_account_clients_endpoint', 'add_mindesk_account_clients_endpoint_content' );
function add_mindesk_account_clients_endpoint_content() {
    require_once(get_template_directory() . '/myaccount/clients.php') ;
}

add_action( 'woocommerce_account_children_endpoint', 'add_mindesk_account_children_endpoint_content' );
function add_mindesk_account_children_endpoint_content() {
    require_once(get_template_directory() . '/myaccount/children.php') ;
}

// Step 4. Endpoint page title
add_filter( 'woocommerce_endpoint_clients_title', 'set_mindesk_account_clients_endpoint_title', 10, 2 );
function set_mindesk_account_clients_endpoint_title( $title, $endpoint ) {
    $title = __("Clients", "woocommerce" );

    return $title;
}

add_filter( 'woocommerce_endpoint_children_title', 'set_mindesk_account_children_endpoint_title', 10, 2 );
function set_mindesk_account_children_endpoint_title( $title, $endpoint ) {
    $title = __( "Children", "woocommerce" );

    return $title;
}

// Step 5. Redirect if not allowed user role
add_action( 'template_redirect', 'redirect_mindesk_account_dashboard' );
function redirect_mindesk_account_dashboard() {
    if ( is_account_page() ) {
        global $wp;

        $item_key = array_keys(get_menu_item_by_user_role());
        $page_url = get_permalink( get_option('woocommerce_myaccount_page_id') );

        if ( empty($item_key) &&  ( isset($wp->query_vars['children']) || isset($wp->query_vars['clients']) ) ) {
            wp_safe_redirect( get_permalink($page_id) );
            exit();
        }
        elseif ( 'clients' == reset($item_key) && isset($wp->query_vars['children']) ) {
            wp_safe_redirect( get_permalink($page_id) . 'clients/' );
            exit();
        }
        elseif ( 'children' == reset($item_key) && isset($wp->query_vars['clients']) ) {
            wp_safe_redirect( get_permalink($page_id) . 'children/'  );
            exit();
        }
    }
}

// Step 6. FLush rewrite rules:
// Go to Settings > Permalinks and click on "Save Changes".

代码进入函数。活动子主题(或活动主题)的php文件。经过测试并正常工作。

相关: WooCommerce My Account custom endpoint menu item