From 9e9ec176586046d9a7f4f42ebcfe022c027197c4 Mon Sep 17 00:00:00 2001
From: Bernhard Kau <b.kau@inpsyde.com>
Date: Fri, 26 Apr 2024 17:24:40 +0200
Subject: [PATCH] Rename and reorder the columns

---
 src/DB_Command.php | 18 +++++++++++-------
 1 file changed, 11 insertions(+), 7 deletions(-)

diff --git a/src/DB_Command.php b/src/DB_Command.php
index e174e216..3c79ca28 100644
--- a/src/DB_Command.php
+++ b/src/DB_Command.php
@@ -1484,12 +1484,16 @@ public function search( $args, $assoc_args ) {
 
 							if ( $format ) {
 								$search_results[] = [
-									'table'  => $table,
-									'column' => $column,
-									'key'    => $primary_key,
-									'value'  => $result->$primary_key,
+									'table'             => $table,
+									'column'            => $column,
 									// Remove the colors for the format output.
-									'match'  => str_replace( [ $colors['match'][0], $colors['match'][1] ], [ '', '' ], $col_val ),
+									'match'             => str_replace(
+										[ $colors['match'][0], $colors['match'][1] ],
+										[ '','' ],
+										$col_val
+									),
+									'primary_key_name'  => $primary_key,
+									'primary_key_value' => $result->$primary_key,
 								];
 							} else {
 								WP_CLI::log( $matches_only ? $col_val : ( $one_line ? "{$table_column_val}:{$pk_val}{$col_val}" : "{$pk_val}{$col_val}" ) );
@@ -1504,7 +1508,7 @@ public function search( $args, $assoc_args ) {
 			$formatter_args   = [
 				'format' => $format,
 			];
-			$formatter_fields = [ 'table', 'column', 'key', 'value', 'match' ];
+			$formatter_fields = [ 'table', 'column', 'match', 'primary_key_name', 'primary_key_value' ];
 
 			if ( $fields ) {
 				$fields           = explode( ',', $assoc_args['fields'] );
@@ -1515,7 +1519,7 @@ public function search( $args, $assoc_args ) {
 				if ( count( $tables ) > 1 ) {
 					WP_CLI::error( 'The "ids" format can only be used for a single table.' );
 				}
-				$search_results = array_column( $search_results, 'value' );
+				$search_results = array_column( $search_results, 'primary_key_value' );
 			}
 
 			$formatter = new Formatter( $formatter_args, $formatter_fields );