<?php
if ( ! defined( 'ABSPATH' ) ) {
	exit; // Exit if accessed directly
}

/**
 * Refund AfterPay invoice
 *
 * Check if refund is possible, then process it. Currently only supports RefundFull.
 *
 * @class WC_AfterPay_Refund
 * @version 1.0.0
 * @package WC_Gateway_AfterPay/Classes
 * @category Class
 * @author Krokedil
 */
class WC_AfterPay_Refund {

	/** @var int */
	private $order_id = '';

	/** @var bool */
	private $testmode = false;

	/**
	 * WC_AfterPay_Refund constructor.
	 */
	public function __construct() {
		$afterpay_settings = get_option( 'woocommerce_afterpay_invoice_settings' );
		$this->testmode = 'yes' == $afterpay_settings['testmode'] ? true : false;
	}

	/**
	 * Grab AfterPay reservation ID.
	 *
	 * @return string
	 */
	public function get_reservation_id() {
		return get_post_meta( $this->order_id, '_afterpay_reservation_id', true );
	}

	/**
	 * Get payment method settings.
	 *
	 * @return array
	 */
	public function get_payment_method_settings() {
		$order                = wc_get_order( $this->order_id );
		$order_payment_method = $order->payment_method;

		$payment_method_settings = get_option( 'woocommerce_' . $order_payment_method . '_settings' );
		return $payment_method_settings;
	}

	/**
	 * Check if order was created using one of AfterPay's payment options.
	 *
	 * @return boolean
	 */
	public function check_if_afterpay_order() {
		$order                = wc_get_order( $this->order_id );
		$order_payment_method = $order->payment_method;

		if ( strpos( $order_payment_method, 'afterpay' ) !== false ) {
			return true;
		}

		return false;
	}

	/**
	 * Process refund.
	 *
	 * @param $order_id
	 * @return boolean
	 */
	public function refund_invoice( $order_id, $amount = null, $reason = '' ) {
		$this->order_id = $order_id;
		$order = wc_get_order( $this->order_id );

		// If this order wasn't created using an AfterPay payment method, bail.
		if ( ! $this->check_if_afterpay_order() ) {
			return;
		}

		// Get settings for payment method used to create this order.
		$payment_method_settings = $this->get_payment_method_settings();

		$order_maintenance_endpoint = $this->testmode ? ARVATO_ORDER_MAINTENANCE_TEST : ARVATO_ORDER_MAINTENANCE_LIVE;

		// Check if logging is enabled
		$this->log_enabled = $payment_method_settings['debug'];

		$refund_args = array(
			'User'       => array(
				'ClientID' => $payment_method_settings['client_id_' . strtolower($order->billing_country)],
				'Username' => $payment_method_settings['username_' . strtolower($order->billing_country)],
				'Password' => $payment_method_settings['password_' . strtolower($order->billing_country)]
			),
			'ReservationID' => $this->get_reservation_id(),
			'InvoiceNumber' => $order->get_transaction_id(),
		);

		$soap_client = new SoapClient( $order_maintenance_endpoint );

		try {
			if ( $amount != $order->get_total() ) {
				$refund_args['OrderDetails']['Amount']            = $amount;
				$refund_args['OrderDetails']['OrderNo']           = $order->get_order_number();
				$refund_args['OrderDetails']['CurrencyCode']      = $order->get_order_currency();
				$refund_args['OrderDetails']['OrderChannelType']  = 'Internet';
				$refund_args['OrderDetails']['OrderDeliveryType'] = 'Normal';

				$response = $soap_client->RefundPartial( $refund_args );
			} else {
				$refund_args['OrderNo'] = $order->get_order_number();
				$response               = $soap_client->RefundFull( $refund_args );
			}

			if ( $response->IsSuccess ) {
				// Add time stamp, used to prevent duplicate cancellations for the same order.
				update_post_meta( $this->order_id, '_afterpay_invoice_refunded', current_time( 'mysql' ) );
				$order->add_order_note( __( 'AfterPay refund was successfully processed.', 'woocommerce-gateway-afterpay' ) );

				return $response;
			} else {
				$order->add_order_note( __( 'AfterPay refund could not be processed.', 'woocommerce-gateway-afterpay' ) );
				WC_Gateway_AfterPay_Factory::log( 'Refund failed.' );
				return new WP_Error( 'afterpay-refund', __( 'Refund failed.', 'woocommerce-gateway-afterpay' ) );
			}
		} catch ( Exception $e ) {
			WC_Gateway_AfterPay_Factory::log( $e->getMessage() );
			echo '<div class="woocommerce-error">';
			echo $e->getMessage();
			echo '</div>';
		}
	}

}
$wc_afterpay_refund = new WC_AfterPay_Refund;