Mirrors/bat
2
0
Fork 0
mirror of https://github.com/sharkdp/bat synced 2024-11-27 22:30:33 +00:00
Code Issues Projects Releases Packages Wiki Activity

Graceful handling of error conditions

Browse source
This commit is contained in:
Josh Mickley 2020-03-25 22:03:10 -04:00 committed by David Peter
parent 376c556862
commit 40a827ebcb
2 changed files with 21 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
src/bin/bat/config.rs
View file

@ -16,28 +16,31 @@ pub fn config_file() -> PathBuf {
.unwrap_or_else(|| PROJECT_DIRS.config_dir().join("config"))
}
pub fn generate_config_file() {
pub fn generate_config_file() -> bat::errors::Result<()> {
let config_file = config_file();
if config_file.exists() {
println!("A config file already exists at: {}", config_file.to_string_lossy());
print!("Overwrite? (y/n): ");
let _ = io::stdout().flush();
print!("Overwrite? (y/N): ");
io::stdout().flush()?;
let mut decision = String::new();
io::stdin().read_line(&mut decision).expect("Failed to read input");
io::stdin().read_line(&mut decision)?;
if !decision.trim().eq_ignore_ascii_case("Y") {
return;
return Ok(());
}
} else {
let config_dir = config_file.parent().unwrap();
if !config_dir.exists() {
fs::create_dir(config_dir).expect("Unable to create config directory");
let config_dir = config_file.parent();
match config_dir {
Some(path) => fs::create_dir_all(path)?,
None => return Ok(Err(format!("Unable to write config file to: {}", config_file.to_string_lossy()))?),
}
}
let default_config = "# Specify desired theme (e.g. \"TwoDark\")
#--theme=\"TwoDark\"
let default_config = r#"# bat config
# Specify desired theme (e.g. "TwoDark"). Issue `bat --list-themes` for a list of all available themes
#--theme="TwoDark"
# Enable this to use italic text on the terminal (not supported on all terminals):
#--italic-text=always
@ -46,14 +49,16 @@ pub fn generate_config_file() {
#--paging=never
# Use C++ syntax for .ino files
#--map-syntax \"*.ino:C++\"
#--map-syntax "*.ino:C++"
# Use \".gitignore\"-style highlighting for \".ignore\" files
#--map-syntax \".ignore:Git Ignore\"
";
# Use ".gitignore"-style highlighting for ".ignore" files
#--map-syntax ".ignore:Git Ignore"
"#;
fs::write(&config_file, default_config).expect("Error writing config file!");
fs::write(&config_file, default_config)?;
println!("Success! Config file written to {}", config_file.to_string_lossy());
return Ok(());
}
pub fn get_args_from_config_file() -> Result<Vec<OsString>, shell_words::ParseError> {

6
src/bin/bat/main.rs
View file

@ -179,19 +179,15 @@ fn run() -> Result<bool> {
if app.matches.is_present("list-languages") {
list_languages(&config)?;
Ok(true)
} else if app.matches.is_present("list-themes") {
list_themes(&config)?;
Ok(true)
} else if app.matches.is_present("config-file") {
println!("{}", config_file().to_string_lossy());
Ok(true)
} else if app.matches.is_present("generate-config-file") {
generate_config_file();
generate_config_file()?;
Ok(true)
} else if app.matches.is_present("config-dir") {
writeln!(io::stdout(), "{}", config_dir())?;