Edit File by line

<?php

[0] Fix | Delete

[1] Fix | Delete

[2] Fix | Delete

namespace NetworkPosts\Components\db;

[3] Fix | Delete

[4] Fix | Delete

use NetworkPosts\db\NetsPostsDbHelper;

[5] Fix | Delete

use WPDB;

[6] Fix | Delete

[7] Fix | Delete

class NetsPostsQuery {

[8] Fix | Delete

private static $TEMP_TABLE_NAME = 'network_posts_results';

[9] Fix | Delete

[10] Fix | Delete

/**

[11] Fix | Delete

* @var WPDB

[12] Fix | Delete

[13] Fix | Delete

protected $db;

[14] Fix | Delete

protected $meta_query;

[15] Fix | Delete

protected $condition_builder;

[16] Fix | Delete

[17] Fix | Delete

protected $blogs = array();

[18] Fix | Delete

[19] Fix | Delete

protected $included_posts = array();

[20] Fix | Delete

protected $excluded_posts = array();

[21] Fix | Delete

protected $limit = 999;

[22] Fix | Delete

protected $offset = 0;

[23] Fix | Delete

protected $meta_keys = array();

[24] Fix | Delete

protected $order_by = false;

[25] Fix | Delete

protected $sort_type = 'DESC';

[26] Fix | Delete

protected $is_random = false;

[27] Fix | Delete

[28] Fix | Delete

protected $load_only_ids = false;

[29] Fix | Delete

protected $posts_without_children = false;

[30] Fix | Delete

/**

[31] Fix | Delete

* @var string

[32] Fix | Delete

[33] Fix | Delete

private $include_posts_table;

[34] Fix | Delete

/**

[35] Fix | Delete

* @var string

[36] Fix | Delete

[37] Fix | Delete

private $exclude_posts_table;

[38] Fix | Delete

private $current_blog_id = 1;

[39] Fix | Delete

[40] Fix | Delete

public function __construct( \WPDB $db ) {

[41] Fix | Delete

$this->set_db( $db );

[42] Fix | Delete

$this->condition_builder = new NetsPostsQueryConditionBuilder( $db->base_prefix );

[43] Fix | Delete

$this->meta_query = new NetsPostsMetaQueryBuilder( $db->base_prefix );

[44] Fix | Delete

}

[45] Fix | Delete

[46] Fix | Delete

public function include_blogs( array $blogs ): void {

[47] Fix | Delete

$this->condition_builder->include_blogs( $blogs );

[48] Fix | Delete

$this->meta_query->include_blogs( $blogs );

[49] Fix | Delete

}

[50] Fix | Delete

[51] Fix | Delete

public function exclude_blog( $blog_id ): void {

[52] Fix | Delete

$this->condition_builder->exclude_blog( $blog_id );

[53] Fix | Delete

}

[54] Fix | Delete

[55] Fix | Delete

public function include_blog_posts( int $blog_id, array $included_posts ): void {

[56] Fix | Delete

$this->condition_builder->include_blog_posts( $blog_id, $included_posts );

[57] Fix | Delete

}

[58] Fix | Delete

[59] Fix | Delete

public function get_included_posts( int $blog_id ): array {

[60] Fix | Delete

return $this->condition_builder->get_included_posts( $blog_id );

[61] Fix | Delete

}

[62] Fix | Delete

[63] Fix | Delete

public function exclude_posts( int $blog_id, array $excluded_posts ): void {

[64] Fix | Delete

$this->condition_builder->exclude_posts( $blog_id, $excluded_posts );

[65] Fix | Delete

}

[66] Fix | Delete

[67] Fix | Delete

public function get_excluded_posts(): array {

[68] Fix | Delete

return $this->condition_builder->get_excluded_posts();

[69] Fix | Delete

}

[70] Fix | Delete

[71] Fix | Delete

public function include_posts_from_table( string $table ) {

[72] Fix | Delete

$this->include_posts_table = $table;

[73] Fix | Delete

}

[74] Fix | Delete

[75] Fix | Delete

public function exclude_posts_from_table( string $table ) {

[76] Fix | Delete

$this->exclude_posts_table = $table;

[77] Fix | Delete

}

[78] Fix | Delete

[79] Fix | Delete

/**

[80] Fix | Delete

* @param int $limit

[81] Fix | Delete

[82] Fix | Delete

public function set_limit( int $limit ): void {

[83] Fix | Delete

$this->limit = $limit;

[84] Fix | Delete

}

[85] Fix | Delete

[86] Fix | Delete

/**

[87] Fix | Delete

* @param int $offset

[88] Fix | Delete

[89] Fix | Delete

public function set_offset( int $offset ): void {

[90] Fix | Delete

$this->condition_builder->set_offset( $offset );

[91] Fix | Delete

}

[92] Fix | Delete

[93] Fix | Delete

[94] Fix | Delete

/**

[95] Fix | Delete

* @param mixed $post_type

[96] Fix | Delete

[97] Fix | Delete

public function set_post_type( $post_type ): void {

[98] Fix | Delete

$this->condition_builder->set_post_type( $post_type );

[99] Fix | Delete

}

[100] Fix | Delete

[101] Fix | Delete

/**

[102] Fix | Delete

* @param string $days

[103] Fix | Delete

[104] Fix | Delete

public function set_days( string $days ): void {

[105] Fix | Delete

$this->condition_builder->set_days( $days );

[106] Fix | Delete

}

[107] Fix | Delete

[108] Fix | Delete

public function filter_after_days( int $after_days ) {

[109] Fix | Delete

$this->condition_builder->filter_after_days( $after_days );

[110] Fix | Delete

}

[111] Fix | Delete

[112] Fix | Delete

/**

[113] Fix | Delete

* @param string $order_by

[114] Fix | Delete

[115] Fix | Delete

public function set_order_by( string $order_by ): void {

[116] Fix | Delete

$this->order_by = $order_by;

[117] Fix | Delete

}

[118] Fix | Delete

[119] Fix | Delete

/**

[120] Fix | Delete

* @param string $type

[121] Fix | Delete

[122] Fix | Delete

public function set_sort_type( string $type ): void {

[123] Fix | Delete

$this->sort_type = $type;

[124] Fix | Delete

}

[125] Fix | Delete

[126] Fix | Delete

/**

[127] Fix | Delete

* @param array $title_keywords

[128] Fix | Delete

[129] Fix | Delete

public function set_title_keywords( array $title_keywords ): void {

[130] Fix | Delete

$this->condition_builder->set_title_keywords( $title_keywords );

[131] Fix | Delete

}

[132] Fix | Delete

[133] Fix | Delete

public function set_author_id( int $author ): void {

[134] Fix | Delete

$this->condition_builder->set_author_id( $author );

[135] Fix | Delete

}

[136] Fix | Delete

[137] Fix | Delete

public function exclude_author_id( int $author ): void {

[138] Fix | Delete

$this->condition_builder->exclude_author_id( $author );

[139] Fix | Delete

}

[140] Fix | Delete

[141] Fix | Delete

public function set_random(): void {

[142] Fix | Delete

$this->is_random = true;

[143] Fix | Delete

}

[144] Fix | Delete

[145] Fix | Delete

[146] Fix | Delete

/**

[147] Fix | Delete

* @param string $acf_date_filter_field

[148] Fix | Delete

[149] Fix | Delete

public function set_acf_date_filter_field( string $acf_date_filter_field ): void {

[150] Fix | Delete

$this->meta_query->set_primary_acf_field( $acf_date_filter_field );

[151] Fix | Delete

}

[152] Fix | Delete

[153] Fix | Delete

/**

[154] Fix | Delete

* @param string $acf_date

[155] Fix | Delete

[156] Fix | Delete

public function filter_acf_date( string $acf_date ): void {

[157] Fix | Delete

$this->meta_query->filter_acf_date( $acf_date );

[158] Fix | Delete

}

[159] Fix | Delete

[160] Fix | Delete

/**

[161] Fix | Delete

* @param string $date

[162] Fix | Delete

[163] Fix | Delete

public function filter_before_acf_date( string $date ): void {

[164] Fix | Delete

$this->meta_query->filter_before_acf_date( $date );

[165] Fix | Delete

}

[166] Fix | Delete

[167] Fix | Delete

/**

[168] Fix | Delete

* @param string $after_acf_date

[169] Fix | Delete

[170] Fix | Delete

public function filter_after_acf_date( string $after_acf_date ): void {

[171] Fix | Delete

$this->meta_query->filter_after_acf_date( $after_acf_date );

[172] Fix | Delete

}

[173] Fix | Delete

[174] Fix | Delete

public function set_primary_acf_field( string $field ): void {

[175] Fix | Delete

$this->meta_query->set_primary_acf_field( $field );

[176] Fix | Delete

}

[177] Fix | Delete

[178] Fix | Delete

public function order_by_acf( string $order ): void {

[179] Fix | Delete

$this->meta_query->order_by_acf( $order );

[180] Fix | Delete

}

[181] Fix | Delete

[182] Fix | Delete

public function get_acf_date_field(): string {

[183] Fix | Delete

return $this->meta_query->get_primary_acf__field();

[184] Fix | Delete

}

[185] Fix | Delete

[186] Fix | Delete

public function without_children() {

[187] Fix | Delete

$this->posts_without_children = true;

[188] Fix | Delete

}

[189] Fix | Delete

[190] Fix | Delete

[191] Fix | Delete

public function get_posts(): array {

[192] Fix | Delete

$results = $this->get_rows();

[193] Fix | Delete

[194] Fix | Delete

if ( $results ) {

[195] Fix | Delete

if ( $this->posts_without_children ) {

[196] Fix | Delete

$ids = array_map( function ( $record ) {

[197] Fix | Delete

return $record['ID'];

[198] Fix | Delete

}, $results );

[199] Fix | Delete

$blogs = $this->condition_builder->get_blogs();

[200] Fix | Delete

foreach( $blogs as $blog ) {

[201] Fix | Delete

$this->current_blog_id = $blog;

[202] Fix | Delete

$parent_ids = $this->get_parents( $ids );

[203] Fix | Delete

[204] Fix | Delete

$results = array_filter( $results, function ( $post ) use ( $parent_ids ) {

[205] Fix | Delete

return ! array_has_value( $post['ID'], $parent_ids );

[206] Fix | Delete

} );

[207] Fix | Delete

$results = array_values( $results );

[208] Fix | Delete

}

[209] Fix | Delete

}

[210] Fix | Delete

}

[211] Fix | Delete

[212] Fix | Delete

return $results;

[213] Fix | Delete

}

[214] Fix | Delete

[215] Fix | Delete

protected function get_rows(): array {

[216] Fix | Delete

[217] Fix | Delete

$table = null;

[218] Fix | Delete

$blogs = $this->condition_builder->get_blogs();

[219] Fix | Delete

foreach ( $blogs as $blog ) {

[220] Fix | Delete

$this->current_blog_id = $blog;

[221] Fix | Delete

$this->meta_query->set_current_blog( $blog );

[222] Fix | Delete

$this->condition_builder->set_current_blog( $blog );

[223] Fix | Delete

$query = $this->build_query();

[224] Fix | Delete

if( $table ){

[225] Fix | Delete

$this->insert_temp_data( $table, $query );

[226] Fix | Delete

} else {

[227] Fix | Delete

$table = $this->create_temporary_table( $query );

[228] Fix | Delete

}

[229] Fix | Delete

}

[230] Fix | Delete

if( $table ) {

[231] Fix | Delete

if( $this->load_only_ids && $this->order_by ) {

[232] Fix | Delete

$query = 'SELECT ID, blog_id FROM ' . $table . ' GROUP BY ID, blog_id';

[233] Fix | Delete

} else {

[234] Fix | Delete

$query = 'SELECT * FROM ' . $table . ' GROUP BY ID, blog_id';

[235] Fix | Delete

}

[236] Fix | Delete

if ( $this->meta_query->has_order() ) {

[237] Fix | Delete

$query .= ' ' . $this->order_posts_by_acf();

[238] Fix | Delete

} else if ( $this->is_random ) {

[239] Fix | Delete

$query .= ' ' . $this->build_random_order();

[240] Fix | Delete

} elseif ( $this->order_by ) {

[241] Fix | Delete

$query .= ' ' . $this->build_order_by();

[242] Fix | Delete

}

[243] Fix | Delete

$query .= ' ' . $this->build_limit();

[244] Fix | Delete

[245] Fix | Delete

$results = $this->db->get_results( $query, ARRAY_A );

[246] Fix | Delete

$this->drop_temp_table( $table );

[247] Fix | Delete

return $results;

[248] Fix | Delete

}

[249] Fix | Delete

return array();

[250] Fix | Delete

}

[251] Fix | Delete

[252] Fix | Delete

public function get_ids(): array {

[253] Fix | Delete

$this->load_only_ids = true;

[254] Fix | Delete

$results = $this->get_rows();

[255] Fix | Delete

[256] Fix | Delete

if ( $results ) {

[257] Fix | Delete

if ( $this->posts_without_children ) {

[258] Fix | Delete

$blogs = $this->condition_builder->get_blogs();

[259] Fix | Delete

foreach ( $blogs as $blog ) {

[260] Fix | Delete

$this->current_blog_id = $blog;

[261] Fix | Delete

$parent_ids = $this->get_parents( $results );

[262] Fix | Delete

$results = array_filter( $results, function ( $id ) use ( $parent_ids ) {

[263] Fix | Delete

return ! array_has_value( $id, $parent_ids );

[264] Fix | Delete

} );

[265] Fix | Delete

}

[266] Fix | Delete

}

[267] Fix | Delete

}

[268] Fix | Delete

[269] Fix | Delete

return $results;

[270] Fix | Delete

}

[271] Fix | Delete

[272] Fix | Delete

protected function set_db( \WPDB $db ): void {

[273] Fix | Delete

$this->db = $db;

[274] Fix | Delete

}

[275] Fix | Delete

[276] Fix | Delete

protected function build_query(): string {

[277] Fix | Delete

$posts_table = $this->get_posts_table();

[278] Fix | Delete

$blog_id = $this->current_blog_id;

[279] Fix | Delete

$collate = 'utf8';

[280] Fix | Delete

if ( $this->load_only_ids ) {

[281] Fix | Delete

$columns = array( 'ID', "$blog_id as blog_id" );

[282] Fix | Delete

if( $this->order_by ){

[283] Fix | Delete

$columns[] = $this->order_by;

[284] Fix | Delete

}

[285] Fix | Delete

} else {

[286] Fix | Delete

$columns = array(

[287] Fix | Delete

'ID',

[288] Fix | Delete

"$blog_id as blog_id",

[289] Fix | Delete

'post_title',

[290] Fix | Delete

'post_excerpt',

[291] Fix | Delete

'post_content',

[292] Fix | Delete

'post_author',

[293] Fix | Delete

'post_date',

[294] Fix | Delete

'post_type',

[295] Fix | Delete

);

[296] Fix | Delete

}

[297] Fix | Delete

$join_tables = '';

[298] Fix | Delete

if ( $this->include_posts_table ) {

[299] Fix | Delete

$include_table = $this->include_posts_table;

[300] Fix | Delete

$join_tables .= " INNER JOIN $include_table ON ($include_table.post_id = $posts_table.ID AND $include_table.blog_id = $blog_id)";

[301] Fix | Delete

}

[302] Fix | Delete

if ( $this->exclude_posts_table ) {

[303] Fix | Delete

$exclude_table = $this->exclude_posts_table;

[304] Fix | Delete

$join_tables .= " OUTER JOIN $exclude_table ON ($exclude_table.post_id = $posts_table.ID AND $exclude_table.blog_id = $blog_id)";

[305] Fix | Delete

}

[306] Fix | Delete

if ( $this->meta_query->is_acf_needed() ) {

[307] Fix | Delete

[308] Fix | Delete

$acf_columns = $this->meta_query->get_columns();

[309] Fix | Delete

foreach ( $acf_columns as $key => $value ){

[310] Fix | Delete

if( is_int( $key ) ) {

[311] Fix | Delete

$columns[] = $value;

[312] Fix | Delete

} else {

[313] Fix | Delete

$columns[] = $key . ' as `' . $value . '`';

[314] Fix | Delete

}

[315] Fix | Delete

}

[316] Fix | Delete

//$columns = array_merge( $columns, $acf_columns );

[317] Fix | Delete

$join_tables .= $this->meta_query->build_join( $posts_table, 'ID' );

[318] Fix | Delete

}

[319] Fix | Delete

[320] Fix | Delete

$columns_string = join( ',', $columns );

[321] Fix | Delete

$query = "SELECT $columns_string FROM ${posts_table} ${join_tables}";

[322] Fix | Delete

$where = $this->build_where_condition();

[323] Fix | Delete

if ( $where ) {

[324] Fix | Delete

$query .= ' WHERE ' . $where;

[325] Fix | Delete

}

[326] Fix | Delete

return $query;

[327] Fix | Delete

}

[328] Fix | Delete

[329] Fix | Delete

protected function get_posts_table(): string {

[330] Fix | Delete

return NetsPostsDbHelper::make_table_name( $this->db->base_prefix, 'posts', $this->current_blog_id );

[331] Fix | Delete

}

[332] Fix | Delete

[333] Fix | Delete

protected function build_where_condition(): string {

[334] Fix | Delete

$condition = $this->condition_builder->build();

[335] Fix | Delete

if ( $this->meta_query->is_acf_needed() ) {

[336] Fix | Delete

if( $condition ){

[337] Fix | Delete

$condition .= ' AND ';

[338] Fix | Delete

}

[339] Fix | Delete

$condition .= $this->meta_query->build_where_condition();

[340] Fix | Delete

}

[341] Fix | Delete

return $condition;

[342] Fix | Delete

}

[343] Fix | Delete

[344] Fix | Delete

protected function build_random_order(): string {

[345] Fix | Delete

return 'ORDER BY RAND()';

[346] Fix | Delete

}

[347] Fix | Delete

[348] Fix | Delete

protected function build_order_by(): string {

[349] Fix | Delete

return 'ORDER BY ' . $this->order_by . ' ' . $this->sort_type;

[350] Fix | Delete

}

[351] Fix | Delete

[352] Fix | Delete

protected function build_limit(): string {

[353] Fix | Delete

return 'LIMIT ' . $this->offset . ', ' . $this->limit;

[354] Fix | Delete

}

[355] Fix | Delete

[356] Fix | Delete

protected function order_posts_by_acf(): string {

[357] Fix | Delete

return $this->meta_query->build_order_by();

[358] Fix | Delete

}

[359] Fix | Delete

[360] Fix | Delete

protected function create_temporary_table( string $query ): string {

[361] Fix | Delete

$table_name = $this->db->prefix . self::$TEMP_TABLE_NAME;

[362] Fix | Delete

$create_table_query = NetsPostsDbHelper::build_temporary_table_query( $table_name, $query );

[363] Fix | Delete

$this->db->query( $create_table_query );

[364] Fix | Delete

return $table_name;

[365] Fix | Delete

}

[366] Fix | Delete

[367] Fix | Delete

protected function insert_temp_data( string $table, string $query ) {

[368] Fix | Delete

$insert_query = "INSERT INTO $table $query";

[369] Fix | Delete

$this->db->query( $insert_query );

[370] Fix | Delete

}

[371] Fix | Delete

[372] Fix | Delete

protected function drop_temp_table( string $table ): void {

[373] Fix | Delete

$this->db->query( 'DROP TABLE IF EXISTS ' . $table );

[374] Fix | Delete

}

[375] Fix | Delete

[376] Fix | Delete

/**

[377] Fix | Delete

* @param int[] $post_ids

[378] Fix | Delete

[379] Fix | Delete

* @return int[]

[380] Fix | Delete

[381] Fix | Delete

protected function get_parents( array $post_ids ): array {

[382] Fix | Delete

$query = "SELECT post_parent FROM " . $this->get_posts_table() . ' WHERE ';

[383] Fix | Delete

$query .= 'post_parent IN (' . join( ',', $post_ids ) . ')';

[384] Fix | Delete

$post_type = $this->condition_builder->get_post_type();

[385] Fix | Delete

if( ! empty( $post_type ) ) {

[386] Fix | Delete

$post_type = array_map( function( $type ){ return '"' . $type . '"'; }, $post_type );

[387] Fix | Delete

$query .= ' AND post_type IN (' . implode( ',', $post_type ) . ')';

[388] Fix | Delete

}

[389] Fix | Delete

$parents = $this->db->get_results( $query, ARRAY_A );

[390] Fix | Delete

[391] Fix | Delete

$ids = array_map( function ( $post ) {

[392] Fix | Delete

return $post['post_parent'];

[393] Fix | Delete

}, $parents );

[394] Fix | Delete

return array_unique( $ids );

[395] Fix | Delete

}

[396] Fix | Delete

[397] Fix | Delete

public function exclude_protected_posts() {

[398] Fix | Delete

$this->condition_builder->exclude_protected();

[399] Fix | Delete

}

[400] Fix | Delete

}

[401] Fix | Delete