// 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 std::sync::{Arc, Mutex};

use serde::{Deserialize, Serialize};

use crate::system::{
    disk::{Disk, PartitionTableType},
    drivers,
};

#[derive(Default, Debug, Copy, Clone, PartialEq, Eq, Hash, Serialize, Deserialize)]
pub enum Mode {
    // Core
    #[default]
    Home,
    Done,
    Failed,
    // Boot Diags
    DiagMenu,
    BootDiags,
    BootScan,
    BootSetup,
    InjectDrivers,
    SetBootMode,
    // Clone
    ScanDisks,
    InstallDrivers,
    SelectDisks,
    SelectTableType,
    Confirm,
    PreClone,
    Clone,
    SelectParts,
    PostClone,
    // WinPE
    PEMenu,
}

#[derive(Debug, Default)]
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<drivers::Driver>,
    pub table_type: Option<PartitionTableType>,
}

impl CloneSettings {
    pub fn new(disk_list: Arc<Mutex<Vec<Disk>>>) -> Self {
        CloneSettings {
            disk_list,
            ..Default::default()
        }
    }

    pub fn scan_drivers(&mut self) {
        self.driver_list = drivers::scan();
    }
}