#![feature(never_type)]
#![feature(exit_status_error)]
use std::collections::HashSet;
use datatypes::Bread;
pub mod args;
pub mod config;
pub mod datatypes;
pub mod fs_operation;
use datatypes::*;
pub fn prompt(mut conf: config::Config) -> Result<Option<HashSet<String>>, String> {
if conf.delete {
let mut rl = rustyline::Editor::<()>::new();
conf.delete = false;
let breads = fs_operation::handle_files(conf).collect::<Vec<_>>();
let mut files_changed = None;
loop {
breads.iter().for_each(|b| println!("{}", b));
match rl.readline("Are you sure you want to delete all crumbs? (y/n/s/i): ") {
Ok(s) => match s.as_str() {
"y" => {
let mut cache = HashSet::new();
for b in breads {
cache.insert(
fs_operation::delete_the_crumbs(b).map_err(|e| e.to_string())?,
);
}
if !cache.is_empty() {
files_changed = Some(cache)
};
}
"n" => (), "s" => continue,
"i" => {
files_changed = Some(prompt_bread(breads.into_iter(), &mut rl, "delete")?)
}
_ => return Err("I don't understand, please give y/n/s/i".to_string()),
},
Err(e) => return Err(format!("error in prompt readline {}", e.to_string())),
}
break;
}
Ok(files_changed)
} else if conf.restore {
let mut rl = rustyline::Editor::<()>::new();
let breads = fs_operation::handle_files(conf).collect::<Vec<_>>();
let mut files_changed = None;
loop {
breads.iter().for_each(|b| println!("{}", b));
match rl.readline("Are you sure you want to restore all crumbs? (y/n/s/i): ") {
Ok(s) => match s.as_str() {
"y" => {
let mut cache = HashSet::new();
for b in breads {
cache.insert(
fs_operation::restore_the_crumb(b).map_err(|e| e.to_string())?,
);
}
if !cache.is_empty() {
files_changed = Some(cache)
};
}
"n" => (), "s" => continue,
"i" => {
files_changed = Some(prompt_bread(breads.into_iter(), &mut rl, "restore")?)
}
_ => return Err("I don't understand, please give y/n/s/i".to_string()),
},
Err(e) => return Err(format!("error in prompt readline {}", e.to_string())),
}
break;
}
Ok(files_changed)
} else {
match conf.output {
config::OutputFormat::None => {
fs_operation::handle_files(conf).for_each(|b| println!("{}", b))
}
config::OutputFormat::Json => {
println!(
"{}",
serde_json::to_string(&fs_operation::handle_files(conf).collect::<Vec<_>>())
.map_err(|e| e.to_string())?
)
}
config::OutputFormat::List => fs_operation::handle_files(conf).for_each(|b| {
b.crumbs
.iter()
.for_each(|crumb| println!("{}:{}", b.file_path, crumb.list_format()))
}),
config::OutputFormat::Range => fs_operation::handle_files(conf).for_each(|b| {
b.crumbs.iter().for_each(|crumb| {
println!(
r#"File path: {}
The mark:
{}
Context around:
{}
=======================
"#,
b.file_path,
crumb,
crumb.range_format()
)
})
}),
}
Ok(None)
}
}
fn prompt_bread(
breads: impl Iterator<Item = Bread>,
rl: &mut rustyline::Editor<()>,
op: &str,
) -> Result<HashSet<String>, String> {
let mut files_changed = HashSet::new();
for b in breads {
loop {
println!("{}", b);
match rl.readline(&format!(
"Are you sure you want to {} this bread {}? (y/n/s/i): ",
op, b.file_path
)) {
Ok(s) => match s.as_str() {
"y" => match op {
"delete" => {
files_changed.insert(
fs_operation::delete_the_crumbs(b).map_err(|e| e.to_string())?,
);
}
"restore" => {
files_changed.insert(
fs_operation::restore_the_crumb(b).map_err(|e| e.to_string())?,
);
}
_ => (),
},
"n" => {}
"s" => {
continue;
}
"i" => {
let go_to_handle = prompt_crumbs(b.crumbs.iter(), rl, op)?;
if go_to_handle.len() != 0 {
match op {
"delete" => {
files_changed.insert(
fs_operation::delete_the_crumbs_on_special_index(
b,
go_to_handle,
)
.map_err(|e| e.to_string())?,
);
}
"restore" => {
files_changed.insert(
fs_operation::restore_the_crumb_on_special_index(
b,
go_to_handle,
)
.map_err(|e| e.to_string())?,
);
}
_ => (),
}
}
}
_ => {
println!("I don't understand, please give y/n/s/i");
}
},
Err(e) => return Err(e.to_string()),
}
break;
}
}
Ok(files_changed)
}
fn prompt_crumbs<'a>(
crumbs: impl Iterator<Item = &'a Crumb>,
rl: &mut rustyline::Editor<()>,
op: &str,
) -> Result<HashSet<usize>, String> {
let mut going_to_handle_crumbs_indexes = HashSet::new();
for (ind, c) in crumbs.enumerate() {
loop {
println!("{}", c);
match rl.readline(&format!(
"Are you sure you want to {} this crumb? (y/n/s): ",
op
)) {
Ok(s) => match s.as_str() {
"y" => {
going_to_handle_crumbs_indexes.insert(ind);
}
"n" => {}
"s" => {
continue;
}
_ => {
println!("I don't understand, please give y/n/s");
}
},
Err(e) => return Err(e.to_string()),
}
break;
}
}
Ok(going_to_handle_crumbs_indexes)
}