HOME

<?php

namespace LottaFramework;

class Query {

	// cache all querys
	protected $querys = [];

	/**
	 * Get All Users
	 *
	 * @return array
	 */
	public function users() {
		if ( isset( $this->querys['users'] ) ) {
			return $this->querys['users'];
		}

		$users = [];

		foreach ( get_users() as $key => $user ) {
			$users[ $user->data->ID ] = $user->data->user_nicename;
		}

		wp_reset_postdata();

		$this->querys['users'] = $users;

		return $users;
	}

	/**
	 * Get Available Custom Post Types or Taxonomies
	 *
	 * @param $query
	 * @param bool $exclude_defaults
	 *
	 * @return array
	 */
	public function customPostTypes( $query, $exclude_defaults = true ) {
		// Taxonomies
		if ( 'tax' === $query ) {
			$custom_types = get_taxonomies( [ 'show_in_nav_menus' => true ], 'objects' );
			// Post Types
		} else {
			$custom_types = get_post_types( [ 'show_in_nav_menus' => true ], 'objects' );
		}

		$custom_type_list = [];

		foreach ( $custom_types as $key => $value ) {
			if ( $exclude_defaults ) {
				if ( $key != 'post' && $key != 'page' && $key != 'category' && $key != 'post_tag' ) {
					$custom_type_list[ $key ] = $value->label;
				}
			} else {
				$custom_type_list[ $key ] = $value->label;
			}
		}

		return $custom_type_list;
	}


	/**
	 * Get Terms of Taxonomy
	 *
	 * @param $slug
	 * @param int $per_page
	 * @param bool $hide_empty
	 *
	 * @return array|null
	 */
	public function termsByTaxonomy( $slug, $per_page = - 1, $hide_empty = false ) {
		$key = 'terms-' . (string) $slug . (string) $per_page . $hide_empty ? '-hide-empty' : '-include-empty';
		if ( isset( $this->querys[ $key ] ) ) {
			return $this->querys[ $key ];
		}

		// Exclude WooCommerce product
		if ( ( 'product_cat' === $slug || 'product_tag' === $slug ) && ! class_exists( 'WooCommerce' ) ) {
			return [];
		}

		$query      = get_terms( $slug, [ 'hide_empty' => $hide_empty, 'posts_per_page' => $per_page ] );
		$taxonomies = [];

		foreach ( $query as $tax ) {
			$taxonomies[ $tax->term_id ] = $tax->name;
		}

		wp_reset_postdata();

		$this->querys[ $key ] = $taxonomies;

		return $taxonomies;
	}

	/**
	 * Get Posts of Post Type
	 *
	 * @param $slug
	 * @param int $per_page
	 *
	 * @return array
	 */
	public function postsByPostType( $slug, $per_page = - 1 ) {
		$key = 'posts-' . (string) $slug . (string) $per_page;
		if ( isset( $this->querys[ $key ] ) ) {
			return $this->querys[ $key ];
		}

		$query = get_posts( [ 'post_type' => $slug, 'posts_per_page' => $per_page ] );
		$posts = [];

		foreach ( $query as $post ) {
			$posts[ $post->ID ] = $post->post_title;
		}

		wp_reset_postdata();

		$this->querys[ $key ] = $posts;

		return $posts;
	}

	/**
	 * Query post list
	 *
	 * @param string $post_type
	 * @param int $limit
	 * @param string $search
	 *
	 * @return array
	 */
	public function postList( $post_type = 'any', $limit = - 1, $search = '' ) {

		global $wpdb;
		$where = '';
		$data  = [];

		if ( - 1 == $limit ) {
			$limit = '';
		} elseif ( 0 == $limit ) {
			$limit = "limit 0,1";
		} else {
			$limit = $wpdb->prepare( " limit 0,%d", esc_sql( $limit ) );
		}

		if ( 'any' === $post_type ) {
			$in_search_post_types = get_post_types( [ 'exclude_from_search' => false ] );
			if ( empty( $in_search_post_types ) ) {
				$where .= ' AND 1=0 ';
			} else {
				$where .= " AND {$wpdb->posts}.post_type IN ('" . join( "', '",
						array_map( 'esc_sql', $in_search_post_types ) ) . "')";
			}
		} elseif ( ! empty( $post_type ) ) {
			$where .= $wpdb->prepare( " AND {$wpdb->posts}.post_type = %s", esc_sql( $post_type ) );
		}

		if ( ! empty( $search ) ) {
			$where .= $wpdb->prepare( " AND {$wpdb->posts}.post_title LIKE %s", '%' . esc_sql( $search ) . '%' );
		}

		$query   = "select post_title,ID  from $wpdb->posts where post_status = 'publish' $where $limit";
		$results = $wpdb->get_results( $query );
		if ( ! empty( $results ) ) {
			foreach ( $results as $row ) {
				$data[ $row->ID ] = $row->post_title;
			}
		}

		return $data;
	}
}