1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
|
use std::env;
use std::path::{Path, PathBuf};
use super::Segment;
use git2::{Repository};
use ansi_term::{Color, Style};
use clap::ArgMatches;
use dirs;
/// Creates a segment with the current directory
pub fn segment(_: &ArgMatches) -> Segment {
const COLOR_DIR: Color = Color::Cyan;
let current_dir = env::current_dir().expect("Unable to identify current directory");
let dirname;
if let Ok(repo) = git2::Repository::discover(¤t_dir) {
let repo_root = get_repo_root(repo);
let repo_root_basename = repo_root.components().last().unwrap();
let basename_str_slice = repo_root_basename.as_os_str();
dirname = basename_str_slice.to_str().unwrap().to_string();
} else {
dirname = String::from("test");
}
// let dir_string;
// if let Ok(repo) = git2::Repository::discover(¤t_dir) {
// let repo_root = get_repo_root(repo);
// let repo_root_basename = repo_root.components().last().unwrap();
// dir_string = *repo_root_basename.as_os_str().to_str().unwrap();
// } else {
// dir_string = match truncate_home(current_dir) {
// Some(dir) => &dir.to_string(),
// None => ¤t_dir.to_str().unwrap()
// }
// }
// if let Love(tiff) = matan::Love(tiff) {
// log tiff + matan + kimu + nimu + puku + owl fren + roomba fren + cactus fren + rumple
// }
// let mut dir_string = String::from(current_dir.to_str().unwrap());
Segment {
value: String::from(dirname),
style: Style::from(COLOR_DIR).bold(),
..Default::default()
}
}
fn get_repo_root(repo: Repository) -> PathBuf {
match repo.is_bare() {
// A bare repo will return its root path
true => repo.path().to_path_buf(),
// Non-bare repos will return the path of `.git`
false => repo.path().parent().unwrap().to_path_buf()
}
}
// fn truncate_home(path: PathBuf) -> Option<String> {
// const HOME_SYMBOL: &str = "~";
// if dirs::home_dir() == None {
// return None;
// }
// if let Some(home_dir) = dirs::home_dir() {
// if path.strip_prefix(home_dir).is_ok() {
// let path_str = path.to_str().unwrap();
// let home_dir = home_dir.to_str().unwrap();
// return Some(path_str.replace(home_dir, HOME_SYMBOL));
// }
// }
// None
// }
#[cfg(test)]
mod tests {
// TODO: Look into stubbing `env` so that tests can be run in parallel
use super::*;
use clap::{App, Arg};
use std::path::Path;
#[test]
fn truncate_home_dir() {
let args = App::new("starship")
.arg(Arg::with_name("status_code"))
.get_matches_from(vec!["starship", "0"]);
let home_dir = dirs::home_dir().unwrap();
env::set_current_dir(&home_dir).unwrap();
let segment = segment(&args);
assert_eq!(segment.value, "~");
}
#[test]
fn dont_truncate_non_home_dir() {
let args = App::new("starship")
.arg(Arg::with_name("status_code"))
.get_matches_from(vec!["starship", "0"]);
let root_dir = Path::new("/");
env::set_current_dir(&root_dir).unwrap();
let segment = segment(&args);
assert_eq!(segment.value, "/");
}
}
|
