$value ) { if ( is_numeric( $name ) ) { $pairs[] = $value; } else { $pairs[] = sprintf( '%s = %s', $name, $value ); } } return 'SET ' . implode( ', ', $pairs ); } /** * @return array[] * @throws Query_Builder_Exception */ public function prepare_parts_callbacks(): array { return array( 'select' => array( $this, 'prepare_select' ), 'join' => array( $this->view(), 'get_prepared_join' ), 'where' => array( $this, 'prepare_where' ), 'order_by' => array( $this, 'prepare_order_by' ), 'limit' => array( $this, 'prepare_limit' ), ); } /** * @return $this * @throws Query_Builder_Exception */ public function prepare_select(): Query_Builder { $this->select = 'SELECT'; $columns = array(); foreach ( $this->view()->select_columns() as $column ) { $columns[] = $this->view()->column( $column ); } $this->select .= " \r\n\t" . implode( ", \r\n\t", $columns ); $this->select .= "\r\nFROM `{$this->view()->table()}`"; return $this; } /** * @return $this * @throws Query_Builder_Exception */ public function prepare_where(): Query_Builder { $custom = $this->view()->custom_where(); if ( $custom ) { $this->where = $custom; return $this; } $this->where = ( new Query_Conditions_Builder() ) ->set_condition( array( 'type' => Query_Conditions_Builder::TYPE_EQUAL_STATIC, 'values' => array( 1, 1 ), ) ) ->set_view( $this->view() ) ->result(); return $this; } /** * @throws Query_Builder_Exception */ public function prepare_limit(): Query_Builder { if ( ! $this->view()->limit() ) { return $this; } $this->limit = 'LIMIT ' . implode( ', ', $this->view()->limit() ); return $this; } /** * @return Query_Builder * @throws Query_Builder_Exception */ public function prepare_order_by(): Query_Builder { if ( ! $this->view()->order_by() ) { return $this; } $columns = array(); foreach ( $this->view()->order_by() as $column ) { $name = $this->view()->column( $column ); $sort = $column['sort'] ?? ''; $columns[] = trim( "$name $sort" ); } $this->order_by = 'ORDER BY ' . " \r\n\t" . implode( ", \r\n\t", $columns ); return $this; } /** * @throws Query_Builder_Exception */ public function maybe_prepare_parts() { foreach ( $this->prepare_parts_callbacks() as $part_name => $callback ) { if ( ! empty( $this->$part_name ) ) { continue; } call_user_func( $callback, $this ); } } /** * @return array * @throws Query_Builder_Exception */ public function get_parts(): array { $parts = array(); foreach ( $this->prepare_parts_callbacks() as $part_name => $callback ) { if ( empty( $this->$part_name ) ) { continue; } $parts[] = $this->$part_name; } return $parts; } public function debug(): Query_Builder { $this->debug = ! $this->debug; return $this; } private function log( ...$something ) { if ( ! $this->debug ) { return; } if ( 1 === count( $something ) ) { do_action( 'qm/debug', $something[0] ); return; } do_action( 'qm/debug', $something ); } public function prepare_sql(): Query_Builder { if ( ! empty( $this->sql ) ) { return $this; } $this->maybe_prepare_parts(); $this->sql = implode( "\r\n", $this->get_parts() ) . ';'; return $this; } /** * @return \QM_DB|\wpdb */ public function db() { global $wpdb; return $wpdb; } /** * @return string */ public function sql(): string { $this->prepare_sql(); $this->log( $this->sql ); return $this->sql; } /** * @throws Query_Builder_Exception */ public function query_all(): array { // phpcs:ignore WordPress.DB $rows = $this->db()->get_results( $this->sql(), ARRAY_A ); if ( empty( $rows ) ) { throw new Query_Builder_Exception( esc_html( "Empty {$this->view()->table()} rows" ) ); } $response = array(); foreach ( $rows as $row ) { $response[] = $this->view()->get_prepared_row( $row ); } return $response; } /** * @param string $format * * @return \Generator * @throws Query_Builder_Exception */ public function generate_all( string $format = OBJECT ): \Generator { // phpcs:ignore WordPress.DB $this->db()->query( $this->sql() ); if ( empty( $this->db()->last_result ) ) { throw new Query_Builder_Exception( esc_html( "Empty {$this->view()->table()} rows" ) ); } foreach ( (array) $this->db()->last_result as $row ) { yield ( OBJECT === $format ? $row : (array) $row ); } } /** * @throws Query_Builder_Exception */ public function query_one(): array { // phpcs:ignore WordPress.DB $row = $this->db()->get_row( $this->sql(), ARRAY_A ); return $this->view()->get_prepared_row( $row ); } /** * @param $column_offset * @param $row_offset * * @return array * @throws Query_Builder_Exception */ public function query_var( $column_offset = 0, $row_offset = 0 ) { // phpcs:ignore WordPress.DB $row = $this->db()->get_var( $this->sql(), $column_offset, $row_offset ); return $this->view()->get_prepared_row( $row ); } /** * @param int $index * * @return array * @throws Query_Builder_Exception */ public function query_col( $index = 0 ) { $values = $this->db()->get_col( $this->sql(), $index ); return $this->view()->get_prepared_values( $values ); } }