417 lines
16 KiB
PHP
417 lines
16 KiB
PHP
<?php
|
|
|
|
/**
|
|
* Affiliate Tracker
|
|
*
|
|
* Handles referral tracking via cookies and order creation.
|
|
*
|
|
* @package WooNooW\Modules\Affiliate
|
|
*/
|
|
|
|
namespace WooNooW\Modules\Affiliate;
|
|
|
|
if (!defined('ABSPATH')) exit;
|
|
|
|
class AffiliateTracker
|
|
{
|
|
const COOKIE_NAME = 'woonoow_ref';
|
|
const COOKIE_EXPIRES = 30; // days
|
|
|
|
/**
|
|
* Initialize tracking hooks
|
|
*/
|
|
public static function init()
|
|
{
|
|
error_log('[AffiliateTracker] init() called');
|
|
|
|
// Intercept referral links on init
|
|
add_action('init', [__CLASS__, 'capture_referral_link']);
|
|
|
|
// Hook into WooCommerce order creation (legacy checkout)
|
|
add_action('woocommerce_checkout_order_processed', [__CLASS__, 'record_referral'], 10, 3);
|
|
|
|
// Hook into WooCommerce 8.3+ block checkout
|
|
add_action('woocommerce_store_api_checkout_order_processed', [__CLASS__, 'record_referral_block'], 10, 1);
|
|
|
|
// Hook into new order creation (universal)
|
|
add_action('woocommerce_new_order', [__CLASS__, 'record_referral_new_order'], 10, 2);
|
|
|
|
// Hook for all order status changes
|
|
add_action('woocommerce_order_status_changed', [__CLASS__, 'handle_order_status_changed'], 10, 4);
|
|
|
|
// Hook for REST API order creation
|
|
add_action('rest_after_insert_shop_order', [__CLASS__, 'handle_rest_order_created'], 10, 3);
|
|
|
|
error_log('[AffiliateTracker] All hooks registered');
|
|
}
|
|
|
|
/**
|
|
* Handle order status change - track completed orders
|
|
*/
|
|
public static function handle_order_status_changed($order_id, $from_status, $to_status, $order)
|
|
{
|
|
if ($to_status === 'completed') {
|
|
self::process_order_for_referral($order_id);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Handle REST API order creation
|
|
*/
|
|
public static function handle_rest_order_created($order, $request, $creating)
|
|
{
|
|
if ($creating) {
|
|
self::process_order_for_referral($order->get_id());
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Record referral from block checkout (Store API)
|
|
*/
|
|
public static function record_referral_block($order)
|
|
{
|
|
if (is_numeric($order)) {
|
|
$order = wc_get_order($order);
|
|
}
|
|
if ($order) {
|
|
self::process_order_for_referral($order->get_id());
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Record referral from woocommerce_new_order hook
|
|
*/
|
|
public static function record_referral_new_order($order_id, $order)
|
|
{
|
|
error_log('[AffiliateTracker] woocommerce_new_order hook fired for order: ' . $order_id);
|
|
self::process_order_for_referral($order_id);
|
|
}
|
|
|
|
/**
|
|
* Capture ?ref=CODE from URL and set cookie
|
|
* Also captures UTM parameters for referral attribution
|
|
*/
|
|
public static function capture_referral_link()
|
|
{
|
|
error_log('[AffiliateTracker] capture_referral_link called, ref=' . ($_GET['ref'] ?? 'none'));
|
|
|
|
$options = [
|
|
'expires' => time() + (self::COOKIE_EXPIRES * DAY_IN_SECONDS),
|
|
'path' => '/',
|
|
'secure' => is_ssl(),
|
|
'samesite' => 'Lax'
|
|
];
|
|
|
|
// Capture referral code
|
|
if (isset($_GET['ref']) && !empty($_GET['ref'])) {
|
|
$referral_code = sanitize_text_field($_GET['ref']);
|
|
$result = setcookie(self::COOKIE_NAME, $referral_code, $options);
|
|
$_COOKIE[self::COOKIE_NAME] = $referral_code;
|
|
error_log('[AffiliateTracker] Set woonoow_ref cookie: ' . $referral_code . ', result=' . ($result ? 'true' : 'false'));
|
|
error_log('[AffiliateTracker] Cookie options: ' . json_encode($options));
|
|
} else {
|
|
// Check if cookie already exists from previous visit
|
|
if (isset($_COOKIE[self::COOKIE_NAME])) {
|
|
error_log('[AffiliateTracker] No ref param, but existing cookie: ' . $_COOKIE[self::COOKIE_NAME]);
|
|
} else {
|
|
error_log('[AffiliateTracker] No ref param and no existing cookie');
|
|
}
|
|
}
|
|
|
|
// Capture UTM parameters
|
|
$utm_params = [];
|
|
$utm_keys = ['utm_source', 'utm_medium', 'utm_campaign', 'utm_content', 'utm_term'];
|
|
|
|
foreach ($utm_keys as $key) {
|
|
if (isset($_GET[$key]) && !empty($_GET[$key])) {
|
|
$utm_params[$key] = sanitize_text_field($_GET[$key]);
|
|
}
|
|
}
|
|
|
|
// Capture referrer URL
|
|
if (isset($_SERVER['HTTP_REFERER']) && !empty($_SERVER['HTTP_REFERER'])) {
|
|
$utm_params['referrer_url'] = esc_url_raw($_SERVER['HTTP_REFERER']);
|
|
}
|
|
|
|
// Store UTM params in cookie if any captured
|
|
if (!empty($utm_params)) {
|
|
$utm_json = json_encode($utm_params);
|
|
setcookie(self::COOKIE_NAME . '_utm', $utm_json, $options);
|
|
$_COOKIE[self::COOKIE_NAME . '_utm'] = $utm_json;
|
|
error_log('[AffiliateTracker] Set woonoow_ref_utm cookie');
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Get active affiliate for an order
|
|
* Checks both coupons and cookies.
|
|
*/
|
|
public static function get_affiliate_for_order($order)
|
|
{
|
|
global $wpdb;
|
|
$table = $wpdb->prefix . 'woonoow_affiliates';
|
|
|
|
// 1. Check if a coupon is used that maps to an affiliate
|
|
$used_coupons = $order->get_coupon_codes();
|
|
if (!empty($used_coupons)) {
|
|
foreach ($used_coupons as $code) {
|
|
$coupon = new \WC_Coupon($code);
|
|
if ($coupon->get_id()) {
|
|
$affiliate = $wpdb->get_row($wpdb->prepare(
|
|
"SELECT * FROM $table WHERE coupon_id = %d AND status = 'active'",
|
|
$coupon->get_id()
|
|
));
|
|
if ($affiliate) return $affiliate;
|
|
}
|
|
}
|
|
}
|
|
|
|
// 2. Check referral code stored on the order by SPA checkout
|
|
$order_referral_code = $order->get_meta('_woonoow_referral_code');
|
|
if (!empty($order_referral_code)) {
|
|
$affiliate = $wpdb->get_row($wpdb->prepare(
|
|
"SELECT * FROM $table WHERE referral_code = %s AND status = 'active'",
|
|
sanitize_text_field($order_referral_code)
|
|
));
|
|
if ($affiliate) return $affiliate;
|
|
}
|
|
|
|
// 3. Fallback to cookie
|
|
if (isset($_COOKIE[self::COOKIE_NAME])) {
|
|
$referral_code = sanitize_text_field($_COOKIE[self::COOKIE_NAME]);
|
|
$affiliate = $wpdb->get_row($wpdb->prepare(
|
|
"SELECT * FROM $table WHERE referral_code = %s AND status = 'active'",
|
|
$referral_code
|
|
));
|
|
if ($affiliate) return $affiliate;
|
|
}
|
|
|
|
return null;
|
|
}
|
|
|
|
/**
|
|
* Get referral record for a specific order
|
|
* Used by admin API to display affiliate info on order details
|
|
*/
|
|
public static function get_referral_for_order($order_id)
|
|
{
|
|
global $wpdb;
|
|
$referrals_table = $wpdb->prefix . 'woonoow_referrals';
|
|
|
|
return $wpdb->get_row($wpdb->prepare(
|
|
"SELECT * FROM $referrals_table WHERE order_id = %d",
|
|
$order_id
|
|
));
|
|
}
|
|
|
|
/**
|
|
* Record referral when order is processed (Checkout)
|
|
*/
|
|
public static function record_referral($order_id, $posted_data, $order)
|
|
{
|
|
self::process_order_for_referral($order_id);
|
|
}
|
|
|
|
/**
|
|
* Record referral when order is created via Admin
|
|
*/
|
|
public static function record_referral_admin($post_id)
|
|
{
|
|
$order = wc_get_order($post_id);
|
|
if ($order) {
|
|
self::process_order_for_referral($post_id);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Process order and create pending referral record
|
|
*/
|
|
private static function process_order_for_referral($order_id)
|
|
{
|
|
error_log('[AffiliateTracker] process_order_for_referral() called for order: ' . $order_id);
|
|
|
|
global $wpdb;
|
|
$order = wc_get_order($order_id);
|
|
if (!$order) {
|
|
error_log('[AffiliateTracker] Order not found: ' . $order_id);
|
|
return;
|
|
}
|
|
|
|
// Check if referral already exists for this order
|
|
$referrals_table = $wpdb->prefix . 'woonoow_referrals';
|
|
$exists = $wpdb->get_var($wpdb->prepare("SELECT id FROM $referrals_table WHERE order_id = %d", $order_id));
|
|
if ($exists) {
|
|
error_log('[AffiliateTracker] Referral already exists for order: ' . $order_id);
|
|
return;
|
|
}
|
|
|
|
$affiliate = self::get_affiliate_for_order($order);
|
|
if (!$affiliate) {
|
|
error_log('[AffiliateTracker] No affiliate found for order: ' . $order_id);
|
|
return;
|
|
}
|
|
|
|
error_log('[AffiliateTracker] Found affiliate: ' . $affiliate->referral_code . ' (rate: ' . ($affiliate->custom_commission_rate ?? 'null') . ')');
|
|
|
|
// Prevent self-referrals (unless admin override is enabled)
|
|
$order_user_id = $order->get_user_id();
|
|
if ((int)$order_user_id === (int)$affiliate->user_id) {
|
|
error_log('[AffiliateTracker] Self-referral detected, blocking');
|
|
$allow_self_referral = get_option('woonoow_affiliate_allow_self_referral', false);
|
|
if (!$allow_self_referral) return;
|
|
}
|
|
|
|
// Calculate commission
|
|
// Priority: Product rate > Affiliate custom rate > Global default
|
|
$global_default_rate = (float) get_option('woonoow_affiliate_default_rate', 10);
|
|
$affiliate_custom_rate = !empty($affiliate->custom_commission_rate) ? (float) $affiliate->custom_commission_rate : null;
|
|
|
|
// Get order items - try WC method first, fallback to direct DB query
|
|
$items = self::get_order_items($order_id);
|
|
error_log('[AffiliateTracker] Order items: ' . count($items));
|
|
$total_commission = 0;
|
|
|
|
if (!empty($items)) {
|
|
foreach ($items as $item) {
|
|
$product_id = (int) $item['product_id'];
|
|
$line_total = (float) $item['line_total'];
|
|
|
|
// Check for product-specific affiliate rate
|
|
$product_rate = get_post_meta($product_id, '_woonoow_affiliate_commission_rate', true);
|
|
$product_enabled = get_post_meta($product_id, '_woonoow_affiliate_enabled', true) === 'yes';
|
|
error_log('[AffiliateTracker] Product ' . $product_id . ': enabled=' . ($product_enabled ? 'yes' : 'no') . ', rate=' . $product_rate . ', line_total=' . $line_total);
|
|
|
|
// Determine rate for this item (only if product affiliate is enabled)
|
|
$item_rate = null;
|
|
if ($product_enabled && $product_rate !== '') {
|
|
$item_rate = (float) $product_rate;
|
|
error_log('[AffiliateTracker] Using product rate: ' . $item_rate);
|
|
} elseif ($affiliate_custom_rate !== null) {
|
|
$item_rate = $affiliate_custom_rate;
|
|
error_log('[AffiliateTracker] Using affiliate rate: ' . $item_rate);
|
|
} else {
|
|
$item_rate = $global_default_rate;
|
|
error_log('[AffiliateTracker] Using global rate: ' . $item_rate);
|
|
}
|
|
|
|
if ($item_rate > 0 && $line_total > 0) {
|
|
$total_commission += ($line_total * $item_rate) / 100;
|
|
error_log('[AffiliateTracker] Added: ' . ($line_total * $item_rate / 100));
|
|
}
|
|
}
|
|
} else {
|
|
// Fallback to subtotal if no items (legacy)
|
|
$subtotal = (float)$order->get_subtotal();
|
|
$commission_rate = $affiliate_custom_rate ?? $global_default_rate;
|
|
if ($commission_rate <= 0) return;
|
|
$total_commission = ($subtotal * $commission_rate) / 100;
|
|
}
|
|
|
|
if ($total_commission <= 0) {
|
|
error_log('[AffiliateTracker] Commission is 0, skipping');
|
|
return;
|
|
}
|
|
|
|
error_log('[AffiliateTracker] Total commission: ' . $total_commission);
|
|
|
|
// Get UTM parameters from cookie
|
|
$utm_data = [];
|
|
if (isset($_COOKIE[self::COOKIE_NAME . '_utm'])) {
|
|
$utm_data = json_decode($_COOKIE[self::COOKIE_NAME . '_utm'], true) ?: [];
|
|
}
|
|
|
|
// Insert pending referral
|
|
$insert_data = [
|
|
'affiliate_id' => $affiliate->id,
|
|
'order_id' => $order_id,
|
|
'customer_id' => $order->get_user_id() ?: null,
|
|
'commission_amount' => $total_commission,
|
|
'currency' => $order->get_currency(),
|
|
'status' => 'pending'
|
|
];
|
|
|
|
// Add UTM data if available
|
|
if (!empty($utm_data)) {
|
|
foreach (['utm_source', 'utm_medium', 'utm_campaign', 'utm_content', 'utm_term', 'referrer_url'] as $utm_key) {
|
|
if (isset($utm_data[$utm_key])) {
|
|
$insert_data[$utm_key] = $utm_data[$utm_key];
|
|
}
|
|
}
|
|
}
|
|
|
|
$inserted = $wpdb->insert($referrals_table, $insert_data);
|
|
if (!$inserted) {
|
|
error_log('[AffiliateTracker] Failed to insert referral for order ' . $order_id . ': ' . $wpdb->last_error);
|
|
return;
|
|
}
|
|
|
|
$referral_id = $wpdb->insert_id;
|
|
|
|
// Fire event for notifications
|
|
do_action('woonoow/affiliate/referral_received', $referral_id, $affiliate, $order);
|
|
|
|
// Trigger email notification to affiliate
|
|
$user = get_userdata($affiliate->user_id);
|
|
if ($user) {
|
|
do_action('woonoow/email/trigger', 'affiliate_new_referral', $user->user_email, [
|
|
'affiliate_name' => $user->display_name,
|
|
'commission_amount' => $total_commission,
|
|
'currency' => $order->get_currency(),
|
|
'order_number' => $order->get_order_number()
|
|
]);
|
|
}
|
|
|
|
// Schedule auto-approval (e.g., 14 days) via Action Scheduler
|
|
if (function_exists('as_schedule_single_action')) {
|
|
$approval_days = get_option('woonoow_affiliate_holding_period', 14);
|
|
$timestamp = time() + ($approval_days * DAY_IN_SECONDS);
|
|
as_schedule_single_action($timestamp, 'woonoow_approve_referral', ['referral_id' => $referral_id], 'woonoow_affiliate');
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Get order items - direct DB query for HPOS compatibility
|
|
*/
|
|
private static function get_order_items($order_id)
|
|
{
|
|
global $wpdb;
|
|
|
|
// First try HPOS order_product_lookup table (more reliable with HPOS)
|
|
// Force fresh query by using wpdb::query with direct table
|
|
$order_id_int = (int) $order_id;
|
|
$items = $wpdb->get_results($wpdb->prepare(
|
|
"SELECT product_id, product_qty, product_net_revenue as line_total
|
|
FROM {$wpdb->prefix}wc_order_product_lookup
|
|
WHERE order_id = %d",
|
|
$order_id_int
|
|
), ARRAY_A);
|
|
|
|
error_log('[AffiliateTracker] get_order_items() HPOS query returned: ' . count($items) . ' items for order ' . $order_id_int);
|
|
|
|
// If no results from HPOS table, try legacy order_items table
|
|
if (empty($items)) {
|
|
$items = $wpdb->get_results($wpdb->prepare(
|
|
"SELECT oim_product.meta_value as product_id, oim_total.meta_value as line_total
|
|
FROM {$wpdb->prefix}woocommerce_order_items oi
|
|
LEFT JOIN {$wpdb->prefix}woocommerce_order_itemmeta oim_product ON oi.order_item_id = oim_product.order_item_id AND oim_product.meta_key = '_product_id'
|
|
LEFT JOIN {$wpdb->prefix}woocommerce_order_itemmeta oim_total ON oi.order_item_id = oim_total.order_item_id AND oim_total.meta_key = '_line_total'
|
|
WHERE oi.order_id = %d AND oi.order_item_type = 'line_item'",
|
|
$order_id
|
|
), ARRAY_A);
|
|
}
|
|
|
|
// Convert to standard format
|
|
$result = [];
|
|
foreach ($items as $item) {
|
|
$product_id = isset($item['product_id']) ? (int) $item['product_id'] : 0;
|
|
if ($product_id > 0) {
|
|
$result[] = [
|
|
'product_id' => $product_id,
|
|
'line_total' => (float) ($item['line_total'] ?? 0)
|
|
];
|
|
}
|
|
}
|
|
return $result;
|
|
}
|
|
}
|