wp-graphql-woocommerce/includes/data/connection/class-tax-rate-connection-resolver.php

<?php
/**
 * ConnectionResolver - Tax_Rate_Connection_Resolver
 *
 * Resolves connections to Tax Rates
 *
 * @package WPGraphQL\WooCommerce\Data\Connection
 * @since 0.0.2
 */

namespace WPGraphQL\WooCommerce\Data\Connection;

use WPGraphQL\Data\Connection\AbstractConnectionResolver;

/**
 * Class Tax_Rate_Connection_Resolver
 *
 * @property \WPGraphQL\WooCommerce\Data\Loader\WC_Db_Loader $loader
 *
 * @package WPGraphQL\WooCommerce\Data\Connection
 */
class Tax_Rate_Connection_Resolver extends AbstractConnectionResolver {
	/**
	 * Return the name of the loader to be used with the connection resolver
	 *
	 * @return string
	 */
	public function get_loader_name() {
		return 'tax_rate';
	}

	/**
	 * Confirms the uses has the privileges to query Tax Rates
	 *
	 * @return bool
	 */
	public function should_execute() {
		if ( ! wc_rest_check_manager_permissions( 'settings', 'read' ) ) {
			graphql_debug(
				__( 'User does not have permission to view tax rates.', 'graphql-for-ecommerce' )
			);
			return false;
		}
		return true;
	}

	/**
	 * Creates query arguments array
	 *
	 * @return array
	 */
	public function get_query_args() {
		$query_args = [];

		// Prepare for later use.
		$last  = ! empty( $this->args['last'] ) ? $this->args['last'] : null;
		$first = ! empty( $this->args['first'] ) ? $this->args['first'] : null;

		/**
		 * Collect the input_fields and sanitize them to prepare them for sending to the WP_Query
		 */
		$input_fields = [];
		if ( ! empty( $this->args['where'] ) ) {
			$input_fields = $this->sanitize_input_fields( $this->args['where'] );
		}

		if ( ! empty( $input_fields ) ) {
			$query_args = array_merge( $query_args, $input_fields );
		}

		/**
		 * Set posts_per_page the highest value of $first and $last, with a (filterable) max of 100
		 */
		$query_args['items_per_page'] = min( max( absint( $first ), absint( $last ), 10 ), $this->query_amount ) + 1;

		/**
		 * Set the cursor args.
		 *
		 * @see \WPGraphQL\Data\Config::graphql_wp_query_cursor_pagination_support
		 */
		$query_args['graphql_after_cursor']   = $this->get_after_offset();
		$query_args['graphql_before_cursor']  = $this->get_before_offset();
		$query_args['graphql_cursor_compare'] = ! empty( $last ) ? '>' : '<';

		/**
		 * If there's no orderby params in the inputArgs, set order based on the first/last argument
		 */
		if ( empty( $query_args['orderby'] ) ) {
			$query_args['orderby'] = 'tax_rate_order';
		}
		if ( empty( $query_args['order'] ) ) {
			$query_args['order'] = ! empty( $last ) ? 'ASC' : 'DESC';
		}

		/**
		 * Filter the $query args to allow folks to customize queries programmatically
		 *
		 * @param array                                $query_args The args that will be passed to the WP_Query
		 * @param mixed                                $source     The source that's passed down the GraphQL queries
		 * @param array<string, mixed>|null            $args       The inputArgs on the field
		 * @param \WPGraphQL\AppContext                $context    The AppContext passed down the GraphQL tree
		 * @param \GraphQL\Type\Definition\ResolveInfo $info       The ResolveInfo passed down the GraphQL tree
		 */
		$query_args = apply_filters( 'graphql_tax_rate_connection_query_args', $query_args, $this->source, $this->args, $this->context, $this->info );

		return $query_args;
	}

	/**
	 * Executes query
	 *
	 * @return array
	 */
	public function get_query() {
		global $wpdb;

		/** @var array<string, mixed> $query_args */
		$query_args = $this->query_args;

		if ( ! empty( $query_args['where'] ) ) {
			$sql_where = $query_args['where'];

			$results = $wpdb->get_results( // @codingStandardsIgnoreStart
				$wpdb->prepare(
					"SELECT rates.tax_rate_id
					FROM {$wpdb->prefix}woocommerce_tax_rates AS rates
					LEFT JOIN {$wpdb->prefix}woocommerce_tax_rate_locations AS locations
					ON rates.tax_rate_id = locations.tax_rate_id
					WHERE {$sql_where}
					GROUP BY rates.tax_rate_id
					ORDER BY %s %s",
					$query_args['orderby'],
					$query_args['order']
				)
			); // @codingStandardsIgnoreEnd
		} else {
			$results = $wpdb->get_results( // @codingStandardsIgnoreStart
				$wpdb->prepare(
					"SELECT tax_rate_id
					FROM {$wpdb->prefix}woocommerce_tax_rates
					ORDER BY %s %s",
					$query_args['orderby'],
					$query_args['order']
				)
			); // @codingStandardsIgnoreEnd
		}//end if

		$results = array_map(
			static function ( $rate ) {
				return $rate->tax_rate_id;
			},
			(array) $results
		);
		return $results;
	}

	/**
	 * Return an array of items from the query
	 *
	 * @return array
	 */
	public function get_ids() {
		return ! empty( $this->query ) ? $this->query : [];
	}

	/**
	 * This sets up the "allowed" args, and translates the GraphQL-friendly keys to WP_Query
	 * friendly keys. There's probably a cleaner/more dynamic way to approach this, but
	 * this was quick. I'd be down to explore more dynamic ways to map this, but for
	 * now this gets the job done.
	 *
	 * @param array $where_args - arguments being used to filter query.
	 *
	 * @return array
	 */
	public function sanitize_input_fields( array $where_args ) {
		$args = [];
		if ( ! empty( $where_args['orderby'] ) ) {
			if ( ! empty( $where_args['orderby']['field'] ) ) {
				$orderby_possibles = [
					'id'    => 'tax_rate_id',
					'order' => 'tax_rate_order',
				];
				$args['orderby']   = $orderby_possibles[ $where_args['orderby']['field'] ];
			}

			if ( ! empty( $where_args['orderby']['order'] ) ) {
				$args['order'] = $where_args['orderby']['order'];
			}
		}

		$args['where'] = '1=1';

		if ( isset( $where_args['class'] ) ) {
			$args['where'] .= ' AND ';

			if ( empty( $where_args['class'] ) ) {
				$args['where'] .= 'tax_rate_class = ""';
			} else {
				$rate_class     = $where_args['class'];
				$args['where'] .= "tax_rate_class = '{$rate_class}'";
			}
		}

		if ( ! empty( $where_args['postCode'] ) ) {
			$args['where'] .= ' AND ';
			$post_code      = $where_args['postCode'];
			$args['where'] .= "location_code = '{$post_code}'";
		}

		if ( ! empty( $where_args['postCodeIn'] ) ) {
			$args['where'] .= ' AND ';

			$args['where'] .= ' (';
			foreach ( $where_args['postCodeIn'] as $i => $post_code ) {
				if ( 0 === $i ) {
					$args['where'] .= "location_code = '{$post_code}' ";
				} else {
					$args['where'] .= "OR location_code = '{$post_code}' ";
				}
			}
			$args['where'] .= ') ';
		}

		return $args;
	}

	/**
	 * Stub function
	 *
	 * @todo Implement pagination on this connection.
	 *
	 * @param integer $offset Tax rate index.
	 *
	 * @return bool
	 */
	public function is_valid_offset( $offset ) {
		return true;
	}
}