wp-graphql-woocommerce/includes/mutation/class-cart-update-item-quantities.php

<?php
/**
 * Mutation - updateItemQuantities
 *
 * Registers mutation for updating cart item quantities.
 *
 * @package WPGraphQL\WooCommerce\Mutation
 * @since 0.1.0
 */

namespace WPGraphQL\WooCommerce\Mutation;

use Exception;
use GraphQL\Error\UserError;
use GraphQL\Type\Definition\ResolveInfo;
use WPGraphQL\AppContext;
use WPGraphQL\WooCommerce\Data\Mutation\Cart_Mutation;

/**
 * Class - Cart_Update_Item_Quantities
 */
class Cart_Update_Item_Quantities {
	/**
	 * Registers mutation
	 *
	 * @return void
	 */
	public static function register_mutation() {
		register_graphql_mutation(
			'updateItemQuantities',
			[
				'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 [
			'items' => [
				'type'        => [ 'list_of' => 'CartItemQuantityInput' ],
				'description' => static function () {
					return __( 'Cart item being updated', 'graphql-for-ecommerce' );
				},
			],
		];
	}

	/**
	 * Defines the mutation output field configuration
	 *
	 * @return array
	 */
	public static function get_output_fields() {
		return [
			'updated' => [
				'type'    => [ 'list_of' => 'CartItem' ],
				'resolve' => static function ( $payload ) {
					$items = [];
					foreach ( $payload['updated'] as $key ) {
						$items[] = WC()->cart->get_cart_item( $key );
					}

					return $items;
				},
			],
			'removed' => [
				'type'    => [ 'list_of' => 'CartItem' ],
				'resolve' => static function ( $payload ) {
					return $payload['removed'];
				},
			],
			'items'   => [
				'type'    => [ 'list_of' => 'CartItem' ],
				'resolve' => static function ( $payload ) {
					$updated = [];
					foreach ( $payload['updated'] as $key ) {
						$updated[] = \WC()->cart->get_cart_item( $key );
					}

					return array_merge( $updated, $payload['removed'] );
				},
			],
			'cart'    => Cart_Mutation::get_cart_field( true ),
		];
	}

	/**
	 * Defines the mutation data modification closure.
	 *
	 * @return callable
	 */
	public static function mutate_and_get_payload() {
		return static function ( $input, AppContext $context, ResolveInfo $info ) {
			Cart_Mutation::check_session_token();

			// Confirm "items" exists.
			if ( empty( $input['items'] ) ) {
				throw new UserError( __( 'No item data provided', 'graphql-for-ecommerce' ) );
			}

			// Confirm "items" is value.
			if ( ! is_array( $input['items'] ) ) {
				throw new UserError( __( 'Provided "items" invalid', 'graphql-for-ecommerce' ) );
			}

			do_action( 'graphql_woocommerce_before_set_item_quantities', $input['items'], $input, $context, $info );

			// Update quantities. If quantity set to 0, the items in removed.
			$removed       = [];
			$updated       = [];
			$removed_items = [];
			foreach ( $input['items'] as $item ) {
				if ( Cart_Mutation::item_is_valid( $item ) ) {
					$key      = $item['key'];
					$quantity = $item['quantity'];
					if ( 0 === $quantity ) {
						$removed_item    = \WC()->cart->get_cart_item( $key );
						$removed_items[] = $removed_item;
						do_action( 'graphql_woocommerce_before_remove_item', $removed_item, 'update_quantity', $input, $context, $info );
						$removed[ $key ] = \WC()->cart->remove_cart_item( $key );
						do_action( 'graphql_woocommerce_after_remove_item', $removed_item, 'update_quantity', $input, $context, $info );
						continue;
					}

					do_action( 'graphql_woocommerce_before_set_item_quantity', \WC()->cart->get_cart_item( $key ), $input, $context, $info );
					$updated[ $key ] = \WC()->cart->set_quantity( $key, $quantity, true );
					do_action( 'graphql_woocommerce_after_set_item_quantity', \WC()->cart->get_cart_item( $key ), $input, $context, $info );
				}
			}

			// Throw failed.
			try {
				$errors = array_keys(
					array_filter(
						array_merge( $removed, $updated ),
						static function ( $value ) {
							return ! $value;
						}
					)
				);
				if ( 0 < count( $errors ) ) {
					throw new Exception(
						sprintf(
							/* translators: %s: Cart item keys */
							__( 'Cart items identified with keys %s failed to update', 'graphql-for-ecommerce' ),
							implode( ', ', $errors )
						)
					);
				}
			} catch ( \Throwable $e ) {
				throw new UserError( $e->getMessage() );
			}//end try

			do_action(
				'graphql_woocommerce_before_set_item_quantities',
				array_keys( $updated ),
				array_keys( $removed ),
				$input,
				$context,
				$info
			);

			do_action( 'woographql_update_session', true );

			return [
				'removed' => $removed_items,
				'updated' => array_keys( $updated ),
			];
		};
	}
}