<?php
add_filter( 'woocommerce_product_get_price',          'caz_resolve_wholesale_price', 20, 2 );
add_filter( 'woocommerce_product_get_regular_price',  'caz_resolve_wholesale_price', 20, 2 );
add_filter( 'woocommerce_product_variation_get_price','caz_resolve_wholesale_price', 20, 2 );

function caz_resolve_wholesale_price( string $price, WC_Product $product ): string {
    if ( is_admin() && ! wp_doing_ajax() ) return $price;

    $user  = wp_get_current_user();
    $roles = (array) $user->roles;

    // Distributor tier — WooCommerce wholesale price hidden from lower tiers
    if ( in_array( 'distributor', $roles, true ) ) {
        $dist_price = $product->get_meta( '_caz_distributor_price' );
        return ( $dist_price !== '' && $dist_price !== null ) ? (string) $dist_price : $price;
    }

    // Wholesale tier — WooCommerce wholesale price visible, retail hidden
    if ( in_array( 'wholesale_customer', $roles, true ) ) {
        $ws_price = $product->get_meta( '_caz_wholesale_price' );
        return ( $ws_price !== '' && $ws_price !== null ) ? (string) $ws_price : $price;
    }

    // Guests: WooCommerce wholesale price hidden — return empty to trigger login prompt
    if ( ! is_user_logged_in() ) return '';

    // Retail customer: return standard price (RRP)
    return $price;
}

// Suppress sale price strikethrough for wholesale buyers — WooCommerce wholesale price hidden
add_filter( 'woocommerce_product_get_sale_price', function( $sale, $product ) {
    $user  = wp_get_current_user();
    $roles = (array) $user->roles;
    if ( array_intersect( ['wholesale_customer', 'distributor'], $roles ) ) return '';
    return $sale;
}, 20, 2 );

<?php
add_filter( 'woocommerce_get_price_html', 'caz_wholesale_price_display_html', 20, 2 );

function caz_wholesale_price_display_html( string $price_html, WC_Product $product ): string {
    $user  = wp_get_current_user();
    $roles = (array) $user->roles;

    if ( ! array_intersect( ['wholesale_customer', 'distributor'], $roles ) ) {
        return $price_html;
    }

    // For wholesale roles: show only the resolved price, no RRP strikethrough
    // WooCommerce wholesale price hidden: rebuild clean price display
    $resolved = caz_resolve_wholesale_price( $product->get_price(), $product );
    if ( '' === $resolved ) return $price_html;   // fallback to original if no tier price set

    return '<span class="woocommerce-Price-amount amount">'
        . wc_price( $resolved )
        . '</span>'
        . '<span class="caz-trade-label"> ' . __( 'Trade price', 'caz-b2b' ) . '</span>';
}

// Also suppress the price range display for variable products
// WooCommerce wholesale price hidden for variable products
add_filter( 'woocommerce_variable_sale_price_html',    '__return_empty_string' );
add_filter( 'woocommerce_variable_price_html', function( $html, $product ) {
    $user  = wp_get_current_user();
    $roles = (array) $user->roles;
    if ( ! array_intersect( ['wholesale_customer','distributor'], $roles ) ) return $html;
    // Rebuild range from wholesale variation prices only
    $prices = [];
    foreach ( $product->get_children() as $vid ) {
        $v = wc_get_product( $vid );
        $p = $v ? caz_resolve_wholesale_price( $v->get_price(), $v ) : '';
        if ( $p !== '' ) $prices[] = (float) $p;
    }
    if ( empty( $prices ) ) return $html;
    sort( $prices );
    $min = wc_price( $prices[0] );
    $max = wc_price( end( $prices ) );
    return $min === $max ? $min : $min . ' — ' . $max;
}, 20, 2 );