إخفاء الأسعار وزر الشراء عن غير المسجلين في ووكومرس (متاجر الجملة)

/**
 * إخفاء الأسعار وزر الشراء عن غير المسجّلين — ووكومرس
 * يوضع في functions.php للقالب الابن
 */

// من يحق له رؤية الأسعار؟ عدّل هنا: 'any' لأي مسجّل، أو اسم دور مثل 'wholesale'
function mirror_can_see_prices() {
    $allowed_role = 'any';

    if ( ! is_user_logged_in() ) {
        return false;
    }
    if ( 'any' === $allowed_role ) {
        return true;
    }
    $user = wp_get_current_user();
    return in_array( $allowed_role, (array) $user->roles, true ) || current_user_can( 'manage_options' );
}

// 1) استبدال السعر برسالة
add_filter( 'woocommerce_get_price_html', function ( $price, $product ) {
    if ( mirror_can_see_prices() ) {
        return $price;
    }
    $login = wp_login_url( get_permalink( $product->get_id() ) );
    return '<a class="mirror-login-to-see" href="' . esc_url( $login ) . '">سجّل الدخول لعرض السعر</a>';
}, 10, 2 );

// 2) منع الشراء (يخفي زر السلة في صفحة المنتج والشبكة)
add_filter( 'woocommerce_is_purchasable', function ( $purchasable, $product ) {
    return mirror_can_see_prices() ? $purchasable : false;
}, 10, 2 );

// 3) إزالة زر "أضف للسلة" من شبكة المتجر نهائيًا للزوّار
add_action( 'init', function () {
    if ( ! mirror_can_see_prices() ) {
        remove_action( 'woocommerce_after_shop_loop_item', 'woocommerce_template_loop_add_to_cart', 10 );
    }
} );

// 4) حماية إضافية: رفض الإضافة للسلة عبر أي طريقة
add_filter( 'woocommerce_add_to_cart_validation', function ( $passed ) {
    if ( ! mirror_can_see_prices() ) {
        wc_add_notice( 'الأسعار والشراء متاحان للعملاء المسجّلين فقط.', 'error' );
        return false;
    }
    return $passed;
} );

// 5) تنسيق بسيط للرسالة
add_action( 'wp_head', function () {
    if ( mirror_can_see_prices() ) {
        return;
    }
    echo '<style>.mirror-login-to-see{display:inline-block;background:#eff6ff;color:#2563eb;font-weight:700;padding:6px 14px;border-radius:999px;font-size:14px;text-decoration:none}</style>';
} );

// ينفَّذ مرة واحدة ثم يُحذف (الأدوار تُحفظ في قاعدة البيانات)
add_action( 'init', function () {
    if ( ! get_role( 'wholesale' ) ) {
        add_role( 'wholesale', 'تاجر جملة', get_role( 'customer' )->capabilities );
    }
} );