diff --git a/src/main.rs b/src/main.rs index 612e130ec5..22f373a2c6 100644 --- a/src/main.rs +++ b/src/main.rs @@ -36,8 +36,14 @@ pub fn main() { let config_options = Options::from_cli(&config.options, opts.command.clone()); if let Some(Command::Setup(ref setup)) = opts.command { - Setup::from_cli(setup, &opts, &config_options).expect("Failed to print to stdout"); - } + Setup::from_cli(setup, &opts, &config_options).map_or_else( + |e| { + eprintln!("{:?}", e); + process::exit(1); + }, + |_| {}, + ); + }; atomic_create_dir(&*ZELLIJ_TMP_DIR).unwrap(); atomic_create_dir(&*ZELLIJ_TMP_LOG_DIR).unwrap(); diff --git a/zellij-utils/src/setup.rs b/zellij-utils/src/setup.rs index 61c330cdb1..aa0bb642b4 100644 --- a/zellij-utils/src/setup.rs +++ b/zellij-utils/src/setup.rs @@ -96,13 +96,25 @@ pub const STRIDER_LAYOUT: &[u8] = include_bytes!(concat!( pub const NO_STATUS_LAYOUT: &[u8] = include_bytes!(concat!( env!("CARGO_MANIFEST_DIR"), "/", - "assets/layouts/default.yaml" + "assets/layouts/disable-status-bar.yaml" )); pub fn dump_default_config() -> std::io::Result<()> { dump_asset(DEFAULT_CONFIG) } +pub fn dump_specified_layout(layout: &str) -> std::io::Result<()> { + match layout { + "strider" => dump_asset(STRIDER_LAYOUT), + "default" => dump_asset(DEFAULT_LAYOUT), + "disable-status" => dump_asset(NO_STATUS_LAYOUT), + not_found => Err(std::io::Error::new( + std::io::ErrorKind::Other, + format!("Layout: {} not found", not_found), + )), + } +} + #[derive(Debug, Default, Clone, StructOpt, Serialize, Deserialize)] pub struct Setup { /// Dump the default configuration file to stdout @@ -117,6 +129,9 @@ pub struct Setup { #[structopt(long)] pub check: bool, + /// Dump the specified layout file to stdout + #[structopt(long)] + pub dump_layout: Option, /// Generates completion for the specified shell #[structopt(long)] pub generate_completion: Option, @@ -144,6 +159,11 @@ impl Setup { std::process::exit(0); } + if let Some(layout) = &self.dump_layout { + dump_specified_layout(layout)?; + std::process::exit(0); + } + Ok(()) }