将产品限制为特定用户角色

Question

我有一个只有两个产品的WooCommerce网站，每种产品都属于其类别；

类别X必须每个人都能看到。

类别y必须仅限于具有用户角色的用户：从业者。

我如何：

（1）使类别y仅对从业者用户角色可见

，并且

（2）使所有人都可以看到类别，但只能通过从业者用户角色购买？

这是我的尝试，尽管无法正常工作。

add_action( 'woocommerce_product_query', 'limit_products_basic', 10, 2 );

function limit_products_basic( $q, $query )
{
 $user = wp_get_current_user();
 if ( !in_array( "practitioner", $user->roles ) ) {
     $tax_query = (array) $q->get( 'tax_query' );
     $tax_query[] = array(
     'taxonomy' => 'product_cat',
     'field' => 'slug',
     'terms' => array( 'app' ),
     'include_children' => true,
     );
 $q->set( 'tax_query', $tax_query );
 }
}

我知道有很多插件可以帮助实现这一目标，但是我希望能保持清洁和“简单”。

谢谢

Answer 1

我发现了Tom Anbinder的这个出色的教程，这就是结果：

add_role( 'practitioner', 'Practitioner', get_role( 'customer' )->capabilities );

if ( ! function_exists( 'wpf_is_current_user_role' ) ) {
    function wpf_is_current_user_role( $roles_to_check ) {
        $current_user       = wp_get_current_user();
        $current_user_roles = ( empty( $current_user->roles ) ? array( '' ) : $current_user->roles );
        $roles_intersect    = array_intersect( $current_user_roles, $roles_to_check );
        return ( ! empty( $roles_intersect ) );
    }
}

if ( ! function_exists( 'wpf_do_hide_product' ) ) {

    function wpf_do_hide_product( $product_id_to_check ) {

        $products_to_hide  = array( 204,); 
        $roles_to_hide_for = array( 'practitioner' ); 

        return (
            in_array( $product_id_to_check, $products_to_hide ) && 
            !wpf_is_current_user_role( $roles_to_hide_for )         
        );
    }
}

add_filter( 'woocommerce_is_purchasable', 'wpf_product_purchasable_by_user_role', PHP_INT_MAX, 2 );
if ( ! function_exists( 'wpf_product_purchasable_by_user_role' ) ) {
    function wpf_product_purchasable_by_user_role( $purchasable, $product ) {
        return ( wpf_do_hide_product( $product->get_id() ) ? false : $purchasable );
    }
}

汤姆网站上的更多详细信息： wpfactory.com