// This file is part of Deja-Vu.
//
// Deja-Vu is free software: you can redistribute it and/or modify it
// under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// Deja-Vu is distributed in the hope that it will be useful, but
// WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
// See the GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with Deja-Vu. If not, see <https://www.gnu.org/licenses/>.
//
use core::{
    action::{Action, DiagResult},
    components::{
        Component,
        footer::Footer,
        fps::FpsCounter,
        left::{Left, SelectionType},
        popup,
        right::Right,
        state::StatefulList,
        title::Title,
    },
    config::Config,
    line::{DVLine, get_disk_description_right, get_part_description},
    state::{CloneSettings, Mode},
    system::{
        boot::{self, SafeMode},
        cpu::get_cpu_name,
        disk::PartitionTableType,
        drivers,
    },
    tasks::{Task, TaskResult, TaskType, Tasks},
    tui::{Event, Tui},
};
use std::{
    env,
    iter::zip,
    path::PathBuf,
    sync::{Arc, Mutex},
};

use color_eyre::Result;
use ratatui::{
    crossterm::event::KeyEvent,
    layout::{Constraint, Direction, Layout},
    prelude::Rect,
    style::Color,
};
use tokio::sync::mpsc;
use tracing::{debug, info};

use crate::diags::{
    DiagGroup, Type as DiagType, get_diag_type, parse_bcd, parse_bitlocker, parse_chkdsk,
    parse_dism, parse_registry_hives, parse_system_files,
};

pub struct App {
    // TUI
    action_rx: mpsc::UnboundedReceiver<Action>,
    action_tx: mpsc::UnboundedSender<Action>,
    components: Vec<Box<dyn Component>>,
    config: Config,
    frame_rate: f64,
    last_tick_key_events: Vec<KeyEvent>,
    should_quit: bool,
    should_suspend: bool,
    tick_rate: f64,
    // App
    clone: CloneSettings,
    cur_mode: Mode,
    diag_groups: Arc<Mutex<Vec<DiagGroup>>>,
    list: StatefulList<Mode>,
    boot_modes: Vec<SafeMode>,
    selections: Vec<Option<usize>>,
    system32: String,
    tasks: Tasks,
}

impl App {
    pub fn new(tick_rate: f64, frame_rate: f64) -> Result<Self> {
        let (action_tx, action_rx) = mpsc::unbounded_channel();
        let diag_groups_arc = Arc::new(Mutex::new(Vec::new()));
        let disk_list_arc = Arc::new(Mutex::new(Vec::new()));
        let tasks = Tasks::new(action_tx.clone(), disk_list_arc.clone());
        let mut list = StatefulList::default();
        list.set_items(vec![
            Mode::BootScan,
            Mode::BootSetup,
            Mode::InjectDrivers,
            Mode::SetBootMode,
        ]);
        Ok(Self {
            // TUI
            action_rx,
            action_tx,
            components: vec![
                Box::new(Title::new("Boot Diagnostics")),
                Box::new(FpsCounter::new()),
                Box::new(Left::new()),
                Box::new(Right::new()),
                Box::new(Footer::new()),
                Box::new(popup::Popup::new()),
                Box::new(crate::components::progress::Progress::new()),
                Box::new(crate::components::logview::LogView::new(
                    diag_groups_arc.clone(),
                )),
            ],
            config: Config::new()?,
            frame_rate,
            last_tick_key_events: Vec::new(),
            should_quit: false,
            should_suspend: false,
            tick_rate,
            // App
            clone: CloneSettings::new(disk_list_arc),
            cur_mode: Mode::Home,
            diag_groups: diag_groups_arc,
            list,
            boot_modes: vec![SafeMode::Enable, SafeMode::Disable],
            system32: String::new(),
            selections: vec![None, None],
            tasks,
        })
    }

    pub fn inject_driver(&mut self, index: usize) {
        if let Some(driver) = self.clone.driver_list.get(index) {
            if let Some(disk_index) = self.clone.disk_index_dest {
                let disk_list = self.clone.disk_list.lock().unwrap();
                if let Some(disk) = disk_list.get(disk_index) {
                    if let Some(os_index) = self.clone.part_index_os {
                        if let Ok(task) = boot::inject_driver(
                            driver,
                            disk.get_part_letter(os_index).as_str(),
                            &self.system32,
                        ) {
                            self.tasks.add(task);
                        }
                    }
                }
            }
        }
    }

    pub fn next_mode(&mut self) -> Mode {
        match self.cur_mode {
            Mode::Home => Mode::ScanDisks,
            Mode::InstallDrivers => Mode::ScanDisks,
            Mode::ScanDisks => Mode::SelectDisks,
            Mode::SelectDisks => Mode::SelectParts,
            Mode::SelectParts => Mode::DiagMenu,
            Mode::BootDiags | Mode::BootSetup => Mode::DiagMenu,
            Mode::BootScan => Mode::BootDiags,
            Mode::InjectDrivers | Mode::SetBootMode => Mode::DiagMenu,
            Mode::Done => Mode::DiagMenu,
            Mode::Failed => Mode::Failed,
            // Default to current mode
            _ => self.cur_mode,
        }
    }

    pub fn set_boot_mode(&mut self, boot_mode: SafeMode) {
        let new_mode = match boot_mode {
            SafeMode::Disable => "Normal",
            SafeMode::Enable => "Safe Mode (minimal)",
        };
        info!("Setting boot mode to: {new_mode}");
        let disk_list = self.clone.disk_list.lock().unwrap();
        if let Some(disk_index) = self.clone.disk_index_dest {
            if let Some(disk) = disk_list.get(disk_index) {
                if let Some(boot_index) = self.clone.part_index_boot {
                    if let Ok(task) = boot::set_mode(
                        disk.get_part_letter(boot_index).as_str(),
                        &boot_mode,
                        &self.system32,
                        &disk.part_type,
                    ) {
                        self.tasks.add(task);
                    };
                }
            }
        }
    }

    pub fn set_mode(&mut self, new_mode: Mode) -> Result<()> {
        info!("Setting mode to {new_mode:?}");
        self.cur_mode = new_mode;
        match new_mode {
            Mode::DiagMenu => {
                self.selections[0] = None;
                self.selections[1] = None;
                self.list.select_first_item();
            }
            Mode::BootScan => {
                self.action_tx.send(Action::DisplayPopup(
                    popup::Type::Info,
                    String::from("Gathering info..."),
                ))?;
                self.queue_boot_scan_tasks()?;
            }
            Mode::InjectDrivers | Mode::InstallDrivers => self.clone.scan_drivers(),
            Mode::ScanDisks => {
                if self.tasks.idle() {
                    self.tasks.add(TaskType::ScanDisks);
                }
                self.action_tx.send(Action::DisplayPopup(
                    popup::Type::Info,
                    String::from("Scanning Disks..."),
                ))?;
            }
            _ => {}
        }
        Ok(())
    }

    pub async fn run(&mut self) -> Result<()> {
        let mut tui = Tui::new()?
            // .mouse(true) // uncomment this line to enable mouse support
            .tick_rate(self.tick_rate)
            .frame_rate(self.frame_rate);
        tui.enter()?;

        for component in &mut self.components {
            component.register_action_handler(self.action_tx.clone())?;
        }
        for component in &mut self.components {
            component.register_config_handler(self.config.clone())?;
        }
        for component in &mut self.components {
            component.init(tui.size()?)?;
        }

        let action_tx = self.action_tx.clone();

        // Late init
        self.system32 = if cfg!(windows) {
            if let Ok(path) = env::var("SYSTEMROOT") {
                format!("{path}\\System32")
            } else {
                self.action_tx.send(Action::Error(String::from(
                    "ERROR\n\n\nFailed to find SYSTEMROOT",
                )))?;
                return Ok(());
            }
        } else {
            String::from(".")
        };
        action_tx.send(Action::SetMode(Mode::ScanDisks))?;

        loop {
            self.handle_events(&mut tui).await?;
            self.handle_actions(&mut tui)?;
            if self.should_suspend {
                tui.suspend()?;
                action_tx.send(Action::Resume)?;
                action_tx.send(Action::ClearScreen)?;
                // tui.mouse(true);
                tui.enter()?;
            } else if self.should_quit {
                tui.stop()?;
                break;
            }
        }
        tui.exit()?;
        Ok(())
    }

    async fn handle_events(&mut self, tui: &mut Tui) -> Result<()> {
        let Some(event) = tui.next_event().await else {
            return Ok(());
        };
        let action_tx = self.action_tx.clone();
        match event {
            Event::Quit => action_tx.send(Action::Quit)?,
            Event::Tick => action_tx.send(Action::Tick)?,
            Event::Render => action_tx.send(Action::Render)?,
            Event::Resize(x, y) => action_tx.send(Action::Resize(x, y))?,
            Event::Key(key) => self.handle_key_event(key)?,
            _ => {}
        }
        for component in &mut self.components {
            if let Some(action) = component.handle_events(Some(event.clone()))? {
                action_tx.send(action)?;
            }
        }
        Ok(())
    }

    fn handle_key_event(&mut self, key: KeyEvent) -> Result<()> {
        let action_tx = self.action_tx.clone();
        let Some(keymap) = self.config.keybindings.get(&self.cur_mode) else {
            return Ok(());
        };
        if let Some(action) = keymap.get(&vec![key]) {
            info!("Got action: {action:?}");
            action_tx.send(action.clone())?;
        } else {
            // If the key was not handled as a single key action,
            // then consider it for multi-key combinations.
            self.last_tick_key_events.push(key);

            // Check for multi-key combinations
            if let Some(action) = keymap.get(&self.last_tick_key_events) {
                info!("Got action: {action:?}");
                action_tx.send(action.clone())?;
            }
        }
        Ok(())
    }

    fn handle_actions(&mut self, tui: &mut Tui) -> Result<()> {
        while let Ok(action) = self.action_rx.try_recv() {
            if action != Action::Tick && action != Action::Render {
                debug!("{action:?}");
            }
            match action {
                Action::Tick => {
                    self.last_tick_key_events.drain(..);
                    // Check background task(s)
                    if let Some(task) = self.tasks.poll()? {
                        self.handle_task(&task)?;
                    }
                }
                Action::Quit => self.should_quit = true,
                Action::Suspend => self.should_suspend = true,
                Action::Resume => self.should_suspend = false,
                Action::ClearScreen => tui.terminal.clear()?,
                Action::KeyUp => self.list.previous(),
                Action::KeyDown => self.list.next(),
                Action::Error(ref msg) => {
                    self.action_tx
                        .send(Action::DisplayPopup(popup::Type::Error, msg.clone()))?;
                    self.action_tx.send(Action::SetMode(Mode::Failed))?;
                }
                Action::BootScan => self.action_tx.send(Action::SetMode(Mode::BootScan))?,
                Action::DiagMainMenu => self.action_tx.send(Action::SetMode(Mode::DiagMenu))?,
                Action::InstallDriver => {
                    self.action_tx.send(Action::SetMode(Mode::InstallDrivers))?;
                }
                Action::NextScreen => {
                    let next_mode = self.next_mode();
                    self.cur_mode = next_mode;
                    self.action_tx.send(Action::DismissPopup)?;
                    self.action_tx.send(Action::SetMode(next_mode))?;
                }
                Action::Process => match self.cur_mode {
                    Mode::BootDiags => {
                        self.action_tx.send(Action::SetMode(Mode::LogView))?;
                    }
                    Mode::DiagMenu => {
                        // Use highlighted entry
                        if let Some(new_mode) = self.list.get_selected() {
                            self.action_tx.send(Action::SetMode(new_mode))?;
                        }
                    }
                    Mode::Done => {
                        self.action_tx.send(Action::NextScreen)?;
                    }
                    Mode::BootSetup => {
                        let new_mode = self.next_mode();
                        self.action_tx.send(Action::SetMode(new_mode))?;
                    }
                    _ => {}
                },
                Action::Resize(w, h) => self.handle_resize(tui, w, h)?,
                Action::Render => self.render(tui)?,
                Action::ScanDisks => self.action_tx.send(Action::SetMode(Mode::ScanDisks))?,
                Action::Select(one, two) => match self.cur_mode {
                    Mode::InjectDrivers => {
                        if let Some(index) = one {
                            self.inject_driver(index);
                        }
                    }
                    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_dest = one;
                    }
                    Mode::SelectParts => {
                        self.clone.part_index_boot = one;
                        self.clone.part_index_os = two;
                    }
                    Mode::SetBootMode => {
                        if let Some(index) = one {
                            if let Some(boot_mode) = self.boot_modes.get(index) {
                                self.set_boot_mode(boot_mode.to_owned());
                            }
                        }
                    }
                    _ => {}
                },
                Action::SetMode(new_mode) => {
                    self.set_mode(new_mode)?;
                    self.action_tx
                        .send(Action::UpdateFooter(build_footer_string(self.cur_mode)))?;
                    self.action_tx.send(build_left_items(self))?;
                    self.action_tx.send(build_right_items(self))?;
                    self.action_tx.send(Action::Select(None, None))?;
                }
                Action::TaskGroupStart(ref title) => {
                    // TODO: Verify this isn't broken and/or unused
                    if self.cur_mode == Mode::BootScan {
                        self.action_tx.send(Action::DiagLineStart {
                            text: title.clone(),
                        })?;
                        if let Ok(mut diag_groups) = self.diag_groups.lock() {
                            diag_groups.push(DiagGroup::new(get_diag_type(&title)));
                        }
                    }
                }
                Action::TasksComplete => {
                    if self.cur_mode == Mode::BootDiags {
                        self.action_tx.send(Action::DismissPopup)?;
                    } else {
                        self.action_tx.send(Action::NextScreen)?;
                    }
                }
                _ => {}
            }
            for component in &mut self.components {
                if let Some(action) = component.update(action.clone())? {
                    self.action_tx.send(action)?;
                };
            }
        }
        Ok(())
    }

    fn handle_resize(&mut self, tui: &mut Tui, w: u16, h: u16) -> Result<()> {
        tui.resize(Rect::new(0, 0, w, h))?;
        self.render(tui)?;
        Ok(())
    }

    fn handle_task(&mut self, task: &Task) -> Result<()> {
        info!("Handling Task: {task:?}");
        if self.cur_mode == Mode::BootScan {
            if let Ok(mut diag_groups) = self.diag_groups.lock() {
                if let Some(current_group) = diag_groups.last_mut() {
                    match current_group.diag_type {
                        DiagType::Bitlocker => {
                            if let Some(task_result) = &task.result {
                                parse_bitlocker(current_group, task_result.clone());
                            }
                        }
                        DiagType::BootConfigData => {
                            if let Some(task_result) = &task.result {
                                parse_bcd(current_group, task_result.clone());
                            }
                        }
                        DiagType::CheckDisk => {
                            if let Some(task_result) = &task.result {
                                parse_chkdsk(current_group, task_result.clone());
                            }
                        }
                        DiagType::ComponentStore => {
                            if let Some(task_result) = &task.result {
                                parse_dism(current_group, task_result.clone());
                            }
                        }
                        DiagType::SystemFiles => {
                            if let Some(task_result) = &task.result {
                                parse_system_files(current_group, task_result.clone());
                            }
                        }
                        DiagType::Registry => {
                            if let Some(task_result) = &task.result {
                                match &task.task_type {
                                    TaskType::CommandWait(_, cmd_args) => {
                                        parse_registry_hives(
                                            current_group,
                                            task_result.clone(),
                                            cmd_args.clone(),
                                        );
                                    }
                                    _ => {}
                                }
                            }
                        }
                        DiagType::Unknown => {
                            panic!("This shouldn't happen?");
                        }
                    }
                    self.action_tx.send(Action::DiagLineUpdate {
                        result: current_group.get_pass_fail_warn(),
                        text: current_group.result.clone(),
                    })?;
                }
            }
            return Ok(());
        }
        match task.task_type {
            TaskType::CommandNoWait(_, _) | TaskType::CommandWait(_, _) | TaskType::Diskpart(_) => {
                // Check result
                if let Some(result) = &task.result {
                    match result {
                        TaskResult::Error(msg) => {
                            self.action_tx
                                .send(Action::Error(format!("{task:?} Failed: {msg}")))?;
                        }
                        TaskResult::Output(stdout, stderr, success) => {
                            if !success {
                                let msg = if !stdout.is_empty() {
                                    stdout.clone()
                                } else if !stderr.is_empty() {
                                    stderr.clone()
                                } else {
                                    String::from("Unknown Error")
                                };
                                self.action_tx
                                    .send(Action::Error(format!("{task:?} Failed: {msg}")))?;
                            }
                        }
                    }
                }
            }
            TaskType::GroupEnd { ref label } => {
                self.action_tx.send(Action::DiagLineEnd {
                    text: label.clone(),
                })?;
            }
            _ => {}
        }
        Ok(())
    }

    fn queue_boot_scan_tasks(&mut self) -> Result<()> {
        if let Ok(mut diag_groups) = self.diag_groups.lock() {
            diag_groups.clear();
        }
        let disk_list = self.clone.disk_list.lock().unwrap();
        if let Some(disk_index) = self.clone.disk_index_dest {
            if let Some(disk) = disk_list.get(disk_index) {
                let table_type = disk.part_type.clone();
                let letter_boot = disk.get_part_letter(self.clone.part_index_boot.unwrap());
                let letter_os = disk.get_part_letter(self.clone.part_index_os.unwrap());

                // Safety check
                if letter_os.is_empty() {
                    if letter_boot.is_empty() {
                        self.action_tx.send(Action::Error(String::from(
                            "ERROR\n\n\nFailed to get drive letters for the boot and OS volumes",
                        )))?;
                    } else {
                        self.action_tx.send(Action::Error(String::from(
                            "ERROR\n\n\nFailed to get drive letter for the OS volume",
                        )))?;
                    }
                    return Ok(());
                }

                // BCD
                if !letter_boot.is_empty() {
                    self.tasks.add_group(
                        DiagType::BootConfigData.to_string().as_str(),
                        vec![TaskType::CommandWait(
                            PathBuf::from(format!("{}\\bcdedit.exe", &self.system32)),
                            vec![
                                String::from("/store"),
                                format!(
                                    "{letter_boot}:{}\\Boot\\BCD",
                                    if table_type == PartitionTableType::Guid {
                                        "\\EFI\\Microsoft"
                                    } else {
                                        ""
                                    }
                                ),
                                String::from("/enum"),
                            ],
                        )],
                    );
                }

                // Bitlocker
                self.tasks.add_group(
                    DiagType::Bitlocker.to_string().as_str(),
                    vec![
                        TaskType::CommandWait(
                            PathBuf::from(format!("{}\\manage-bde.exe", &self.system32)),
                            vec![String::from("-status"), format!("{letter_os}:")],
                        ),
                        TaskType::CommandWait(
                            PathBuf::from(format!("{}\\manage-bde.exe", &self.system32)),
                            vec![
                                String::from("-protectors"),
                                String::from("-get"),
                                format!("{letter_os}:"),
                            ],
                        ),
                    ],
                );

                // Filesystem Health
                self.tasks.add_group(
                    DiagType::CheckDisk.to_string().as_str(),
                    vec![TaskType::CommandWait(
                        PathBuf::from(format!("{}\\chkdsk.exe", &self.system32)),
                        vec![format!("{letter_os}:")],
                    )],
                );

                // DISM Health
                self.tasks.add_group(
                    DiagType::ComponentStore.to_string().as_str(),
                    vec![TaskType::CommandWait(
                        PathBuf::from(format!("{}\\dism.exe", &self.system32)),
                        vec![
                            format!("/Image:{letter_os}:"),
                            String::from("/Cleanup-Image"),
                            String::from("/ScanHealth"),
                        ],
                    )],
                );

                // Critical Files/Folders
                let paths: Vec<PathBuf> = [
                    // Files/Folders
                    "Users",
                    "Program Files",
                    "Program Files (x86)",
                    "ProgramData",
                    "Windows\\System32\\config",
                ]
                .iter()
                .map(|s| PathBuf::from(format!("{letter_os}:\\{s}")))
                .collect();
                self.tasks.add_group(
                    DiagType::SystemFiles.to_string().as_str(),
                    vec![TaskType::TestPaths(paths)],
                );

                // Registry
                self.tasks.add_group(
                    DiagType::Registry.to_string().as_str(),
                    vec![
                        TaskType::CommandWait(
                            PathBuf::from(format!("{}\\reg.exe", &self.system32)),
                            vec![
                                String::from("load"),
                                String::from("HKLM\\TmpSoftware"),
                                format!("{letter_os}:\\Windows\\System32\\config\\SOFTWARE"),
                            ],
                        ),
                        TaskType::CommandWait(
                            PathBuf::from(format!("{}\\reg.exe", &self.system32)),
                            vec![
                                String::from("load"),
                                String::from("HKLM\\TmpSystem"),
                                format!("{letter_os}:\\Windows\\System32\\config\\SYSTEM"),
                            ],
                        ),
                        TaskType::CommandWait(
                            PathBuf::from(format!("{}\\reg.exe", &self.system32)),
                            vec![
                                String::from("load"),
                                String::from("HKU\\TmpDefault"),
                                format!("{letter_os}:\\Windows\\System32\\config\\DEFAULT"),
                            ],
                        ),
                        TaskType::CommandWait(
                            PathBuf::from(format!("{}\\reg.exe", &self.system32)),
                            vec![String::from("unload"), String::from("HKLM\\TmpSoftware")],
                        ),
                        TaskType::CommandWait(
                            PathBuf::from(format!("{}\\reg.exe", &self.system32)),
                            vec![String::from("unload"), String::from("HKLM\\TmpSystem")],
                        ),
                        TaskType::CommandWait(
                            PathBuf::from(format!("{}\\reg.exe", &self.system32)),
                            vec![String::from("unload"), String::from("HKU\\TmpDefault")],
                        ),
                    ],
                );
                self.tasks.add(TaskType::Sleep); // NOTE: DELETEME
            }
        }
        Ok(())
    }

    fn render(&mut self, tui: &mut Tui) -> Result<()> {
        tui.draw(|frame| {
            if let [header, _body, footer, left, right, popup, progress, results] =
                get_chunks(frame.area())[..]
            {
                let component_areas = vec![
                    header, // Title Bar
                    header, // FPS Counter
                    left, right, footer, popup, // core
                    progress, results, // boot-diags
                ];
                for (component, area) in zip(self.components.iter_mut(), component_areas) {
                    if let Err(err) = component.draw(frame, area) {
                        let _ = self
                            .action_tx
                            .send(Action::Error(format!("Failed to draw: {err:?}")));
                    }
                }
            };
        })?;
        Ok(())
    }
}

fn centered_rect(percent_x: u16, percent_y: u16, r: Rect) -> Rect {
    // Cut the given rectangle into three vertical pieces
    let popup_layout = Layout::default()
        .direction(Direction::Vertical)
        .constraints([
            Constraint::Percentage((100 - percent_y) / 2),
            Constraint::Percentage(percent_y),
            Constraint::Percentage((100 - percent_y) / 2),
        ])
        .split(r);

    // Then cut the middle vertical piece into three width-wise pieces
    Layout::default()
        .direction(Direction::Horizontal)
        .constraints([
            Constraint::Percentage((100 - percent_x) / 2),
            Constraint::Percentage(percent_x),
            Constraint::Percentage((100 - percent_x) / 2),
        ])
        .split(popup_layout[1])[1] // Return the middle chunk
}

fn get_chunks(r: Rect) -> Vec<Rect> {
    let mut chunks: Vec<Rect> = Vec::with_capacity(6);

    // Main sections
    chunks.extend(
        Layout::default()
            .direction(Direction::Vertical)
            .constraints([
                Constraint::Length(3),
                Constraint::Min(1),
                Constraint::Length(3),
            ])
            .split(r)
            .to_vec(),
    );

    // Left/Right
    chunks.extend(
        Layout::default()
            .direction(Direction::Horizontal)
            .constraints([Constraint::Percentage(50), Constraint::Percentage(50)])
            .split(centered_rect(90, 90, chunks[1]))
            .to_vec(),
    );

    // Popup
    chunks.push(centered_rect(60, 25, r));

    // Progress
    chunks.push(centered_rect(60, 70, r));

    // Results
    chunks.push(centered_rect(60, 70, r));

    // Done
    chunks
}

fn build_footer_string(cur_mode: Mode) -> String {
    match cur_mode {
        Mode::BootDiags => {
            String::from("(Enter) to select / (m) for menu / (s) to start over / (q) to quit")
        }
        Mode::BootScan | Mode::BootSetup | Mode::Home | Mode::ScanDisks => {
            String::from("(q) to quit")
        }
        Mode::DiagMenu | Mode::SelectParts => {
            String::from("(Enter) to select / (s) to start over / (q) to quit")
        }
        Mode::Done => String::from("(Enter) to continue / (q) to quit"),
        Mode::InstallDrivers | Mode::InjectDrivers | Mode::SetBootMode => {
            String::from("(Enter) to select / (q) to quit")
        }
        Mode::LogView => {
            String::from("(Enter | Esc) to close log / (up | down) to scroll / (q) to quit")
        }
        Mode::SelectDisks => String::from(
            "(Enter) to select / / (i) to install driver / (r) to rescan / (q) to quit",
        ),
        Mode::Failed => String::from("(Enter) or (q) to quit"),
        // Invalid states
        Mode::Confirm
        | Mode::Clone
        | Mode::PEMenu
        | Mode::PreClone
        | Mode::PostClone
        | Mode::SelectTableType => {
            panic!("This shouldn't happen?")
        }
    }
}

fn build_left_items(app: &App) -> Action {
    let mut items = Vec::new();
    let mut labels = vec![String::new(), String::new()];
    let select_type: SelectionType;
    let title: String;
    match app.cur_mode {
        Mode::Home => {
            select_type = SelectionType::Loop;
            title = String::from("Home");
        }
        Mode::DiagMenu => {
            select_type = SelectionType::Loop;
            title = String::from("Troubleshooting");
            app.list.items.iter().for_each(|mode| {
                let (name, _) = get_mode_strings(*mode);
                items.push(name);
            });
        }
        Mode::InstallDrivers => {
            select_type = SelectionType::One;
            title = String::from("Install Drivers");
            app.clone
                .driver_list
                .iter()
                .for_each(|driver| items.push(driver.to_string()));
        }
        Mode::InjectDrivers => {
            select_type = SelectionType::One;
            title = String::from("Select Drivers");
            app.clone
                .driver_list
                .iter()
                .for_each(|driver| items.push(driver.to_string()));
        }
        Mode::SelectDisks => {
            select_type = SelectionType::One;
            title = String::from("Select Disk");
            let disk_list = app.clone.disk_list.lock().unwrap();
            disk_list
                .iter()
                .for_each(|disk| items.push(disk.description.to_string()));
        }
        Mode::BootScan | Mode::ScanDisks => {
            select_type = SelectionType::Loop;
            title = String::from("Processing");
        }
        Mode::SelectParts => {
            select_type = SelectionType::Two;
            title = String::from("Select Boot and OS Partitions");
            labels[0] = String::from("boot");
            labels[1] = 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::BootDiags | Mode::LogView => {
            select_type = SelectionType::Loop;
            let (new_title, _) = get_mode_strings(app.cur_mode);
            title = new_title;
            if let Ok(diag_groups) = app.diag_groups.lock() {
                let labels: Vec<String> = diag_groups
                    .iter()
                    .map(|group| {
                        let label = group.diag_type.to_string();
                        let status = if group.get_pass_fail_warn() == DiagResult::Pass {
                            "" // Leave blank if OK
                        } else {
                            "  -- Issue(s) detected"
                        };
                        format!("{label}{status}")
                    })
                    .collect();
                items.extend(labels.into_iter());
            }
        }
        Mode::BootSetup => {
            select_type = SelectionType::Loop;
            let (new_title, _) = get_mode_strings(app.cur_mode);
            title = new_title;
        }
        Mode::SetBootMode => {
            select_type = SelectionType::One;
            title = String::from("Set Boot Mode");
            app.boot_modes.iter().for_each(|entry| {
                items.push(format!("{:?} Safe Mode", entry));
            });
        }
        Mode::Done | Mode::Failed => {
            select_type = SelectionType::Loop;
            title = String::from("Done");
        }
        // Invalid states
        Mode::SelectTableType
        | Mode::PEMenu
        | Mode::Confirm
        | Mode::PreClone
        | Mode::Clone
        | Mode::PostClone => {
            panic!("This shouldn't happen?")
        }
    };
    Action::UpdateLeft(title, labels, items, select_type)
}

fn build_right_items(app: &App) -> Action {
    let mut items: Vec<Vec<DVLine>> = Vec::new();
    let mut labels: Vec<Vec<DVLine>> = Vec::new();
    let mut start_index = 0;
    match app.cur_mode {
        Mode::DiagMenu => {
            let header_lines = get_disk_header(app);
            if !header_lines.is_empty() {
                items.push(header_lines);
                start_index += 1;
            }
            app.list.items.iter().for_each(|mode| {
                let (name, description) = get_mode_strings(*mode);
                items.push(vec![
                    DVLine {
                        line_parts: vec![name],
                        line_colors: vec![Color::Cyan],
                    },
                    DVLine {
                        line_parts: vec![String::new()],
                        line_colors: vec![Color::Reset],
                    },
                    DVLine {
                        line_parts: vec![description],
                        line_colors: vec![Color::Reset],
                    },
                ]);
            });
        }
        Mode::BootDiags => {
            let header_lines = get_disk_header(app);
            if !header_lines.is_empty() {
                items.push(header_lines);
                start_index += 1;
            }
            if let Ok(diag_groups) = app.diag_groups.lock() {
                let mut summary: Vec<DVLine> = Vec::new();
                diag_groups
                    .iter()
                    .for_each(|group| summary.extend(group.get_logs_summary().into_iter()));
                items.push(summary);
            }
        }
        Mode::InjectDrivers | Mode::InstallDrivers => {
            items.push(vec![DVLine {
                line_parts: vec![String::from("CPU")],
                line_colors: vec![Color::Cyan],
            }]);
            items.push(vec![DVLine {
                line_parts: vec![get_cpu_name()],
                line_colors: vec![Color::Reset],
            }]);
            start_index += 2;
        }
        Mode::SelectDisks => {
            let dest_dv_line = DVLine {
                line_parts: vec![String::from("Disk")],
                line_colors: vec![Color::Cyan],
            };
            labels.push(vec![dest_dv_line]);
            let disk_list = app.clone.disk_list.lock().unwrap();
            disk_list
                .iter()
                .for_each(|disk| items.push(get_disk_description_right(disk, &None)));
        }
        Mode::SelectParts => {
            ["Boot", "OS"].iter().for_each(|s| {
                labels.push(vec![DVLine {
                    line_parts: vec![String::from(*s)],
                    line_colors: vec![Color::Cyan],
                }])
            });
            if let Some(index) = app.clone.disk_index_dest {
                start_index += 1;
                let disk_list = app.clone.disk_list.lock().unwrap();
                if let Some(disk) = disk_list.get(index) {
                    // Disk Details
                    items.push(get_disk_description_right(disk, &None));

                    // Partition Details
                    disk.parts
                        .iter()
                        .for_each(|part| items.push(get_part_description(part)));
                }
            }
        }
        Mode::SetBootMode => {
            app.boot_modes.iter().for_each(|mode| {
                match mode {
                    SafeMode::Disable => {
                        items.push(vec![DVLine {
                            line_parts: vec![String::from("Disable Safe Mode")],
                            line_colors: vec![Color::Reset],
                        }]);
                    }
                    SafeMode::Enable => {
                        items.push(vec![DVLine {
                            line_parts: vec![String::from("Enable Safe Mode (minimal)")],
                            line_colors: vec![Color::Reset],
                        }]);
                    }
                };
            });
        }
        _ => {}
    }
    Action::UpdateRight(labels, start_index, items)
}

fn get_disk_header(app: &App) -> Vec<DVLine> {
    let mut header_lines: Vec<DVLine> = Vec::new();
    if let Some(index) = app.clone.disk_index_dest {
        let disk_list = app.clone.disk_list.lock().unwrap();
        if let Some(disk) = disk_list.get(index) {
            let mut parts: Vec<usize> = Vec::new();
            if let Some(index) = app.clone.part_index_boot {
                parts.push(index);
            }
            if let Some(index) = app.clone.part_index_os {
                parts.push(index);
            }
            // Disk Details
            header_lines.append(&mut vec![
                DVLine {
                    line_parts: vec![String::from("Disk")],
                    line_colors: vec![Color::Cyan],
                },
                DVLine {
                    line_parts: vec![String::new()],
                    line_colors: vec![Color::Reset],
                },
            ]);
            header_lines.append(&mut get_disk_description_right(disk, &Some(parts)));
        }
    }
    header_lines
}

fn get_mode_strings(mode: Mode) -> (String, String) {
    match mode {
        Mode::BootScan | Mode::BootDiags | Mode::LogView => (
            String::from("Boot Diagnostics"),
            String::from("Check for common Windows boot issues"),
        ),
        Mode::BootSetup => (
            String::from("Boot Setup"),
            String::from("Create or recreate boot files"),
        ),
        Mode::InjectDrivers => (
            String::from("Inject Drivers"),
            String::from("Inject drivers into existing Windows environment"),
        ),
        Mode::SetBootMode => (
            String::from("Toggle Safe Mode"),
            String::from("Enable or disable safe mode"),
        ),
        _ => panic!("This shouldn't happen"),
    }
}