<?php
namespace AISP_OTOOLS;

if ( ! defined( 'ABSPATH' ) ) {
	exit;
}

class Admin {

		public function __construct() {
		$allow_admin_pay = get_option( 'aisp_otools_allow_on_hold_payment', 'no' );

		if ( 'yes' === $allow_admin_pay ) {
			add_filter( 'woocommerce_order_needs_payment', [ $this, 'force_admin_can_pay' ], 999, 2 );
			add_filter( 'user_has_cap', [ $this, 'grant_admin_pay_cap' ], 10, 3 );
			add_action( 'wp', [ $this, 'bypass_ownership_on_order_pay' ], 9 );
			add_filter( 'woocommerce_valid_order_statuses_for_payment', [ $this, 'maybe_allow_admin_payment_statuses' ], 10, 2 );
			
		}
	}

	public function force_admin_can_pay( $needs_payment, $order ) {
		if ( ! $order instanceof \WC_Order ) {
			return $needs_payment;
		}

		$allow_payment = get_option( 'aisp_otools_allow_on_hold_payment', 'no' );
		if ( 'yes' !== $allow_payment ) {
			return $needs_payment;
		}

		if ( current_user_can( 'manage_woocommerce' ) ) {
			$status = $order->get_status();
			$total  = (float) $order->get_total();

			if ( in_array( $status, [ 'pending', 'on-hold', 'failed' ], true ) && $total > 0 ) {
				return true;
			}
		}

		return $needs_payment;
	}

	public function grant_admin_pay_cap( $allcaps, $caps, $args ) {
		if ( isset( $args[0] ) && 'pay_for_order' === $args[0] && current_user_can( 'manage_woocommerce' ) ) {
			$allcaps['pay_for_order'] = true;
		}
		return $allcaps;
	}

	public function bypass_ownership_on_order_pay() {
		if ( is_wc_endpoint_url( 'order-pay' ) && is_user_logged_in() && current_user_can( 'manage_woocommerce' ) ) {
			remove_action( 'woocommerce_order_pay', 'wc_order_is_valid_for_payment_check', 10 );
		}
	}

	
	public function maybe_allow_admin_payment_statuses( $statuses, $order ) {
		$allow_payment = get_option( 'aisp_otools_allow_on_hold_payment', 'no' );

		if ( 'yes' !== $allow_payment ) {
			return $statuses;
		}

		$statuses[] = 'on-hold';
		$statuses[] = 'pending';
		$statuses[] = 'failed';

		return array_unique( $statuses );
	}
	
}