根据WooCommerce中的用户角色，通过占位符替换产品图像

Question

在WooCommerce中，我使用用户角色来定义每个用户角色可以看到的内容。如果用户是“客户”或“管理员”，他可以看到产品图像，否则他会看到WooCommerce默认占位符图像。

为此我使用下面的代码：

function woocommerce_product_get_image_id_callback( $value ) {
    global $current_user;

    if ( in_array( 'customer', (array) $current_user->roles )|| in_array( 'administrator', (array) $current_user->roles )) {
        return $value;
    } else {
        return false;
    }
}
add_filter( 'woocommerce_product_get_image_id', 'woocommerce_product_get_image_id_callback', 10, 1 );

它适用于简单的产品，但不适用于可变产品： 对于未选择变体的变量产品，将显示占位符，但当用户选择任何变体时，将显示图像。

有人可以帮忙解决这个问题吗？也许禁用在if中选择产品变体？

Answer 1

更新：以下替换代码将处理所有产品类型（甚至可变产品及其变体）：

add_filter( 'woocommerce_product_variation_get_image_id', 'woocommerce_product_get_image_id_callback', 10, 2 );
add_filter( 'woocommerce_product_get_image_id', 'woocommerce_product_get_image_id_callback', 10, 2 );
function woocommerce_product_get_image_id_callback( $image_id, $product ) {
    if ( ! ( current_user_can('customer') || current_user_can('administrator') ) ) {
        $image_id = '';
    }
    return $image_id;
}

代码位于活动子主题（或活动主题）的function.php文件中。经过测试和工作。