ultimatemember/includes/core/class-query.php

<?php
namespace um\core;

// Exit if accessed directly.
if ( ! defined( 'ABSPATH' ) ) exit;

if ( ! class_exists( 'Query' ) ) {
    class Query {

        public $wp_pages = array();
        public $roles = array();

        function __construct() {


        }


        function ajax_paginate() {
            extract( $_REQUEST );

            ob_start();

            do_action( "um_ajax_load_posts__{$hook}", $args );

            $output = ob_get_contents();
            ob_end_clean();

            die( $output );
        }


        /***
         ***	@get wp pages
         ***/
        function wp_pages() {
            global $wpdb;

            if( isset( $this->wp_pages ) && ! empty( $this->wp_pages ) ){
                return $this->wp_pages;
            }

            $count_pages = wp_count_posts('page');

            if ( $count_pages->publish > 300 ){
                return 'reached_maximum_limit';
            }


            $pages = $wpdb->get_results('SELECT * FROM '.$wpdb->posts.' WHERE post_type = "page" AND post_status = "publish" ', OBJECT);

            $array = array();
            if( $wpdb->num_rows > 0 ){
                foreach ($pages as $page_data) {
                    $array[ $page_data->ID ] = $page_data->post_title;
                }
            }

            $this->wp_pages = $array;

            return $array;
        }

        /***
         ***	@get all forms
         ***/
        function forms() {

            $args = array(
                'post_type' => 'um_form',
                'posts_per_page' => 200,
                'post_status' => array('publish')
            );

            $query = new \WP_Query( $args );
            foreach( $query->posts as $post ) {
                setup_postdata( $post );
                $results[ $post->ID ] = $post->post_title;
            }
            return $results;

        }
        

        /***
         ***	@Do custom queries
         ***/
        function make( $args ) {

            $defaults = array(
                'post_type' => 'post',
                'post_status' => array('publish')
            );
            $args = wp_parse_args( $args, $defaults );

            if ( isset( $args['post__in'] ) && empty( $args['post__in'] ) )
                return false;

            extract( $args );

            if ( $post_type == 'comment' ) { // comments

                unset( $args['post_type'] );

                $args['type__not_in'] = apply_filters( 'um_excluded_comment_types', array('') );

                $comments = get_comments($args);
                return $comments;

            } else {

                $custom_posts = new \WP_Query();
                $custom_posts->query( $args );
                return $custom_posts;

            }

        }

        /***
         ***	@Get last users
         ***/
        function get_recent_users($number = 5){
            $args = array( 'fields' => 'ID', 'number' => $number, 'orderby' => 'user_registered', 'order' => 'desc' );

            $users = new \WP_User_Query( $args );
            return $users->results;
        }

        /***
         ***	@Count users by status
         ***/
        function count_users_by_status( $status ) {
            $args = array( 'fields' => 'ID', 'number' => 0 );
            if ( $status == 'unassigned' ) {
                $args['meta_query'][] = array(array('key' => 'account_status','compare' => 'NOT EXISTS'));
                $users = new \WP_User_Query( $args );
                foreach( $users->results as $user ) {
                    update_user_meta( $user, 'account_status', 'approved' );
                }
            } else {
                $args['meta_query'][] = array(array('key' => 'account_status','value' => $status,'compare' => '='));
            }
            $users = new \WP_User_Query( $args );
            return count($users->results);
        }

        /***
         ***	@Get users by status
         ***/
        function get_users_by_status($status, $number = 5){
            global $wpdb;

            $args = array( 'fields' => 'ID', 'number' => $number, 'orderby' => 'user_registered', 'order' => 'desc' );

            $args['meta_query'][] = array(
                array(
                    'key'     => 'account_status',
                    'value'   => $status,
                    'compare' => '='
                )
            );

            $users = new \WP_User_Query( $args );
            return $users->results;
        }


        /***
         ***	@Count all users
         ***/
        function count_users(){
            $result = count_users();
            return $result['total_users'];
        }


        /***
         ***	@Using wpdb instead of update_post_meta
         ***/
        function update_attr( $key, $post_id, $new_value ){
            update_post_meta( $post_id, '_um_' . $key, $new_value );
        }

        /***
         ***	@get data
         ***/
        function get_attr( $key, $post_id ){
            $meta = get_post_meta( $post_id, '_um_' . $key, true );
            return $meta;
        }

        /***
         ***	@delete data
         ***/
        function delete_attr( $key, $post_id ){
            $meta = delete_post_meta( $post_id, '_um_' . $key );
            return $meta;
        }

        /***
         ***	@Checks if post has a specific meta key
         ***/
        function has_post_meta($key, $value=null, $post_id=null ){
            if (!$post_id){
                global $post;
                $post_id = $post->ID;
            }
            if ($value ){
                if ( get_post_meta($post_id, $key, true) == $value )
                    return true;
            } else {
                if ( get_post_meta($post_id, $key, true) )
                    return true;
            }
            return false;
        }

        /***
         ***	@Get posts with specific meta key/value
         ***/
        function find_post_id($post_type, $key, $value){
            $posts = get_posts( array( 'post_type' => $post_type, 'meta_key' => $key, 'meta_value' => $value ) );
            if ( isset($posts[0]) && !empty($posts) )
                return $posts[0]->ID;
            return false;
        }


        /***
         ***	@Get post data
         ***/
        function post_data( $post_id ) {
            $array['form_id'] = $post_id;
            $mode = $this->get_attr('mode', $post_id);
            $meta = get_post_custom( $post_id );
            foreach ($meta as $k => $v){
                if ( strstr($k, '_um_'.$mode.'_' ) ) {
                    $k = str_replace('_um_'.$mode.'_', '', $k);
                    $array[$k] = $v[0];
                } elseif ($k == '_um_mode'){
                    $k = str_replace('_um_', '', $k);
                    $array[$k] = $v[0];
                } elseif ( strstr($k, '_um_') ) {
                    $k = str_replace('_um_', '', $k);
                    $array[$k] = $v[0];
                }

            }

            foreach( $array as $k => $v ) {
                if ( strstr( $k, 'login_') || strstr( $k, 'register_' ) || strstr( $k, 'profile_' ) ){
                    if ( $mode != 'directory' ) {
                        unset($array[$k]);
                    }
                }
            }
            return $array;
        }

        /***
         ***	@Capture selected value
         ***/
        function get_meta_value( $key, $array_key = null, $fallback = null ) {
            global $post;
            $post_id = get_the_ID();
            $try = get_post_meta( $post_id, $key, true );

            if ( ! empty( $try ) )
                if ( is_array( $try ) && in_array( $array_key, $try ) ) {
                    return $array_key;
                } else if ( is_array( $try ) ) {
                    return '';
                } else {
                    return $try;
                }

            if ($fallback == 'na') {
                $fallback = 0;
                $none = '';
            } else {
                $none = 0;
            }
            return (!empty($fallback)) ? $fallback : $none;
        }

        /***
         ***	@Checks if its a core page of UM
         ***/
        function is_core( $post_id ){
            $is_core = get_post_meta($post_id, '_um_core', true);
            if ( $is_core != '' ) {
                return $is_core;
            } else {
                return false;
            }
        }

    }
}