candifloss/PrettyPrompt
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

Git repo name & path abrev

Browse Source
This commit is contained in:
Candifloss 2024-08-17 23:57:14 +05:30
parent 27b0a77202
commit e43fd9d7a1
1 changed files with 16 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
src/main.rs
View File

@ -1,7 +1,7 @@
use std::env::{current_dir,var_os,args};
use std::path::PathBuf;
use colored::{Colorize,ColoredString};
use std::process::Command;
//use std::path::PathBuf;
//use std::ffi::OsString;
fn get_shell_char (shell: String) -> String {
@ -71,15 +71,14 @@ fn get_git_repo_name (git_repo_root: String) -> String {
fn get_git_char (git_branch: String) -> ColoredString {
match git_branch.as_str() {
"" => "".clear(),
"main" => " 󰊢 ".truecolor(178,98,44),
"master" => " 󰊢 ".truecolor(196,132,29),
_ => " 󰊢 ".truecolor(82,82,82),
}
}
fn abrev_path (pwd: PathBuf, homedir: String) -> String {
let mut path = pwd.display().to_string();
path = path.replace(&homedir, "~"); // Abreviate homedir with "~"
fn abrev_path (path: String) -> String {
let mut short_dir = path.clone();
let slashes = path.matches("/").count();
@ -106,13 +105,9 @@ fn main() -> std::io::Result<()> {
let _user = var_os("USER").expect("UnknownUser").to_str().expect("UnknownUser").to_string();
/*
if user == "root" {
println!("roo_user");
println!("root_user");
}*/
//pwd
let pwd = current_dir()?;
let homedir = var_os("HOME").expect("UnknownDir").to_str().expect("UnknownDir").to_string();
//Shell symbol
let args: Vec<String> = args().collect();
let shell: String;
@ -137,15 +132,24 @@ fn main() -> std::io::Result<()> {
//Git status
let git_branch = get_git_branch();//.clear();
let git_repo_root = get_git_root();
let git_repo_name = get_git_repo_name(git_repo_root);
let git_repo_name = get_git_repo_name(git_repo_root.clone()).truecolor(200, 150, 80);
let git_char = get_git_char(git_branch); //.to_string());
//pwd
let homedir = var_os("HOME").expect("UnknownDir").to_str().expect("UnknownDir").to_string();
let pwd = current_dir()?;
let mut cur_dir = pwd.display().to_string();
cur_dir = cur_dir.replace(&git_repo_root, ""); // Remove git repo root
cur_dir = cur_dir.replace(&homedir, "~"); // Abreviate homedir with "~"
cur_dir = abrev_path(cur_dir);
print!("{}{}{}{}{}{} ",
ssh_char,
get_shell_char(shell).truecolor(75,75,75),
git_repo_name,
git_char,
abrev_path(pwd,homedir).italic().truecolor(82,82,82),
//abrev_path(pwd,homedir).italic().truecolor(82,82,82),
cur_dir.italic().truecolor(82,82,82),
angle.truecolor(0, 255, 180),
);