Merge pull request #238 from kidunot89/feature/customer-downloadable-items

Adds Customer to DownloadableItems connection

Author: kidunot89

docs/config.js

@@ -10,9 +10,12 @@ const config = {
 		"title": "WooGraphQL Documentation",
 		"githubUrl": "https://github.com/wp-graphql/wp-graphql-woocommerce",
 		"helpUrl": "https://github.com/wp-graphql/wp-graphql-woocommerce/issues",
-		"tweetText": "",
+		"tweetText": "Take a look at the amazing Docs Site for @woographql https://woographql.axistaylor.com",
 		"links": [
-			{ "text": "", "link": ""}
+			{
+				"text": "AxisTaylor",
+				"link": "https://axistaylor.com"
+			}
 		],
 		"search": {
 			"enabled": false,

includes/connection/class-downloadable-items.php

@@ -22,6 +22,13 @@ class Downloadable_Items {
 	public static function register_connections() {
 		// From Order.
 		register_graphql_connection( self::get_connection_config() );
+
+		// From Customer.
+		register_graphql_connection(
+			self::get_connection_config(
+				array( 'fromType' => 'Customer' )
+			)
+		);
 	}
 
 	/**

includes/data/connection/class-downloadable-item-connection-resolver.php

@@ -15,6 +15,7 @@
 use GraphQLRelay\Connection\ArrayConnection;
 use WPGraphQL\AppContext;
 use WPGraphQL\Data\Connection\AbstractConnectionResolver;
+use WPGraphQL\WooCommerce\Model\Customer;
 
 /**
  * Class Downloadable_Item_Connection_Resolver
@@ -103,7 +104,12 @@ public function get_query_args() {
 	 * @return \WP_Query
 	 */
 	public function get_query() {
-		$items = $this->source->downloadable_items;
+		$items = 0;
+		if ( is_a( $this->source, Customer::class ) ) {
+			$items = wc_get_customer_available_downloads( $this->source->ID );
+		} else {
+			$items = $this->source->downloadable_items;
+		}
 
 		if ( empty( $items ) ) {
 			return array();

tests/_support/Helper/crud-helpers/customer.php

@@ -205,4 +205,29 @@ public function print_failed_query( $id ) {
 			'jwtRefreshToken'       => null,
 		);
 	}
+
+	public function print_downloadables( $id ) {
+		$items = wc_get_customer_available_downloads( $id );
+
+		if ( empty( $items ) ) {
+			return array();
+		}
+		
+		$nodes = array();
+		foreach ( $items as $item ) {
+			$nodes[] = array(
+				'url'                => $item['download_url'],
+				'accessExpires'      => $item['access_expires'],
+				'downloadId'         => $item['download_id'],
+				'downloadsRemaining' => isset( $item['downloads_remaining'] ) && 'integer' === gettype( $item['downloads_remaining'] )
+					? $item['downloads_remaining']
+					: null,
+				'name'               => $item['download_name'],
+				'product'            => array( 'productId' => $item['product_id'] ),
+				'download'           => array( 'downloadId' => $item['download_id'] ),
+			);
+		}
+
+		return $nodes;
+	}
 }

tests/wpunit/DownloadableItemQueriesTest.php

@@ -320,4 +320,72 @@ function( $product_id ) {
 		$this->assertEquals( $expected, $actual );
     }
 
+    public function testCustomerToDownloadableItemsQuery() {
+        $downloadable_product = $this->products->create_simple(
+            array(
+                'downloadable' => true,
+                'downloads'    => array( $this->products->create_download() )
+            )
+        );
+        $order_id = $this->orders->create(
+            array(
+                'status'      => 'completed',
+                'customer_id' => $this->customer,
+            ),
+            array(
+                'line_items' => array(
+                    array(
+                        'product' => $downloadable_product,
+                        'qty'     => 1,
+                    ),
+                ),
+            )
+        );
+
+        // Force download permission updated.
+        wc_downloadable_product_permissions( $order_id, true );
+        
+        $query = '
+            query {
+                customer {
+                    downloadableItems(first: 1) {
+                        nodes {
+                            url
+                            accessExpires
+                            downloadId
+                            downloadsRemaining
+                            name
+                            product {
+                                productId
+                            }
+                            download {
+                                downloadId
+                            }
+                        }
+                    }
+                }
+            }
+        ';
+
+        /**
+		 * Assertion One
+		 * 
+		 * tests query results
+		 */
+		wp_set_current_user( $this->customer );
+		$actual   = graphql( array( 'query' => $query ) );
+		$expected = array(
+            'data' => array(
+                'customer' => array(
+                    'downloadableItems' => array( 'nodes' => $this->customers->print_downloadables( $this->customer ) ),
+                ),
+            ),
+        );
+
+		// use --debug flag to view.
+		codecept_debug( $actual );
+
+		$this->assertEquals( $expected, $actual );
+    }
+
 }