| @@ -1,7 +1,16 @@ | ||
| 1 | 1 | <?php |
| 2 | +/** | |
| 3 | + * Queries the database for stream records. | |
| 4 | + * | |
| 5 | + * @package WP_Stream | |
| 6 | + */ | |
| 7 | + | |
| 2 | 8 | namespace WP_Stream; |
| 3 | 9 | |
| 10 | +/** | |
| 11 | + * Class - Query | |
| 12 | + */ | |
| 4 | 13 | class Query { |
| 5 | 14 | /** |
| 6 | 15 | * Hold the number of records found |
| 7 | 16 | * |
| @@ -11,9 +20,9 @@ | ||
| 11 | 20 | |
| 12 | 21 | /** |
| 13 | 22 | * Query records |
| 14 | 23 | * |
| 15 | - * @param array Query args | |
| 24 | + * @param array $args Arguments to filter the records by. | |
| 16 | 25 | * |
| 17 | 26 | * @return array Stream Records |
| 18 | 27 | */ |
| 19 | 28 | public function query( $args ) { |
| @@ -45,11 +54,11 @@ | ||
| 45 | 54 | $where .= $wpdb->prepare( " AND $wpdb->stream.user_role = %s", $args['user_role'] ); |
| 46 | 55 | } |
| 47 | 56 | |
| 48 | 57 | if ( ! empty( $args['search'] ) ) { |
| 49 | - $field = ! empty( $args['search_field'] ) ? $args['search_field'] : 'summary'; | |
| 58 | + $field = ! empty( $args['search_field'] ) ? $args['search_field'] : 'summary'; | |
| 50 | 59 | |
| 51 | - // Sanitize field | |
| 60 | + // Sanitize field. | |
| 52 | 61 | $allowed_fields = array( 'ID', 'site_id', 'blog_id', 'object_id', 'user_id', 'user_role', 'created', 'summary', 'connector', 'context', 'action', 'ip' ); |
| 53 | 62 | if ( in_array( $field, $allowed_fields, true ) ) { |
| 54 | 63 | $where .= $wpdb->prepare( " AND $wpdb->stream.{$field} LIKE %s", "%{$args['search']}%" ); // @codingStandardsIgnoreLine can't prepare column name |
| 55 | 64 | } |
| @@ -73,33 +82,33 @@ | ||
| 73 | 82 | |
| 74 | 83 | /** |
| 75 | 84 | * PARSE DATE PARAM FAMILY |
| 76 | 85 | */ |
| 86 | + if ( ! empty( $args['date'] ) ) { | |
| 87 | + $args['date_from'] = $args['date']; | |
| 88 | + $args['date_to'] = $args['date']; | |
| 89 | + } | |
| 90 | + | |
| 77 | 91 | if ( ! empty( $args['date_from'] ) ) { |
| 78 | - $date = get_gmt_from_date( date( 'Y-m-d H:i:s', strtotime( $args['date_from'] . ' 00:00:00' ) ) ); | |
| 92 | + $date = get_gmt_from_date( gmdate( 'Y-m-d H:i:s', strtotime( $args['date_from'] . ' 00:00:00' ) ) ); | |
| 79 | 93 | $where .= $wpdb->prepare( " AND DATE($wpdb->stream.created) >= %s", $date ); |
| 80 | 94 | } |
| 81 | 95 | |
| 82 | 96 | if ( ! empty( $args['date_to'] ) ) { |
| 83 | - $date = get_gmt_from_date( date( 'Y-m-d H:i:s', strtotime( $args['date_to'] . ' 23:59:59' ) ) ); | |
| 97 | + $date = get_gmt_from_date( gmdate( 'Y-m-d H:i:s', strtotime( $args['date_to'] . ' 23:59:59' ) ) ); | |
| 84 | 98 | $where .= $wpdb->prepare( " AND DATE($wpdb->stream.created) <= %s", $date ); |
| 85 | 99 | } |
| 86 | 100 | |
| 87 | 101 | if ( ! empty( $args['date_after'] ) ) { |
| 88 | - $date = get_gmt_from_date( date( 'Y-m-d H:i:s', strtotime( $args['date_after'] ) ) ); | |
| 102 | + $date = get_gmt_from_date( gmdate( 'Y-m-d H:i:s', strtotime( $args['date_after'] ) ) ); | |
| 89 | 103 | $where .= $wpdb->prepare( " AND DATE($wpdb->stream.created) > %s", $date ); |
| 90 | 104 | } |
| 91 | 105 | |
| 92 | 106 | if ( ! empty( $args['date_before'] ) ) { |
| 93 | - $date = get_gmt_from_date( date( 'Y-m-d H:i:s', strtotime( $args['date_before'] ) ) ); | |
| 107 | + $date = get_gmt_from_date( gmdate( 'Y-m-d H:i:s', strtotime( $args['date_before'] ) ) ); | |
| 94 | 108 | $where .= $wpdb->prepare( " AND DATE($wpdb->stream.created) < %s", $date ); |
| 95 | 109 | } |
| 96 | 110 | |
| 97 | - if ( ! empty( $args['date'] ) ) { | |
| 98 | - $args['date_from'] = date( 'Y-m-d', strtotime( $args['date'] ) ) . ' 00:00:00'; | |
| 99 | - $args['date_to'] = date( 'Y-m-d', strtotime( $args['date'] ) ) . ' 23:59:59'; | |
| 100 | - } | |
| 101 | - | |
| 102 | 111 | /** |
| 103 | 112 | * PARSE __IN PARAM FAMILY |
| 104 | 113 | */ |
| 105 | 114 | $ins = array(); |
| @@ -193,9 +202,9 @@ | ||
| 193 | 202 | $selects = array(); |
| 194 | 203 | |
| 195 | 204 | if ( ! empty( $fields ) ) { |
| 196 | 205 | foreach ( $fields as $field ) { |
| 197 | - // We'll query the meta table later | |
| 206 | + // We'll query the meta table later. | |
| 198 | 207 | if ( 'meta' === $field ) { |
| 199 | 208 | continue; |
| 200 | 209 | } |
| 201 | 210 | |
| @@ -207,11 +216,21 @@ | ||
| 207 | 216 | |
| 208 | 217 | $select = implode( ', ', $selects ); |
| 209 | 218 | |
| 210 | 219 | /** |
| 220 | + * Filters query WHERE statement as an alternative to filtering | |
| 221 | + * the $query using the hook below. | |
| 222 | + * | |
| 223 | + * @param string $where WHERE statement. | |
| 224 | + * | |
| 225 | + * @return string | |
| 226 | + */ | |
| 227 | + $where = apply_filters( 'wp_stream_db_query_where', $where ); | |
| 228 | + | |
| 229 | + /** | |
| 211 | 230 | * BUILD THE FINAL QUERY |
| 212 | 231 | */ |
| 213 | - $query = "SELECT SQL_CALC_FOUND_ROWS {$select} | |
| 232 | + $query = "SELECT {$select} | |
| 214 | 233 | FROM $wpdb->stream |
| 215 | 234 | {$join} |
| 216 | 235 | WHERE 1=1 {$where} |
| 217 | 236 | {$orderby} |
| @@ -226,50 +245,31 @@ | ||
| 226 | 245 | * @return string |
| 227 | 246 | */ |
| 228 | 247 | $query = apply_filters( 'wp_stream_db_query', $query, $args ); |
| 229 | 248 | |
| 230 | - $result = array(); | |
| 249 | + // Build result count query. | |
| 250 | + $count_query = "SELECT COUNT(*) as found | |
| 251 | + FROM $wpdb->stream | |
| 252 | + {$join} | |
| 253 | + WHERE 1=1 {$where}"; | |
| 254 | + | |
| 231 | 255 | /** |
| 256 | + * Filter allows the result count query to be modified before execution. | |
| 257 | + * | |
| 258 | + * @param string $query | |
| 259 | + * @param array $args | |
| 260 | + * | |
| 261 | + * @return string | |
| 262 | + */ | |
| 263 | + $count_query = apply_filters( 'wp_stream_db_count_query', $count_query, $args ); | |
| 264 | + | |
| 265 | + /** | |
| 232 | 266 | * QUERY THE DATABASE FOR RESULTS |
| 233 | 267 | */ |
| 234 | - $result['items'] = $wpdb->get_results( $query ); // @codingStandardsIgnoreLine $query already prepared | |
| 235 | - $result['count'] = $result['items'] ? absint( $wpdb->get_var( 'SELECT FOUND_ROWS()' ) ) : 0; | |
| 268 | + $result = array( | |
| 269 | + 'items' => $wpdb->get_results( $query ), // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared | |
| 270 | + 'count' => absint( $wpdb->get_var( $count_query ) ), // phpcs:ignore WordPress.DB.PreparedSQL.NotPrepared | |
| 271 | + ); | |
| 236 | 272 | |
| 237 | - // Add meta to the records, when applicable | |
| 238 | - if ( empty( $fields ) || in_array( 'meta', $fields, true ) ) { | |
| 239 | - $result['items'] = $this->add_record_meta( $result['items'] ); | |
| 240 | - } | |
| 241 | - | |
| 242 | 273 | return $result; |
| 243 | - } | |
| 244 | - | |
| 245 | - /** | |
| 246 | - * Add meta to a set of records | |
| 247 | - * | |
| 248 | - * @param array $records | |
| 249 | - * | |
| 250 | - * @return array | |
| 251 | - */ | |
| 252 | - public function add_record_meta( $records ) { | |
| 253 | - global $wpdb; | |
| 254 | - | |
| 255 | - $record_ids = array_map( 'absint', wp_list_pluck( $records, 'ID' ) ); | |
| 256 | - | |
| 257 | - if ( empty( $record_ids ) ) { | |
| 258 | - return (array) $records; | |
| 259 | - } | |
| 260 | - | |
| 261 | - $sql_meta = sprintf( | |
| 262 | - "SELECT * FROM $wpdb->streammeta WHERE record_id IN ( %s )", | |
| 263 | - implode( ',', $record_ids ) | |
| 264 | - ); | |
| 265 | - | |
| 266 | - $meta = $wpdb->get_results( $sql_meta ); // @codingStandardsIgnoreLine prepare okay | |
| 267 | - $ids_f = array_flip( $record_ids ); | |
| 268 | - | |
| 269 | - foreach ( $meta as $meta_record ) { | |
| 270 | - $records[ $ids_f[ $meta_record->record_id ] ]->meta[ $meta_record->meta_key ] = maybe_unserialize( $meta_record->meta_value ); | |
| 271 | - } | |
| 272 | - | |
| 273 | - return (array) $records; | |
| 274 | 274 | } |
| 275 | 275 | } |
