"*product" added to "RootQuery"

Author: kidunot89

includes/type/object/class-product-types.php

@@ -212,6 +212,9 @@ private static function register_simple_product_type() {
 				),
 			)
 		);
+
+		// Register "simpleProduct" query.
+		self::register_product_query( 'simple' );
 	}
 
 	/**
@@ -231,6 +234,9 @@ private static function register_variable_product_type() {
 				),
 			)
 		);
+
+		// Register "variableProduct" query.
+		self::register_product_query( 'variable' );
 	}
 
 	/**
@@ -258,6 +264,9 @@ private static function register_external_product_type() {
 				),
 			)
 		);
+
+		// Register "externalProduct" query.
+		self::register_product_query( 'external' );
 	}
 
 	/**
@@ -284,5 +293,81 @@ public static function register_group_product_type() {
 				),
 			)
 		);
+
+		// Register "groupProduct" query.
+		self::register_product_query( 'group' );
+	}
+
+	/**
+	 * Register product query
+	 *
+	 * @param string $type  Product type.
+	 */
+	private static function register_product_query( $type ) {
+		$field_name = "{$type}Product";
+		$type_name  = ucfirst( $type ) . 'Product';
+		register_graphql_field(
+			'RootQuery',
+			$field_name,
+			array(
+				'type'        => $type_name,
+				'description' => __( 'A simple product object', 'wp-graphql-woocommerce' ),
+				'args'        => array(
+					'id'        => array(
+						'type'        => 'ID',
+						'description' => __( 'Get the product by its global ID', 'wp-graphql-woocommerce' ),
+					),
+					'productId' => array(
+						'type'        => 'Int',
+						'description' => __( 'Get the product by its database ID', 'wp-graphql-woocommerce' ),
+					),
+					'slug'      => array(
+						'type'        => 'String',
+						'description' => __( 'Get the product by its slug', 'wp-graphql-woocommerce' ),
+					),
+					'sku'       => array(
+						'type'        => 'String',
+						'description' => __( 'Get the product by its sku', 'wp-graphql-woocommerce' ),
+					),
+				),
+				'resolve'     => function ( $source, array $args, AppContext $context, ResolveInfo $info ) use ( $type ) {
+					$product_id = 0;
+					$id_type = '';
+					if ( ! empty( $args['id'] ) ) {
+						$id_components = Relay::fromGlobalId( $args['id'] );
+						if ( empty( $id_components['id'] ) || empty( $id_components['type'] ) ) {
+							throw new UserError( __( 'The "id" is invalid', 'wp-graphql-woocommerce' ) );
+						}
+						$product_id = absint( $id_components['id'] );
+						$id_type = 'ID';
+					} elseif ( ! empty( $args['productId'] ) ) {
+						$product_id = absint( $args['productId'] );
+						$id_type = 'product ID';
+					} elseif ( ! empty( $args['slug'] ) ) {
+						$post = get_page_by_path( $args['slug'], OBJECT, 'product' );
+						$product_id = ! empty( $post ) ? absint( $post->ID ) : 0;
+						$id_type = 'slug';
+					} elseif ( ! empty( $args['sku'] ) ) {
+						$product_id = \wc_get_product_id_by_sku( $args['sku'] );
+						$id_type = 'sku';
+					}
+
+					if ( empty( $product_id ) ) {
+						/* translators: %1$s: ID type, %2$s: ID value */
+						throw new UserError( sprintf( __( 'No product ID was found corresponding to the %1$s: %2$s' ), $id_type, $product_id ) );
+					} elseif ( \WC()->product_factory->get_product_type( $product_id ) !== $type ) {
+						/* translators: Invalid product type message %1$s: Product ID, %2$s: Product type */
+						throw new UserError( sprintf( __( 'This product of ID %1$s is not a %2$s product' ), $product_id, $type ) );
+					} elseif ( get_post( $product_id )->post_type !== 'product' ) {
+						/* translators: %1$s: ID type, %2$s: ID value */
+						throw new UserError( sprintf( __( 'No product exists with the %1$s: %2$s' ), $id_type, $product_id ) );
+					}
+
+					$product = Factory::resolve_crud_object( $product_id, $context );
+
+					return $product;
+				},
+			)
+		);
 	}
 }