如何让WooCommerce自定义我的账户端点仅对指定角色可见并拦截非法访问

当然可以实现这个需求！下面是具体的步骤和代码示例，帮你完成这个功能：

核心思路 #

我们需要在页面模板加载前检查当前访问用户的角色，如果用户不是shop_manager且正在访问你的自定义端点，就触发404页面状态，让用户看到404内容（这种方式比直接重定向更符合HTTP规范）。

代码实现 #

将以下代码添加到你的主题functions.php文件或者自定义插件中：

情况1：自定义端点通过add_rewrite_endpoint添加 #

假设你是用官方推荐的add_rewrite_endpoint创建的自定义端点（比如端点名为custom-shop-manager-dashboard）：

// 确保自定义端点已正确注册（如果已经添加过这段可以忽略）
add_action( 'init', 'register_my_account_custom_endpoint' );
function register_my_account_custom_endpoint() {
    add_rewrite_endpoint( 'custom-shop-manager-dashboard', EP_ROOT | EP_PAGES );
}

// 拦截非授权用户访问
add_action( 'template_redirect', 'restrict_custom_endpoint_to_shop_manager' );
function restrict_custom_endpoint_to_shop_manager() {
    // 检查当前是否在我的账户页面，且访问的是目标自定义端点
    if ( is_account_page() && get_query_var( 'custom-shop-manager-dashboard', false ) !== false ) {
        $current_user = wp_get_current_user();
        
        // 如果用户未登录，或不是shop_manager角色，触发404
        if ( ! is_user_logged_in() || ! in_array( 'shop_manager', $current_user->roles ) ) {
            global $wp_query;
            $wp_query->set_404();
            status_header( 404 );
            get_template_part( '404' );
            exit;
        }
    }
}

情况2：自定义端点通过?action=参数实现 #

如果你的自定义端点是通过my-account/?action=custom-endpoint这种方式访问的，代码调整如下：

add_action( 'template_redirect', 'restrict_action_based_endpoint_to_shop_manager' );
function restrict_action_based_endpoint_to_shop_manager() {
    if ( is_account_page() && isset( $_GET['action'] ) && $_GET['action'] === 'custom-endpoint' ) {
        $current_user = wp_get_current_user();
        
        if ( ! is_user_logged_in() || ! in_array( 'shop_manager', $current_user->roles ) ) {
            global $wp_query;
            $wp_query->set_404();
            status_header( 404 );
            get_template_part( '404' );
            exit;
        }
    }
}

关键注意事项 #

1. 刷新固定链接：添加或修改端点相关代码后，一定要进入WordPress后台的「设置」→「固定链接」，直接点击「保存更改」，确保Rewrite规则生效。
2. 灵活调整拦截逻辑：如果你希望未登录用户跳转到登录页而非404，可以把! is_user_logged_in()的判断替换成重定向代码，比如：wp_safe_redirect( wc_get_page_permalink( 'myaccount' ) ); exit;。
3. 端点名称替换：记得把代码中的custom-shop-manager-dashboard或custom-endpoint替换成你实际的自定义端点名称。

内容的提问来源于stack exchange，提问作者Nicola