Files
wp-graphql-woocommerce/includes/data/connection/class-product-attribute-connection-resolver.php
T

98 lines
2.6 KiB
PHP

<?php
/**
* ConnectionResolver - Product_Attribute_Connection_Resolver
*
* Resolves connections to ProductAttributes
*
* @package WPGraphQL\WooCommerce\Data\Connection
* @since 0.0.1
*/
namespace WPGraphQL\WooCommerce\Data\Connection;
use GraphQL\Error\UserError;
use GraphQL\Type\Definition\ResolveInfo;
use GraphQLRelay\Relay;
use WPGraphQL\AppContext;
use WPGraphQL\WooCommerce\Model\Product;
const GLOBAL_ID_DELIMITER = ':';
/**
* Class Product_Attribute_Connection_Resolver
*/
class Product_Attribute_Connection_Resolver {
/**
* Builds Product attribute items
*
* @param array $attributes Array of WC_Product_Attributes instances.
* @param Product $source Parent product model.
* @param array $args Connection arguments.
* @param AppContext $context AppContext object.
* @param ResolveInfo $info ResolveInfo object.
*
* @throws UserError Invalid product attribute enumeration value.
* @return array
*/
private function get_items( $attributes, $source, $args, $context, $info ) {
$items = array();
foreach ( $attributes as $attribute_name => $data ) {
$data->_relay_id = base64_encode(
$attribute_name
. GLOBAL_ID_DELIMITER
. $source->ID
. GLOBAL_ID_DELIMITER
. $data->get_name()
);
$items[] = $data;
}
if ( ! empty( $args['type'] ) ) {
switch ( $args['type'] ) {
case 'local':
$items = array_filter(
$items,
function( $item ) {
return ! $item->is_taxonomy();
}
);
break;
case 'global':
$items = array_filter(
$items,
function( $item ) {
return $item->is_taxonomy();
}
);
break;
default:
throw new UserError( __( 'Invalid product attribute type provided', 'wp-graphql-woocommerce' ) );
}
}
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 ) {
$attributes = $this->get_items( $source->attributes, $source, $args, $context, $info );
$connection = Relay::connectionFromArray( $attributes, $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( $attributes ) ? $connection : null;
}
}