<?php
class WPAM_Commission_Tracking {
    
    public static function handle_commission_tracking_hook($args)
    {
        /*
        * The args array must have the following 3 details
        * $args['txn_id']
        * $args['amount']
        * $args['aff_id']
        */
   
        WPAM_Logger::log_debug('handle_commission_tracking_hook() - Txn ID : '.$args['txn_id'].', Amount: '.$args['amount'].', Affiliate ID: '.$args['aff_id']);
        WPAM_Commission_Tracking::award_commission($args);
    }
            
    public static function award_commission($args){
        global $wpdb;
        $txn_id = $args['txn_id'];
        $amount = $args['amount'];
        $aff_id = $args['aff_id'];
        $buyer_email = '';
        if(isset($args['email']) && !empty($args['email'])){
            $buyer_email = $args['email'];
        }
        $affiliate = '';
        if(isset($aff_id) && is_numeric($aff_id)){  //aff_id contains affiliate ID from the new cookie system (wpam_id)
            $query = "SELECT * FROM ".WPAM_AFFILIATES_TBL." WHERE affiliateId = %d";        
            $affiliate = $wpdb->get_row($wpdb->prepare($query, $aff_id));    
        }
        else{ //TODO start - We only need this code for now to get the affiliate ID for a purchase. Later with the new tracking system it can be deleted            
            $query = "
            SELECT a.*
            FROM ".WPAM_TRACKING_TOKENS_PURCHASE_LOGS_TBL." pl
            INNER JOIN ".WPAM_TRACKING_TOKENS_TBL." tt ON (tt.trackingTokenId = pl.trackingTokenId)
            INNER JOIN ".WPAM_AFFILIATES_TBL." a ON (a.affiliateId = tt.sourceAffiliateId)
            WHERE
            pl.purchaseLogId = %s
            ";        
            $affiliate = $wpdb->get_row($wpdb->prepare($query, $txn_id));              
        } //TODO end - later affiliate ID can be tracked directly from the cookie instead of ref_key
        
        if($affiliate != null && $affiliate->status == "active") {
            if(isset($args['comm_override']) && !empty($args['comm_override'])){
                //Filter for overriding the commission from an addon/plugin
                $override = "";
                $override = apply_filters('wpam_commission_tracking_override', $override, $affiliate, $args);
                if (!empty($override)) {
                    //commission has been overriden by another addon/plugin
                    WPAM_Logger::log_debug('*** Commission for this sale has been overriden by an addon/plugin via filter. ***'); 
                    return;
                }
            }
            $creditAmount = 0;
            if ($affiliate->bountyType == 'percent')
            {
                $creditAmount = $amount * ($affiliate->bountyAmount / 100.0);
            }
            else if ($affiliate->bountyType == 'fixed')
            {
                $creditAmount = (float) $affiliate->bountyAmount;
            }
            else
            {
                //No commission type for this affiliate (e.g. added directly in the database): no commission is calculated
                WPAM_Logger::log_debug('Affiliate ID: '.$affiliate->affiliateId.' has no commission type set, so the commission amount is 0. Set a commission type for this affiliate.', 3);
            }
            $creditAmount = round($creditAmount, 2);
            //checking to see if "Disable Own Referrals" option is enabled
            if(get_option(WPAM_PluginConfig::$DisableOwnReferrals) == 1){
                if (!empty($buyer_email)) {
                    if (wpam_check_if_buyer_is_referrer($aff_id, $buyer_email)) {
                        WPAM_Logger::log_debug('The buyer ('.$buyer_email.') is the referrer (affiliate id: '.$aff_id.') so this sale is not eligible for generating any commission.', 4);
                        return;
                    }
                } 
                else {
                    WPAM_Logger::log_debug('Buyer email data is missing so the plugin cannot check for own referrals', 2);
                }
            }
            //checking to see if "do not record zero amount commission" option is enabled
            if(get_option(WPAM_PluginConfig::$AffdoNotRecordZeroAmtCommission) == 1){              
                if($creditAmount <= 0){
                    WPAM_Logger::log_debug('The commission amount for this transaction is 0 or less so this will not be recorded.');
                    return;
                }
            }
            $creditAmount = apply_filters('wpam_credit_amount', $creditAmount, $args);
            if(isset($args['manual_comm_amount']) && is_numeric($args['manual_comm_amount'])){
                $creditAmount = round($args['manual_comm_amount'], 2);;  //override the commission amount if provided e.g. manual commission
            }
            $currency = WPAM_MoneyHelper::getCurrencyCode();
            $description = sprintf(__('Credit for sale of %s %s (PURCHASE LOG ID = %s)', 'affiliates-manager'), $amount, $currency, $txn_id);
            $query = "
            SELECT *
            FROM ".WPAM_TRANSACTIONS_TBL."
            WHERE referenceId = %s    
            ";
            $txn_record = $wpdb->get_row($wpdb->prepare($query, $txn_id));
            if($txn_record != null) {  //found a record
                WPAM_Logger::log_debug('Commission for this sale has already been awarded. PURCHASE LOG ID: '.$txn_id.', Purchase amount: '.$amount);        
            } 
            else {
                $table = WPAM_TRANSACTIONS_TBL;
                $data = array();
                $data['dateModified'] = (isset($args['date_modified']) && !empty($args['date_modified'])) ? $args['date_modified'] : date("Y-m-d H:i:s", time());
                $data['dateCreated'] = (isset($args['date_created']) && !empty($args['date_created'])) ? $args['date_created'] : date("Y-m-d H:i:s", time());
                $data['referenceId'] = $txn_id;
                $data['affiliateId'] = $affiliate->affiliateId;
                $data['type'] = 'credit';
                $data['description'] = $description;
                $data['amount'] = $creditAmount;
                $data['email'] = $buyer_email;
                $wpdb->insert( $table, $data);
                
                //
                $data['sale_amount'] = $amount;
                do_action('wpam_affiliate_commission_added', $data);
                /*
                if($strRefKey){
                    $db->getEventRepository()->quickInsert( time(), $binConverter->stringToBin( $strRefKey ), 'purchase' );
                }
                */
                //checking to see if Send Commission Notification to affiliate option is enabled
                if(get_option(WPAM_PluginConfig::$SendAffCommissionNotification) == 1){
                    //override from email & name
                    add_filter('wp_mail_from', 'wpam_filter_from_email');
                    add_filter('wp_mail_from_name', 'wpam_filter_from_name');
                    add_filter('wp_mail_content_type', 'wpam_filter_mail_content_type');
                    $address = $affiliate->email;
                    $subject = __("You just earned a commission!", 'affiliates-manager');               
                    $message = WPAM_MessageHelper::GetMessage('affiliate_commission_notification_email');
                    $aff_id = $affiliate->affiliateId;
                    $aff_first_name = $affiliate->firstName;
                    $aff_last_name = $affiliate->lastName; 
                    $aff_email = $affiliate->email;
                    $tags = array("{aff_id}","{aff_first_name}","{aff_last_name}","{aff_email}");
                    $vals = array($aff_id, $aff_first_name, $aff_last_name, $aff_email);
                    $body = str_replace($tags,$vals,$message);
                    WPAM_Logger::log_debug("Sending a commission notification to affiliate email ".$address);
                    $mail_sent = wp_mail( $address, $subject, $body );
                    if($mail_sent==true){
                        WPAM_Logger::log_debug("Email was sent successfully by WordPress");
                    }
                    else{
                        WPAM_Logger::log_debug("Email could not be sent by WordPress");
                    }
                    remove_filter('wp_mail_content_type', 'wpam_filter_mail_content_type');
                    remove_filter('wp_mail_from', 'wpam_filter_from_email');
                    remove_filter('wp_mail_from_name', 'wpam_filter_from_name');
                }
                //checking to see if Send Commission Notification to admin option is enabled
                if(get_option(WPAM_PluginConfig::$SendAdminAffCommissionNotification) == 1){
                    //override from email & name
                    add_filter('wp_mail_from', 'wpam_filter_from_email');
                    add_filter('wp_mail_from_name', 'wpam_filter_from_name');
                    add_filter('wp_mail_content_type', 'wpam_filter_mail_content_type');
                    $address = get_option(WPAM_PluginConfig::$AdminAffCommissionNotificationEmail);
                    if(!isset($address) || empty($address)){
                        $address = get_option('admin_email');
                    }
                    $subject = __("Affiliate commission notification", 'affiliates-manager');               
                    $message = WPAM_MessageHelper::GetMessage('admin_affiliate_commission_notification_email');
                    $aff_id = $affiliate->affiliateId;
                    $aff_first_name = $affiliate->firstName;
                    $aff_last_name = $affiliate->lastName; 
                    $aff_email = $affiliate->email;
                    $tags = array("{aff_id}","{aff_first_name}","{aff_last_name}","{aff_email}");
                    $vals = array($aff_id, $aff_first_name, $aff_last_name, $aff_email);
                    $body = str_replace($tags,$vals,$message);
                    WPAM_Logger::log_debug("Sending a commission notification to admin email ".$address);
                    $mail_sent = wp_mail( $address, $subject, $body );
                    if($mail_sent==true){
                        WPAM_Logger::log_debug("Email was sent successfully by WordPress");
                    }
                    else{
                        WPAM_Logger::log_debug("Email could not be sent by WordPress");
                    }
                    remove_filter('wp_mail_content_type', 'wpam_filter_mail_content_type');
                    remove_filter('wp_mail_from', 'wpam_filter_from_email');
                    remove_filter('wp_mail_from_name', 'wpam_filter_from_name');
                }
            }
        }
        else{
            WPAM_Logger::log_debug('Affiliate account is not active. This commission cannot be rewarded', 4);
        }
    }
    
    /*
     * Cancels the commission of a sale that was refunded, by adding a "refund" line of minus the commission.
     * For a partial refund, pass the total refunded so far and the amount paid: only that part of the commission is
     * cancelled (e.g. $50 refunded on a $250 sale cancels 20% of the commission). Refund lines already added for this
     * sale are taken into account, so a later refund only adds the rest, and the same refund sent again adds nothing.
     */
    public static function refund_commission($txn_id, $total_refunded = null, $payment_amount = null){
        WPAM_Logger::log_debug('Commission refund handler function has been invoked for PURCHASE LOG ID: '.$txn_id);
        global $wpdb;
        $table = WPAM_TRANSACTIONS_TBL;
        //find the commission record
        $query = "
        SELECT *
        FROM ".WPAM_TRANSACTIONS_TBL."
        WHERE referenceId = %s
        AND type = 'credit'
        ";
        $txn_record = $wpdb->get_row($wpdb->prepare($query, $txn_id));
        if($txn_record == null) {
            WPAM_Logger::log_debug('No commission record found for PURCHASE LOG ID: '.$txn_id.'. Commission cannot be refunded!');
            return;
        }
        //The part of the commission to cancel: all of it, unless a partial refund is given
        $part = 1;
        if(is_numeric($total_refunded) && is_numeric($payment_amount) && $payment_amount > 0 && $total_refunded < $payment_amount){
            $part = $total_refunded / $payment_amount;
        }
        $to_cancel = round($txn_record->amount * $part, 2);
        //The commission already cancelled for this sale
        $query = "
        SELECT SUM(amount)
        FROM ".WPAM_TRANSACTIONS_TBL."
        WHERE referenceId = %s
        AND amount < 0
        AND type = 'refund'
        ";
        $already_cancelled = round(-1 * (float) $wpdb->get_var($wpdb->prepare($query, $txn_id)), 2);
        $amount = round($to_cancel - $already_cancelled, 2);
        if($amount <= 0) {
            WPAM_Logger::log_debug('Commission for this sale has already been refunded. PURCHASE LOG ID: '.$txn_id);
            return;
        }
        $amount = number_format($amount, 2, '.', ''); //e.g. "62.50", as the commission is shown
        $currency = WPAM_MoneyHelper::getCurrencyCode();
        $description = sprintf(__('Refund of %s %s (PURCHASE LOG ID = %s)', 'affiliates-manager'), $amount, $currency, $txn_id);
        $data = array();
        $data['dateModified'] = date("Y-m-d H:i:s", time());
        $data['dateCreated'] = date("Y-m-d H:i:s", time());
        $data['referenceId'] = $txn_id;
        $data['affiliateId'] = $txn_record->affiliateId;
        $data['type'] = 'refund';
        $data['description'] = $description;
        $data['amount'] = '-'.$amount;
        if($wpdb->insert( $table, $data) === false) {
            WPAM_Logger::log_debug('Error! The commission refund ('.$amount.') could not be saved for PURCHASE LOG ID: '.$txn_id.'. Database error: '.$wpdb->last_error, 4);
            return;
        }
        WPAM_Logger::log_debug('Commission refunded ('.$amount.') for PURCHASE LOG ID: '.$txn_id.', Affiliate ID: '.$txn_record->affiliateId);
    }
    
    /*
     * Gets total transaction count for a given affiliate.
     * $args array requires at least 3 elements 
     * aff_id - affiliate ID
     * start_date - start date 
     * end_date - end date
     */
    public static function get_transaction_count($args){ 
        global $wpdb;
        $table = WPAM_TRANSACTIONS_TBL;
        $count = 0;
        $result = $wpdb->get_var( $wpdb->prepare( 
	"
		SELECT COUNT(*) 
		FROM $table 
		WHERE affiliateId = %d
                AND dateCreated >= %s
                AND dateCreated < %s
                AND type = 'credit'
	",
        $args['aff_id'],        
	$args['start_date'],
        $args['end_date']        
        ) );
        if($result != null){
            $count = $result;
        }
        return $count;
    }
    /*
     * Gets all time total transaction count for a given affiliate.
     * $args array requires only 1 element
     * aff_id - affiliate ID
     */
    public static function get_all_time_transaction_count($args){ //$args() at least requires affiliate ID
        global $wpdb;
        $table = WPAM_TRANSACTIONS_TBL;
        $count = 0;
        $result = $wpdb->get_var( $wpdb->prepare( 
	"
		SELECT COUNT(*) 
		FROM $table 
		WHERE affiliateId = %d
                AND type = 'credit'
	",
        $args['aff_id']               
        ) );
        if($result != null){
            $count = $result;
        }
        return $count;
    }
    
    /*
     * Gets total commission amount for a given affiliate.
     * $args array requires at least 3 elements 
     * aff_id - affiliate ID
     * start_date - start date 
     * end_date - end date
     */    
    public static function get_total_commission_amount($args){ 
        global $wpdb;
        $table = WPAM_TRANSACTIONS_TBL;
        $total_commission = $wpdb->get_var( $wpdb->prepare( 
	"
		SELECT COALESCE(SUM(IF(type = 'credit', amount, 0)),0) 
		FROM $table 
		WHERE affiliateId = %d
                AND dateCreated >= %s
                AND dateCreated < %s
	",
        $args['aff_id'],        
	$args['start_date'],
        $args['end_date']        
        ) );
        return $total_commission;
    }
}