array( 'callback' => array( $this, 'build_equal_static' ), ), self::TYPE_EQUAL => array( 'callback' => array( $this, 'build_equal_column' ), ), self::TYPE_EQUAL_COLUMNS => array( 'callback' => array( $this, 'build_equal_two_columns' ), ), self::TYPE_LIKE => array( 'callback' => array( $this, 'build_like' ), ), self::TYPE_NOT_LIKE => array( 'callback' => array( $this, 'build_not_like' ), ), self::TYPE_MORE_STATIC => array( 'callback' => array( $this, 'build_more_static' ), ), self::TYPE_LESS_STATIC => array( 'callback' => array( $this, 'build_less_static' ), ), self::TYPE_IN => array( 'callback' => array( $this, 'build_in' ), ), /** * @since 3.1.0 */ self::TYPE_NOT_EQUAL => array( 'callback' => array( $this, 'build_not_equal_column' ), ), self::TYPE_LIKE_END => array( 'callback' => array( $this, 'build_like_end' ), ), self::TYPE_NOT_LIKE_END => array( 'callback' => array( $this, 'build_not_like_end' ), ), self::TYPE_LESS_OR_EQUAL_STATIC => array( 'callback' => array( $this, 'build_less_or_equal_static' ), ), self::TYPE_MORE_OR_EQUAL_STATIC => array( 'callback' => array( $this, 'build_more_or_equal_static' ), ), /** * @since 3.2.0 */ self::TYPE_IS_NULL => array( 'callback' => array( $this, 'build_is_null' ), ), ); } /** * @param array|Query_Conditions_Builder $condition * * @return $this */ public function set_condition( $condition ): Query_Conditions_Builder { if ( ! is_array( $condition ) && ! ( $condition instanceof Query_Conditions_Builder ) ) { return $this; } $this->conditions[] = $condition; return $this; } public function set_conditions( array $conditions ): Query_Conditions_Builder { foreach ( $conditions as $condition ) { $this->set_condition( $condition ); } return $this; } public function after_set_view() { try { $view = $this->view(); } catch ( Query_Builder_Exception $exception ) { return; } $this->set_conditions( $view->conditions() ); } /** * @return string */ public function result(): string { if ( ! $this->conditions ) { return ''; } return 'WHERE ' . $this->prepare_conditions(); } /** * @return string */ public function prepare_conditions(): string { $this->generator = $this->generate_conditions(); $prepared = array(); foreach ( $this->generator as $condition ) { $prepared[] = $this->prepare(); } return implode( sprintf( "\r\n\t%s ", $this->relation_type ), $prepared ); } /** * @return mixed */ public function prepare() { $condition = $this->generator->current(); if ( $condition instanceof Query_Conditions_Builder ) { return sprintf( "(\r\n%s\r\n)", $condition->prepare_conditions() ); } $type = $this->get_condition_type(); $values = $this->get_condition_values(); $callback = $this->get_types()[ $type ]['callback']; return call_user_func( $callback, $values[0], $values[1] ?? false ); } /** * @return string */ private function get_condition_type(): string { $condition = $this->generator->current(); $type = $condition['type'] ?? false; if ( ! $type || ! in_array( $type, array_keys( $this->get_types() ), true ) ) { // phpcs:ignore WordPress.Security.EscapeOutput.OutputNotEscaped _doing_it_wrong( __METHOD__, "Undefined condition type: $type", '2.0.0' ); wp_die(); } return $type; } public function build_equal_static( $first, $second ): string { global $wpdb; return $wpdb->prepare( '%s = %s', $first, $second ); } /** * @param $column_name * @param $second * * @return string */ public function build_equal_column( $column_name, $second ): string { global $wpdb; $format = $this->get_format_value( $second ); $format_col = $this->get_format_column(); $column_name = Db_Tools::sanitize_column( $column_name ); try { $column_name = $this->view()->column( $column_name ); } catch ( Query_Builder_Exception $exception ) { // silence } return $wpdb->prepare( // phpcs:ignore WordPress.DB sprintf( $format_col, $column_name ) . " = {$format}", $second ); } /** * @param $column_name * @param $second * * @return string * @since 3.1.0 */ public function build_not_equal_column( $column_name, $second ): string { global $wpdb; $format = $this->get_format_value( $second ); $format_col = $this->get_format_column(); $column_name = Db_Tools::sanitize_column( $column_name ); try { $column_name = $this->view()->column( $column_name ); } catch ( Query_Builder_Exception $exception ) { // silence } return $wpdb->prepare( // phpcs:ignore WordPress.DB sprintf( $format_col, $column_name ) . " != {$format}", $second ); } /** * @param $column_name * @param $second * * @return string */ public function build_equal_two_columns( $column_name, $second ): string { $column_name = Db_Tools::sanitize_column( $column_name ); $second = Db_Tools::sanitize_column( $second ); try { $column_name = $this->view()->column( $column_name ); $second = $this->view()->column( $second ); } catch ( Query_Builder_Exception $exception ) { // silence } return "{$column_name} = {$second}"; } /** * @param $column_name * @param $second * * @return string */ public function build_like( $column_name, $second ): string { $second = esc_sql( $second ); $column_name = Db_Tools::sanitize_column( $column_name ); try { $column_name = $this->view()->column( $column_name ); } catch ( Query_Builder_Exception $exception ) { // silence } return "{$column_name} LIKE '%{$second}%'"; } /** * @param $column_name * @param $second * * @return string */ public function build_not_like( $column_name, $second ): string { $second = esc_sql( $second ); $column_name = Db_Tools::sanitize_column( $column_name ); try { $column_name = $this->view()->column( $column_name ); } catch ( Query_Builder_Exception $exception ) { // silence } return "{$column_name} NOT LIKE '%{$second}%'"; } /** * @param $column_name * @param $second * * @return string */ public function build_like_end( $column_name, $second ): string { $second = esc_sql( $second ); $column_name = Db_Tools::sanitize_column( $column_name ); try { $column_name = $this->view()->column( $column_name ); } catch ( Query_Builder_Exception $exception ) { // silence } return "{$column_name} LIKE '{$second}%'"; } /** * @param $column_name * @param $second * * @return string */ public function build_not_like_end( $column_name, $second ): string { $second = esc_sql( $second ); $column_name = Db_Tools::sanitize_column( $column_name ); try { $column_name = $this->view()->column( $column_name ); } catch ( Query_Builder_Exception $exception ) { // silence } return "{$column_name} NOT LIKE '{$second}%'"; } /** * @param $column_name * @param $second * * @return string */ public function build_more_static( $column_name, $second ): string { global $wpdb; $format = $this->get_format_value( $second ); $format_col = $this->get_format_column(); $column_name = Db_Tools::sanitize_column( $column_name ); try { $column_name = $this->view()->column( $column_name ); } catch ( Query_Builder_Exception $exception ) { // silence } // phpcs:disable WordPress.DB return $wpdb->prepare( sprintf( $format_col, $column_name ) . ' > ' . $format, $second ); // phpcs:enable WordPress.DB } /** * @param $column_name * @param $second * * @return string */ public function build_more_or_equal_static( $column_name, $second ): string { global $wpdb; $format = $this->get_format_value( $second ); $format_col = $this->get_format_column(); $column_name = Db_Tools::sanitize_column( $column_name ); try { $column_name = $this->view()->column( $column_name ); } catch ( Query_Builder_Exception $exception ) { // silence } // phpcs:disable WordPress.DB return $wpdb->prepare( sprintf( $format_col, $column_name ) . ' >= ' . $format, $second ); // phpcs:enable WordPress.DB } /** * .wrap > *:not(h1, #poststuff) { * display: none; * } */ /** * @param $column_name * @param $second * * @return string */ public function build_less_static( $column_name, $second ): string { global $wpdb; $format = $this->get_format_value( $second ); $format_col = $this->get_format_column(); $column_name = Db_Tools::sanitize_column( $column_name ); try { $column_name = $this->view()->column( $column_name ); } catch ( Query_Builder_Exception $exception ) { // silence } // phpcs:disable WordPress.DB return $wpdb->prepare( sprintf( $format_col, $column_name ) . ' < ' . $format, $second ); // phpcs:enable WordPress.DB } /** * @param $column_name * @param $second * * @return string */ public function build_less_or_equal_static( $column_name, $second ): string { global $wpdb; $format = $this->get_format_value( $second ); $format_col = $this->get_format_column(); $column_name = Db_Tools::sanitize_column( $column_name ); try { $column_name = $this->view()->column( $column_name ); } catch ( Query_Builder_Exception $exception ) { // silence } // phpcs:disable WordPress.DB return $wpdb->prepare( sprintf( $format_col, $column_name ) . ' <= ' . $format, $second ); // phpcs:enable WordPress.DB } /** * @param $column_name * @param $second * * @return string */ public function build_in( $column_name, $second ): string { if ( ! is_array( $second ) ) { $second = array( $second ); } $in_list = array(); foreach ( $second as $in_item ) { if ( ! is_scalar( $in_item ) ) { continue; } $in_list[] = is_numeric( $in_item ) ? $in_item : sprintf( "'%s'", sanitize_key( $in_item ) ); } // Fix: Return FALSE condition when IN list is empty to prevent SQL error if ( empty( $in_list ) ) { $column_name = Db_Tools::sanitize_column( $column_name ); try { $column_name = $this->view()->column( $column_name ); } catch ( Query_Builder_Exception $exception ) { // silence } return "{$column_name} IN (NULL)"; } $right_part = implode( ', ', $in_list ); $column_name = Db_Tools::sanitize_column( $column_name ); try { $column_name = $this->view()->column( $column_name ); } catch ( Query_Builder_Exception $exception ) { // silence } return "{$column_name} IN ({$right_part})"; } /** * @param $column_name * * @return string * @since 3.2.0 */ public function build_is_null( $column_name ): string { $column_name = Db_Tools::sanitize_column( $column_name ); try { $column_name = $this->view()->column( $column_name ); } catch ( Query_Builder_Exception $exception ) { // silence } // phpcs:disable WordPress.DB return sprintf( '%s IS NULL', $column_name ); // phpcs:enable WordPress.DB } /** * @return array */ public function get_condition_values(): array { $condition = $this->generator->current(); return $condition['values'] ?? array(); } /** * @return $this * @since 3.1.0 */ public function set_relation_and(): Query_Conditions_Builder { return $this->set_relation_type( 'AND' ); } /** * @return $this * @since 3.1.0 */ public function set_relation_or(): Query_Conditions_Builder { return $this->set_relation_type( 'OR' ); } protected function set_relation_type( string $type ): Query_Conditions_Builder { if ( ! in_array( $type, self::RELATIONS_TYPES, true ) ) { return $this; } $this->relation_type = $type; return $this; } protected function generate_conditions(): \Generator { foreach ( $this->conditions as $condition ) { yield $condition; } } protected function get_format_value( $value ) { /** @var array $condition */ $condition = $this->generator->current(); if ( ! empty( $condition['format'] ) ) { return $condition['format']; } return is_numeric( $value ) ? '%d' : '%s'; } protected function get_format_column() { /** @var array $condition */ $condition = $this->generator->current(); return empty( $condition['format_col'] ) ? '%s' : $condition['format_col']; } }