<?php
// Add to functions.php or a custom plugin to hide price for guests in WooCommerce

// Step 1: Replace price display with login prompt for guests
add_filter( 'woocommerce_get_price_html', 'caz_hide_price_for_guests', 10, 2 );

function caz_hide_price_for_guests( string $price, WC_Product $product ): string {
    // Only hide price for guests — logged-in users see the real price
    if ( is_user_logged_in() ) return $price;

    return '<a href="' . esc_url( wp_login_url( get_permalink() ) ) . '" class="caz-login-to-see-price">'
        . __( 'Login to see prices', 'caz-b2b' )
        . '</a>';
}

// Step 2: Remove add to cart for guests — WooCommerce hide price for guests
add_filter( 'woocommerce_is_purchasable', 'caz_hide_add_to_cart_for_guests', 10, 2 );

function caz_hide_add_to_cart_for_guests( bool $purchasable, WC_Product $product ): bool {
    if ( ! is_user_logged_in() ) return false;
    return $purchasable;
}

// Step 3: Replace archive add-to-cart button with login link
add_filter( 'woocommerce_loop_add_to_cart_link', 'caz_loop_login_link_for_guests', 10, 2 );

function caz_loop_login_link_for_guests( string $button, WC_Product $product ): string {
    if ( is_user_logged_in() ) return $button;

    return '<a href="' . esc_url( wp_login_url( get_permalink( $product->get_id() ) ) ) . '"
        class="button caz-login-btn">'
        . __( 'Login to purchase', 'caz-b2b' )
        . '</a>';
}

function caz_hide_price_for_guests( string $price, WC_Product $product ): string {
    if ( is_user_logged_in() ) return $price;

    // Only hide price for guests in the 'trade-products' category
    $trade_categories = apply_filters( 'caz_hide_price_categories', [ 'trade-products', 'wholesale' ] );
    $product_cats     = wp_get_post_terms( $product->get_id(), 'product_cat', [ 'fields' => 'slugs' ] );

    if ( empty( array_intersect( $trade_categories, $product_cats ) ) ) {
        return $price;  // Not a trade category — show price to all
    }

    return '<a href="' . esc_url( wp_login_url( get_permalink() ) ) . '" class="caz-login-price">'
        . __( 'Login to see trade prices', 'caz-b2b' )
        . '</a>';
}

function caz_hide_price_for_guests_and_retail( string $price, WC_Product $product ): string {
    $approved_roles = apply_filters( 'caz_approved_price_roles', [
        'wholesale_customer',
        'trade_customer',
        'administrator',
    ]);

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

    // Hide price for guests (not logged in) AND for unapproved roles
    if ( empty( array_intersect( $approved_roles, $roles ) ) ) {
        $register_url = get_permalink( wc_get_page_id( 'myaccount' ) ) . 'register/';
        return '<a href="' . esc_url( $register_url ) . '" class="caz-register-prompt">'
            . __( 'Register for trade account to see prices', 'caz-b2b' )
            . '</a>';
    }

    return $price;
}