Merge pull request #774 from xwp/feature/issue-770

Add helper function to DB class

Author: Luke Carbis

classes/class-cli.php

@@ -111,7 +111,7 @@ public function query( $args, $assoc_args ) {
 
 		$query_args['fields'] = implode( ',', $fields );
 
-		$records = wp_stream_get_instance()->db->query->query( $query_args );
+		$records = wp_stream_get_instance()->db->query( $query_args );
 
 		// Make structure Formatter compatible
 		foreach ( (array) $records as $key => $record ) {
@@ -212,7 +212,7 @@ private function csv_format( $array ) {
 	 * @return void
 	 */
 	private function connection() {
-		$query = wp_stream_get_instance()->db->query->query( array( 'records_per_page' => 1, 'fields' => 'created' ) );
+		$query = wp_stream_get_instance()->db->query( array( 'records_per_page' => 1, 'fields' => 'created' ) );
 
 		if ( ! $query ) {
 			WP_CLI::error( esc_html__( 'SITE IS DISCONNECTED', 'stream' ) );

classes/class-db.php

@@ -198,4 +198,17 @@ function existing_records( $column ) {
 
 		return isset( $this->plugin->connectors->term_labels[ $column ] ) ? $this->plugin->connectors->term_labels[ $column ] : array();
 	}
+
+	/**
+	 * Helper function for calling $this->query->query()
+	 *
+	 * @see Query->query()
+	 *
+	 * @param array Query args
+	 *
+	 * @return array Stream Records
+	 */
+	function query( $args ) {
+		return $this->query->query( $args );
+	}
 }

classes/class-list-table.php

@@ -204,7 +204,7 @@ function get_records() {
 			$args['records_per_page'] = $this->get_items_per_page( 'edit_stream_per_page', 20 );
 		}
 
-		$items = $this->plugin->db->query->query( $args );
+		$items = $this->plugin->db->query( $args );
 
 		return $items;
 	}