use std::{fs, process}; use std::path::{Path, PathBuf}; use std::io::{self, Write}; use colored::Colorize; pub fn init() { let ventodir: PathBuf = env_config(); if ventodir.is_dir() { let mut answer = String::new(); print!("⚠️ {} {}", format!("WARNING:").bold().red(), "Vento has already been initialized. Reinitializing will delete all files on the directory for Vento. Do you wish to proceed? (y/N) "); let _ = io::stdout().flush(); io::stdin() .read_line(&mut answer) .expect("❌ Failed to read input"); match answer.as_str().trim() { "y" | "Y" => {fs::remove_dir_all(&ventodir).expect("❌ Vento was unable to initalize. Do you have the correct permissions?");}, "n" | "N" | _ => process::exit(0) }; }; create_slots(ventodir); } pub fn list(slot: &str) { let ventodir: PathBuf = env_config(); let active: PathBuf = [ventodir.to_path_buf(), Path::new("active").to_path_buf()].iter().collect(); let inactive: PathBuf = [ventodir.to_path_buf(), Path::new("inactive").to_path_buf()].iter().collect(); let slotdir: PathBuf; match slot { "inactive" => slotdir = inactive, "active" | _ => slotdir = active }; println!("🗃️ {}", format!("Files in {} inventory:", match slot { "active" => format!("{}", slot).bold(), "inactive" | _ => format!("{}", slot).blue().bold() }).green()); for file in fs::read_dir(&slotdir).unwrap() { println!("- {}", file.unwrap().path().file_name().unwrap().to_os_string().into_string().unwrap()); }; } pub fn switch() { let ventodir: PathBuf = env_config(); let active: PathBuf = [ventodir.to_path_buf(), Path::new("active").to_path_buf()].iter().collect(); let temp: PathBuf = [ventodir.to_path_buf(), Path::new("temp").to_path_buf()].iter().collect(); let inactive: PathBuf = [ventodir.to_path_buf(), Path::new("inactive").to_path_buf()].iter().collect(); fs::rename(&active, &temp).expect("❌ Vento was unable to switch slots. Try running vento init and try again"); fs::rename(&inactive, &active).expect("❌ Vento was unable to switch slots. Try running vento init and try again"); fs::rename(&temp, &inactive).expect("❌ Vento was unable to switch slots. Try running vento init and try again"); println!("🎉 {}", format!("Switched inventory slots!").green()); } fn env_config() -> PathBuf { let emptypath = PathBuf::new(); let home = match dirs::home_dir() { Option::Some(dir) => dir, _ => PathBuf::new() }; if home == emptypath { println!("❌ {}", format!("Vento was unable to detect your home folder. Have you configured your environment correctly?").red()); process::exit(0); } else { return [home, Path::new(".vento").to_path_buf()].iter().collect(); }; } fn create_slots(dir: PathBuf) { let active: PathBuf = [dir.to_path_buf(), Path::new("active").to_path_buf()].iter().collect(); let inactive: PathBuf = [dir.to_path_buf(), Path::new("inactive").to_path_buf()].iter().collect(); fs::create_dir_all(active).expect("❌ Vento was unable to initalize. Do you have the correct permissions?"); fs::create_dir_all(inactive).expect("❌ Vento was unable to initalize. Do you have the correct permissions?"); println!("🎉 {}", format!("Vento has been succesfully initialized!").green()); }