<?php

/**

 * Site API: WP_Site_Query class

 *

 * @package WordPress

 * @subpackage Sites

 * @since 4.6.0

 */

/**

 * Core class used for querying sites.

 *

 * @since 4.6.0

 *

 * @see WP_Site_Query::__construct() for accepted arguments.

 */

class WP_Site_Query {

	/**

	 * SQL for database query.

	 *

	 * @since 4.6.0

	 * @var string

	 */

	public $request;

	/**

	 * SQL query clauses.

	 *

	 * @since 4.6.0

	 * @var array

	 */

	protected $sql_clauses = array(

		'select'  => '',

		'from'    => '',

		'where'   => array(),

		'groupby' => '',

		'orderby' => '',

		'limits'  => '',

	);

	/**

	 * Metadata query container.

	 *

	 * @since 5.1.0

	 * @var WP_Meta_Query

	 */

	public $meta_query = false;

	/**

	 * Metadata query clauses.

	 *

	 * @since 5.1.0

	 * @var array

	 */

	protected $meta_query_clauses;

	/**

	 * Date query container.

	 *

	 * @since 4.6.0

	 * @var object WP_Date_Query

	 */

	public $date_query = false;

	/**

	 * Query vars set by the user.

	 *

	 * @since 4.6.0

	 * @var array

	 */

	public $query_vars;

	/**

	 * Default values for query vars.

	 *

	 * @since 4.6.0

	 * @var array

	 */

	public $query_var_defaults;

	/**

	 * List of sites located by the query.

	 *

	 * @since 4.6.0

	 * @var array

	 */

	public $sites;

	/**

	 * The amount of found sites for the current query.

	 *

	 * @since 4.6.0

	 * @var int

	 */

	public $found_sites = 0;

	/**

	 * The number of pages.

	 *

	 * @since 4.6.0

	 * @var int

	 */

	public $max_num_pages = 0;

	/**

	 * Sets up the site query, based on the query vars passed.

	 *

	 * @since 4.6.0

	 * @since 4.8.0 Introduced the 'lang_id', 'lang__in', and 'lang__not_in' parameters.

	 * @since 5.1.0 Introduced the 'update_site_meta_cache', 'meta_query', 'meta_key',

	 *              'meta_value', 'meta_type' and 'meta_compare' parameters.

	 *

	 * @param string|array $query {

	 *     Optional. Array or query string of site query parameters. Default empty.

	 *

	 *     @type array        $site__in               Array of site IDs to include. Default empty.

	 *     @type array        $site__not_in           Array of site IDs to exclude. Default empty.

	 *     @type bool         $count                  Whether to return a site count (true) or array of site objects.

	 *                                                Default false.

	 *     @type array        $date_query             Date query clauses to limit sites by. See WP_Date_Query.

	 *                                                Default null.

	 *     @type string       $fields                 Site fields to return. Accepts 'ids' (returns an array of site IDs)

	 *                                                or empty (returns an array of complete site objects). Default empty.

	 *     @type int          $ID                     A site ID to only return that site. Default empty.

	 *     @type int          $number                 Maximum number of sites to retrieve. Default 100.

	 *     @type int          $offset                 Number of sites to offset the query. Used to build LIMIT clause.

	 *                                                Default 0.

	 *     @type bool         $no_found_rows          Whether to disable the `SQL_CALC_FOUND_ROWS` query. Default true.

	 *     @type string|array $orderby                Site status or array of statuses. Accepts 'id', 'domain', 'path',

	 *                                                'network_id', 'last_updated', 'registered', 'domain_length',

	 *                                                'path_length', 'site__in' and 'network__in'. Also accepts false,

	 *                                                an empty array, or 'none' to disable `ORDER BY` clause.

	 *                                                Default 'id'.

	 *     @type string       $order                  How to order retrieved sites. Accepts 'ASC', 'DESC'. Default 'ASC'.

	 *     @type int          $network_id             Limit results to those affiliated with a given network ID. If 0,

	 *                                                include all networks. Default 0.

	 *     @type array        $network__in            Array of network IDs to include affiliated sites for. Default empty.

	 *     @type array        $network__not_in        Array of network IDs to exclude affiliated sites for. Default empty.

	 *     @type string       $domain                 Limit results to those affiliated with a given domain. Default empty.

	 *     @type array        $domain__in             Array of domains to include affiliated sites for. Default empty.

	 *     @type array        $domain__not_in         Array of domains to exclude affiliated sites for. Default empty.

	 *     @type string       $path                   Limit results to those affiliated with a given path. Default empty.

	 *     @type array        $path__in               Array of paths to include affiliated sites for. Default empty.

	 *     @type array        $path__not_in           Array of paths to exclude affiliated sites for. Default empty.

	 *     @type int          $public                 Limit results to public sites. Accepts '1' or '0'. Default empty.

	 *     @type int          $archived               Limit results to archived sites. Accepts '1' or '0'. Default empty.

	 *     @type int          $mature                 Limit results to mature sites. Accepts '1' or '0'. Default empty.

	 *     @type int          $spam                   Limit results to spam sites. Accepts '1' or '0'. Default empty.

	 *     @type int          $deleted                Limit results to deleted sites. Accepts '1' or '0'. Default empty.

	 *     @type int          $lang_id                Limit results to a language ID. Default empty.

	 *     @type array        $lang__in               Array of language IDs to include affiliated sites for. Default empty.

	 *     @type array        $lang__not_in           Array of language IDs to exclude affiliated sites for. Default empty.

	 *     @type string       $search                 Search term(s) to retrieve matching sites for. Default empty.

	 *     @type array        $search_columns         Array of column names to be searched. Accepts 'domain' and 'path'.

	 *                                                Default empty array.

	 *     @type bool         $update_site_cache      Whether to prime the cache for found sites. Default true.

	 *     @type bool         $update_site_meta_cache Whether to prime the metadata cache for found sites. Default true.

	 *     @type array        $meta_query             Meta query clauses to limit retrieved sites by. See `WP_Meta_Query`.

	 *                                                Default empty.

	 *     @type string       $meta_key               Limit sites to those matching a specific metadata key.

	 *                                                Can be used in conjunction with `$meta_value`. Default empty.

	 *     @type string       $meta_value             Limit sites to those matching a specific metadata value.

	 *                                                Usually used in conjunction with `$meta_key`. Default empty.

	 *     @type string       $meta_type              Data type that the `$meta_value` column will be CAST to for

	 *                                                comparisons. Default empty.

	 *     @type string       $meta_compare           Comparison operator to test the `$meta_value`. Default empty.

	 * }

	 */

	public function __construct( $query = '' ) {

		$this->query_var_defaults = array(

			'fields'                 => '',

			'ID'                     => '',

			'site__in'               => '',

			'site__not_in'           => '',

			'number'                 => 100,

			'offset'                 => '',

			'no_found_rows'          => true,

			'orderby'                => 'id',

			'order'                  => 'ASC',

			'network_id'             => 0,

			'network__in'            => '',

			'network__not_in'        => '',

			'domain'                 => '',

			'domain__in'             => '',

			'domain__not_in'         => '',

			'path'                   => '',

			'path__in'               => '',

			'path__not_in'           => '',

			'public'                 => null,

			'archived'               => null,

			'mature'                 => null,

			'spam'                   => null,

			'deleted'                => null,

			'lang_id'                => null,

			'lang__in'               => '',

			'lang__not_in'           => '',

			'search'                 => '',

			'search_columns'         => array(),

			'count'                  => false,

			'date_query'             => null, // See WP_Date_Query

			'update_site_cache'      => true,

			'update_site_meta_cache' => true,

			'meta_query'             => '',

			'meta_key'               => '',

			'meta_value'             => '',

			'meta_type'              => '',

			'meta_compare'           => '',

		);

		if ( ! empty( $query ) ) {

			$this->query( $query );

		}

	}

	/**

	 * Parses arguments passed to the site query with default query parameters.

	 *

	 * @since 4.6.0

	 *

	 * @see WP_Site_Query::__construct()

	 *

	 * @param string|array $query Array or string of WP_Site_Query arguments. See WP_Site_Query::__construct().

	 */

	public function parse_query( $query = '' ) {

		if ( empty( $query ) ) {

			$query = $this->query_vars;

		}

		$this->query_vars = wp_parse_args( $query, $this->query_var_defaults );

		/**

		 * Fires after the site query vars have been parsed.

		 *

		 * @since 4.6.0

		 *

		 * @param WP_Site_Query $this The WP_Site_Query instance (passed by reference).

		 */

		do_action_ref_array( 'parse_site_query', array( &$this ) );

	}

	/**

	 * Sets up the WordPress query for retrieving sites.

	 *

	 * @since 4.6.0

	 *

	 * @param string|array $query Array or URL query string of parameters.

	 * @return array|int List of WP_Site objects, a list of site ids when 'fields' is set to 'ids',

	 *                   or the number of sites when 'count' is passed as a query var.

	 */

	public function query( $query ) {

		$this->query_vars = wp_parse_args( $query );

		return $this->get_sites();

	}

	/**

	 * Retrieves a list of sites matching the query vars.

	 *

	 * @since 4.6.0

	 *

	 * @global wpdb $wpdb WordPress database abstraction object.

	 *

	 * @return array|int List of WP_Site objects, a list of site ids when 'fields' is set to 'ids',

	 *                   or the number of sites when 'count' is passed as a query var.

	 */

	public function get_sites() {

		global $wpdb;

		$this->parse_query();

		// Parse meta query.

		$this->meta_query = new WP_Meta_Query();

		$this->meta_query->parse_query_vars( $this->query_vars );

		/**

		 * Fires before sites are retrieved.

		 *

		 * @since 4.6.0

		 *

		 * @param WP_Site_Query $this Current instance of WP_Site_Query (passed by reference).

		 */

		do_action_ref_array( 'pre_get_sites', array( &$this ) );

		// Reparse query vars, in case they were modified in a 'pre_get_sites' callback.

		$this->meta_query->parse_query_vars( $this->query_vars );

		if ( ! empty( $this->meta_query->queries ) ) {

			$this->meta_query_clauses = $this->meta_query->get_sql( 'blog', $wpdb->blogs, 'blog_id', $this );

		}

		$site_ids = null;

		/**

		 * Filter the sites array before the query takes place.

		 *

		 * Return a non-null value to bypass WordPress's default site queries.

		 *

		 *

		 * @since 5.2.0

		 *

		 * @param array|null    $site_ids Return an array of site data to short-circuit WP's site query,

		 *                                or null to allow WP to run its normal queries.

		 * @param WP_Site_Query $this The WP_Site_Query instance, passed by reference.

		 */

		$site_ids = apply_filters_ref_array( 'sites_pre_query', array( $site_ids, &$this ) );

		if ( null === $site_ids ) {

			// $args can include anything. Only use the args defined in the query_var_defaults to compute the key.

			$_args = wp_array_slice_assoc( $this->query_vars, array_keys( $this->query_var_defaults ) );

			// Ignore the $fields argument as the queried result will be the same regardless.

			unset( $_args['fields'] );

			$key          = md5( serialize( $_args ) );

			$last_changed = wp_cache_get_last_changed( 'sites' );

			$cache_key   = "get_sites:$key:$last_changed";

			$cache_value = wp_cache_get( $cache_key, 'sites' );

			if ( false === $cache_value ) {

				$site_ids = $this->get_site_ids();

				if ( $site_ids ) {

					$this->set_found_sites();

				}

				$cache_value = array(

					'site_ids'    => $site_ids,

					'found_sites' => $this->found_sites,

				);

				wp_cache_add( $cache_key, $cache_value, 'sites' );

			} else {

				$site_ids          = $cache_value['site_ids'];

				$this->found_sites = $cache_value['found_sites'];

			}

		}

		if ( $this->found_sites && $this->query_vars['number'] ) {

			$this->max_num_pages = ceil( $this->found_sites / $this->query_vars['number'] );

		}

		// If querying for a count only, there's nothing more to do.

		if ( $this->query_vars['count'] ) {

			// $site_ids is actually a count in this case.

			return intval( $site_ids );

		}

		$site_ids = array_map( 'intval', $site_ids );

		if ( 'ids' == $this->query_vars['fields'] ) {

			$this->sites = $site_ids;

			return $this->sites;

		}

		// Prime site network caches.

		if ( $this->query_vars['update_site_cache'] ) {

			_prime_site_caches( $site_ids, $this->query_vars['update_site_meta_cache'] );

		}

		// Fetch full site objects from the primed cache.

		$_sites = array();

		foreach ( $site_ids as $site_id ) {

			if ( $_site = get_site( $site_id ) ) {

				$_sites[] = $_site;

			}

		}

		/**

		 * Filters the site query results.

		 *

		 * @since 4.6.0

		 *

		 * @param WP_Site[]     $_sites An array of WP_Site objects.

		 * @param WP_Site_Query $this   Current instance of WP_Site_Query (passed by reference).

		 */

		$_sites = apply_filters_ref_array( 'the_sites', array( $_sites, &$this ) );

		// Convert to WP_Site instances.

		$this->sites = array_map( 'get_site', $_sites );

		return $this->sites;

	}

	/**

	 * Used internally to get a list of site IDs matching the query vars.

	 *

	 * @since 4.6.0

	 *

	 * @global wpdb $wpdb WordPress database abstraction object.

	 *

	 * @return int|array A single count of site IDs if a count query. An array of site IDs if a full query.

	 */

	protected function get_site_ids() {

		global $wpdb;

		$order = $this->parse_order( $this->query_vars['order'] );

		// Disable ORDER BY with 'none', an empty array, or boolean false.

		if ( in_array( $this->query_vars['orderby'], array( 'none', array(), false ), true ) ) {

			$orderby = '';

		} elseif ( ! empty( $this->query_vars['orderby'] ) ) {

			$ordersby = is_array( $this->query_vars['orderby'] ) ?

				$this->query_vars['orderby'] :

				preg_split( '/[,\s]/', $this->query_vars['orderby'] );

			$orderby_array = array();

			foreach ( $ordersby as $_key => $_value ) {

				if ( ! $_value ) {

					continue;

				}

				if ( is_int( $_key ) ) {

					$_orderby = $_value;

					$_order   = $order;

				} else {

					$_orderby = $_key;

					$_order   = $_value;

				}

				$parsed = $this->parse_orderby( $_orderby );

				if ( ! $parsed ) {

					continue;

				}

				if ( 'site__in' === $_orderby || 'network__in' === $_orderby ) {

					$orderby_array[] = $parsed;

					continue;

				}

				$orderby_array[] = $parsed . ' ' . $this->parse_order( $_order );

			}

			$orderby = implode( ', ', $orderby_array );

		} else {

			$orderby = "{$wpdb->blogs}.blog_id $order";

		}