daknuett/make_prompt_rs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

works

Browse Source
This commit is contained in:
Daniel Knuettel 2024-04-16 20:34:30 +02:00
parent 8fac4e5aff
commit 73be916538
2 changed files with 36 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
source.sh
View File

@ -3,6 +3,8 @@ bin_path=$(pwd)/target/debug/make_prompt
PROMPT_COMMAND='make_prompt' PROMPT_COMMAND='make_prompt'
make_prompt () { make_prompt () {
local prompt_pref=$($bin_path $?) local lastexit=$?
local bgjobs=$(jobs | wc -l)
local prompt_pref=$($bin_path $lastexit $bgjobs)
PS1="$prompt_pref" PS1="$prompt_pref"
} }

46
src/make_prompt.rs
View File

@ -5,7 +5,7 @@ use std::io::BufRead;
fn fixed_width(input: &String, len: usize) -> String fn fixed_width(input: &String, len: usize) -> String
{ {
let clen = input.len(); let clen = input.chars().count();
const PLACE_HOLDER: &str = "🤄"; const PLACE_HOLDER: &str = "🤄";
if clen > len if clen > len
@ -21,8 +21,8 @@ fn fixed_width(input: &String, len: usize) -> String
else else
{ {
let n_pad = len - clen; let n_pad = len - clen;
let n_pad_left = n_pad / 2; let n_pad_right = n_pad / 2;
let n_pad_right = n_pad_left + { if n_pad % 2 == 0 {0} else {1}}; let n_pad_left = n_pad_right + { if n_pad % 2 == 0 {0} else {1}};
let mut result: String = str::repeat(" ", n_pad_left).to_owned(); let mut result: String = str::repeat(" ", n_pad_left).to_owned();
result.push_str(input); result.push_str(input);
result.push_str(&str::repeat(" ", n_pad_right)); result.push_str(&str::repeat(" ", n_pad_right));
@ -197,15 +197,28 @@ fn main()
} }
}; };
let raw_cwd = getenv("PWD");
let home = getenv("HOME");
let cwd = { let cwd = {
if raw_cwd.starts_with(&home)
{
let everything_after_home = &raw_cwd[home.len()..];
let mut res = "~".to_string();
res.push_str(everything_after_home);
res
}
else
{
raw_cwd.to_string()
}
};
let bgjobs = {
if args.len() <= 2 if args.len() <= 2
{ {
let promise = std::env::current_dir(); "0".to_string()
match promise
{
Ok(v) => v.display().to_string(),
Err(_reason) => panic!("failed to get cwd")
}
} }
else else
{ {
@ -219,18 +232,23 @@ fn main()
print!("{}{}", shell_color_str(&"┌[", SH_GREEN) print!("{}{}", shell_color_str(&"┌[", SH_GREEN)
, battery_info(12)); , battery_info(12));
print!("{}{}", shell_color_str(&"|", SH_GREEN) print!("{}{}", shell_color_str(&"|", SH_GREEN)
, fixed_width(&get_git_branch(& std::path::Path::new(&cwd)), 12)); , fixed_width(&get_git_branch(& std::path::Path::new(&raw_cwd)), 10));
print!("{}{}", shell_color_str(&"|", SH_GREEN) print!("{}{}", shell_color_str(&"|", SH_GREEN)
, fixed_width(&get_editor(), 10)); , fixed_width(&get_editor(), 12));
print!("{}{}", shell_color_str(&"|", SH_GREEN) print!("{}{}", shell_color_str(&"|", SH_GREEN)
, fixed_width(lastexit, 4)); , fixed_width(lastexit, 3));
println!("{}", shell_color_str(&"]┐", SH_GREEN)); println!("{}", shell_color_str(&"]┐", SH_GREEN));
} }
{ {
print!("{}{}", shell_color_str(&"╞[", SH_GREEN) print!("{}{}", shell_color_str(&"╞[", SH_GREEN)
, fixed_width(&getenv("HOSTNAME"), 12)); , shell_color_str(&fixed_width(&getenv("HOSTNAME"), 12), SH_CYAN));
print!("{}{}", shell_color_str(&"|", SH_GREEN) print!("{}{}", shell_color_str(&"|", SH_GREEN)
, fixed_width(&getenv("USER"), 10)); , shell_color_str(&fixed_width(&getenv("USER"), 10), SH_RED));
print!("{}{}", shell_color_str(&"|", SH_GREEN)
, fixed_width(&cpu_arch, 12));
print!("{}{}", shell_color_str(&"|", SH_GREEN)
, fixed_width(&bgjobs, 3));
println!("{}", shell_color_str(&"]┘", SH_GREEN)); println!("{}", shell_color_str(&"]┘", SH_GREEN));
} }
{ {