Merge pull request #189 from kidunot89/bugfix/union_type_resolver

More support provided for WPGraphQL ACF

Author: kidunot89

includes/class-acf-schema-filters.php

@@ -8,6 +8,10 @@
 
 namespace WPGraphQL\WooCommerce;
 
+use WPGraphQL\WooCommerce\Model\Coupon;
+use WPGraphQL\WooCommerce\Model\Order;
+use WPGraphQL\WooCommerce\Model\Product;
+
 /**
  * Class ACF_Schema_Filters
  */
@@ -19,6 +23,7 @@ class ACF_Schema_Filters {
 	public static function add_filters() {
 		// Registers WooCommerce CPTs && taxonomies.
 		add_filter( 'graphql_acf_get_root_id', array( __CLASS__, 'resolve_crud_root_id' ), 10, 2 );
+		add_filter( 'graphql_acf_post_object_source', array( __CLASS__, 'resolve_post_object_source' ), 10, 2 );
 	}
 
 	/**
@@ -38,4 +43,32 @@ public static function resolve_crud_root_id( $id, $root ) {
 
 		return $id;
 	}
+
+	/**
+	 * Filters ACF "post_object" field type resolver to ensure that
+	 * the proper Type source is provided for WooCommerce CPTs.
+	 *
+	 * @param mixed|null $source  source of the data being provided.
+	 * @param mixed|null $value  Post ID.
+	 *
+	 * @return mixed|null
+	 */
+	public static function resolve_post_object_source( $source, $value ) {
+		$post = get_post( $value );
+		if ( $post instanceof \WP_Post ) {
+			switch ( $post->post_type ) {
+				case 'shop_coupon':
+					$source = new Coupon( $post->ID );
+					break;
+				case 'shop_order':
+					$source = new Order( $post->ID );
+					break;
+				case 'product':
+					$source = new Product( $post->ID );
+					break;
+			}
+		}
+
+		return $source;
+	}
 }

includes/class-core-schema-filters.php

@@ -302,12 +302,15 @@ function( $type ) {
 	 * @param \WPGraphQL\Type\WPUnionType  $wp_union  WPUnion object.
 	 */
 	public static function inject_union_type_resolver( $type, $value, $wp_union ) {
-		if ( 'product' === $value->post_type ) {
-			$node     = new \WPGraphQL\WooCommerce\Model\Product( $value->ID );
-			$new_type = Factory::resolve_node_type( '', $node );
-			if ( $new_type ) {
-				$type = $wp_union->type_registry->get_type( $new_type );
-			}
+		switch ( get_class( $value ) ) {
+			case 'WPGraphQL\WooCommerce\Model\Product':
+			case 'WPGraphQL\WooCommerce\Model\Coupon':
+			case 'WPGraphQL\WooCommerce\Model\Order':
+				$new_type = Factory::resolve_node_type( $type, $value );
+				if ( $new_type ) {
+					$type = $wp_union->type_registry->get_type( $new_type );
+				}
+				break;
 		}
 
 		return $type;