git.delta.rocks / jrsonnet / refs/commits / 06bf1b38ec36

1use 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::{apply_tla, error::LocError, throw, ResultExt, State, Val};1011#[cfg(feature = "mimalloc")]12#[global_allocator]13static GLOBAL: mimallocator::Mimalloc = mimallocator::Mimalloc;1415#[derive(Parser)]16enum SubOpts {17	/// Generate completions for specified shell18	Generate {19		/// Target shell name20		shell: Shell,21	},22}2324#[derive(Parser)]25#[clap(next_help_heading = "DEBUG")]26struct DebugOpts {27	/// Required OS stack size.28	/// This shouldn't be changed unless jrsonnet is failing with stack overflow error.29	#[clap(long, name = "size")]30	pub os_stack: Option<usize>,31}3233#[derive(Parser)]34#[clap(next_help_heading = "INPUT")]35struct InputOpts {36	/// Treat input as code, evaluate them instead of reading file37	#[clap(long, short = 'e')]38	pub exec: bool,3940	/// Path to the file to be compiled if `--evaluate` is unset, otherwise code itself41	pub input: Option<String>,42}4344#[derive(Parser)]45#[clap(args_conflicts_with_subcommands = true, disable_version_flag = true)]46struct Opts {47	#[clap(subcommand)]48	sub: Option<SubOpts>,4950	#[clap(flatten)]51	input: InputOpts,52	#[clap(flatten)]53	general: GeneralOpts,5455	#[clap(flatten)]56	trace: TraceOpts,57	#[clap(flatten)]58	manifest: ManifestOpts,59	#[clap(flatten)]60	output: OutputOpts,61	#[clap(flatten)]62	debug: DebugOpts,63}6465fn main() {66	let opts: Opts = Opts::parse();6768	if let Some(sub) = opts.sub {69		match sub {70			SubOpts::Generate { shell } => {71				use clap_complete::generate;72				let app = &mut Opts::command();73				let buf = &mut std::io::stdout();74				generate(shell, app, "jrsonnet", buf);75				std::process::exit(0)76			}77		}78	}7980	let success = if let Some(size) = opts.debug.os_stack {81		std::thread::Builder::new()82			.stack_size(size * 1024 * 1024)83			.spawn(|| main_catch(opts))84			.expect("new thread spawned")85			.join()86			.expect("thread finished successfully")87	} else {88		main_catch(opts)89	};90	if !success {91		std::process::exit(1);92	}93}9495#[derive(thiserror::Error, Debug)]96enum Error {97	// Handled differently98	#[error("evaluation error")]99	Evaluation(LocError),100	#[error("io error")]101	Io(#[from] std::io::Error),102	#[error("input is not utf8 encoded")]103	Utf8(#[from] std::str::Utf8Error),104	#[error("missing input argument")]105	MissingInputArgument,106}107impl From<LocError> for Error {108	fn from(e: LocError) -> Self {109		Self::Evaluation(e)110	}111}112impl From<jrsonnet_evaluator::error::Error> for Error {113	fn from(e: jrsonnet_evaluator::error::Error) -> Self {114		Self::from(LocError::from(e))115	}116}117118fn main_catch(opts: Opts) -> bool {119	let s = State::default();120	let trace = opts121		.trace122		.configure(&s)123		.expect("this configurator doesn't fail");124	if let Err(e) = main_real(&s, opts) {125		if let Error::Evaluation(e) = e {126			let mut out = String::new();127			trace.write_trace(&mut out, &e).expect("format error");128			eprintln!("{out}")129		} else {130			eprintln!("{}", e);131		}132		return false;133	}134	true135}136137fn main_real(s: &State, opts: Opts) -> Result<(), Error> {138	let (_stack_guard, tla, _gc_guard) = opts.general.configure(s)?;139	let manifest_format = opts.manifest.configure(s)?;140141	let input = opts.input.input.ok_or(Error::MissingInputArgument)?;142	let val = if opts.input.exec {143		s.evaluate_snippet("<cmdline>".to_owned(), &input as &str)?144	} else if input == "-" {145		let mut input = Vec::new();146		std::io::stdin().read_to_end(&mut input)?;147		let input_str = std::str::from_utf8(&input)?;148		s.evaluate_snippet("<stdin>".to_owned(), input_str)?149	} else {150		s.import(&input)?151	};152153	let val = apply_tla(s.clone(), &tla, val)?;154155	if let Some(multi) = opts.output.multi {156		if opts.output.create_output_dirs {157			let mut dir = multi.clone();158			dir.pop();159			create_dir_all(dir)?;160		}161		let Val::Obj(obj) = val else {162			throw!("value should be object for --multi manifest, got {}", val.value_type())163		};164		for (field, data) in obj.iter(165			#[cfg(feature = "exp-preserve-order")]166			opts.manifest.preserve_order,167		) {168			let data = data.with_description(|| format!("getting field {field} for manifest"))?;169170			let mut path = multi.clone();171			path.push(&field as &str);172			if opts.output.create_output_dirs {173				let mut dir = path.clone();174				dir.pop();175				create_dir_all(dir)?;176			}177			println!("{}", path.to_str().expect("path"));178			let mut file = File::create(path)?;179			writeln!(180				file,181				"{}",182				data.manifest(&manifest_format)183					.with_description(|| format!("manifesting {field}"))?184			)?;185		}186	} else if let Some(path) = opts.output.output_file {187		if opts.output.create_output_dirs {188			let mut dir = path.clone();189			dir.pop();190			create_dir_all(dir)?;191		}192		let mut file = File::create(path)?;193		writeln!(file, "{}", val.manifest(manifest_format)?)?;194	} else {195		let output = val.manifest(manifest_format)?;196		if !output.is_empty() {197			println!("{}", output);198		}199	}200201	Ok(())202}