"cart" field add to mutation output.

Author: kidunot89

includes/data/mutation/class-cart-mutation.php

@@ -14,6 +14,25 @@
  * 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 = \WC()->cart;
+				}
+				return $cart;
+			},
+		);
+	}
+
 	/**
 	 * Returns a cart item.
 	 *

includes/mutation/class-cart-add-fee.php

@@ -75,6 +75,7 @@ public static function get_output_fields() {
 					return $fees[ $payload['id'] ];
 				},
 			),
+			'cart'    => Cart_Mutation::get_cart_field( true ),
 		);
 	}
 

includes/mutation/class-cart-add-item.php

@@ -80,6 +80,7 @@ public static function get_output_fields() {
 					return $item;
 				},
 			),
+			'cart'     => Cart_Mutation::get_cart_field( true ),
 		);
 	}
 

includes/mutation/class-cart-apply-coupon.php

@@ -56,12 +56,7 @@ public static function get_input_fields() {
 	 */
 	public static function get_output_fields() {
 		return array(
-			'cart' => array(
-				'type'    => 'Cart',
-				'resolve' => function ( $payload ) {
-					return $payload['cart'];
-				},
-			),
+			'cart' => Cart_Mutation::get_cart_field(),
 		);
 	}
 

includes/mutation/class-cart-empty.php

@@ -40,12 +40,7 @@ public static function register_mutation() {
 	 */
 	public static function get_output_fields() {
 		return array(
-			'cart' => array(
-				'type'    => 'Cart',
-				'resolve' => function ( $payload ) {
-					return $payload['cart'];
-				},
-			),
+			'cart' => Cart_Mutation::get_cart_field(),
 		);
 	}
 

includes/mutation/class-cart-remove-coupons.php

@@ -56,12 +56,7 @@ public static function get_input_fields() {
 	 */
 	public static function get_output_fields() {
 		return array(
-			'cart' => array(
-				'type'    => 'Cart',
-				'resolve' => function ( $payload ) {
-					return $payload['cart'];
-				},
-			),
+			'cart' => Cart_Mutation::get_cart_field(),
 		);
 	}
 

includes/mutation/class-cart-remove-items.php

@@ -66,6 +66,7 @@ public static function get_output_fields() {
 					return $payload['items'];
 				},
 			),
+			'cart'      => Cart_Mutation::get_cart_field( true ),
 		);
 	}
 

includes/mutation/class-cart-update-item-quantities.php

@@ -82,6 +82,7 @@ public static function get_output_fields() {
 					return array_merge( $updated, $payload['removed'] );
 				},
 			),
+			'cart'    => Cart_Mutation::get_cart_field( true ),
 		);
 	}
 

includes/mutation/class-cart-update-shipping-method.php

@@ -53,12 +53,7 @@ public static function get_input_fields() {
 	 */
 	public static function get_output_fields() {
 		return array(
-			'cart' => array(
-				'type'    => 'Cart',
-				'resolve' => function () {
-					return \WC()->cart;
-				},
-			),
+			'cart' => Cart_Mutation::get_cart_field( true ),
 		);
 	}