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/// Jsonnet commandline interpreter (Rust implementation)49#[derive(Parser)]50#[clap(51 args_conflicts_with_subcommands = true,52 disable_version_flag = true,53 version,54 author55)]56struct Opts {57 #[clap(subcommand)]58 sub: Option<SubOpts>,5960 #[clap(flatten)]61 input: InputOpts,62 #[clap(flatten)]63 general: GeneralOpts,6465 #[clap(flatten)]66 trace: TraceOpts,67 #[clap(flatten)]68 manifest: ManifestOpts,69 #[clap(flatten)]70 output: OutputOpts,71 #[clap(flatten)]72 debug: DebugOpts,73}7475fn main() {76 let opts: Opts = Opts::parse();7778 if let Some(sub) = opts.sub {79 match sub {80 SubOpts::Generate { shell } => {81 use clap_complete::generate;82 let app = &mut Opts::command();83 let buf = &mut std::io::stdout();84 generate(shell, app, "jrsonnet", buf);85 std::process::exit(0)86 }87 }88 }8990 let success = if let Some(size) = opts.debug.os_stack {91 std::thread::Builder::new()92 .stack_size(size * 1024 * 1024)93 .spawn(|| main_catch(opts))94 .expect("new thread spawned")95 .join()96 .expect("thread finished successfully")97 } else {98 main_catch(opts)99 };100 if !success {101 std::process::exit(1);102 }103}104105#[derive(thiserror::Error, Debug)]106enum Error {107 // Handled differently108 #[error("evaluation error")]109 Evaluation(JrError),110 #[error("io error")]111 Io(#[from] std::io::Error),112 #[error("input is not utf8 encoded")]113 Utf8(#[from] std::str::Utf8Error),114 #[error("missing input argument")]115 MissingInputArgument,116}117impl From<JrError> for Error {118 fn from(e: JrError) -> Self {119 Self::Evaluation(e)120 }121}122impl From<ErrorKind> for Error {123 fn from(e: ErrorKind) -> Self {124 Self::from(JrError::from(e))125 }126}127128fn main_catch(opts: Opts) -> bool {129 let s = State::default();130 let trace = opts131 .trace132 .configure(&s)133 .expect("this configurator doesn't fail");134 if let Err(e) = main_real(&s, opts) {135 if let Error::Evaluation(e) = e {136 let mut out = String::new();137 trace.write_trace(&mut out, &e).expect("format error");138 eprintln!("{out}")139 } else {140 eprintln!("{}", e);141 }142 return false;143 }144 true145}146147fn main_real(s: &State, opts: Opts) -> Result<(), Error> {148 let (tla, _gc_guard) = opts.general.configure(s)?;149 let manifest_format = opts.manifest.configure(s)?;150151 let input = opts.input.input.ok_or(Error::MissingInputArgument)?;152 let val = if opts.input.exec {153 s.evaluate_snippet("<cmdline>".to_owned(), &input as &str)?154 } else if input == "-" {155 let mut input = Vec::new();156 std::io::stdin().read_to_end(&mut input)?;157 let input_str = std::str::from_utf8(&input)?;158 s.evaluate_snippet("<stdin>".to_owned(), input_str)?159 } else {160 s.import(&input)?161 };162163 let val = apply_tla(s.clone(), &tla, val)?;164165 if let Some(multi) = opts.output.multi {166 if opts.output.create_output_dirs {167 let mut dir = multi.clone();168 dir.pop();169 create_dir_all(dir)?;170 }171 let Val::Obj(obj) = val else {172 throw!("value should be object for --multi manifest, got {}", val.value_type())173 };174 for (field, data) in obj.iter(175 #[cfg(feature = "exp-preserve-order")]176 opts.manifest.preserve_order,177 ) {178 let data = data.with_description(|| format!("getting field {field} for manifest"))?;179180 let mut path = multi.clone();181 path.push(&field as &str);182 if opts.output.create_output_dirs {183 let mut dir = path.clone();184 dir.pop();185 create_dir_all(dir)?;186 }187 println!("{}", path.to_str().expect("path"));188 let mut file = File::create(path)?;189 writeln!(190 file,191 "{}",192 data.manifest(&manifest_format)193 .with_description(|| format!("manifesting {field}"))?194 )?;195 }196 } else if let Some(path) = opts.output.output_file {197 if opts.output.create_output_dirs {198 let mut dir = path.clone();199 dir.pop();200 create_dir_all(dir)?;201 }202 let mut file = File::create(path)?;203 writeln!(file, "{}", val.manifest(manifest_format)?)?;204 } else {205 let output = val.manifest(manifest_format)?;206 if !output.is_empty() {207 println!("{}", output);208 }209 }210211 Ok(())212}