<?php
/**
 * Plugin Name: WeArePlanet
 * Author: Planet Merchant Services Ltd
 * Text Domain: weareplanet
 * Domain Path: /languages/
 *
 * WeArePlanet
 * This plugin will add support for all WeArePlanet payments methods and connect the WeArePlanet servers to your WooCommerce webshop (https://www.weareplanet.com/).
 *
 * @category Class
 * @package  WeArePlanet
 * @author   Planet Merchant Services Ltd (https://www.weareplanet.com)
 * @license  http://www.apache.org/licenses/LICENSE-2.0 Apache Software License (ASL 2.0)
 */

defined( 'ABSPATH' ) || exit;

/**
 * Class WC_WeArePlanet_Order_Status_Adapter.
 *
 * This class handles the mapping and updating of order statuses in WooCommerce
 * based on the transaction statuses in WeArePlanet.
 */
class WC_WeArePlanet_Order_Status_Adapter
{
	/**
	 * Constants for WeArePlanet transaction statuses.
	 */
	const WEAREPLANET_STATUS_PENDING = 'pending';
	const WEAREPLANET_STATUS_CONFIRMED = 'confirmed';
	const WEAREPLANET_STATUS_PROCESSING = 'processing';
	const WEAREPLANET_STATUS_AUTHORIZED = 'authorized';
	const WEAREPLANET_STATUS_COMPLETED = 'completed';
	const WEAREPLANET_STATUS_FAILED = 'failed';
	const WEAREPLANET_STATUS_VOIDED = 'voided';
	const WEAREPLANET_STATUS_FULFILL = 'fulfill';
	const WEAREPLANET_STATUS_DECLINE = 'decline';

	const WEAREPLANET_ORDER_STATUS_MAPPING_PREFIX = 'weareplanet_order_status_mapping_';
	const WEAREPLANET_CUSTOM_ORDER_STATUS_PREFIX = 'weareplanet_custom_order_status_';
	/**
	 * Stores the status mappings loaded from the database.
	 *
	 * @var array $settings Array of status mappings loaded from the database.
	 */
	private $settings;

	/**
	 * WC_WeArePlanet_Order_Status_Adapter constructor.
	 *
	 * Initializes the settings and adds the filter for updating order status.
	 */
	public function __construct()
	{
		$this->initialize_filters();
		$this->initialize_status_mappings();
	}

	/**
	 * Initialise filters and actions.
	 */
	public static function init(): void
	{
		add_action( 'plugins_loaded', array( __CLASS__, 'register_weareplanet_service_status_adapter' ) );
	}

	/**
	 * Registers the WeArePlanet service status adapter.
	 *
	 * This function initializes an instance of the WC_WeArePlanet_Order_Status_Adapter class.
	 * It is hooked to the 'plugins_loaded' action to ensure it runs after all plugins are loaded.
	 *
	 * @return void
	 */
	public static function register_weareplanet_service_status_adapter(): void
	{
		new WC_WeArePlanet_Order_Status_Adapter();
	}

	/**
	 * Initialize filters.
	 */
	private function initialize_filters(): void
	{
		add_filter( 'weareplanet_default_order_status_mappings', array( $this, 'get_default_status_mappings' ) );
		add_filter( 'weareplanet_woocommerce_statuses', array( $this, 'get_all_woocommerce_statuses' ) );
		add_filter( 'weareplanet_order_statuses', array( $this, 'get_weareplanet_statuses' ) );
		add_filter( 'weareplanet_order_update_status', array( $this, 'update_order_status' ), 10, 5 );
		add_filter( 'weareplanet_wc_status_for_transaction', array( $this, 'get_wc_status_for_transaction' ), 10, 1 );
		add_filter( 'wc_order_statuses', array( $this, 'add_order_statuses' ) );
		add_filter( 'woocommerce_valid_order_statuses_for_payment', array( $this, 'valid_order_statuses_for_payment' ) );
		add_filter( 'woocommerce_payment_complete_order_status', array( $this, 'get_order_status_on_payment_complete' ), 10, 3 );

		//tests.
		// CPT-based orders.
		add_filter( 'bulk_actions-edit-shop_order', array($this, 'bulk_actions_shop_order'), 20, 1 );
		add_action( 'handle_bulk_actions-edit-shop_order', array($this, 'bulk_process_custom_status'), 20, 3 );
		// HPOS orders.
		add_filter( 'bulk_actions-woocommerce_page_wc-orders', array($this, 'bulk_actions_shop_order'), 20, 1 );
		add_filter( 'handle_bulk_actions-woocommerce_page_wc-orders', array($this, 'bulk_process_custom_status'), 20, 3 );
	}

	/**
	 * Modifies the bulk actions available in the WooCommerce orders list.
	 *
	 * @param array $bulk_actions The existing bulk actions.
	 * @return array The modified bulk actions.
	 */
	public function bulk_actions_shop_order( $bulk_actions ) {
		return $bulk_actions;
	}

	/**
	 * Handles custom bulk actions for WooCommerce orders.
	 *
	 * @param string $redirect The redirect URL after processing bulk actions.
	 * @param string $doaction The action being performed.
	 * @param array  $object_ids The IDs of the selected orders.
	 * @return string The modified redirect URL.
	 */
	public function bulk_process_custom_status( $redirect, $doaction, $object_ids ) {
		return $redirect;
	}

	/**
	 * Add order statuses.
	 *
	 * @param mixed $order_statuses order statuses.
	 * @return mixed
	 */
	public function add_order_statuses( $order_statuses ) {
		global $wpdb;

		$prefix = self::WEAREPLANET_CUSTOM_ORDER_STATUS_PREFIX . '%'; //We use % as wildcard for LIKE.
		$results = $wpdb->get_results( $wpdb->prepare( "SELECT option_name, option_value FROM {$wpdb->options} WHERE option_name LIKE %s", $prefix ), ARRAY_A );

		//Build the array with the desired structure.
		foreach ( $results as $row ) {
			$status_label = ucfirst( str_replace( array( 'wc-', '_' ), array( '', ' ' ), $row['option_value'] ) );
			// translators: %s represents the dynamically generated order status label.
			$order_statuses[ $row['option_value'] ] = _x( $status_label, 'Order status', 'woocommerce' ); // phpcs:ignore
		}

		return $order_statuses;
	}


	/**
	 * Add order statuses.
	 *
	 * @param mixed $order_statuses order statuses.
	 * @return mixed
	 */
	public function valid_order_statuses_for_payment( $order_statuses ) {
		$default_mappings = array(
			self::WEAREPLANET_STATUS_PENDING => 'wc-pending',
			self::WEAREPLANET_STATUS_CONFIRMED => 'wc-on-hold',
			self::WEAREPLANET_STATUS_PROCESSING => 'wc-on-hold',
			self::WEAREPLANET_STATUS_COMPLETED => 'wc-processing',
			self::WEAREPLANET_STATUS_FULFILL => 'wc-completed',
		);

		$order_statuses_without_prefix = array_map( function( $status ) {
			return str_replace( 'wc-', '', $status );
		}, array_values( $default_mappings ) );

		return array_merge($order_statuses, $order_statuses_without_prefix);
	}

	/**
	 * Loads the status mappings from the database or initializes them if not present.
	 */
	private function initialize_status_mappings(): void
	{
		$is_custom_mapping_enabled = WC_WeArePlanet_Helper::is_custom_status_mapping_enabled();
		$default_mappings = apply_filters( 'weareplanet_default_order_status_mappings', array() );

		if ( ! is_array( $default_mappings ) ) {
			$default_mappings = array();
		}

		$this->settings = array();

		foreach ( $default_mappings as $transaction_status => $default_order_status ) {
			$option_key = self::WEAREPLANET_ORDER_STATUS_MAPPING_PREFIX . $transaction_status;
			$stored_order_status = $this->ensure_status_has_prefix( get_option( $option_key, $default_order_status ) );

			if ( ! $is_custom_mapping_enabled ) {
				$stored_order_status = $this->ensure_status_has_prefix( $default_order_status );
			}

			$this->settings[] = array(
				'transaction_status' => $transaction_status,
				'order_status' => $stored_order_status,
			);
		}
	}

	/**
	 * Determines whether the provided order status belongs to the custom mapping set.
	 *
	 * @param string $order_status Order status, potentially prefixed with `wc-`.
	 * @return bool
	 */
	private function is_custom_order_status( string $order_status ): bool
	{
		return strpos( $order_status, 'wc-wearep-' ) === 0 || strpos( $order_status, 'wearep-' ) === 0;
	}

	/**
	 * Ensures a status slug contains the `wc-` prefix when required.
	 *
	 * @param string $status Status slug.
	 * @return string
	 */
	private function ensure_status_has_prefix( string $status ): string
	{
		if ( '' === $status ) {
			return $status;
		}

		return strpos( $status, 'wc-' ) === 0 ? $status : 'wc-' . $status;
	}

	/**
	 * Stores the default order status mappings in the database if they are not already set.
	 *
	 * This method is called from a migration process to ensure that WooCommerce order status
	 * mappings are initialized correctly. If no custom mappings exist, it saves the default
	 * values in the `wp_options` table using the option name pattern:
	 * `weareplanet_order_update_status_<status>`.
	 *
	 * @since 1.0.0
	 * @see apply_filters() Allows developers to modify default status mappings before storing them.
	 *
	 * @return void
	 */
	public function store_default_status_mappings_on_database(): void
	{
		$default_mappings = $this->get_default_status_mappings();

		foreach ( $default_mappings as $key => $value ) {
			$result = update_option( self::WEAREPLANET_ORDER_STATUS_MAPPING_PREFIX . $key, $value );
		}
	}

	/**
	 * Gets the default status mappings.
	 * WooCommerce statuses dont have a constant, so we use the woocommerce key.
	 *
	 * This method defines the default mappings between WeArePlanet transaction
	 * statuses and WooCommerce order statuses. These mappings are saved in the WordPress
	 * `wp_options` table as `weareplanet_order_status_mapping_<status>` when the
	 * plugin is installed or initialized.
	 *
	 * WooCommerce introduced the `OrderInternalStatus` constants in version 9.6.0.
 	 * To maintain compatibility with earlier versions, string values are used as a fallback.
	 * This is the interface to use in next versions Automattic\WooCommerce\Enums\OrderInternalStatus
	 *
	 * Example of saved options in `wp_options` (custom mapping enabled):
	 * - weareplanet_order_status_mapping_pending -> wc-pending -> OrderInternalStatus::PENDING
	 * - weareplanet_order_status_mapping_confirmed -> wc-on-hold -> OrderInternalStatus::ON_HOLD
	 * - weareplanet_order_status_mapping_processing -> wc-on-hold -> OrderInternalStatus::ON_HOLD
	 * - weareplanet_order_status_mapping_authorized -> wc-on-hold -> OrderInternalStatus::ON_HOLD
	 * - weareplanet_order_status_mapping_completed -> wc-wearep-waiting -> OrderInternalStatus::ON_HOLD
	 * - weareplanet_order_status_mapping_failed -> wc-failed -> OrderInternalStatus::FAILED
	 * - weareplanet_order_status_mapping_voided -> wc-cancelled or 'wc-refunded' -> OrderInternalStatus::CANCELLED
	 * - weareplanet_order_status_mapping_fulfill -> wc-processing -> OrderInternalStatus::PROCESSING
	 *
	 * Example when relying on WooCommerce core statuses:
	 * - weareplanet_order_status_mapping_processing -> wc-processing -> OrderInternalStatus::PROCESSING
	 * - weareplanet_order_status_mapping_completed -> wc-processing -> OrderInternalStatus::PROCESSING
	 *
	 * These defaults are used if no custom mappings are provided by the user.
	 *
	 * @return array
	 */
	public function get_default_status_mappings() : array
	{
		$default_mappings = array(
			self::WEAREPLANET_STATUS_PENDING => 'wc-pending',
			self::WEAREPLANET_STATUS_CONFIRMED => 'wc-on-hold',
			self::WEAREPLANET_STATUS_PROCESSING => 'wc-on-hold',
			self::WEAREPLANET_STATUS_AUTHORIZED => 'wc-on-hold',
			self::WEAREPLANET_STATUS_COMPLETED => 'wc-on-hold',
			self::WEAREPLANET_STATUS_FAILED => 'wc-failed',
			self::WEAREPLANET_STATUS_VOIDED => 'wc-cancelled',
			self::WEAREPLANET_STATUS_FULFILL => 'wc-processing',
			self::WEAREPLANET_STATUS_DECLINE => 'wc-cancelled',
		);

		if ( WC_WeArePlanet_Helper::is_custom_status_mapping_enabled() ) {
			$default_mappings[ self::WEAREPLANET_STATUS_COMPLETED ] = 'wc-wearep-waiting';
		}

		return $default_mappings;
	}

	/**
	 * Gets the legacy status mappings.
	 *
	 * Example of saved options in `wp_options`:
	 * - weareplanet_order_status_mapping_confirmed -> wearep-redirected
	 * - weareplanet_order_status_mapping_processing -> wearep-redirected
	 * - weareplanet_order_status_mapping_completed -> wearep-waiting
	 *
	 * These defaults are used if no custom mappings are provided by the user.
	 *
	 * @return array
	 */
	public function get_legacy_default_status_mappings() : array
	{
		return array(
			self::WEAREPLANET_STATUS_CONFIRMED => 'wearep-redirected',
			self::WEAREPLANET_STATUS_PROCESSING => 'wearep-redirected',
			self::WEAREPLANET_STATUS_COMPLETED => 'wearep-waiting',
		);
	}

	/**
	 * Gets the WooCommerce statuses.
	 *
	 * @return array
	 */
	public function get_all_woocommerce_statuses(): array
	{
		return wc_get_order_statuses();
	}

	/**
	 * Gets the WeArePlanet statuses.
	 *
	 * @return array
	 */
	public function get_weareplanet_statuses(): array
	{
		return array(
			self::WEAREPLANET_STATUS_PENDING => ucwords( self::WEAREPLANET_STATUS_PENDING ),
			self::WEAREPLANET_STATUS_CONFIRMED => ucwords( self::WEAREPLANET_STATUS_CONFIRMED ),
			self::WEAREPLANET_STATUS_PROCESSING => ucwords( self::WEAREPLANET_STATUS_PROCESSING ),
			self::WEAREPLANET_STATUS_AUTHORIZED => ucwords( self::WEAREPLANET_STATUS_AUTHORIZED ),
			self::WEAREPLANET_STATUS_COMPLETED => ucwords( self::WEAREPLANET_STATUS_COMPLETED ),
			self::WEAREPLANET_STATUS_FAILED => ucwords( self::WEAREPLANET_STATUS_FAILED ),
			self::WEAREPLANET_STATUS_VOIDED => ucwords( self::WEAREPLANET_STATUS_VOIDED ),
			self::WEAREPLANET_STATUS_FULFILL => ucwords( self::WEAREPLANET_STATUS_FULFILL ),
			self::WEAREPLANET_STATUS_DECLINE => ucwords( self::WEAREPLANET_STATUS_DECLINE ),
		);
	}

	/**
	 * Gets the WooCommerce status corresponding to a WeArePlanet status.
	 *
	 * @param string|null $status The WeArePlanet transaction status.
	 * @return string|null The corresponding WooCommerce order status or null if not found.
	 */
	private function map_weareplanet_status_to_woocommerce( ?string $status ): ?string
	{
		if ( null === $status || '' === $status ) {
			return $status;
		}

		if ( empty( $this->settings ) ) {
			return $status; //Return the current status if there are no mappings available.
		}

		//Search in 'transaction_status' first.
		foreach ( $this->settings as $setting ) {
			if ( $setting['transaction_status'] === strtolower( $status ) ) {
				return str_replace( 'wc-', '', $setting['order_status'] ); //Return the mapped WooCommerce order status.
			}
		}

		//Fallback to legacy mappings if no match was found.
		$legacy_mappings = $this->get_legacy_default_status_mappings();
		$transaction_status_key = array_search( $status, $legacy_mappings, true );

		if ( !empty( $transaction_status_key ) ) {
			foreach ( $this->settings as $setting ) {
				if ( $setting['transaction_status'] === $transaction_status_key ) {
					return str_replace( 'wc-', '', $setting['order_status'] ); //Return the mapped WooCommerce order status.
				}
			}
		}
		
		return str_replace( 'wc-', '', $status ); //Return legacy status or original (ensuring WC prefix is stripped).
	}

	/**
	 * Updates the status of a WooCommerce order based on the WeArePlanet status.
	 *
	 * @param WC_Order $order $order The WooCommerce order.
	 * @param string|null $status The WeArePlanet transaction status.
	 * @param string|null $default status The WeArePlanet transaction status by default.
	 * @param string $note Optional note to add when updating the status.
	 * @param @param bool $manual Whether this is a manual order status change.
	 */
	public function update_order_status( WC_Order $order, ?string $status, ?string $default_status, string $note = '', bool $manual = false ): void
	{
		// If status is empty.
		if ( $status === null && $order !== null ) {
			$transaction_info = WC_WeArePlanet_Entity_Transaction_Info::load_by_order_id( $order->get_id() );
			if ( $transaction_info ) {
				$status = $transaction_info->get_state();
			}
		}

		// Apply a pre-update filter to allow modifications before processing the status.
		$status = apply_filters( 'weareplanet_pre_order_update_status', $status, $order, $note, $manual );
		$change_setting_enabled = 'yes' === get_option( WooCommerce_WeArePlanet::WEAREPLANET_CK_CHANGE_ORDER_STATUS );
		$is_virtual_order = WC_WeArePlanet_Helper::is_order_virtual( $order );
		$should_force_completed = (
			$change_setting_enabled
			&& $is_virtual_order
			&& null !== $status
			&& strtolower( (string) $status ) === strtolower( \WeArePlanet\Sdk\Model\TransactionState::FULFILL )
		);

		// Preserve WooCommerce's completed status only once the transaction reaches FULFILL.
		$new_status = $should_force_completed ? 'completed' : $this->map_weareplanet_status_to_woocommerce( $status );

		if ( !empty( $new_status ) && !empty( $order ) ) {
			$order->update_status( $new_status, $note, $manual );
			// Apply a post-update filter to allow modifications after updating the status.
			apply_filters( 'weareplanet_post_order_update_status', $status, $order, $note, $manual );
		}
	}

	/**
	 * Listens for the WooCommerce payment complete order status filter
	 * and return the order status accordingly when the order is fulfill.
	 *
	 * @param string $status The default order status (processing or completed).
	 * @param int $order_id The WooCommerce order ID.
	 * @param WC_Order $order The WooCommerce order object.
	 * @return string The updated order status.
	 */
	public function get_order_status_on_payment_complete( string $status, int $order_id, WC_Order $order ): string
	{
		$change_setting_enabled = 'yes' === get_option( WooCommerce_WeArePlanet::WEAREPLANET_CK_CHANGE_ORDER_STATUS );
		$is_virtual_order = WC_WeArePlanet_Helper::is_order_virtual( $order );
		$transaction_info = WC_WeArePlanet_Entity_Transaction_Info::load_by_order_id( $order_id );
		$is_fulfilled = $transaction_info
			&& strtolower( $transaction_info->get_state() ) === strtolower( \WeArePlanet\Sdk\Model\TransactionState::FULFILL );

		// Preserve WooCommerce's completed status only when the transaction is fulfilled.
		if ( $change_setting_enabled && $is_virtual_order && $is_fulfilled ) {
			return self::WEAREPLANET_STATUS_COMPLETED;
		}
		
		// Check if the transaction status is mapped in WooCommerce
		$mapped_status = $this->map_weareplanet_status_to_woocommerce( \WeArePlanet\Sdk\Model\TransactionState::FULFILL );

		if ( ! empty( $mapped_status ) ) {
			return $mapped_status; // Return the mapped WooCommerce status
		}

		return $status; // Return the default status if no mapping exists.
	}

	/**
	 * Retrieves the mapped WooCommerce order status for a given WeArePlanet transaction status.
	 *
	 * This method fetches the corresponding WooCommerce order status that has been mapped
	 * to a transaction status from the WeArePlanet portal. The mapping is stored
	 * in the `wp_options` table using the naming pattern:
	 * `weareplanet_order_status_mapping_<transaction_status>`.
	 *
	 * Expected values for `$weareplanet_status`: Pending, Confirmed, Processing, Failed, Authorized, Voided, Completed, Fulfill, Decline.
	 *
	 * @since 1.0.0
	 * @param string|null $weareplanet_status The transaction status from WeArePlanet.
	 * @return string|null The mapped WooCommerce order status, or null if no mapping is found.
	 */
	public function get_wc_status_for_transaction( ?string $weareplanet_status ): ?string
	{
		if ( null === $weareplanet_status || '' === $weareplanet_status ) {
			return null;
		}

		$status = get_option( self::WEAREPLANET_ORDER_STATUS_MAPPING_PREFIX . strtolower( $weareplanet_status ), null );
		if ( ! WC_WeArePlanet_Helper::is_custom_status_mapping_enabled() && $status && $this->is_custom_order_status( $status ) ) {
			$default_mappings = $this->get_default_status_mappings();
			$status_key = strtolower( $weareplanet_status );
			$status = $default_mappings[ $status_key ] ?? $status;
		}
		return is_null( $status ) ? null : str_replace( 'wc-', '', $status );
	}
}

WC_WeArePlanet_Order_Status_Adapter::init();
