How I Implemented Role-Based Discount System in WooCommerce [Case Study]

// Custom user roles registration
function register_custom_roles() {
    // Special Customer role
    add_role('special_customer', 'Special Customer', array(
        'read' => true,
        // Other capabilities...
    ));
    
    // Reseller role
    add_role('reseller', 'Reseller', array(
        'read' => true,
        // Other capabilities...
    ));
    
    // Super Reseller role
    add_role('super_reseller', 'Super Reseller', array(
        'read' => true,
        // Other capabilities...
    ));
}

function add_simple_product_discount_fields() {
global $post;

// Get current discount values
$special_customer_discount = get_post_meta($post->ID, '_special_customer_discount', true);
$reseller_discount = get_post_meta($post->ID, '_reseller_discount', true);
// More code...

echo '<div class="options_group show_if_simple">';

// Special Customer Discount Fields
echo '<h4>Set Price for Special Customers</h4>';
woocommerce_wp_text_input(
    array(
        'id' => '_special_customer_discount',
        'label' => 'Discount (%)',
        'desc_tip' => true,
        'description' => 'Enter the discount percentage for Special Customers.',
        // More parameters...
    )
);

// More discount fields for other roles...

}

function add_variation_discount_fields($loop, $variation_data, $variation) {
    // Get current discount values
    $special_customer_discount = get_post_meta($variation->ID, '_special_customer_discount', true);
    $reseller_discount = get_post_meta($variation->ID, '_reseller_discount', true);
    // More code...
    
    // Output custom fields
    echo '<div class="role-based-pricing">';
    
    echo '<p><strong>Set Discount Percentages for User Roles</strong></p>';
    
    // Special Customer Discount
    echo '<div class="discount-field-group">';
    // Fields implementation...
    echo '</div>';
    
    // More discount fields for other roles...
}

function set_default_discounts_for_product($post_id) {
    // Check if this is a new product
    if (get_post_meta($post_id, '_default_discounts_set', true) === 'yes') {
        return;
    }
    
    // Get product object
    $product = wc_get_product($post_id);
    
    if (!$product) {
        return;
    }
    
    // Get product type and name
    $product_type = $product->get_type();
    $product_name = $product->get_name();
    
    // Default discounts based on product name/type
    $special_customer_discount = 10; // 10% for all products
    $reseller_discount = 20; // 20% default for resellers
    $super_reseller_discount = 25; // 25% default for super resellers
    
    // Set specific discounts based on product
    // More conditional logic for different products...
}

function display_discount_info() {
    global $product;
    
    // Get current user
    $user = wp_get_current_user();
    $user_role = $user->roles[0] ?? 'customer';
    
    // If user is not logged in or is a regular customer, don't show discount info
    if (!is_user_logged_in() || $user_role === 'customer') {
        return;
    }
    
    // For variable products, handle differently with JavaScript
    if ($product->is_type('variable')) {
        // JavaScript implementation for variations...
    } else {
        // For simple products, display static discount info
        // Implementation details...
    }
}

// Client-side JavaScript
jQuery(document).ready(function($) {
    // Function to update discount info
    function updateDiscountInfo(variation) {
        // AJAX call to get the discount for this variation
        $.ajax({
            url: admin_ajax_url,
            type: 'POST',
            data: {
                'action': 'get_variation_discount',
                'variation_id': variation.variation_id,
                'user_role': userRole
            },
            success: function(response) {
                // Update the display with discount info
                // More code...
            }
        });
    }
    
    // Listen for variation changes
    $('.variations_form').on('found_variation', function(event, variation) {
        updateDiscountInfo(variation);
    });
});

// Server-side AJAX handler
function get_variation_discount() {
    $variation_id = isset($_POST['variation_id']) ? intval($_POST['variation_id']) : 0;
    $user_role = isset($_POST['user_role']) ? sanitize_text_field($_POST['user_role']) : '';
    
    // Get discount based on user role
    $discount = 0;
    
    if ($user_role === 'special_customer') {
        $discount = get_post_meta($variation_id, '_special_customer_discount', true);
    } elseif ($user_role === 'reseller') {
        $discount = get_post_meta($variation_id, '_reseller_discount', true);
    } elseif ($user_role === 'super_reseller') {
        $discount = get_post_meta($variation_id, '_super_reseller_discount', true);
    }
    
    wp_send_json_success(array(
        'discount' => $discount
    ));
}