46 lines
1.1 KiB
Rust
46 lines
1.1 KiB
Rust
|
use std::env::current_dir as current_dir;
|
|||
|
|
//use std::env::var_os as var_os;
|
|||
|
|
use std::env::args as args;
|
|||
|
|
use colored::Colorize;
|
|||
|
|
|
|||
|
|
fn main() -> std::io::Result<()> {
|
|||
|
|
let angle = "❯";
|
|||
|
|
let pwd = current_dir()?;
|
|||
|
|
let dir = pwd.display().to_string();
|
|||
|
|
let args: Vec<String> = args().collect();
|
|||
|
|
let shell: String;
|
|||
|
|
|
|||
|
|
if args.len() > 1 {
|
|||
|
|
shell = args[1].clone();
|
|||
|
|
}
|
|||
|
|
else {
|
|||
|
|
shell = "none".to_string();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
let shell_char = match shell.as_str() {
|
|||
|
|
"bash" => "",
|
|||
|
|
"zsh" => "",
|
|||
|
|
"fish" => "",
|
|||
|
|
"nushell" => "",
|
|||
|
|
"ion" => "",
|
|||
|
|
_ => "",
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
/*
|
|||
|
|
match var_os(USER).as_str() {
|
|||
|
|
"root" => {
|
|||
|
|
shell_char = shell_char.truecolor(255,53,94);
|
|||
|
|
},
|
|||
|
|
_ => {
|
|||
|
|
shell_char = shell_char.truecolor(54,178,91);
|
|||
|
|
}
|
|||
|
|
}*/
|
|||
|
|
|
|||
|
|
print!("{} {}{} ",
|
|||
|
|
shell_char.truecolor(75,75,75),
|
|||
|
|
dir.italic().truecolor(75,75,75),
|
|||
|
|
angle.truecolor(0, 255, 180),
|
|||
|
|
);
|
|||
|
|
Ok(())
|
|||
|
|
}
|