difftreelog
fix cli description
in: master
1 file changed
cmds/jrsonnet/src/main.rsdiffbeforeafterboth1use std::{2 fs::{create_dir_all, File},3 io::{Read, Write},4};56use clap::{CommandFactory, Parser};7use clap_complete::Shell;8use jrsonnet_cli::{ConfigureState, GeneralOpts, ManifestOpts, OutputOpts, TraceOpts};9use jrsonnet_evaluator::{10 apply_tla,11 error::{Error as JrError, ErrorKind},12 throw, ResultExt, State, Val,13};1415#[cfg(feature = "mimalloc")]16#[global_allocator]17static GLOBAL: mimallocator::Mimalloc = mimallocator::Mimalloc;1819#[derive(Parser)]20enum SubOpts {21 /// Generate completions for specified shell22 Generate {23 /// Target shell name24 shell: Shell,25 },26}2728#[derive(Parser)]29#[clap(next_help_heading = "DEBUG")]30struct DebugOpts {31 /// Required OS stack size.32 /// This shouldn't be changed unless jrsonnet is failing with stack overflow error.33 #[clap(long, name = "size")]34 pub os_stack: Option<usize>,35}3637#[derive(Parser)]38#[clap(next_help_heading = "INPUT")]39struct InputOpts {40 /// Treat input as code, evaluate them instead of reading file41 #[clap(long, short = 'e')]42 pub exec: bool,4344 /// Path to the file to be compiled if `--evaluate` is unset, otherwise code itself45 pub input: Option<String>,46}4748#[derive(Parser)]49#[clap(args_conflicts_with_subcommands = true, disable_version_flag = true)]50struct Opts {51 #[clap(subcommand)]52 sub: Option<SubOpts>,5354 #[clap(flatten)]55 input: InputOpts,56 #[clap(flatten)]57 general: GeneralOpts,5859 #[clap(flatten)]60 trace: TraceOpts,61 #[clap(flatten)]62 manifest: ManifestOpts,63 #[clap(flatten)]64 output: OutputOpts,65 #[clap(flatten)]66 debug: DebugOpts,67}6869fn main() {70 let opts: Opts = Opts::parse();7172 if let Some(sub) = opts.sub {73 match sub {74 SubOpts::Generate { shell } => {75 use clap_complete::generate;76 let app = &mut Opts::command();77 let buf = &mut std::io::stdout();78 generate(shell, app, "jrsonnet", buf);79 std::process::exit(0)80 }81 }82 }8384 let success = if let Some(size) = opts.debug.os_stack {85 std::thread::Builder::new()86 .stack_size(size * 1024 * 1024)87 .spawn(|| main_catch(opts))88 .expect("new thread spawned")89 .join()90 .expect("thread finished successfully")91 } else {92 main_catch(opts)93 };94 if !success {95 std::process::exit(1);96 }97}9899#[derive(thiserror::Error, Debug)]100enum Error {101 // Handled differently102 #[error("evaluation error")]103 Evaluation(JrError),104 #[error("io error")]105 Io(#[from] std::io::Error),106 #[error("input is not utf8 encoded")]107 Utf8(#[from] std::str::Utf8Error),108 #[error("missing input argument")]109 MissingInputArgument,110}111impl From<JrError> for Error {112 fn from(e: JrError) -> Self {113 Self::Evaluation(e)114 }115}116impl From<ErrorKind> for Error {117 fn from(e: ErrorKind) -> Self {118 Self::from(JrError::from(e))119 }120}121122fn main_catch(opts: Opts) -> bool {123 let s = State::default();124 let trace = opts125 .trace126 .configure(&s)127 .expect("this configurator doesn't fail");128 if let Err(e) = main_real(&s, opts) {129 if let Error::Evaluation(e) = e {130 let mut out = String::new();131 trace.write_trace(&mut out, &e).expect("format error");132 eprintln!("{out}")133 } else {134 eprintln!("{}", e);135 }136 return false;137 }138 true139}140141fn main_real(s: &State, opts: Opts) -> Result<(), Error> {142 let (tla, _gc_guard) = opts.general.configure(s)?;143 let manifest_format = opts.manifest.configure(s)?;144145 let input = opts.input.input.ok_or(Error::MissingInputArgument)?;146 let val = if opts.input.exec {147 s.evaluate_snippet("<cmdline>".to_owned(), &input as &str)?148 } else if input == "-" {149 let mut input = Vec::new();150 std::io::stdin().read_to_end(&mut input)?;151 let input_str = std::str::from_utf8(&input)?;152 s.evaluate_snippet("<stdin>".to_owned(), input_str)?153 } else {154 s.import(&input)?155 };156157 let val = apply_tla(s.clone(), &tla, val)?;158159 if let Some(multi) = opts.output.multi {160 if opts.output.create_output_dirs {161 let mut dir = multi.clone();162 dir.pop();163 create_dir_all(dir)?;164 }165 let Val::Obj(obj) = val else {166 throw!("value should be object for --multi manifest, got {}", val.value_type())167 };168 for (field, data) in obj.iter(169 #[cfg(feature = "exp-preserve-order")]170 opts.manifest.preserve_order,171 ) {172 let data = data.with_description(|| format!("getting field {field} for manifest"))?;173174 let mut path = multi.clone();175 path.push(&field as &str);176 if opts.output.create_output_dirs {177 let mut dir = path.clone();178 dir.pop();179 create_dir_all(dir)?;180 }181 println!("{}", path.to_str().expect("path"));182 let mut file = File::create(path)?;183 writeln!(184 file,185 "{}",186 data.manifest(&manifest_format)187 .with_description(|| format!("manifesting {field}"))?188 )?;189 }190 } else if let Some(path) = opts.output.output_file {191 if opts.output.create_output_dirs {192 let mut dir = path.clone();193 dir.pop();194 create_dir_all(dir)?;195 }196 let mut file = File::create(path)?;197 writeln!(file, "{}", val.manifest(manifest_format)?)?;198 } else {199 let output = val.manifest(manifest_format)?;200 if !output.is_empty() {201 println!("{}", output);202 }203 }204205 Ok(())206}