<?php
/**
 * Factory
 *
 * This class serves as a factory for all the resolvers of queries and mutations.
 *
 * @package WPGraphQL\WooCommerce\Data
 * @since   0.0.1
 */

namespace WPGraphQL\WooCommerce\Data;

use GraphQL\Deferred;
use GraphQL\Error\UserError;
use GraphQL\Type\Definition\ResolveInfo;
use function WC;
use WPGraphQL\AppContext;
use WPGraphQL\WooCommerce\Data\Connection\Coupon_Connection_Resolver;
use WPGraphQL\WooCommerce\Data\Connection\Customer_Connection_Resolver;
use WPGraphQL\WooCommerce\Data\Connection\Order_Connection_Resolver;
use WPGraphQL\WooCommerce\Data\Connection\Order_Item_Connection_Resolver;
use WPGraphQL\WooCommerce\Data\Connection\Product_Connection_Resolver;
use WPGraphQL\WooCommerce\Data\Connection\Product_Attribute_Connection_Resolver;
use WPGraphQL\WooCommerce\Data\Connection\Variation_Attribute_Connection_Resolver;
use WPGraphQL\WooCommerce\Data\Connection\Product_Download_Connection_Resolver;
use WPGraphQL\WooCommerce\Data\Connection\Refund_Connection_Resolver;
use WPGraphQL\WooCommerce\Data\Connection\Tax_Rate_Connection_Resolver;
use WPGraphQL\WooCommerce\Data\Connection\Shipping_Method_Connection_Resolver;
use WPGraphQL\WooCommerce\Data\Connection\Cart_Item_Connection_Resolver;
use WPGraphQL\WooCommerce\Data\Connection\Payment_Gateway_Connection_Resolver;
use WPGraphQL\WooCommerce\Data\Connection\Downloadable_Item_Connection_Resolver;
use WPGraphQL\WooCommerce\Model\Order_Item;
use WPGraphQL\WooCommerce\Model\Product;
use WPGraphQL\WooCommerce\Model\Customer;
use WPGraphQL\WooCommerce\Model\Tax_Rate;
use WPGraphQL\WooCommerce\Model\Shipping_Method;

/**
 * Class Factory
 */
class Factory {
	/**
	 * Returns the current woocommerce customer object tied to the current session.
	 *
	 * @return Customer
	 * @access public
	 */
	public static function resolve_session_customer() {
		return new Customer();
	}

	/**
	 * Returns the Customer store object for the provided user ID
	 *
	 * @param int        $id      - user ID of the customer being retrieved.
	 * @param AppContext $context - AppContext object.
	 *
	 * @return Deferred object
	 * @access public
	 */
	public static function resolve_customer( $id, AppContext $context ) {
		if ( empty( $id ) || ! absint( $id ) ) {
			return null;
		}
		$customer_id = absint( $id );
		$loader      = $context->getLoader( 'wc_customer' );
		$loader->buffer( array( $customer_id ) );
		return new Deferred(
			function () use ( $loader, $customer_id ) {
				return $loader->load( $customer_id );
			}
		);
	}

	/**
	 * Returns the WooCommerce CRUD object for the post ID
	 *
	 * @param int        $id      - post ID of the crud object being retrieved.
	 * @param AppContext $context - AppContext object.
	 *
	 * @return Deferred object
	 * @access public
	 */
	public static function resolve_crud_object( $id, AppContext $context ) {
		if ( empty( $id ) || ! absint( $id ) ) {
			return null;
		}

		$context->getLoader( 'wc_cpt' )->buffer( array( $id ) );
		return new Deferred(
			function () use ( $id, $context ) {
				return $context->getLoader( 'wc_cpt' )->load( $id );
			}
		);
	}

	/**
	 * Returns the order item Model for the order item.
	 *
	 * @param int $item - order item crud object instance.
	 *
	 * @return Order_Item
	 * @access public
	 * @throws UserError Invalid object.
	 */
	public static function resolve_order_item( $item ) {
		/**
		 * If $id is an instance of WC_Order_Item
		 */
		if ( is_a( $item, \WC_Order_Item::class ) ) {
			return new Order_Item( $item );
		} else {
			throw new UserError( __( 'Object provided to order item resolver is an invalid type', 'wp-graphql-woocommerce' ) );
		}
	}

	/**
	 * Returns the tax rate Model for the tax rate ID.
	 *
	 * @param string $id - Tax rate ID.
	 * @param AppContext $context - AppContext object.
	 *
	 * @return Deferred object
	 */
	public static function resolve_tax_rate( $id, AppContext $context ) {
		if ( empty( $id ) || ! is_numeric( $id ) ) {
			return null;
		}

		$id = absint( $id );
		$loader    = $context->getLoader( 'tax_rate' );
		$loader->buffer( array( $id ) );
		return new Deferred(
			function () use ( $loader, $id ) {
				return $loader->load( $id );
			}
		);
	}

	/**
	 * Returns the shipping method Model for the shipping method ID.
	 *
	 * @param int $id - Shipping method ID.
	 *
	 * @return Shipping_Method
	 * @access public
	 * @throws UserError Invalid object.
	 */
	public static function resolve_shipping_method( $id ) {
		$wc_shipping = \WC_Shipping::instance();
		$methods     = $wc_shipping->get_shipping_methods();
		if ( empty( $methods[ $id ] ) ) {
			throw new UserError(
				/* translators: shipping method ID */
				sprintf( __( 'No Shipping Method assigned to ID %s was found ', 'wp-graphql-woocommerce' ), $id )
			);
		}

		$method = $methods[ $id ];

		return new Shipping_Method( $method );
	}

	/**
	 * Resolves the WooCommerce cart instance.
	 *
	 * @return \WC_Cart
	 */
	public static function resolve_cart() {
		return WC()->cart;
	}

	/**
	 * Resolves a cart item by key.
	 *
	 * @param string     $key      Cart item key.
	 * @param AppContext $context  AppContext object.
	 *
	 * @return Deferred object
	 */
	public static function resolve_cart_item( $key, AppContext $context ) {
		if ( empty( $key ) ) {
			return null;
		}

		$context->getLoader( 'cart_item' )->buffer( array( $key ) );
		return new Deferred(
			function () use ( $key, $context ) {
				return $context->getLoader( 'cart_item' )->load( $key );
			}
		);
	}

	/**
	 * Resolves a fee object by ID.
	 *
	 * @param int $id Fee object generated ID.
	 *
	 * @return object
	 */
	public static function resolve_cart_fee( $id ) {
		if ( ! empty( self::resolve_cart()->get_fees()[ $id ] ) ) {
			return self::resolve_cart()->get_fees()[ $id ];
		}

		return null;
	}

	/**
	 * Resolves a downloadable item by ID.
	 *
	 * @param int        $id       Downloadable item ID.
	 * @param AppContext $context  AppContext object.
	 *
	 * @return Deferred object
	 */
	public static function resolve_downloadable_item( $id, AppContext $context ) {
		if ( empty( $id ) || ! absint( $id ) ) {
			return null;
		}
		$object_id = absint( $id );
		$loader    = $context->getLoader( 'downloadable_item' );
		$loader->buffer( array( $object_id ) );
		return new Deferred(
			function () use ( $loader, $object_id ) {
				return $loader->load( $object_id );
			}
		);
	}

	/**
	 * Resolves Relay node for some WooGraphQL types.
	 *
	 * @param mixed      $node     Node object.
	 * @param string     $id       Object unique ID.
	 * @param string     $type     Node type.
	 * @param AppContext $context  AppContext instance.
	 *
	 * @return mixed
	 */
	public static function resolve_node( $node, $id, $type, $context ) {
		switch ( $type ) {
			case 'customer':
				$node = self::resolve_customer( $id, $context );
				break;
			case 'shop_coupon':
			case 'shop_order':
			case 'shop_order_refund':
			case 'product':
			case 'product_variation':
				$node = self::resolve_crud_object( $id, $context );
				break;
			case 'shipping_method':
				$node = self::resolve_shipping_method( $id );
				break;
			case 'tax_rate':
				$node = self::resolve_tax_rate( $id, $context );
				break;
		}

		return $node;
	}

	/**
	 * Resolves Relay node type for some WooGraphQL types.
	 *
	 * @param string|null $type  Node type.
	 * @param mixed       $node  Node object.
	 *
	 * @return string|null
	 */
	public static function resolve_node_type( $type, $node ) {
		switch ( true ) {
			case is_a( $node, Coupon::class ):
				$type = 'Coupon';
				break;
			case is_a( $node, Customer::class ):
				$type = 'Customer';
				break;
			case is_a( $node, Order::class ):
				$type = 'Order';
				break;
			case is_a( $node, Product::class ) && 'simple' === $node->type:
				$type = 'SimpleProduct';
				break;
			case is_a( $node, Product::class ) && 'variable' === $node->type:
				$type = 'VariableProduct';
				break;
			case is_a( $node, Product::class ) && 'external' === $node->type:
				$type = 'ExternalProduct';
				break;
			case is_a( $node, Product::class ) && 'grouped' === $node->type:
				$type = 'GroupProduct';
				break;
			case is_a( $node, Product_Variation::class ):
				$type = 'ProductVariation';
				break;
			case is_a( $node, Refund::class ):
				$type = 'Refund';
				break;
			case is_a( $node, Shipping_Method::class ):
				$type = 'ShippingMethod';
				break;
			case is_a( $node, Tax_Rate::class ):
				$type = 'TaxRate';
				break;
		}

		return $type;
	}

	/**
	 * Resolves Coupon connections
	 *
	 * @param mixed       $source     - Data resolver for connection source.
	 * @param array       $args       - Connection arguments.
	 * @param AppContext  $context    - AppContext object.
	 * @param ResolveInfo $info       - ResolveInfo object.
	 *
	 * @return array
	 * @access public
	 */
	public static function resolve_coupon_connection( $source, array $args, AppContext $context, ResolveInfo $info ) {
		$resolver = new Coupon_Connection_Resolver( $source, $args, $context, $info );
		return $resolver->get_connection();
	}

	/**
	 * Resolves Customer connections
	 *
	 * @param mixed       $source     - Connection parent resolver.
	 * @param array       $args       - Connection arguments.
	 * @param AppContext  $context    - AppContext object.
	 * @param ResolveInfo $info       - ResolveInfo object.
	 *
	 * @return array
	 * @access public
	 */
	public static function resolve_customer_connection( $source, array $args, AppContext $context, ResolveInfo $info ) {
		$resolver = new Customer_Connection_Resolver( $source, $args, $context, $info );
		return $resolver->get_connection();
	}

	/**
	 * Resolves Order connections
	 *
	 * @param mixed       $source     - Data resolver for connection source.
	 * @param array       $args       - Connection arguments.
	 * @param AppContext  $context    - AppContext object.
	 * @param ResolveInfo $info       - ResolveInfo object.
	 *
	 * @return array
	 * @access public
	 */
	public static function resolve_order_connection( $source, array $args, AppContext $context, ResolveInfo $info ) {
		$resolver = new Order_Connection_Resolver( $source, $args, $context, $info );
		return $resolver->get_connection();
	}

	/**
	 * Resolves Order connections
	 *
	 * @param mixed       $source     - Data resolver for connection source.
	 * @param array       $args       - Connection arguments.
	 * @param AppContext  $context    - AppContext object.
	 * @param ResolveInfo $info       - ResolveInfo object.
	 *
	 * @return array
	 * @access public
	 */
	public static function resolve_order_item_connection( $source, array $args, AppContext $context, ResolveInfo $info ) {
		$resolver = new Order_Item_Connection_Resolver();
		return $resolver->resolve( $source, $args, $context, $info );
	}

	/**
	 * Resolves Product connections
	 *
	 * @param mixed       $source     - Data resolver for connection source.
	 * @param array       $args       - Connection arguments.
	 * @param AppContext  $context    - AppContext object.
	 * @param ResolveInfo $info       - ResolveInfo object.
	 *
	 * @return array
	 * @access public
	 */
	public static function resolve_product_connection( $source, array $args, AppContext $context, ResolveInfo $info ) {
		$resolver = new Product_Connection_Resolver( $source, $args, $context, $info );
		return $resolver->get_connection();
	}

	/**
	 * Resolves ProductAttribute connections
	 *
	 * @param mixed       $source     - Data resolver for connection source.
	 * @param array       $args       - Connection arguments.
	 * @param AppContext  $context    - AppContext object.
	 * @param ResolveInfo $info       - ResolveInfo object.
	 *
	 * @return array
	 * @access public
	 */
	public static function resolve_product_attribute_connection( $source, array $args, AppContext $context, ResolveInfo $info ) {
		$resolver = new Product_Attribute_Connection_Resolver();
		return $resolver->resolve( $source, $args, $context, $info );
	}

	/**
	 * Resolves VariationAttribute connections
	 *
	 * @param mixed       $source     - Data resolver for connection source.
	 * @param array       $args       - Connection arguments.
	 * @param AppContext  $context    - AppContext object.
	 * @param ResolveInfo $info       - ResolveInfo object.
	 *
	 * @return array
	 * @access public
	 */
	public static function resolve_variation_attribute_connection( $source, array $args, AppContext $context, ResolveInfo $info ) {
		$resolver = new Variation_Attribute_Connection_Resolver();
		return $resolver->resolve( $source, $args, $context, $info );
	}

	/**
	 * Resolves ProductDownload connections
	 *
	 * @param mixed       $source     - Data resolver for connection source.
	 * @param array       $args       - Connection arguments.
	 * @param AppContext  $context    - AppContext object.
	 * @param ResolveInfo $info       - ResolveInfo object.
	 *
	 * @return array
	 * @access public
	 */
	public static function resolve_product_download_connection( $source, array $args, AppContext $context, ResolveInfo $info ) {
		$resolver = new Product_Download_Connection_Resolver();
		return $resolver->resolve( $source, $args, $context, $info );
	}

	/**
	 * Resolves Refund connections
	 *
	 * @param mixed       $source     - Data resolver for connection source.
	 * @param array       $args       - Connection arguments.
	 * @param AppContext  $context    - AppContext object.
	 * @param ResolveInfo $info       - ResolveInfo object.
	 *
	 * @return array
	 * @access public
	 */
	public static function resolve_refund_connection( $source, array $args, AppContext $context, ResolveInfo $info ) {
		$resolver = new Refund_Connection_Resolver( $source, $args, $context, $info );
		return $resolver->get_connection();
	}

	/**
	 * Resolves TaxRate connections
	 *
	 * @param mixed       $source     - Data resolver for connection source.
	 * @param array       $args       - Connection arguments.
	 * @param AppContext  $context    - AppContext object.
	 * @param ResolveInfo $info       - ResolveInfo object.
	 *
	 * @return array
	 * @access public
	 */
	public static function resolve_tax_rate_connection( $source, array $args, AppContext $context, ResolveInfo $info ) {
		$resolver = new Tax_Rate_Connection_Resolver( $source, $args, $context, $info );
		return $resolver->get_connection();
	}

	/**
	 * Resolves ShippingMethod connections
	 *
	 * @param mixed       $source     - Data resolver for connection source.
	 * @param array       $args       - Connection arguments.
	 * @param AppContext  $context    - AppContext object.
	 * @param ResolveInfo $info       - ResolveInfo object.
	 *
	 * @return array
	 * @access public
	 */
	public static function resolve_shipping_method_connection( $source, array $args, AppContext $context, ResolveInfo $info ) {
		$resolver = new Shipping_Method_Connection_Resolver();
		return $resolver->resolve( $source, $args, $context, $info );
	}

	/**
	 * Resolves CartItem connections
	 *
	 * @param mixed       $source     - Data resolver for connection source.
	 * @param array       $args       - Connection arguments.
	 * @param AppContext  $context    - AppContext object.
	 * @param ResolveInfo $info       - ResolveInfo object.
	 *
	 * @return array
	 * @access public
	 */
	public static function resolve_cart_item_connection( $source, array $args, AppContext $context, ResolveInfo $info ) {
		$resolver = new Cart_Item_Connection_Resolver( $source, $args, $context, $info );
		return $resolver->get_connection();
	}

	/**
	 * Resolves DownloadableItem connections
	 *
	 * @param mixed       $source     - Data resolver for connection source.
	 * @param array       $args       - Connection arguments.
	 * @param AppContext  $context    - AppContext object.
	 * @param ResolveInfo $info       - ResolveInfo object.
	 *
	 * @return array
	 * @access public
	 */
	public static function resolve_downloadable_item_connection( $source, array $args, AppContext $context, ResolveInfo $info ) {
		$resolver = new Downloadable_Item_Connection_Resolver( $source, $args, $context, $info );
		return $resolver->get_connection();
	}

	/**
	 * Resolves PaymentGateway connections
	 *
	 * @param mixed       $source     - Data resolver for connection source.
	 * @param array       $args       - Connection arguments.
	 * @param AppContext  $context    - AppContext object.
	 * @param ResolveInfo $info       - ResolveInfo object.
	 *
	 * @return array
	 * @access public
	 */
	public static function resolve_payment_gateway_connection( $source, array $args, AppContext $context, ResolveInfo $info ) {
		$resolver = new Payment_Gateway_Connection_Resolver();
		return $resolver->resolve( $source, $args, $context, $info );
	}
}