Files
wp-graphql-woocommerce/includes/data/connection/class-order-item-connection-resolver.php
T
2019-10-25 19:13:36 -04:00

82 lines
2.1 KiB
PHP

<?php
/**
* ConnectionResolver - Order_Item_Connection_Resolver
*
* Resolves connections to Orders
*
* @package WPGraphQL\WooCommerce\Data\Connection
* @since 0.0.2
*/
namespace WPGraphQL\WooCommerce\Data\Connection;
use GraphQL\Type\Definition\ResolveInfo;
use GraphQLRelay\Relay;
use WPGraphQL\AppContext;
/**
* Class Order_Item_Connection_Resolver
*/
class Order_Item_Connection_Resolver {
/**
* Prepare items
*
* @param mixed $source - Connection source Model instance.
* @param array $args - Connection arguments.
* @param AppContext $context - AppContext object.
* @param ResolveInfo $info - ResolveInfo object.
*
* @return WC_Order_Item[]
*/
public function query( $source, array $args, AppContext $context, ResolveInfo $info ) {
// @codingStandardsIgnoreLine
switch ( $info->fieldName ) {
case 'taxLines':
$type = 'tax';
break;
case 'shippingLines':
$type = 'shipping';
break;
case 'feeLines':
$type = 'fee';
break;
case 'couponLines':
$type = 'coupon';
break;
default:
$type = 'line_item';
break;
}
$items = array();
foreach ( $source->get_items( $type ) as $id => $item ) {
$item->cached_order = $source;
$items[] = $item;
}
return $items;
}
/**
* Creates connection
*
* @param mixed $source - Connection source Model instance.
* @param array $args - Connection arguments.
* @param AppContext $context - AppContext object.
* @param ResolveInfo $info - ResolveInfo object.
*/
public function resolve( $source, array $args, AppContext $context, ResolveInfo $info ) {
$items = $this->query( $source, $args, $context, $info );
$connection = Relay::connectionFromArray( $items, $args );
$nodes = array();
if ( ! empty( $connection['edges'] ) && is_array( $connection['edges'] ) ) {
foreach ( $connection['edges'] as $edge ) {
$nodes[] = ! empty( $edge['node'] ) ? $edge['node'] : null;
}
}
$connection['nodes'] = ! empty( $nodes ) ? $nodes : null;
return ! empty( $items ) ? $connection : null;
}
}