Skip to content

Instantly share code, notes, and snippets.

@goranefbl

goranefbl/class-wpgens-raf-checkout.php

Created January 15, 2024 20:17
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save goranefbl/85611be2f73b5f49a29a9710f04d9346 to your computer and use it in GitHub Desktop.
Save goranefbl/85611be2f73b5f49a29a9710f04d9346 to your computer and use it in GitHub Desktop.
Download ZIP
remove ref check
Raw
class-wpgens-raf-checkout.php
<?php
/**
* Hook into checkout
*
* @since 2.0.0
*/
if (!defined('ABSPATH')) {
exit;
}
class WPGens_RAF_Checkout
{
/**
* Constructor.
*
*/
public function __construct()
{
// Save RAF ID in Order Meta after Order is Complete
add_action('woocommerce_checkout_update_order_meta', array($this, 'maybe_save_raf_id'));
//Remove Cookie after checkout if Setting is set
add_action('woocommerce_thankyou', array($this, 'remove_cookie_after'));
// Hide auto applied coupon codes from showing.
add_filter('woocommerce_cart_totals_coupon_label', array($this, 'hide_coupon_code'), 10, 2);
// Auto apply RAF Coupons on cart for referrals. Also apply on checkout if cart is skipped.
add_action('woocommerce_before_cart', array($this, 'apply_matched_coupons')); // woocommerce_before_checkout_form
add_action('woocommerce_before_checkout_form', array($this, 'apply_matched_coupons')); // woocommerce_before_checkout_form
// add_action( 'woocommerce_ajax_added_to_cart', array( $this,'apply_matched_coupons') ); //woocomerce ajax add to cart but wc_print_notices is problem.
add_action('woocommerce_checkout_update_order_review', array($this, 'checkout_form_check'));
add_filter('woocommerce_get_shop_coupon_data', array($this, 'add_referral_via_coupon_field'), 10, 2);
add_action('woocommerce_applied_coupon', array($this, 'add_referral_apply_coupon_referral_code'), 10, 2);
add_action('woocommerce_removed_coupon', array($this, 'wc_removed_coupon'), 10, 1);
}
/**
* Before version 3, settings field were saved as "yes", after new react admin its "1".
*/
public function settings_field_active($field)
{
$value = get_option($field);
if ($value === "yes" || $value === "1") {
return true;
}
return false;
}
public function wc_removed_coupon($coupon)
{
$guest_coupon_code = get_option('gens_raf_guest_coupon_code');
if ($coupon === $guest_coupon_code || substr($coupon, 0, 3) === "ref") {
// unset($_COOKIE['gens_raf']);
// setcookie('gens_raf', '', time() - 3600, '/');
}
}
public function add_referral_apply_coupon_referral_code($coupon_code)
{
$guest_coupon_code = get_option('gens_raf_guest_coupon_code');
if ($coupon_code !== $guest_coupon_code && $this->settings_field_active('gens_raf_guest_enable')) {
$user_id = $this->get_id_from_referral_code($coupon_code);
if ($user_id) {
$time = 1;
if (get_current_user_id() != $user_id) {
if (get_option('gens_raf_cookie_time') != '') {
$time = intval(get_option('gens_raf_cookie_time'));
}
do_action('new_raf_data', 'coupon_applied', array('user' => get_current_user_id(), 'referral' => $user_id, 'type' => 'code'));
setcookie('gens_raf', $coupon_code, time() + 60 * 60 * 24 * $time, '/');
}
}
}
}
public function add_referral_via_coupon_field($data, $coupon_code)
{
// if (substr($coupon_code, 0, 3) != "ref") {
// return $data;
// }
$guest_coupon_code = get_option('gens_raf_guest_coupon_code');
if (!$this->settings_field_active('gens_raf_referral_codes') || is_admin() || (isset(WC()->cart) && WC()->cart->has_discount($guest_coupon_code))) {
return $data;
}
$user_id = $this->get_id_from_referral_code($coupon_code);
if ($user_id) {
if (get_current_user_id() != $user_id) {
$coupon_post_obj = get_page_by_title($guest_coupon_code, OBJECT, 'shop_coupon');
$coupon_id = $coupon_post_obj->ID;
$discount_type = get_post_meta($coupon_id, "discount_type", true);
$amount = get_post_meta($coupon_id, "coupon_amount", true);
$product_ids = get_post_meta($coupon_id, "product_ids", true);
$exclude_product_ids = get_post_meta($coupon_id, "exclude_product_ids", true);
$exclude_product_categories = get_post_meta($coupon_id, "exclude_product_categories", true);
$product_categories = get_post_meta($coupon_id, "product_categories", true);
$minimum_amount = get_post_meta($coupon_id, "minimum_amount", true);
$individual_use = get_post_meta($coupon_id, "individual_use", true);
$exclude_sale_items = get_post_meta($coupon_id, "exclude_sale_items", true);
$free_shipping = get_post_meta($coupon_id, "free_shipping", true);
$data = array(
'discount_type' => $discount_type,
'amount' => $amount,
'free_shipping' => $free_shipping,
'minimum_amount' => $minimum_amount,
'individual_use' => $individual_use === "yes" ? true : false,
'exclude_sale_items' => $exclude_sale_items === "yes" ? true : false,
'product_ids' => array_map('intval', (array) explode(',', $product_ids)),
'excluded_product_ids' => array_map('intval', (array) explode(',', $exclude_product_ids)),
'product_categories' => array_map('intval', (array) $product_categories),
'excluded_product_categories' => array_map('intval', (array) $exclude_product_categories),
'id' => true
);
}
}
return $data;
}
/**
* Save RAF(User) ID in Order Meta after Order is Complete
* woocommerce_checkout_update_order_meta hook
*
* @since 2.0.0
* @return string
*/
public function maybe_save_raf_id($order_id)
{
//1. Check cookie & get referrer or exit
$referrer_id = $this->check_referrer_cookie();
if (!$referrer_id) {
return $order_id;
}
//2. Check filter & is plugin active
$disable = apply_filters('gens_raf_disable', $this->settings_field_active('gens_raf_disable'), $order_id, $referrer_id);
if ($disable) {
return $order_id;
}
// 3. Save referral id and then work on security.
if (filter_var($referrer_id, FILTER_VALIDATE_EMAIL)) {
$rafID = $referrer_id;
} else {
$rafID = get_user_meta($referrer_id, "gens_referral_id", true);
}
$raf_info = $this->security_check($order_id, $referrer_id);
$order = wc_get_order($order_id);
do_action('new_raf_data', 'user_referred', array('user' => get_current_user_id() === 0 ? $order->get_billing_email() : get_current_user_id(), 'referrer' => $referrer_id));
do_action('new_raf_data', 'new_order', array_merge(array('user' => get_current_user_id(), 'referral' => $referrer_id, 'order' => $order_id), $raf_info));
$order->update_meta_data('_raf_meta', $raf_info); // will be depricated
$order->update_meta_data('_wpgens_raf_meta', $raf_info);
$order->update_meta_data('_raf_id', esc_attr($rafID)); // will be depricated
$order->update_meta_data('_wpgens_raf_id', esc_attr($rafID));
$order->save();
return $order_id;
}
public function security_check($order_id, $referrer_id)
{
$minimum_amount = apply_filters('gens_raf_minimum_order_amount', get_option('gens_raf_min_ref_order'), $order_id);
$nth_coupon = intval(get_option('gens_raf_nth_coupon'));
// Prevent user from checkout as a guest using his email
$order = wc_get_order($order_id);
$aelia_order_total = $order->get_meta('_order_total_base_currency', true);
$order_total = $order->get_total();
$user_email = (defined('WC_VERSION') && version_compare(WC_VERSION, '2.7', '<')) ? $order->billing_email : $order->get_billing_email();
$user_address = (defined('WC_VERSION') && version_compare(WC_VERSION, '2.7', '<')) ? $order->billing_address_1 : $order->get_billing_address_1();
$num_referrals = intval(get_user_meta($referrer_id, "gens_num_friends", true)) + 1;
// Check if user exists with this email already in the system?
$order_statuses = apply_filters('gens_raf_fraud_order_status', array_keys(wc_get_order_statuses()));
$email_exists = wc_get_orders(array(
'limit' => 1,
'customer' => $user_email,
'status' => $order_statuses,
'exclude' => array($order_id)
));
$customer_ip_address = $order->get_customer_ip_address();
$ip_address_exists = wc_get_orders(array(
'limit' => 1,
'customer_ip_address' => $customer_ip_address,
'status' => $order_statuses,
'exclude' => array($order_id)
));
$custom_msg = "";
$raf_info['custom_msg'] = apply_filters('gens_raf_custom_message', $custom_msg);
if (!empty($email_exists) && !$this->settings_field_active('gens_raf_allow_existing')) {
$raf_info = array("info" => __("Potential Fraud Detected. Referred customer has previous orders with " . $user_email . " address. Check plugin fraud settings to change this.", "gens-raf"), "generate" => "false", "increase_referrals" => "false", "email_notify" => "true");
} else if (!empty($ip_address_exists) && !$this->settings_field_active('gens_raf_allow_existing') && !$this->settings_field_active('gens_raf_fraud_disable_ip')) {
$raf_info = array("info" => __("Potential Fraud Detected. Referred customer has previous orders with " . $customer_ip_address . " ip address. Check plugin fraud settings to change this.", "gens-raf"), "generate" => "false", "increase_referrals" => "false", "email_notify" => "true");
} else if (get_current_user_id() === $referrer_id || $user_email === $referrer_id) {
$raf_info = array("info" => __("Potential Fraud Detected. User is trying to refer himself.", "gens-raf"), "generate" => "false", "increase_referrals" => "false", "email_notify" => "true");
} else if ($this->user_has_orders() && !$this->settings_field_active('gens_raf_allow_existing')) {
$raf_info = array("info" => __("This is not a new customer, and settings are set to disable coupons for existing customers.", "gens-raf"), "generate" => "false", "increase_referrals" => "false", "email_notify" => "false");
} else if (($user_address == get_user_meta($referrer_id, "billing_address_1", true)) && $user_address != '' && !$this->settings_field_active('gens_raf_fraud_disable_street')) {
$raf_info = array("info" => __("Potential Fraud Detected. Referer and referre have the same billing address. Investigate.", "gens-raf"), "generate" => "false", "increase_referrals" => "false", "email_notify" => "true");
} else if ($minimum_amount && (($aelia_order_total === '' && $minimum_amount > $order_total) || ($aelia_order_total !== '' && $minimum_amount > $aelia_order_total))) {
$raf_info = array("info" => __("Order minimum amount of " . $minimum_amount . " has not been met.", "gens-raf"), "generate" => "false", "increase_referrals" => "false", "email_notify" => "false");
} else if (!empty($nth_coupon) && ($nth_coupon !== 1) && ($num_referrals % $nth_coupon != 0)) {
$raf_info = array("info" => __("Coupons wont be generated due to nth coupon option.", "gens-raf"), "generate" => "false", "increase_referrals" => "true", "email_notify" => "false");
} else {
$raf_info = array("info" => "Referral is fine. Coupon will be generated on the order complete.", "generate" => "true", "increase_referrals" => "true", "email_notify" => "false");
}
// Notify admin of potential fraud detection.
if ($raf_info['email_notify'] == 'true') {
if (!$fraud_email_address) {
$fraud_email_address = get_option('admin_email');
}
$subject = 'Refer a friend - Potential fraud detected for a new referral';
$body = $raf_info['info'] . ' Order: ' . get_admin_url(null, 'post.php?post=' . $order_id . '&action=edit');
$headers = array('Content-Type: text/html; charset=UTF-8');
wp_mail($fraud_email_address, $subject, $body, $headers);
}
return apply_filters("gens_raf_order_info", $raf_info, $order, $referrer_id);
}
/**
* Returning number of orders customer has.
*
* @since 2.0.0
*/
public function user_has_orders($user_email = false)
{
$user_id = get_current_user_id();
if ($user_id == 0) {
if ($user_email !== false && $user_email != '') {
$email_exists = wc_get_orders(array(
'limit' => 1,
'customer' => $user_email,
'status' => array('wc-processing', 'wc-completed')
));
return count($email_exists);
}
return 0;
}
$customer_orders = wc_get_orders(array(
'numberposts' => 5,
'customer_id' => $user_id,
'status' => array('wc-completed'),
));
return count($customer_orders);
}
public function check_referrer_cookie()
{
// First check cookie
if (isset($_COOKIE["gens_raf"])) {
$user_id = $this->get_id_from_referral_code($_COOKIE["gens_raf"]);
if ($user_id) {
return $user_id;
}
}
// Then check if referral was applied through coupon.
if (!empty(WC()->cart->applied_coupons)) {
$coupons = WC()->cart->applied_coupons;
foreach ($coupons as $coupon) {
// if (substr($coupon, 0, 3) === "ref") {
$user_id = $this->get_id_from_referral_code($coupon);
if ($user_id) {
return $user_id;
}
// }
}
}
if (isset($_COOKIE["gens_raf"]) && filter_var($_COOKIE["gens_raf"], FILTER_VALIDATE_EMAIL)) {
return $_COOKIE["gens_raf"];
}
// Nothing? Return false.
return false;
}
/**
* Remove Cookie after checkout if Setting is set
* woocommerce_thankyou hook
*
* @since 1.0.0
*/
public function remove_cookie_after($order_id)
{
if (isset($_COOKIE['gens_raf']) && $this->settings_field_active('gens_raf_cookie_remove')) {
unset($_COOKIE['gens_raf']);
setcookie('gens_raf', '', time() - 3600, '/');
}
}
public function hide_coupon_code($text, $coupon)
{
$guest_coupon_code = get_option('gens_raf_guest_coupon_code');
if (method_exists($coupon, "get_code")) { // support for older version of woo
$coupon_code = $coupon->get_code();
} else {
$coupon_code = $coupon->code;
}
if ($coupon_code == strtolower($guest_coupon_code)) {
return apply_filters("gens_raf_coupon_applied", __('Coupon Applied!', 'gens-raf'));
} else {
return $text;
}
}
/**
* Auto apply coupons at the cart page for referred person, if chosen.
*
* @since 1.1.0
*/
public function apply_matched_coupons()
{
$guest_coupon_code = get_option('gens_raf_guest_coupon_code');
$guest_coupon_msg = get_option('gens_raf_guest_coupon_msg');
$referrer_id = $this->check_referrer_cookie();
if (!$referrer_id || get_current_user_id() === $referrer_id) {
return false;
}
do_action('gens_raf_auto_apply_coupon', $referrer_id);
// Do not allow guest links without single order.
if (filter_var($referrer_id, FILTER_VALIDATE_EMAIL)) {
// $user = get_user_by('email', $referrer_id);
// $customer_orders = wc_get_orders(array(
// 'numberposts' => 1,
// 'customer_id' => $user->ID,
// 'status' => array_keys(wc_get_order_statuses()),
// ));
// if (count($customer_orders) < 1) {
// return false;
// }
} else {
$user_info = get_userdata($referrer_id);
if ($user_info->first_name != '') {
$user_name = $user_info->first_name . ' ' . $user_info->last_name;
} else {
$user_name = __("Your friend", "gens-raf");
}
$guest_coupon_msg = str_replace('{{name}}', $user_name, $guest_coupon_msg);
}
if (!empty(WC()->cart->applied_coupons) || empty($guest_coupon_code) || $this->settings_field_active('gens_raf_disable') || WC()->cart->cart_contents_count < 1 || !$this->settings_field_active('gens_raf_guest_enable') || ($this->user_has_orders() >= 1 && !$this->settings_field_active('gens_raf_allow_existing'))) {
return false;
}
do_action('new_raf_data', 'coupon_applied', array('user' => get_current_user_id(), 'referral' => $referrer_id, 'type' => 'link'));
if (WC()->cart->add_discount($guest_coupon_code)) {
wc_add_notice($guest_coupon_msg);
wc_print_notices();
}
}
/**
* Remove coupon if user wants to abuse it by adding it as a guest then logging in at the checkout.
*
* @since 1.1.0
*/
public function checkout_form_check($post_data)
{
$guest_coupon_code = get_option('gens_raf_guest_coupon_code');
$email_exists = false;
parse_str($post_data, $data);
$referrer_id = $this->check_referrer_cookie();
if (!$referrer_id) {
return false;
}
$advocate_address = get_user_meta($referrer_id, "billing_address_1", true);
if (isset($data['billing_email']) && $data['billing_email'] != "") {
// This email already exists? Remove coupon.
$email_exists = wc_get_orders(array(
'numberposts' => 1,
'billing_email' => $data['billing_email'],
'status' => array('wc-processing', 'wc-completed'),
));
// This phone already exists? Remove coupon.
$phone_exists = false;
if ($data['billing_phone']) {
$orders_with_phone = wc_get_orders(array(
'numberposts' => 1,
'billing_phone' => $data['billing_phone'],
'status' => array('wc-processing', 'wc-completed'),
));
$phone_exists = !empty($orders_with_phone);
}
// In case referral is using his own coupon code.
$user = get_user_by('email', $data['billing_email']);
if ($user) {
$raf_code = get_user_meta($user->ID, 'gens_referral_id', true);
if (WC()->cart->has_discount($raf_code)) {
WC()->cart->remove_coupon($raf_code);
}
}
}
do_action('gens_raf_checkout_check', $guest_coupon_code, $referrer_id);
$should_remove_coupon = (isset($_COOKIE["gens_raf_guest"]) && $data['billing_email'] === $_COOKIE["gens_raf_guest"]) ||
get_current_user_id() === $referrer_id ||
(($this->user_has_orders($data['billing_email']) > 0 || !empty($email_exists) || !empty($phone_exists) || ($advocate_address === $data['billing_address_1'] && $advocate_address !== '')) && !$this->settings_field_active('gens_raf_allow_existing'));
$should_remove_coupon = apply_filters('gens_raf_should_remove_coupon', $should_remove_coupon, $guest_coupon_code, $referrer_id);
$coupon_removal_message = apply_filters('gens_raf_coupon_removal_message', __('The coupon has been removed from your cart. You are an existing customer and are not entitled to a referral discount.', 'gens-raf'));
if ($should_remove_coupon) {
if (WC()->cart->has_discount($guest_coupon_code)) {
WC()->cart->remove_coupon($guest_coupon_code);
wc_add_notice($coupon_removal_message, 'error');
}
if (isset($_COOKIE["gens_raf"]) && !empty($_COOKIE["gens_raf"]) && WC()->cart->has_discount($_COOKIE["gens_raf"])) {
WC()->cart->remove_coupon($_COOKIE["gens_raf"]);
wc_add_notice($coupon_removal_message, 'error');
}
}
// Because we cant remove multiple coupons during applying of them....
if (WC()->cart->has_discount() && count(WC()->cart->get_applied_coupons()) > 1) {
$coupons = WC()->cart->get_applied_coupons();
$i = 0;
foreach ($coupons as $coupon) {
if (substr($coupon, 0, 3) === "ref" || $coupon == $guest_coupon_code) {
$i++;
}
if ($i > 1) {
WC()->cart->remove_coupon($coupon);
}
}
}
}
public function get_id_from_referral_code($referral_code)
{
global $wpdb;
$results = $wpdb->get_results($wpdb->prepare("SELECT user_id FROM $wpdb->usermeta WHERE meta_key='gens_referral_id' AND meta_value = %s", $referral_code));
if (!empty($results)) {
return (int)$results[0]->user_id;
}
return false;
}
}
$wpgens_raf_checkout = new WPGens_RAF_Checkout();
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment