Refactor left pane

Just use Strings instead of the various structs

deja_vu/src/action.rs

@@ -16,15 +16,7 @@
 use serde::{Deserialize, Serialize};
 use strum::Display;
 
-use crate::{
+use crate::{components::popup::Type, line::DVLine, state::Mode, system::disk::Disk};
-    components::popup::Type,
-    line::DVLine,
-    state::Mode,
-    system::{
-        disk::{Disk, PartitionTableType},
-        drivers::Driver,
-    },
-};
 
 #[derive(Debug, Clone, PartialEq, Eq, Display, Serialize, Deserialize)]
 pub enum Action {
@@ -33,12 +25,11 @@ pub enum Action {
     InstallDriver,
     Process,
     ScanDisks,
-    Select(Option<usize>, Option<usize>), // indicies for (source, dest) or (boot, os)
+    Select(Option<usize>, Option<usize>), // indicies for (source, dest) etc
-    SelectDriver(Driver),
+    SelectRight(Option<usize>, Option<usize>), // indicies for right info pane
-    SelectTableType(PartitionTableType),
     UpdateDiskList(Vec<Disk>),
-    UpdateLeft(Vec<Vec<DVLine>>, usize, Vec<Vec<DVLine>>), // (labels, start_index, lines) - lines before start are always shown
+    UpdateLeft(String, Vec<String>, Vec<String>, bool), // (title, labels, items, select_one)
-    UpdateRight(Vec<Vec<DVLine>>, usize, Vec<Vec<DVLine>>), // Same as above
+    UpdateRight(Vec<Vec<DVLine>>, usize, Vec<Vec<DVLine>>), // (labels, start_index, items) - items before start are always shown
     // Screens
     DismissPopup,
     DisplayPopup(Type, String),

deja_vu/src/app.rs

@@ -101,44 +101,43 @@ impl App {
         })
     }
 
-    pub fn next_mode(&mut self) -> (Option<Mode>, Option<Mode>) {
+    pub fn prev_mode(&mut self) -> Option<Mode> {
-        let new_mode = match (self.prev_mode, self.cur_mode) {
+        let new_mode = match self.cur_mode {
-            (_, Mode::InstallDrivers) => Mode::ScanDisks,
+            Mode::Failed => Some(Mode::Failed),
-            (_, Mode::ScanDisks) => Mode::SelectDisks,
+            Mode::Done => Some(Mode::Done),
-            (_, Mode::SelectDisks | Mode::SelectTableType | Mode::SelectParts) => {
+            Mode::SelectParts => Some(Mode::SelectParts),
-                if self.selections[1].is_some() {
+            Mode::Confirm => Some(Mode::SelectTableType),
-                    Mode::Confirm
+            Mode::SelectTableType => Some(Mode::SelectDisks),
-                } else {
+            Mode::SelectDisks => Some(Mode::ScanDisks),
-                    self.cur_mode
+            //
-                }
+            Mode::InstallDrivers
-            }
+            | Mode::ScanDisks
-            (Mode::SelectDisks, Mode::Confirm) => Mode::SelectTableType,
+            | Mode::PreClone
-            (Mode::SelectTableType, Mode::Confirm) => Mode::PreClone,
+            | Mode::Clone
-            (_, Mode::PreClone) => Mode::Clone,
+            | Mode::PostClone => None,
-            (_, Mode::Clone) => Mode::SelectParts,
-            (Mode::SelectParts, Mode::Confirm) => Mode::PostClone,
-            (_, Mode::PostClone | Mode::Done) => Mode::Done,
-            (_, Mode::Failed) => Mode::Failed,
-            // Invalid states
-            (_, Mode::Confirm) => panic!("This shouldn't happen."),
         };
+        new_mode
+    }
 
-        let prev_mode = {
+    pub fn next_mode(&mut self) -> Option<Mode> {
-            match self.cur_mode {
+        let new_mode = match self.cur_mode {
-                Mode::Confirm => Some(self.prev_mode),
+            Mode::InstallDrivers => Mode::ScanDisks,
-                Mode::PreClone | Mode::Clone | Mode::PostClone | Mode::Done => {
+            Mode::ScanDisks => Mode::SelectDisks,
-                    // Override since we're past the point of no return
+            Mode::SelectDisks => Mode::SelectTableType,
-                    Some(self.cur_mode)
+            Mode::SelectTableType => Mode::Confirm,
-                }
+            Mode::Confirm => Mode::PreClone,
-                _ => Some(self.cur_mode),
+            Mode::PreClone => Mode::Clone,
-            }
+            Mode::Clone => Mode::SelectParts,
+            Mode::SelectParts => Mode::PostClone,
+            Mode::PostClone | Mode::Done => Mode::Done,
+            Mode::Failed => Mode::Failed,
         };
 
         if new_mode == self.cur_mode {
             // No mode change needed
-            (None, None)
+            None
         } else {
-            (prev_mode, Some(new_mode))
+            Some(new_mode)
         }
     }
 
@@ -146,6 +145,7 @@ impl App {
         info!("Setting mode to {new_mode:?}");
         self.cur_mode = new_mode;
         match new_mode {
+            Mode::InstallDrivers => self.clone.scan_drivers(),
             Mode::ScanDisks => {
                 self.prev_mode = self.cur_mode;
                 if self.tasks.idle() {
@@ -244,10 +244,7 @@ impl App {
                     String::from("COMPLETE\n\n\nThank you for using this tool!"),
                 ))?;
             }
-            _ => {
+            _ => {}
-                //TODO
-                //FIXME
-            }
         }
         Ok(())
     }
@@ -361,37 +358,41 @@ impl App {
                 Action::InstallDriver => {
                     self.action_tx.send(Action::SetMode(Mode::InstallDrivers))?;
                 }
-                Action::SelectDriver(ref driver) => {
+                Action::Process => match self.cur_mode {
-                    self.clone.driver = Some(driver.clone());
+                    Mode::Confirm => {
-                    drivers::load(&driver.inf_paths);
+                        self.action_tx.send(Action::NextScreen)?;
-                    self.action_tx.send(Action::NextScreen)?;
+                    }
-                }
+                    _ => {}
-                Action::SelectTableType(ref table_type) => {
+                },
-                    self.clone.table_type = Some(table_type.clone());
-                    self.action_tx.send(Action::NextScreen)?;
-                }
                 Action::Resize(w, h) => self.handle_resize(tui, w, h)?,
                 Action::Render => self.render(tui)?,
                 Action::PrevScreen => {
-                    let new_mode = match (self.prev_mode, self.cur_mode) {
+                    if let Some(new_mode) = self.prev_mode() {
-                        (Mode::SelectTableType, Mode::SelectTableType) => Mode::SelectDisks,
+                        self.prev_mode = new_mode;
-                        (_, _) => self.prev_mode,
+                        self.cur_mode = new_mode;
-                    };
+                        self.action_tx.send(Action::SetMode(new_mode))?;
-                    self.action_tx.send(Action::SetMode(new_mode))?;
+                    }
                 }
                 Action::NextScreen => match self.next_mode() {
-                    (None, None) => {}
+                    None => {}
-                    (Some(prev), Some(next)) => {
+                    Some(next) => {
-                        self.prev_mode = prev;
+                        self.prev_mode = self.cur_mode;
                         self.cur_mode = next;
                         self.action_tx.send(Action::DismissPopup)?;
                         self.action_tx.send(Action::SetMode(next))?;
                     }
-                    _ => panic!("Failed to determine next mode"),
                 },
                 Action::ScanDisks => self.action_tx.send(Action::SetMode(Mode::ScanDisks))?,
                 Action::Select(one, two) => {
                     match self.cur_mode {
+                        Mode::InstallDrivers => {
+                            if let Some(index) = one {
+                                if let Some(driver) = self.clone.driver_list.get(index).cloned() {
+                                    drivers::load(&driver.inf_paths);
+                                    self.clone.driver = Some(driver);
+                                }
+                            }
+                        }
                         Mode::SelectDisks => {
                             self.clone.disk_index_source = one;
                             self.clone.disk_index_dest = two;
@@ -400,6 +401,21 @@ impl App {
                             self.clone.part_index_boot = one;
                             self.clone.part_index_os = two;
                         }
+                        Mode::SelectTableType => {
+                            self.clone.table_type = {
+                                if let Some(index) = one {
+                                    match index {
+                                        0 => Some(PartitionTableType::Guid),
+                                        1 => Some(PartitionTableType::Legacy),
+                                        index => {
+                                            panic!("Failed to select PartitionTableType: {}", index)
+                                        }
+                                    }
+                                } else {
+                                    None
+                                }
+                            }
+                        }
                         _ => {}
                     }
                     self.selections[0] = one;
@@ -407,13 +423,22 @@ impl App {
                 }
                 Action::SetMode(new_mode) => {
                     self.set_mode(new_mode)?;
-                    let (labels, start, lines) =
+                    let (title, labels, items, select_one) = build_left_items(self, self.cur_mode);
-                        get_right_selections(self, self.prev_mode, self.cur_mode);
                     self.action_tx
-                        .send(Action::UpdateRight(labels, start, lines))?;
+                        .send(Action::UpdateLeft(title, labels, items, select_one))?;
+                    let (labels, start, items) = build_right_items(self, self.cur_mode);
+                    self.action_tx
+                        .send(Action::UpdateRight(labels, start, items))?;
                     match new_mode {
-                        Mode::InstallDrivers | Mode::SelectDisks => {
+                        // Mode::InstallDrivers | Mode::SelectDisks => {
-                            self.action_tx.send(Action::Select(None, None))?
+                        //     self.action_tx.send(Action::Select(None, None))?
+                        // }
+                        Mode::SelectTableType | Mode::Confirm => {
+                            // Select source/dest disks
+                            self.action_tx.send(Action::SelectRight(
+                                self.clone.disk_index_source,
+                                self.clone.disk_index_dest,
+                            ))?;
                         }
                         Mode::SelectParts => {
                             // Select first partition as boot partition
@@ -525,28 +550,76 @@ fn get_chunks(r: Rect) -> Vec<Rect> {
     chunks
 }
 
-fn get_right_selections(
+fn build_left_items(app: &App, cur_mode: Mode) -> (String, Vec<String>, Vec<String>, bool) {
-    app: &App,
+    let title: String;
-    prev_mode: Mode,
+    let mut items = Vec::new();
-    cur_mode: Mode,
+    let mut labels: Vec<String> = Vec::new();
-) -> (Vec<Vec<DVLine>>, usize, Vec<Vec<DVLine>>) {
+    let mut select_one = false;
+    match cur_mode {
+        Mode::InstallDrivers => {
+            title = String::from("Install Drivers");
+            select_one = true;
+            app.clone
+                .driver_list
+                .iter()
+                .for_each(|driver| items.push(driver.to_string()));
+        }
+        Mode::SelectDisks => {
+            title = String::from("Select Source and Destination Disks");
+            labels.push(String::from("source"));
+            labels.push(String::from("dest"));
+            let disk_list = app.clone.disk_list.lock().unwrap();
+            disk_list
+                .iter()
+                .for_each(|disk| items.push(disk.description.to_string()));
+        }
+        Mode::SelectTableType => {
+            title = String::from("Select Partition Table Type");
+            select_one = true;
+            items.push(format!("{}", PartitionTableType::Guid));
+            items.push(format!("{}", PartitionTableType::Legacy));
+        }
+        Mode::Confirm => {
+            title = String::from("Confirm Selections");
+        }
+        Mode::PreClone | Mode::Clone | Mode::PostClone | Mode::ScanDisks => {
+            title = String::from("Processing");
+        }
+        Mode::SelectParts => {
+            title = String::from("Select Boot and OS Partitions");
+            labels.push(String::from("boot"));
+            labels.push(String::from("os"));
+            let disk_list = app.clone.disk_list.lock().unwrap();
+            if let Some(index) = app.clone.disk_index_dest {
+                if let Some(disk) = disk_list.get(index) {
+                    disk.get_parts().iter().for_each(|part| {
+                        items.push(part.to_string());
+                    });
+                }
+            }
+        }
+        Mode::Done | Mode::Failed => title = String::from("Done"),
+    };
+    (title, labels, items, select_one)
+}
+
+fn build_right_items(app: &App, cur_mode: Mode) -> (Vec<Vec<DVLine>>, usize, Vec<Vec<DVLine>>) {
+    let mut items = Vec::new();
     let mut labels: Vec<Vec<DVLine>> = Vec::new();
-    let mut selections = Vec::new();
     let mut start_index = 0;
-    match (prev_mode, cur_mode) {
+    match cur_mode {
-        (_, Mode::InstallDrivers) => {
+        Mode::InstallDrivers => {
-            selections.push(vec![DVLine {
+            items.push(vec![DVLine {
                 line_parts: vec![String::from("CPU")],
                 line_colors: vec![Color::Cyan],
             }]);
-            selections.push(vec![DVLine {
+            items.push(vec![DVLine {
                 line_parts: vec![get_cpu_name()],
                 line_colors: vec![Color::Reset],
             }]);
             start_index = 2;
         }
-        (_, Mode::SelectDisks | Mode::SelectTableType)
+        Mode::SelectDisks | Mode::SelectTableType | Mode::Confirm => {
-        | (Mode::SelectDisks | Mode::SelectTableType, Mode::Confirm) => {
             // Labels
             labels.push(vec![DVLine {
                 line_parts: vec![String::from("Source")],
@@ -559,31 +632,28 @@ fn get_right_selections(
                 ],
                 line_colors: vec![Color::Cyan, Color::Red],
             };
-            match (prev_mode, cur_mode) {
+            if let Some(table_type) = &app.clone.table_type {
-                (Mode::SelectTableType, Mode::Confirm) => {
+                // Show table type
-                    if let Some(table_type) = &app.clone.table_type {
+                let type_str = match table_type {
-                        // Show table type
+                    PartitionTableType::Guid => "GPT",
-                        let type_str = match table_type {
+                    PartitionTableType::Legacy => "MBR",
-                            PartitionTableType::Guid => "GPT",
+                };
-                            PartitionTableType::Legacy => "MBR",
+                labels.push(vec![
-                        };
+                    dest_dv_line,
-                        labels.push(vec![
+                    DVLine {
-                            dest_dv_line,
+                        line_parts: vec![format!("      (Will be formatted {type_str})")],
-                            DVLine {
+                        line_colors: vec![Color::Yellow],
-                                line_parts: vec![format!("      (Will be formatted {type_str})")],
+                    },
-                                line_colors: vec![Color::Yellow],
+                ]);
-                            },
+            } else {
-                        ]);
+                labels.push(vec![dest_dv_line]);
-                    }
-                }
-                _ => labels.push(vec![dest_dv_line]),
             }
             let disk_list = app.clone.disk_list.lock().unwrap();
             disk_list
                 .iter()
-                .for_each(|disk| selections.push(get_disk_description_right(&disk)));
+                .for_each(|disk| items.push(get_disk_description_right(&disk)));
         }
-        (_, Mode::SelectParts) | (Mode::SelectParts, Mode::Confirm) => {
+        Mode::SelectParts => {
             vec!["Boot", "OS"].iter().for_each(|s| {
                 labels.push(vec![DVLine {
                     line_parts: vec![String::from(*s)],
@@ -595,16 +665,16 @@ fn get_right_selections(
                 let disk_list = app.clone.disk_list.lock().unwrap();
                 if let Some(disk) = disk_list.get(index) {
                     // Disk Details
-                    selections.push(get_disk_description_right(&disk));
+                    items.push(get_disk_description_right(&disk));
 
                     // Partition Details
                     disk.parts
                         .iter()
-                        .for_each(|part| selections.push(get_part_description(&part)));
+                        .for_each(|part| items.push(get_part_description(&part)));
                 }
             }
         }
         _ => {}
     }
-    (labels, start_index, selections)
+    (labels, start_index, items)
 }

deja_vu/src/components/left.rs

@@ -20,42 +20,37 @@ use ratatui::{
     widgets::{Block, Borders, HighlightSpacing, List, ListItem, Padding, Paragraph},
 };
 use tokio::sync::mpsc::UnboundedSender;
-use tracing::info;
 
-use super::{popup, state::StatefulList, Component};
+use super::{state::StatefulList, Component};
-use crate::{
+use crate::{action::Action, config::Config};
-    action::Action,
-    config::Config,
-    state::Mode,
-    system::{
-        disk::{Disk, Partition, PartitionTableType},
-        drivers::{self, Driver},
-    },
-};
 
 #[derive(Default)]
 pub struct Left {
     command_tx: Option<UnboundedSender<Action>>,
     config: Config,
-    disk_id_dest: Option<usize>,
+    labels: Vec<String>,
-    table_type: Option<PartitionTableType>,
+    list: StatefulList<String>,
-    title_text: String,
+    select_one: bool,
-    list_disks: StatefulList<Disk>,
-    list_drivers: StatefulList<Driver>,
-    list_parts: StatefulList<Partition>,
-    list_table_types: StatefulList<PartitionTableType>,
-    mode: Mode,
     selections: Vec<Option<usize>>,
+    selections_saved: Vec<Option<usize>>,
+    title_text: String,
 }
 
 impl Left {
     pub fn new() -> Self {
         Self {
+            select_one: false,
+            labels: vec![String::from("one"), String::from("two")],
             selections: vec![None, None],
+            selections_saved: vec![None, None],
             title_text: String::from("Home"),
             ..Default::default()
         }
     }
+
+    fn set_highlight(&mut self, index: usize) {
+        self.list.select(index);
+    }
 }
 
 impl Component for Left {
@@ -76,180 +71,58 @@ impl Component for Left {
 
     fn update(&mut self, action: Action) -> Result<Option<Action>> {
         match action {
-            Action::KeyUp => match self.mode {
+            Action::Highlight(index) => self.set_highlight(index),
-                Mode::InstallDrivers => self.list_drivers.previous(),
+            Action::KeyUp => self.list.previous(),
-                Mode::SelectDisks => self.list_disks.previous(),
+            Action::KeyDown => self.list.next(),
-                Mode::SelectTableType => self.list_table_types.previous(),
+            Action::Process => {
-                Mode::SelectParts => self.list_parts.previous(),
+                if let Some(command_tx) = self.command_tx.clone() {
-                _ => {}
+                    match (self.selections[0], self.selections[1]) {
-            },
+                        (None, None) => {
-            Action::KeyDown => match self.mode {
+                            // Making first selection
-                Mode::InstallDrivers => self.list_drivers.next(),
+                            command_tx.send(Action::Select(self.list.selected(), None))?;
-                Mode::SelectDisks => self.list_disks.next(),
+                            if self.select_one {
-                Mode::SelectTableType => self.list_table_types.next(),
+                                // Confirm selection
-                Mode::SelectParts => self.list_parts.next(),
+                                command_tx.send(Action::NextScreen)?;
-                _ => {}
+                            }
-            },
+                        }
-            Action::Process => match self.mode {
+                        (Some(first_index), None) => {
-                // NOTE: Remove all variants except Mode::Confirm?
+                            if let Some(second_index) = self.list.selected() {
-                Mode::Confirm => {
+                                // Making second selection
-                    if let Some(command_tx) = self.command_tx.clone() {
+                                if first_index == second_index {
-                        command_tx.send(Action::NextScreen)?;
+                                    // Toggle first selection
-                    }
+                                    command_tx.send(Action::Select(None, None))?;
-                }
+                                } else {
-                Mode::InstallDrivers
+                                    // Both selections made, proceed to next screen
-                | Mode::SelectDisks
+                                    command_tx.send(Action::Select(
-                | Mode::SelectTableType
+                                        Some(first_index),
-                | Mode::SelectParts => {
+                                        Some(second_index),
-                    // Menu selection sections
+                                    ))?;
-                    let selection: Option<usize> = match self.mode {
+                                    command_tx.send(Action::NextScreen)?;
-                        Mode::InstallDrivers => self.list_drivers.selected(),
-                        Mode::SelectDisks => self.list_disks.selected(),
-                        Mode::SelectTableType => self.list_table_types.selected(),
-                        Mode::SelectParts => self.list_parts.selected(),
-                        _ => panic!("This shouldn't happen!"),
-                    };
-                    if let Some(index) = selection {
-                        if let Some(command_tx) = self.command_tx.clone() {
-                            let mut selection_one: Option<usize> = None;
-                            let mut selection_two: Option<usize> = None;
-
-                            // Get selection(s)
-                            if self.selections[0].is_none() {
-                                // First selection
-                                selection_one = Some(index);
-                                selection_two = None;
-                            } else {
-                                // Second selection
-                                if let Some(source_index) = self.selections[0] {
-                                    if index == source_index {
-                                        // Toggle first selection
-                                        selection_one = None;
-                                        self.selections[0] = None;
-                                    } else {
-                                        selection_one = self.selections[0];
-                                        selection_two = Some(index);
-                                    }
                                 }
                             }
-
-                            // Send selection(s) if needed
-                            // NOTE: This is needed to keep the app and all components in sync
-                            match self.mode {
-                                Mode::InstallDrivers => {
-                                    // Only need to select one entry
-                                    if let Some(driver) = self.list_drivers.get_selected() {
-                                        command_tx.send(Action::SelectDriver(driver.clone()))?;
-                                    }
-                                }
-                                Mode::SelectTableType => {
-                                    // Only need to select one entry
-                                    if let Some(table_type) = self.list_table_types.get_selected() {
-                                        self.table_type = Some(table_type.clone());
-                                        command_tx.send(Action::SelectTableType(table_type))?;
-                                    }
-                                }
-                                Mode::SelectDisks | Mode::SelectParts => {
-                                    command_tx
-                                        .send(Action::Select(selection_one, selection_two))?;
-
-                                    // Advance screen if both selections made
-                                    if selection_two.is_some() {
-                                        command_tx.send(Action::NextScreen)?;
-                                    }
-                                }
-                                _ => {}
-                            };
                         }
+                        _ => panic!("This shouldn't happen?"),
                     }
                 }
-                _ => {}
-            },
-            Action::Select(Some(index), None) => self.selections[0] = Some(index),
-            Action::Select(_, Some(index)) => {
-                if self.mode == Mode::SelectDisks {
-                    self.disk_id_dest = Some(index);
-                }
             }
-            Action::SetMode(new_mode) => {
+            Action::Select(one, two) => {
-                let prev_mode = self.mode;
+                self.selections[0] = one;
-                self.mode = new_mode;
+                self.selections[1] = two;
-                match (prev_mode, new_mode) {
+                self.selections_saved[0] = one;
-                    (_, Mode::ScanDisks) => {
+                self.selections_saved[1] = two;
-                        self.list_disks.clear_items();
-                        self.title_text = String::new();
-                    }
-                    (_, Mode::InstallDrivers) => {
-                        self.list_drivers.set_items(drivers::scan());
-                        self.selections[0] = None;
-                        self.selections[1] = None;
-                        self.title_text = String::from("Install Drivers");
-                        if self.list_drivers.is_empty() {
-                            if let Some(command_tx) = self.command_tx.clone() {
-                                command_tx.send(Action::DisplayPopup(
-                                    popup::Type::Error,
-                                    String::from("No drivers available to install"),
-                                ))?;
-                            }
-                        }
-                    }
-                    (_, Mode::SelectDisks) => {
-                        self.selections[0] = None;
-                        self.selections[1] = None;
-                        self.title_text = String::from("Select Source and Destination Disks");
-                    }
-                    (_, Mode::SelectTableType) => {
-                        self.list_table_types
-                            .set_items(vec![PartitionTableType::Guid, PartitionTableType::Legacy]);
-                        self.selections[0] = None;
-                        self.selections[1] = None;
-                        self.title_text = String::from("Select Partition Table Type");
-                    }
-                    (_, Mode::PreClone | Mode::Clone | Mode::PostClone) => {
-                        self.title_text = String::from("Processing");
-                    }
-                    (_, Mode::SelectParts) => {
-                        self.title_text = String::from("Select Boot and OS Partitions");
-                    }
-                    (Mode::SelectDisks | Mode::SelectParts, Mode::Confirm) => {
-                        self.title_text = String::from("Confirm Selections");
-                    }
-                    (Mode::SelectTableType, Mode::Confirm) => {
-                        self.title_text = String::from("Confirm Selections (Again)");
-                    }
-                    (_, Mode::Done | Mode::Failed) => self.title_text = String::from("Done"),
-                    // Invalid states
-                    (_, Mode::Confirm) => panic!("This shouldn't happen."),
-                }
             }
-            Action::UpdateDiskList(disks) => {
+            Action::SetMode(_) => {
-                info!("Updating disk list");
+                self.selections[0] = None;
-                self.list_disks.set_items(disks);
+                self.selections[1] = None;
-                if self.mode == Mode::Clone {
+            }
-                    if let Some(index) = self.disk_id_dest {
+            Action::UpdateLeft(title, labels, items, select_one) => {
-                        if let Some(disk) = self.list_disks.get(index) {
+                self.title_text = title;
-                            self.list_parts.set_items(disk.get_parts());
+                self.labels = labels
-
+                    .iter()
-                            // Auto-select first partition and highlight likely OS partition
+                    .map(|label| format!("      ~{}~", label.to_lowercase()))
-                            if let Some(table_type) = &self.table_type {
+                    .collect();
-                                match table_type {
+                self.list.set_items(items);
-                                    PartitionTableType::Guid => {
+                self.select_one = select_one;
-                                        if disk.num_parts() >= 3 {
-                                            self.selections[0] = Some(0);
-                                            self.list_parts.select(2);
-                                        }
-                                    }
-                                    PartitionTableType::Legacy => {
-                                        if disk.num_parts() >= 2 {
-                                            self.selections[0] = Some(0);
-                                            self.list_parts.select(1);
-                                        }
-                                    }
-                                }
-                            }
-                        }
-                    }
-                }
             }
             _ => {}
         }
@@ -263,126 +136,70 @@ impl Component for Left {
             .areas(area);
 
         // Title
+        let title_text = if self.selections[1].is_some() || self.select_one {
+            "Confirm Selections"
+        } else {
+            self.title_text.as_str()
+        };
         let title =
-            Paragraph::new(Line::from(Span::styled(&self.title_text, Style::default())).centered())
+            Paragraph::new(Line::from(Span::styled(title_text, Style::default())).centered())
                 .block(Block::default().borders(Borders::NONE));
         frame.render_widget(title, title_area);
 
-        // Body
+        // Body (Blank)
-        match self.mode {
+        if self.list.is_empty() {
-            Mode::ScanDisks
+            // Leave blank
-            | Mode::PreClone
+            let paragraph = Paragraph::new(String::new()).block(
-            | Mode::Clone
+                Block::default()
-            | Mode::PostClone
+                    .borders(Borders::ALL)
-            | Mode::Done
+                    .padding(Padding::new(1, 1, 1, 1)),
-            | Mode::Failed => {
+            );
-                // Leave blank
+            frame.render_widget(paragraph, body_area);
-                let paragraph = Paragraph::new(String::new()).block(
-                    Block::default()
-                        .borders(Borders::ALL)
-                        .padding(Padding::new(1, 1, 1, 1)),
-                );
-                frame.render_widget(paragraph, body_area);
 
-                // Bail early
+            // Bail early
-                return Ok(());
+            return Ok(());
-            }
-            Mode::Confirm => {
-                // Nag the user
-                let paragraph = Paragraph::new(String::from("Are the listed selections correct?"))
-                    .block(
-                        Block::default()
-                            .borders(Borders::ALL)
-                            .padding(Padding::new(1, 1, 1, 1)),
-                    );
-                frame.render_widget(paragraph, body_area);
-
-                // Bail early
-                return Ok(());
-            }
-            Mode::InstallDrivers
-            | Mode::SelectDisks
-            | Mode::SelectTableType
-            | Mode::SelectParts => {
-                // List modes
-                let mut list_items = Vec::<ListItem>::new();
-                let list_items_strings: Vec<String> = match self.mode {
-                    Mode::InstallDrivers => self
-                        .list_drivers
-                        .items
-                        .iter()
-                        .map(|i| format!("{i}"))
-                        .collect(),
-                    Mode::SelectDisks => self
-                        .list_disks
-                        .items
-                        .iter()
-                        .map(|i| format!("{i}"))
-                        .collect(),
-                    Mode::SelectTableType => self
-                        .list_table_types
-                        .items
-                        .iter()
-                        .map(|i| format!("{i}"))
-                        .collect(),
-                    Mode::SelectParts => self
-                        .list_parts
-                        .items
-                        .iter()
-                        .map(|i| format!("{i}"))
-                        .collect(),
-                    _ => panic!("This shouldn't happen."),
-                };
-                if !list_items_strings.is_empty() {
-                    for (index, item) in list_items_strings.iter().enumerate() {
-                        let mut item_style = Style::default();
-                        let mut item_text_tail = "";
-                        if let Some(selection_one) = self.selections[0] {
-                            if selection_one == index {
-                                item_style = Style::new().yellow();
-                                item_text_tail = match self.mode {
-                                    Mode::SelectDisks => "      ~source disk~",
-                                    Mode::SelectParts => "      ~boot volume~",
-                                    _ => "",
-                                }
-                            }
-                        }
-                        list_items.push(
-                            ListItem::new(format!("    {item}\n{item_text_tail}\n\n"))
-                                .style(item_style),
-                        );
-                    }
-                }
-                let list = List::new(list_items)
-                    .block(
-                        Block::default()
-                            .borders(Borders::ALL)
-                            .padding(Padding::new(1, 1, 1, 1)),
-                    )
-                    .highlight_spacing(HighlightSpacing::Always)
-                    .highlight_style(Style::new().green().bold())
-                    .highlight_symbol(" --> ")
-                    .repeat_highlight_symbol(false);
-                match self.mode {
-                    Mode::InstallDrivers => {
-                        frame.render_stateful_widget(list, body_area, &mut self.list_drivers.state);
-                    }
-                    Mode::SelectDisks => {
-                        frame.render_stateful_widget(list, body_area, &mut self.list_disks.state);
-                    }
-                    Mode::SelectTableType => frame.render_stateful_widget(
-                        list,
-                        body_area,
-                        &mut self.list_table_types.state,
-                    ),
-                    Mode::SelectParts => {
-                        frame.render_stateful_widget(list, body_area, &mut self.list_parts.state);
-                    }
-                    _ => panic!("This shouldn't happen."),
-                }
-            }
         }
 
+        // Body
+        let list_items: Vec<ListItem> = self
+            .list
+            .items
+            .iter()
+            .enumerate()
+            .map(|(index, item)| {
+                let mut style = Style::default();
+                let text = if self.selections[0].is_some_and(|first_index| first_index == index) {
+                    if let Some(label) = self.labels.get(0) {
+                        style = style.yellow();
+                        label.as_str()
+                    } else {
+                        ""
+                    }
+                } else if self.selections[1].is_some_and(|second_index| second_index == index) {
+                    if let Some(label) = self.labels.get(1) {
+                        style = style.yellow();
+                        label.as_str()
+                    } else {
+                        ""
+                    }
+                } else {
+                    ""
+                };
+                ListItem::new(format!("    {item}\n{text}\n\n")).style(style)
+            })
+            .collect();
+        let list = List::new(list_items)
+            .block(
+                Block::default()
+                    .borders(Borders::ALL)
+                    .padding(Padding::new(1, 1, 1, 1)),
+            )
+            .highlight_spacing(HighlightSpacing::Always)
+            .highlight_style(Style::new().green().bold())
+            .highlight_symbol(" --> ")
+            .repeat_highlight_symbol(false);
+        frame.render_stateful_widget(list, body_area, &mut self.list.state);
+
         // Done
         Ok(())
     }

deja_vu/src/components/right.rs

@@ -22,13 +22,12 @@ use ratatui::{
 use tokio::sync::mpsc::UnboundedSender;
 
 use super::{state::StatefulList, Component};
-use crate::{action::Action, config::Config, line::DVLine, state::Mode};
+use crate::{action::Action, config::Config, line::DVLine};
 
 #[derive(Default)]
 pub struct Right {
     command_tx: Option<UnboundedSender<Action>>,
     config: Config,
-    cur_mode: Mode,
     list_header: Vec<DVLine>,
     list_labels: Vec<Vec<DVLine>>,
     list: StatefulList<Vec<DVLine>>,
@@ -98,13 +97,22 @@ impl Component for Right {
             Action::Select(one, two) => {
                 self.selections[0] = one;
                 self.selections[1] = two;
+                if self.selections_saved[1].is_none() {
+                    // Update selections_saved
+                    // Conversely, if both selections set then preserve previous choices
+                    self.selections_saved[0] = one;
+                    self.selections_saved[1] = two;
+                }
+            }
+            Action::SelectRight(one, two) => {
                 self.selections_saved[0] = one;
                 self.selections_saved[1] = two;
             }
-            Action::SetMode(new_mode) => {
+            Action::SetMode(_) => {
-                self.cur_mode = new_mode;
                 self.selections[0] = None;
                 self.selections[1] = None;
+                self.selections_saved[0] = None;
+                self.selections_saved[1] = None;
             }
             Action::UpdateRight(labels, start_index, list) => {
                 self.list_header = list[..start_index].iter().flatten().cloned().collect();
@@ -142,9 +150,11 @@ impl Component for Right {
         // First selection
         if let Some(first_index) = self.get_first() {
             if let Some(first_desc) = self.list.get(first_index) {
-                self.list_labels[0]
+                if let Some(label) = self.list_labels.get(0) {
-                    .iter()
+                    label
-                    .for_each(|dv| body_text.push(dv.as_line().bold()));
+                        .iter()
+                        .for_each(|dv| body_text.push(dv.as_line().bold()));
+                }
                 body_text.push(Line::from(""));
                 first_desc
                     .iter()
@@ -159,9 +169,11 @@ impl Component for Right {
                 body_text.push(Line::from(""));
                 body_text.push(Line::from(str::repeat("─", (body_area.width - 4) as usize)));
                 body_text.push(Line::from(""));
-                self.list_labels[1]
+                if let Some(label) = self.list_labels.get(1) {
-                    .iter()
+                    label
-                    .for_each(|dv| body_text.push(dv.as_line().bold()));
+                        .iter()
+                        .for_each(|dv| body_text.push(dv.as_line().bold()));
+                }
                 body_text.push(Line::from(""));
                 second_desc
                     .iter()

deja_vu/src/state.rs

@@ -20,7 +20,7 @@ use serde::{Deserialize, Serialize};
 
 use crate::system::{
     disk::{Disk, PartitionTableType},
-    drivers::Driver,
+    drivers,
 };
 
 #[derive(Default, Debug, Copy, Clone, PartialEq, Eq, Hash, Serialize, Deserialize)]
@@ -44,9 +44,10 @@ pub struct CloneSettings {
     pub disk_index_dest: Option<usize>,
     pub disk_index_source: Option<usize>,
     pub disk_list: Arc<Mutex<Vec<Disk>>>,
+    pub driver_list: Vec<drivers::Driver>,
     pub part_index_boot: Option<usize>,
     pub part_index_os: Option<usize>,
-    pub driver: Option<Driver>,
+    pub driver: Option<drivers::Driver>,
     pub table_type: Option<PartitionTableType>,
 }
 
@@ -57,4 +58,8 @@ impl CloneSettings {
             ..Default::default()
         }
     }
+
+    pub fn scan_drivers(&mut self) {
+        self.driver_list = drivers::scan();
+    }
 }