<?php
add_filter( 'woocommerce_get_price_html', 'caz_b2b_login_to_see_price', 10, 2 );

function caz_b2b_login_to_see_price( string $price, WC_Product $product ): string {
    // Approved wholesale roles can see price — WooCommerce B2B login to see price
    $trade_roles = ['wholesale_customer', 'trade_customer', 'administrator'];
    $user        = wp_get_current_user();
    if ( ! empty( array_intersect( $trade_roles, (array) $user->roles ) ) ) {
        return $price;
    }

    // Logged-in but pending approval
    if ( is_user_logged_in() ) {
        return '<span class="caz-pending-approval">'
            . __( 'Your trade account is pending approval. We will notify you by email.', 'caz-b2b' )
            . '</span>';
    }

    // Guest: show WooCommerce B2B login to see price prompt
    $register_url = add_query_arg( 'redirect_to', urlencode( get_permalink() ),
        get_permalink( wc_get_page_id( 'myaccount' ) ) . 'register/' );
    $login_url    = wp_login_url( get_permalink() );

    return '<div class="caz-b2b-price-gate">
        <p class="caz-gate-heading">' . __( 'Trade prices available to registered customers', 'caz-b2b' ) . '</p>
        <p class="caz-gate-benefits">' . __( 'Apply for a free trade account to access wholesale pricing, minimum order discounts, and priority despatch.', 'caz-b2b' ) . '</p>
        <a href="' . esc_url( $register_url ) . '" class="button caz-register-btn">'
            . __( 'Apply for trade account', 'caz-b2b' )
        . '</a>
        <a href="' . esc_url( $login_url ) . '" class="caz-login-link">'
            . __( 'Already registered? Login here', 'caz-b2b' )
        . '</a>
    </div>';
}

<?php
// Add fields to WooCommerce registration form
add_action( 'woocommerce_register_form', 'caz_b2b_registration_fields' );

function caz_b2b_registration_fields(): void {
    ?>
    <p class="form-row">
        <label><?php esc_html_e( 'Company name', 'caz-b2b' ); ?> <span class="required">*</span></label>
        <input type="text" name="caz_company" value="<?php echo esc_attr( $_POST['caz_company'] ?? '' ); ?>" required>
    </p>
    <p class="form-row">
        <label><?php esc_html_e( 'VAT registration number', 'caz-b2b' ); ?></label>
        <input type="text" name="caz_vat" value="<?php echo esc_attr( $_POST['caz_vat'] ?? '' ); ?>">
    </p>
    <?php
}

// Save trade fields and set pending role for WooCommerce B2B login to see price
add_action( 'woocommerce_created_customer', 'caz_save_b2b_registration_fields' );

function caz_save_b2b_registration_fields( int $user_id ): void {
    if ( isset( $_POST['caz_company'] ) ) {
        update_user_meta( $user_id, 'caz_company',     sanitize_text_field( $_POST['caz_company'] ) );
        update_user_meta( $user_id, 'caz_vat',         sanitize_text_field( $_POST['caz_vat'] ?? '' ) );
        update_user_meta( $user_id, 'caz_trade_status', 'pending' );
    }
    // Send admin notification for WooCommerce B2B login to see price approval
    wp_mail( get_option('admin_email'),
        'New trade account application',
        'A new trade account has been requested. Review at: ' . admin_url('users.php')
    );
}

<?php
// Add "Approve Trade Account" action to WP Admin Users list
add_filter( 'user_row_actions', 'caz_approve_trade_action', 10, 2 );

function caz_approve_trade_action( array $actions, WP_User $user ): array {
    if ( 'pending' !== get_user_meta( $user->ID, 'caz_trade_status', true ) ) return $actions;

    $url = add_query_arg([
        'action'  => 'caz_approve_trade',
        'user_id' => $user->ID,
        '_nonce'  => wp_create_nonce( 'caz_approve_' . $user->ID ),
    ], admin_url('users.php') );

    $actions['caz_approve'] = '<a href="' . esc_url( $url ) . '" style="color:#7C3AED;font-weight:bold">'
        . __( 'Approve Trade Account', 'caz-b2b' ) . '</a>';
    return $actions;
}

// Handle the approval — assign wholesale role for WooCommerce B2B login to see price
add_action( 'admin_init', function() {
    if ( $_GET['action'] ?? '' !== 'caz_approve_trade' ) return;
    $user_id = absint( $_GET['user_id'] ?? 0 );
    check_admin_referer( 'caz_approve_' . $user_id, '_nonce' );

    $user = get_user_by( 'id', $user_id );
    if ( ! $user ) return;

    $user->set_role( 'wholesale_customer' );  // triggers WooCommerce B2B login to see price
    update_user_meta( $user_id, 'caz_trade_status', 'approved' );

    // Notify customer their WooCommerce B2B login to see price account is active
    wp_mail( $user->user_email,
        'Your trade account is approved',
        'Your trade account has been approved. You can now log in to see trade pricing at: ' . get_home_url()
    );

    wp_redirect( admin_url('users.php?caz_approved=1') );
    exit;
});