`a7408d3bc96e`

source

crates/jrsonnet-cli/src/lib.rs3.6 KiBsource history
1mod ext;2mod manifest;3mod tla;4mod trace;56pub use ext::*;7pub use manifest::*;8pub use tla::*;9pub use trace::*;1011use clap::Clap;12use jrsonnet_evaluator::{error::Result, EvaluationState, FileImportResolver};13use std::{env, path::PathBuf};1415pub trait ConfigureState {16	fn configure(&self, state: &EvaluationState) -> Result<()>;17}1819#[derive(Clap)]20#[clap(help_heading = "INPUT")]21pub struct InputOpts {22	#[clap(23		long,24		short = 'e',25		about = "Treat input as code, evaluate them instead of reading file"26	)]27	pub exec: bool,2829	#[clap(30		about = "Path to the file to be compiled if `--evaluate` is unset, otherwise code itself"31	)]32	pub input: String,33}3435#[derive(Clap)]36#[clap(help_heading = "OPTIONS")]37pub struct MiscOpts {38	/// Disable standard library.39	/// By default standard library will be available via global `std` variable.40	/// Note that standard library will still be loaded41	/// if chosen manifestification method is not `none`.42	#[clap(long)]43	no_stdlib: bool,4445	/// Maximal allowed number of stack frames,46	/// stack overflow error will be raised if this number gets exceeded.47	#[clap(long, short = 's', default_value = "200")]48	max_stack: usize,4950	/// Library search dirs. (right-most wins)51	/// Any not found `imported` file will be searched in these.52	/// This can also be specified via `JSONNET_PATH` variable,53	/// which should contain a colon-separated (semicolon-separated on Windows) list of directories.54	#[clap(long, short = 'J', multiple_occurrences = true)]55	jpath: Vec<PathBuf>,56}57impl ConfigureState for MiscOpts {58	fn configure(&self, state: &EvaluationState) -> Result<()> {59		if !self.no_stdlib {60			state.with_stdlib();61		}6263		let mut library_paths = self.jpath.clone();64		library_paths.reverse();65		if let Some(path) = env::var_os("JSONNET_PATH") {66			library_paths.extend(env::split_paths(path.as_os_str()));67		}6869		state.set_import_resolver(Box::new(FileImportResolver { library_paths }));7071		state.set_max_stack(self.max_stack);72		Ok(())73	}74}7576/// General configuration of jsonnet77#[derive(Clap)]78#[clap(name = "jrsonnet", version, author)]79pub struct GeneralOpts {80	#[clap(flatten)]81	misc: MiscOpts,8283	#[clap(flatten)]84	tla: TLAOpts,85	#[clap(flatten)]86	ext: ExtVarOpts,8788	#[clap(flatten)]89	trace: TraceOpts,90}9192impl ConfigureState for GeneralOpts {93	fn configure(&self, state: &EvaluationState) -> Result<()> {94		// Configure trace first, because tla-code/ext-code can throw95		self.trace.configure(state)?;96		self.misc.configure(state)?;97		self.tla.configure(state)?;98		self.ext.configure(state)?;99		Ok(())100	}101}102103#[derive(Clap)]104#[clap(help_heading = "GARBAGE COLLECTION")]105pub struct GcOpts {106	/// Do not skip gc on exit107	#[clap(long)]108	gc_collect_on_exit: bool,109	/// Print gc stats before exit110	#[clap(long)]111	gc_print_stats: bool,112	/// Force garbage collection before printing stats113	/// Useful for checking for memory leaks114	/// Does nothing useless --gc-print-stats is specified115	#[clap(long)]116	gc_collect_before_printing_stats: bool,117}118impl GcOpts {119	pub fn configure_global(&self) {120		if !self.gc_collect_on_exit {121			gcmodule::set_thread_collect_on_drop(false)122		}123	}124	pub fn stats_printer(&self) -> Option<GcStatsPrinter> {125		self.gc_print_stats.then(|| GcStatsPrinter {126			collect_before_printing_stats: self.gc_collect_before_printing_stats,127		})128	}129}130131pub struct GcStatsPrinter {132	collect_before_printing_stats: bool,133}134impl Drop for GcStatsPrinter {135	fn drop(&mut self) {136		eprintln!("=== GC STATS ===");137		if self.collect_before_printing_stats {138			let collected = gcmodule::collect_thread_cycles();139			eprintln!("Collected: {}", collected);140		}141		eprintln!("Tracked: {}", gcmodule::count_thread_tracked())142	}143}