<?php

[0] Fix | Delete

[1] Fix | Delete

namespace Elementor\Modules\WpRest\Classes;

[2] Fix | Delete

[3] Fix | Delete

use Elementor\Core\Utils\Collection;

[4] Fix | Delete

use Elementor\Modules\WpRest\Base\Query as Base;

[5] Fix | Delete

[6] Fix | Delete

if ( ! defined( 'ABSPATH' ) ) {

[7] Fix | Delete

exit; // Exit if accessed directly.

[8] Fix | Delete

}

[9] Fix | Delete

[10] Fix | Delete

class Post_Query extends Base {

[11] Fix | Delete

const ENDPOINT = 'post';

[12] Fix | Delete

const SEARCH_FILTER_ACCEPTED_ARGS = 2;

[13] Fix | Delete

const DEFAULT_FORBIDDEN_POST_TYPES = [ 'e-floating-buttons', 'e-landing-page', 'elementor_library', 'attachment', 'revision', 'nav_menu_item', 'custom_css', 'customize_changeset' ];

[14] Fix | Delete

[15] Fix | Delete

/**

[16] Fix | Delete

* @param string $search_term The original search query.

[17] Fix | Delete

* @param \WP_Query $wp_query The WP_Query instance.

[18] Fix | Delete

* @return string Modified search query.

[19] Fix | Delete

*/

[20] Fix | Delete

public function customize_post_query( string $search_term, \WP_Query $wp_query ) {

[21] Fix | Delete

$term = $wp_query->get( 'search_term' ) ?? '';

[22] Fix | Delete

$is_custom_search = $wp_query->get( 'custom_search' ) ?? false;

[23] Fix | Delete

[24] Fix | Delete

if ( $is_custom_search && ! empty( $term ) ) {

[25] Fix | Delete

$escaped = esc_sql( $term );

[26] Fix | Delete

$search_term .= ' AND (';

[27] Fix | Delete

$search_term .= "post_title LIKE '%{$escaped}%'";

[28] Fix | Delete

if ( ctype_digit( $term ) ) {

[29] Fix | Delete

$search_term .= ' OR ID = ' . intval( $term );

[30] Fix | Delete

} else {

[31] Fix | Delete

$search_term .= " OR ID LIKE '%{$escaped}%'";

[32] Fix | Delete

}

[33] Fix | Delete

$search_term .= ')';

[34] Fix | Delete

}

[35] Fix | Delete

[36] Fix | Delete

return $search_term;

[37] Fix | Delete

}

[38] Fix | Delete

[39] Fix | Delete

/**

[40] Fix | Delete

* @param \WP_REST_Request $request

[41] Fix | Delete

* @return \WP_REST_Response

[42] Fix | Delete

*/

[43] Fix | Delete

protected function get( \WP_REST_Request $request ) {

[44] Fix | Delete

$params = $request->get_params();

[45] Fix | Delete

$term = trim( $params[ self::SEARCH_TERM_KEY ] ?? '' );

[46] Fix | Delete

[47] Fix | Delete

if ( empty( $term ) ) {

[48] Fix | Delete

return new \WP_REST_Response( [

[49] Fix | Delete

'success' => true,

[50] Fix | Delete

'data' => [

[51] Fix | Delete

'value' => [],

[52] Fix | Delete

],

[53] Fix | Delete

], 200 );

[54] Fix | Delete

}

[55] Fix | Delete

[56] Fix | Delete

$keys_format_map = $params[ self::KEYS_CONVERSION_MAP_KEY ];

[57] Fix | Delete

$requested_count = $params[ self::ITEMS_COUNT_KEY ] ?? 0;

[58] Fix | Delete

$validated_count = max( $requested_count, 1 );

[59] Fix | Delete

$post_count = min( $validated_count, self::MAX_RESPONSE_COUNT );

[60] Fix | Delete

$is_public_only = $params[ self::IS_PUBLIC_KEY ] ?? true;

[61] Fix | Delete

$post_types = $this->get_post_types_from_params( $params );

[62] Fix | Delete

[63] Fix | Delete

$query_args = [

[64] Fix | Delete

'post_type' => array_keys( $post_types ),

[65] Fix | Delete

'numberposts' => $post_count,

[66] Fix | Delete

'suppress_filters' => false,

[67] Fix | Delete

'custom_search' => true,

[68] Fix | Delete

'search_term' => $term,

[69] Fix | Delete

'post_status' => $is_public_only ? 'publish' : 'any',

[70] Fix | Delete

'orderby' => 'ID',

[71] Fix | Delete

'order' => 'ASC',

[72] Fix | Delete

];

[73] Fix | Delete

[74] Fix | Delete

if ( ! empty( $params[ self::META_QUERY_KEY ] ) && is_array( $params[ self::META_QUERY_KEY ] ) ) {

[75] Fix | Delete

$query_args['meta_query'] = $params[ self::META_QUERY_KEY ];

[76] Fix | Delete

}

[77] Fix | Delete

[78] Fix | Delete

if ( ! empty( $params[ self::TAX_QUERY_KEY ] ) && is_array( $params[ self::TAX_QUERY_KEY ] ) ) {

[79] Fix | Delete

$query_args['tax_query'] = $params[ self::TAX_QUERY_KEY ];

[80] Fix | Delete

}

[81] Fix | Delete

[82] Fix | Delete

$this->add_filter_to_customize_query();

[83] Fix | Delete

$posts = new Collection( get_posts( $query_args ) );

[84] Fix | Delete

$this->remove_filter_to_customize_query();

[85] Fix | Delete

[86] Fix | Delete

$post_type_labels = ( new Collection( $post_types ) )

[87] Fix | Delete

->map( function ( $pt ) {

[88] Fix | Delete

return $pt->label;

[89] Fix | Delete

} )

[90] Fix | Delete

->all();

[91] Fix | Delete

[92] Fix | Delete

return new \WP_REST_Response( [

[93] Fix | Delete

'success' => true,

[94] Fix | Delete

'data' => [

[95] Fix | Delete

'value' => $posts

[96] Fix | Delete

->map( function ( $post ) use ( $keys_format_map, $post_type_labels ) {

[97] Fix | Delete

$post_object = (array) $post;

[98] Fix | Delete

[99] Fix | Delete

if ( isset( $post_object['post_type'] ) ) {

[100] Fix | Delete

$pt_name = $post_object['post_type'];

[101] Fix | Delete

if ( isset( $post_type_labels[ $pt_name ] ) ) {

[102] Fix | Delete

$post_object['post_type'] = $post_type_labels[ $pt_name ];

[103] Fix | Delete

}

[104] Fix | Delete

}

[105] Fix | Delete

[106] Fix | Delete

return $this->translate_keys( $post_object, $keys_format_map );

[107] Fix | Delete

} )

[108] Fix | Delete

->all(),

[109] Fix | Delete

],

[110] Fix | Delete

], 200 );

[111] Fix | Delete

}

[112] Fix | Delete

[113] Fix | Delete

/**

[114] Fix | Delete

* @return void

[115] Fix | Delete

*/

[116] Fix | Delete

private function add_filter_to_customize_query() {

[117] Fix | Delete

$priority = self::SEARCH_FILTER_PRIORITY;

[118] Fix | Delete

$accepted_args = self::SEARCH_FILTER_ACCEPTED_ARGS;

[119] Fix | Delete

[120] Fix | Delete

add_filter( 'posts_search', [ $this, 'customize_post_query' ], $priority, $accepted_args );

[121] Fix | Delete

}

[122] Fix | Delete

[123] Fix | Delete

/**

[124] Fix | Delete

* @return void

[125] Fix | Delete

*/

[126] Fix | Delete

private function remove_filter_to_customize_query() {

[127] Fix | Delete

$priority = self::SEARCH_FILTER_PRIORITY;

[128] Fix | Delete

$accepted_args = self::SEARCH_FILTER_ACCEPTED_ARGS;

[129] Fix | Delete

[130] Fix | Delete

remove_filter( 'posts_search', [ $this, 'customize_post_query' ], $priority, $accepted_args );

[131] Fix | Delete

}

[132] Fix | Delete

[133] Fix | Delete

protected function get_endpoint_registration_args(): array {

[134] Fix | Delete

return [

[135] Fix | Delete

self::INCLUDED_TYPE_KEY => [

[136] Fix | Delete

'description' => 'Included post types',

[137] Fix | Delete

'type' => 'array',

[138] Fix | Delete

'required' => false,

[139] Fix | Delete

'default' => null,

[140] Fix | Delete

'sanitize_callback' => fn ( ...$args ) => self::sanitize_string_array( ...$args ),

[141] Fix | Delete

],

[142] Fix | Delete

self::EXCLUDED_TYPE_KEY => [

[143] Fix | Delete

'description' => 'Post type to exclude',

[144] Fix | Delete

'type' => 'array',

[145] Fix | Delete

'required' => false,

[146] Fix | Delete

'default' => self::DEFAULT_FORBIDDEN_POST_TYPES,

[147] Fix | Delete

'sanitize_callback' => fn ( ...$args ) => self::sanitize_string_array( ...$args ),

[148] Fix | Delete

],

[149] Fix | Delete

self::SEARCH_TERM_KEY => [

[150] Fix | Delete

'description' => 'Posts to search',

[151] Fix | Delete

'type' => 'string',

[152] Fix | Delete

'required' => false,

[153] Fix | Delete

'default' => '',

[154] Fix | Delete

'sanitize_callback' => 'sanitize_text_field',

[155] Fix | Delete

],

[156] Fix | Delete

self::KEYS_CONVERSION_MAP_KEY => [

[157] Fix | Delete

'description' => 'Specify keys to extract and convert, i.e. ["key_1" => "new_key_1"].',

[158] Fix | Delete

'type' => 'array',

[159] Fix | Delete

'required' => false,

[160] Fix | Delete

'default' => [

[161] Fix | Delete

'ID' => 'id',

[162] Fix | Delete

'post_title' => 'label',

[163] Fix | Delete

'post_type' => 'groupLabel',

[164] Fix | Delete

],

[165] Fix | Delete

'sanitize_callback' => fn ( ...$args ) => self::sanitize_string_array( ...$args ),

[166] Fix | Delete

],

[167] Fix | Delete

self::ITEMS_COUNT_KEY => [

[168] Fix | Delete

'description' => 'Posts per page',

[169] Fix | Delete

'type' => 'integer',

[170] Fix | Delete

'required' => false,

[171] Fix | Delete

'default' => self::MAX_RESPONSE_COUNT,

[172] Fix | Delete

],

[173] Fix | Delete

self::IS_PUBLIC_KEY => [

[174] Fix | Delete

'description' => 'Whether to include only public post types',

[175] Fix | Delete

'type' => 'boolean',

[176] Fix | Delete

'required' => false,

[177] Fix | Delete

'default' => true,

[178] Fix | Delete

],

[179] Fix | Delete

self::META_QUERY_KEY => [

[180] Fix | Delete

'description' => 'WP_Query meta_query array',

[181] Fix | Delete

'type' => 'array',

[182] Fix | Delete

'required' => false,

[183] Fix | Delete

'default' => null,

[184] Fix | Delete

'sanitize_callback' => fn ( ...$args ) => self::sanitize_string_array( ...$args ),

[185] Fix | Delete

],

[186] Fix | Delete

self::TAX_QUERY_KEY => [

[187] Fix | Delete

'description' => 'WP_Query tax_query array',

[188] Fix | Delete

'type' => 'array',

[189] Fix | Delete

'required' => false,

[190] Fix | Delete

'default' => null,

[191] Fix | Delete

'sanitize_callback' => fn ( ...$args ) => self::sanitize_string_array( ...$args ),

[192] Fix | Delete

],

[193] Fix | Delete

];

[194] Fix | Delete

}

[195] Fix | Delete

[196] Fix | Delete

protected static function get_allowed_param_keys(): array {

[197] Fix | Delete

return [

[198] Fix | Delete

self::EXCLUDED_TYPE_KEY,

[199] Fix | Delete

self::INCLUDED_TYPE_KEY,

[200] Fix | Delete

self::KEYS_CONVERSION_MAP_KEY,

[201] Fix | Delete

self::META_QUERY_KEY,

[202] Fix | Delete

self::TAX_QUERY_KEY,

[203] Fix | Delete

self::IS_PUBLIC_KEY,

[204] Fix | Delete

self::ITEMS_COUNT_KEY,

[205] Fix | Delete

];

[206] Fix | Delete

}

[207] Fix | Delete

[208] Fix | Delete

protected static function get_keys_to_encode(): array {

[209] Fix | Delete

return [

[210] Fix | Delete

self::EXCLUDED_TYPE_KEY,

[211] Fix | Delete

self::INCLUDED_TYPE_KEY,

[212] Fix | Delete

self::KEYS_CONVERSION_MAP_KEY,

[213] Fix | Delete

self::META_QUERY_KEY,

[214] Fix | Delete

self::TAX_QUERY_KEY,

[215] Fix | Delete

];

[216] Fix | Delete

}

[217] Fix | Delete

[218] Fix | Delete

private function get_post_types_from_params( $params ) {

[219] Fix | Delete

$included_types = $params[ self::INCLUDED_TYPE_KEY ];

[220] Fix | Delete

$excluded_types = $params[ self::EXCLUDED_TYPE_KEY ];

[221] Fix | Delete

$post_type_query_args = [

[222] Fix | Delete

'public' => true,

[223] Fix | Delete

];

[224] Fix | Delete

[225] Fix | Delete

$post_types = get_post_types( $post_type_query_args, 'objects' );

[226] Fix | Delete

[227] Fix | Delete

return Collection::make( $post_types )

[228] Fix | Delete

->filter( function ( $slug, $post_type ) use ( $included_types, $excluded_types ) {

[229] Fix | Delete

return ( empty( $included_types ) || in_array( $post_type, $included_types ) ) &&

[230] Fix | Delete

( empty( $excluded_types ) || ! in_array( $post_type, $excluded_types ) );

[231] Fix | Delete

} )->all();

[232] Fix | Delete

}

[233] Fix | Delete

}

[234] Fix | Delete

[235] Fix | Delete