git.delta.rocks / jrsonnet / refs/commits / dcb0efaafd9f

1use clap::Clap;2use jrsonnet_cli::{ConfigureState, GeneralOpts, InputOpts, ManifestOpts, OutputOpts};3use jrsonnet_evaluator::{error::LocError, EvaluationState};4use std::{5	fs::{create_dir_all, File},6	io::Read,7	io::Write,8	path::PathBuf,9	rc::Rc,10};1112#[cfg(feature = "mimalloc")]13#[global_allocator]14static GLOBAL: mimallocator::Mimalloc = mimallocator::Mimalloc;1516#[derive(Clap)]17// #[clap(help_heading = "DEBUG")]18struct DebugOpts {19	/// Required OS stack size.20	/// This shouldn't be changed unless jrsonnet is failing with stack overflow error.21	#[clap(long, name = "size")]22	pub os_stack: Option<usize>,23}2425#[derive(Clap)]26struct Opts {27	#[clap(flatten)]28	input: InputOpts,29	#[clap(flatten)]30	general: GeneralOpts,31	#[clap(flatten)]32	manifest: ManifestOpts,33	#[clap(flatten)]34	output: OutputOpts,35	#[clap(flatten)]36	debug: DebugOpts,37}3839fn main() {40	let opts: Opts = Opts::parse();41	if let Some(size) = opts.debug.os_stack {42		std::thread::Builder::new()43			.stack_size(size * 1024 * 1024)44			.spawn(|| main_catch(opts))45			.expect("new thread spawned")46			.join()47			.expect("thread finished successfully");48	} else {49		main_catch(opts)50	}51}5253#[derive(thiserror::Error, Debug)]54enum Error {55	// Handled differently56	#[error("evaluation error")]57	Evaluation(jrsonnet_evaluator::error::LocError),58	#[error("io error")]59	Io(#[from] std::io::Error),60	#[error("input is not utf8 encoded")]61	Utf8(#[from] std::str::Utf8Error),62}63impl From<LocError> for Error {64	fn from(e: LocError) -> Self {65		Self::Evaluation(e)66	}67}6869fn main_catch(opts: Opts) {70	let state = EvaluationState::default();71	if let Err(e) = main_real(&state, opts) {72		if let Error::Evaluation(e) = e {73			println!("{}", state.stringify_err(&e));74		} else {75			println!("{}", e);76		}77	}78}7980fn main_real(state: &EvaluationState, opts: Opts) -> Result<(), Error> {81	opts.general.configure(&state)?;82	opts.manifest.configure(&state)?;8384	let val = if opts.input.exec {85		state.evaluate_snippet_raw(86			Rc::new(PathBuf::from("args")),87			(&opts.input.input as &str).into(),88		)?89	} else if opts.input.input == "-" {90		let mut input = Vec::new();91		std::io::stdin().read_to_end(&mut input)?;92		let input_str = std::str::from_utf8(&input)?.into();93		state.evaluate_snippet_raw(Rc::new(PathBuf::from("<stdin>")), input_str)?94	} else {95		state.evaluate_file_raw(&PathBuf::from(opts.input.input))?96	};9798	let val = state.with_tla(val)?;99100	if let Some(multi) = opts.output.multi {101		if opts.output.create_output_dirs {102			let mut dir = multi.clone();103			dir.pop();104			create_dir_all(dir)?;105		}106		for (file, data) in state.manifest_multi(val)?.iter() {107			let mut path = multi.clone();108			path.push(&file as &str);109			if opts.output.create_output_dirs {110				let mut dir = path.clone();111				dir.pop();112				create_dir_all(dir)?;113			}114			println!("{}", path.to_str().expect("path"));115			let mut file = File::create(path)?;116			writeln!(file, "{}", data)?;117		}118	} else if let Some(path) = opts.output.output_file {119		if opts.output.create_output_dirs {120			let mut dir = path.clone();121			dir.pop();122			create_dir_all(dir)?;123		}124		let mut file = File::create(path)?;125		writeln!(file, "{}", state.manifest(val)?)?;126	} else {127		println!("{}", state.manifest(val)?);128	}129130	Ok(())131}