atuin/src/command/init.rs

use eyre::Result;
use structopt::StructOpt;

#[derive(StructOpt)]
pub enum Cmd {
    #[structopt(about = "zsh setup")]
    Zsh,
    #[structopt(about = "bash setup")]
    Bash,
}

fn init_zsh() {
    let full = include_str!("../shell/atuin.zsh");
    println!("{}", full);
}

fn init_bash() {
    let full = include_str!("../shell/atuin.bash");
    println!("{}", full);
}

impl Cmd {
    pub fn run(&self) -> Result<()> {
        match self {
            Self::Zsh => init_zsh(),
            Self::Bash => init_bash(),
        }
        Ok(())
    }
}