code_it_later_rs/

lib.rs

#![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 {
        // only delete is true gonna triger the prompt
        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" => (), // do nothing
                    "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" => (), // do nothing
                    "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 {
            // incase need show again
            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)
}