wp-graphql-woocommerce/includes/mutation/class-order-delete-items.php

<?php
/**
 * Mutation - deleteOrderItems
 *
 * Registers mutation for delete an items from an order.
 *
 * @package WPGraphQL\WooCommerce\Mutation
 * @since 0.2.0
 */

namespace WPGraphQL\WooCommerce\Mutation;

use GraphQL\Error\UserError;
use GraphQL\Type\Definition\ResolveInfo;
use WPGraphQL\AppContext;
use WPGraphQL\Utils\Utils;
use WPGraphQL\WooCommerce\Data\Mutation\Order_Mutation;
use WPGraphQL\WooCommerce\Model\Order;

/**
 * Class Order_Delete_Items
 */
class Order_Delete_Items {
	/**
	 * Registers mutation
	 *
	 * @return void
	 */
	public static function register_mutation() {
		register_graphql_mutation(
			'deleteOrderItems',
			[
				'inputFields'         => self::get_input_fields(),
				'outputFields'        => self::get_output_fields(),
				'mutateAndGetPayload' => self::mutate_and_get_payload(),
			]
		);
	}

	/**
	 * Defines the mutation input field configuration
	 *
	 * @return array
	 */
	public static function get_input_fields() {
		return array_merge(
			[
				'id'      => [
					'type'        => 'ID',
					'description' => __( 'Database ID or global ID of the order', 'wp-graphql-woocommerce' ),
				],
				'orderId' => [
					'type'              => 'Int',
					'description'       => __( 'Order WP ID', 'wp-graphql-woocommerce' ),
					'deprecationReason' => __( 'Use "id" field instead.', 'wp-graphql-woocommerce' ),
				],
				'itemIds' => [
					'type'        => [ 'list_of' => 'Int' ],
					'description' => __( 'ID Order items being deleted', 'wp-graphql-woocommerce' ),
				],
			]
		);
	}

	/**
	 * Defines the mutation output field configuration
	 *
	 * @return array
	 */
	public static function get_output_fields() {
		return [
			'order' => [
				'type'    => 'Order',
				'resolve' => static function ( $payload ) {
					return $payload['order'];
				},
			],
		];
	}

	/**
	 * Defines the mutation data modification closure.
	 *
	 * @return callable
	 */
	public static function mutate_and_get_payload() {
		return static function ( $input, AppContext $context, ResolveInfo $info ) {
			// Retrieve order ID.
			$order_id = null;
			if ( ! empty( $input['id'] ) ) {
				$order_id = Utils::get_database_id_from_id( $input['id'] );
			} elseif ( ! empty( $input['orderId'] ) ) {
				$order_id = absint( $input['orderId'] );
			} else {
				throw new UserError( __( 'Order ID provided is missing or invalid. Please check input and try again.', 'wp-graphql-woocommerce' ) );
			}

			if ( ! $order_id ) {
				throw new UserError( __( 'Order ID provided is invalid. Please check input and try again.', 'wp-graphql-woocommerce' ) );
			}

			// Check if authorized to delete items on this order.
			if ( ! Order_Mutation::authorized( $input, $context, $info, 'delete-items', $order_id ) ) {
				throw new UserError( __( 'User does not have the capabilities necessary to delete order items.', 'wp-graphql-woocommerce' ) );
			}

			// Confirm item IDs.
			if ( empty( $input['itemIds'] ) ) {
				throw new UserError( __( 'No item IDs provided.', 'wp-graphql-woocommerce' ) );
			} elseif ( ! is_array( $input['itemIds'] ) ) {
				throw new UserError( __( 'The "itemIds" provided is invalid', 'wp-graphql-woocommerce' ) );
			}
			$ids = $input['itemIds'];

			// Get Order model instance for output.
			/**
			 * Order model instance.
			 *
			 * @var \WC_Order $order
			 */
			$order = new Order( $order_id );

			// Cache items to prevent null value errors.
			// @codingStandardsIgnoreStart
			$order->get_downloadable_items();
			$order->get_items();
			$order->get_items( 'fee' );
			$order->get_items( 'shipping' );
			$order->get_items( 'tax' );
			$order->get_items( 'coupon' );
			// @codingStandardsIgnoreEnd.

			/**
			 * Working order model instance.
			 *
			 * @var \WC_Order $working_order
			 */
			$working_order = new Order( $order_id );

			/**
			 * Action called before order is deleted.
			 *
			 * @param array           $item_ids  Order item IDs of items being deleted.
			 * @param \WC_Order|\WPGraphQL\WooCommerce\Model\Order $order     Order model instance.
			 * @param array           $input     Input data describing order.
			 * @param \WPGraphQL\AppContext      $context   Request AppContext instance.
			 * @param \GraphQL\Type\Definition\ResolveInfo     $info      Request ResolveInfo instance.
			 */
			do_action( 'graphql_woocommerce_before_order_items_delete', $ids, $working_order, $input, $context, $info );

			// Delete order.
			$errors = '';
			foreach ( $ids as $id ) {
				$working_order->remove_item( $id );
			}
			$working_order->save();

			/**
			 * Action called before order is deleted.
			 *
			 * @param array           $item_ids  Order item IDs of items being deleted.
			 * @param \WC_Order|\WPGraphQL\WooCommerce\Model\Order $order     Order model instance.
			 * @param array           $input     Input data describing order
			 * @param \WPGraphQL\AppContext      $context   Request AppContext instance.
			 * @param \GraphQL\Type\Definition\ResolveInfo     $info      Request ResolveInfo instance.
			 */
			do_action( 'graphql_woocommerce_after_order_delete', $ids, $working_order, $input, $context, $info );

			return [ 'order' => $order ];
		};
	}
}