emplace 0.2.11-alpha.0

use crate::{package::PackageSource, public_clap_app};
use std::{env, error::Error, io};
use strum::IntoEnumIterator;

pub fn init_main(shell_name: &str) -> Result<(), Box<dyn Error>> {
    let exe_path = env::current_exe()?
        .into_os_string()
        .into_string()
        .expect("Could not convert path to string");

    let (check_str, setup_script, shell) = match shell_name {
        "bash" => (BASH_CHECK, BASH_INIT, clap::Shell::Bash),
        "zsh" => (ZSH_CHECK, ZSH_INIT, clap::Shell::Zsh),
        "fish" => (FISH_CHECK, FISH_INIT, clap::Shell::Fish),
        _ => panic!("Shell \"{}\" is not supported at the moment", shell_name),
    };
    // Get all the different package sources and replace them into the check strings
    let checks: String = PackageSource::iter()
        .filter(|s| s.command() != "")
        .map(|s| check_str.replace("## EMPLACE ##", s.command()))
        .collect::<Vec<String>>()
        .join(" || ");

    let script = setup_script
        .replace("## EMPLACE ##", &format!("\"{}\"", exe_path))
        .replace("## EMPLACE_CHECKS ##", &*checks);
    println!("{}", script);

    // Print the completions
    public_clap_app().gen_completions_to("emplace", shell, &mut io::stdout());

    Ok(())
}

const BASH_INIT: &str = r##"
emplace_postexec_invoke_exec () {
    # quit when the previous command failed
    [ -z "$!" ] && exit $?

    local hist=`history 1`
    # optimization to check quickly for strings
    ## EMPLACE_CHECKS ## || $(exit $?);

    local this_command=`HISTTIMEFORMAT= echo $hist | sed -e "s/^[ ]*[0-9]*[ ]*//"`;
    ## EMPLACE ## catch "$this_command"
}
PROMPT_COMMAND="emplace_postexec_invoke_exec;$PROMPT_COMMAND"
"##;

const BASH_CHECK: &str = "[[ $hist == *\"## EMPLACE ##\"* ]]";

const ZSH_INIT: &str = r##"
emplace_precmd() {
    # quit when the previous command failed
    [ -z "$!" ] && exit

    local hist=`history -1`
    # optimization to check quickly for strings
    ## EMPLACE_CHECKS ## || return;

    local this_command=`HISTTIMEFORMAT= echo $hist | sed -e "s/^[ ]*[0-9]*[ ]*//"`;
    ## EMPLACE ## catch "$this_command"
}
# Don't hook them double in nested shells
if [[ ${precmd_functions[(ie)emplace_precmd]} -gt ${#precmd_functions} ]]; then
    precmd_functions+=(emplace_precmd)
fi
"##;

const ZSH_CHECK: &str = "[[ $hist == *\"## EMPLACE ##\"* ]]";

const FISH_INIT: &str = r##"
function emplace_postcmd --on-event fish_postexec
    # quit when the previous command failed
    if test $status -gt 0
        return
    end

    # optimization disabled

    ## EMPLACE ## catch "$argv"
end
"##;

const FISH_CHECK: &str = "";