// Remove prices for non-logged in users
function remove_prices_for_non_logged_users() {
    if (!is_user_logged_in()) {
        // Remove price from shop and category pages
        remove_action('woocommerce_after_shop_loop_item_title', 'woocommerce_template_loop_price', 10);
        
        // Remove price from single product page
        remove_action('woocommerce_single_product_summary', 'woocommerce_template_single_price', 10);
        
        // Add custom message for non-logged in users
        add_action('woocommerce_single_product_summary', 'add_login_message', 10);
        add_action('woocommerce_after_shop_loop_item_title', 'add_login_message', 10);
    }
}
add_action('init', 'remove_prices_for_non_logged_users');

// Custom message function
function add_login_message() {
    echo '<div class="login-to-see-price">';
    echo '<a href="' . get_permalink(get_option('woocommerce_myaccount_page_id')) . '">';
    echo 'Login to see prices';
    echo '</a>';
    echo '</div>';
}

// Add custom CSS
function add_custom_price_css() {
    if (!is_user_logged_in()) {
        ?>
        <style>
            .login-to-see-price {
                margin: 10px 0;
                font-weight: bold;
            }
            .login-to-see-price a {
                color: #333;
                text-decoration: underline;
            }
            /* Hide price in related products and widgets */
            .related.products .price,
            .widget_products .price {
                display: none !important;
            }
        </style>
        <?php
    }
}
add_action('wp_head', 'add_custom_price_css');