Add support for --format flag, which can be table, json or json_pretty

Author: frankiejarrett

includes/wp-cli.php

@@ -19,7 +19,7 @@ class WP_Stream_WP_CLI_Command extends WP_CLI_Command {
 	 * : One or more args to pass to wp_stream_query.
 	 *
 	 * [--format=<format>]
-	 * : Accepted values: table, json, csv. Default: table
+	 * : Accepted values: table, json, json_pretty. Default: table
 	 *
 	 * ## AVAILABLE FIELDS TO QUERY
 	 *
@@ -106,13 +106,29 @@ public function query( $args, $assoc_args ) {
 		}
 
 		foreach ( $assoc_args as $key => $value ) {
+			if ( 'format' === $key ) {
+				continue;
+			}
+
 			$query_args[ $key ] = $value;
 		}
 
 		$query_args['fields'] = implode( ',', $fields );
 
 		$records = wp_stream_query( $query_args );
 
+		if ( isset( $assoc_args['format'] ) ) {
+			if ( 'json' === $assoc_args['format'] ) {
+				$output = json_encode( $records );
+			} elseif ( 'json_pretty' === $assoc_args['format'] ) {
+				$output = json_encode( $records, JSON_PRETTY_PRINT );
+			}
+
+			echo $output . "\n";
+
+			return;
+		}
+
 		// Make structure Formatter compatible
 		foreach ( (array) $records as $key => $record ) {
 			$formatted_records[ $key ] = array();
@@ -133,7 +149,7 @@ public function query( $args, $assoc_args ) {
 		$formatter->display_items( $formatted_records );
 
 		if ( 0 === ( $found = count( $records ) ) ) {
-			WP_CLI::line( 'No records found.' );
+			WP_CLI::line( __( 'No records found.', 'stream' ) );
 		} else {
 			WP_CLI::line( sprintf( _n( '1 record found.', '%s records found.', $found, 'stream' ), number_format( $found ) ) );
 		}