wp-graphql-woocommerce/includes/data/mutation/class-cart-mutation.php

<?php
/**
 * Defines helper functions for executing mutations related to the cart.
 *
 * @package WPGraphQL\WooCommerce\Data\Mutation
 * @since 0.1.0
 */

namespace WPGraphQL\WooCommerce\Data\Mutation;

use GraphQL\Error\UserError;
use WPGraphQL\WooCommerce\Data\Factory;

/**
 * Class - Cart_Mutation
 */
class Cart_Mutation {
	/**
	 * Retrieve `cart` output field defintion
	 *
	 * @param bool $fallback  Should cart be retrieved, if not provided in payload.
	 * @return array
	 */
	public static function get_cart_field( $fallback = false ) {
		return array(
			'type'    => 'Cart',
			'resolve' => function ( $payload ) use ( $fallback ) {
				$cart = ! empty( $payload['cart'] ) ? $payload['cart'] : null;
				if ( is_null( $cart ) && $fallback ) {
					$cart = Factory::resolve_cart();
				}
				return $cart;
			},
		);
	}

	/**
	 * Returns a cart item.
	 *
	 * @param array       $input   Input data describing cart item.
	 * @param AppContext  $context AppContext instance.
	 * @param ResolveInfo $info    Query info.
	 *
	 * @return array
	 */
	public static function prepare_cart_item( $input, $context, $info ) {
		$cart_item_args   = array( $input['productId'] );
		$cart_item_args[] = ! empty( $input['quantity'] ) ? $input['quantity'] : 1;
		$cart_item_args[] = ! empty( $input['variationId'] ) ? $input['variationId'] : 0;
		$cart_item_args[] = ! empty( $input['variation'] ) ? self::prepare_attributes( $input['productId'], $input['variation'] ) : array();
		$cart_item_args[] = ! empty( $input['extraData'] )
			? json_decode( $input['extraData'], true )
			: array();

		return apply_filters( 'graphql_woocommerce_new_cart_item_data', $cart_item_args, $input, $context, $info );
	}

	/**
	 * Processes the provided variation attributes data for the cart.
	 *
	 * @param array $variation_data  Variation data.
	 * @return array
	 */
	private static function prepare_attributes( $product_id, array $variation_data = array() ) {
		$product         = wc_get_product( $product_id );
		$attribute_names = array_keys( $product->get_attributes() );

		$attributes = array();
		foreach( $variation_data as $attribute ) {
			$attribute_name  = $attribute['attributeName'];
			if ( in_array( "pa_{$attribute_name}", $attribute_names, true ) ) {
				$attribute_name = "pa_{$attribute_name}";
			} elseif ( ! in_array( $attribute_name, $attribute_names, true ) ) {
				throw new UserError(
					sprintf(
						__( '%s is not a valid attribute of the product: %s.', 'wp-graphql-woocommerce' ),
						$attribute_name,
						$product->get_name()
					)
				);
			}

			$attribute_value = ! empty( $attribute['attributeValue'] ) ? $attribute['attributeValue'] : '';
			$attribute_key = "attribute_{$attribute_name}";

			$attributes[ $attribute_key ] = $attribute_value;
		}

		return $attributes;
	}

	/**
	 * Returns an array of cart items.
	 *
	 * @param array       $input    Input data describing cart items.
	 * @param AppContext  $context  AppContext instance.
	 * @param ResolveInfo $info     Query info.
	 * @param string      $mutation Mutation type.
	 *
	 * @return array
	 * @throws UserError Cart item not found message.
	 */
	public static function retrieve_cart_items( $input, $context, $info, $mutation = '' ) {
		if ( ! empty( $input['all'] ) && $input['all'] ) {
			$items = array_values( \WC()->cart->get_cart() );
		}

		if ( ! empty( $input['keys'] ) && ! isset( $items ) ) {
			$items = array();
			foreach ( $input['keys'] as $key ) {
				$item = \WC()->cart->get_cart_item( $key );
				if ( empty( $item ) ) {
					/* translators: Cart item not found message */
					throw new UserError( sprintf( __( 'No cart item found with the key: %s', 'wp-graphql-woocommerce' ), $key ) );
				}
				$items[] = $item;
			}
		}

		return apply_filters( 'graphql_woocommerce_retrieve_cart_items', $items, $input, $context, $info, $mutation );
	}

	/**
	 * Return array of data to be when defining a cart fee.
	 *
	 * @param array       $input   input data describing cart item.
	 * @param AppContext  $context AppContext instance.
	 * @param ResolveInfo $info    query info.
	 *
	 * @return array
	 */
	public static function prepare_cart_fee( $input, $context, $info ) {
		$cart_item_args = array(
			$input['name'],
			$input['amount'],
			! empty( $input['taxable'] ) ? $input['taxable'] : false,
			! empty( $input['taxClass'] ) ? $input['taxClass'] : '',
		);

		return apply_filters( 'graphql_woocommerce_new_cart_fee_data', $cart_item_args, $input, $context, $info );
	}

	/**
	 * Validate CartItemQuantityInput item.
	 *
	 * @param array $item  CartItemQuantityInput object.
	 *
	 * @return boolean
	 */
	public static function item_is_valid( array $item ) {
		if ( empty( $item['key'] ) ) {
			return false;
		}
		if ( ! isset( $item['quantity'] ) || ! is_numeric( $item['quantity'] ) ) {
			return false;
		}
		return true;
	}

	/**
	 * Checks for errors thrown by the QL_Session_Handler during session token validation.
	 *
	 * @throws UserError If GRAPHQL_DEBUG is set to true and errors found.
	 */
	public static function check_session_token() {
		$token_invalid = apply_filters( 'graphql_woocommerce_session_token_errors', null );
		if ( $token_invalid ) {
			throw new UserError( $token_invalid );
		}
	}
}