Merge pull request #15 from AndPuQing/agent/pi/cdee7cf3

feat: interactive KB-unit toggle and configurable columns

Author: AndPuQing

README.md

@@ -157,15 +157,24 @@ When running in interactive mode (default), you can use the following keyboard s
 | `a` / `A` | Toggle between bandwidth and accumulated I/O |
 | `p` / `P` | Toggle between showing processes and threads |
 | `r` / `R` | Reverse the current sort order |
-| `Space` | Pause/resume display updates |
-| `Left` / `Right` | Cycle through sort columns |
+| `k` / `K` | Toggle KB units (also enabled by `-k` at startup) |
+| `c` / `C` | Enter/exit column-selection mode (see below) |
+| `Space` | Pause/resume display updates (toggle column in column mode) |
+| `Left` / `Right` | Cycle through sort columns (move cursor in column mode) |
 | `Up` / `Down` | Scroll through process list |
 | `PageUp` / `PageDown` | Scroll by 10 rows |
 | `Home` | Jump to first sort column (or first row with Ctrl) |
 | `End` | Jump to last sort column (or last row with Ctrl) |
 
 Mouse wheel scrolling is also supported for navigating the process list.
 
+### Column Selection Mode
+
+Press `c` to enter column-selection mode. The top bar shows every available column;
+`Left` / `Right` move the cursor, `Space` toggles the selected column on/off, and
+`Esc` or `c` exits. Hidden columns are shown dimmed with `~` markers. The last
+remaining visible column cannot be hidden.
+
 ## Architecture
 
 This implementation uses:

src/main.rs

@@ -179,6 +179,7 @@ async fn run_interactive_mode(process_list: &mut ProcessList, args: &Args) -> Re
     state.only_active = args.only;
     state.accumulated = args.accumulated;
     state.show_processes = args.processes;
+    state.kilobytes = args.kilobytes;
 
     // Start async data stream
     let mut data_cancel_token = CancellationToken::new();
@@ -248,8 +249,33 @@ async fn run_interactive_mode(process_list: &mut ProcessList, args: &Args) -> Re
                             state.selection_mode = false;
                             state.selected_row = None;
                         }
+                        KeyCode::Char('k') | KeyCode::Char('K') => {
+                            state.kilobytes = !state.kilobytes;
+                        }
+                        KeyCode::Char('c') | KeyCode::Char('C') => {
+                            state.column_mode = !state.column_mode;
+                            if state.column_mode {
+                                // Start the cursor on the current sort column.
+                                state.column_cursor = state.sort_column;
+                            }
+                        }
                         KeyCode::Char(' ') => {
-                            state.paused = !state.paused;
+                            if state.column_mode {
+                                // Toggle the selected column; never hide the last one.
+                                if state.visible_columns.is_visible(state.column_cursor) {
+                                    if state.visible_columns.count(has_delay_acct) > 1 {
+                                        state
+                                            .visible_columns
+                                            .set_visible(state.column_cursor, false);
+                                    }
+                                } else {
+                                    state
+                                        .visible_columns
+                                        .set_visible(state.column_cursor, true);
+                                }
+                            } else {
+                                state.paused = !state.paused;
+                            }
                         }
                         KeyCode::Char('p') | KeyCode::Char('P') => {
                             state.show_processes = !state.show_processes;
@@ -269,16 +295,28 @@ async fn run_interactive_mode(process_list: &mut ProcessList, args: &Args) -> Re
                             );
                         }
                         KeyCode::Left => {
-                            state.sort_column = state.sort_column.cycle_backward(has_delay_acct);
-                            state.scroll_offset = 0;
-                            state.selection_mode = false;
-                            state.selected_row = None;
+                            if state.column_mode {
+                                state.column_cursor =
+                                    state.column_cursor.cycle_backward(has_delay_acct);
+                            } else {
+                                state.sort_column =
+                                    state.sort_column.cycle_backward(has_delay_acct);
+                                state.scroll_offset = 0;
+                                state.selection_mode = false;
+                                state.selected_row = None;
+                            }
                         }
                         KeyCode::Right => {
-                            state.sort_column = state.sort_column.cycle_forward(has_delay_acct);
-                            state.scroll_offset = 0;
-                            state.selection_mode = false;
-                            state.selected_row = None;
+                            if state.column_mode {
+                                state.column_cursor =
+                                    state.column_cursor.cycle_forward(has_delay_acct);
+                            } else {
+                                state.sort_column =
+                                    state.sort_column.cycle_forward(has_delay_acct);
+                                state.scroll_offset = 0;
+                                state.selection_mode = false;
+                                state.selected_row = None;
+                            }
                         }
                         KeyCode::Up => {
                             if !state.selection_mode {
@@ -349,6 +387,7 @@ async fn run_interactive_mode(process_list: &mut ProcessList, args: &Args) -> Re
                             }
                         }
                         KeyCode::Esc => {
+                            state.column_mode = false;
                             state.selection_mode = false;
                             state.selected_row = None;
                         }