git.delta.rocks / jrsonnet / refs/commits / 4ea8dc2d6da9

1use jrsonnet_evaluator::{2	create_error, create_error_result, Error, EvaluationState, ImportResolver, LazyBinding,3	LazyVal, ObjMember, ObjValue, Result, Val,4};5use jrsonnet_parser::Visibility;6use libc::{c_char, c_double, c_int, c_uint};7use std::{8	alloc::Layout,9	any::Any,10	cell::RefCell,11	collections::BTreeMap,12	ffi::{CStr, CString},13	fs::File,14	io::Read,15	path::PathBuf,16	rc::Rc,17};1819#[no_mangle]20pub extern "C" fn jsonnet_version() -> &'static [u8; 8] {21	b"v0.16.0\0"22}2324#[derive(Default)]25struct NativeImportResolver {26	library_paths: RefCell<Vec<PathBuf>>,27}28impl NativeImportResolver {29	fn add_jpath(&self, path: PathBuf) {30		self.library_paths.borrow_mut().push(path);31	}32}33impl ImportResolver for NativeImportResolver {34	fn resolve_file(&self, from: &PathBuf, path: &PathBuf) -> Result<Rc<PathBuf>> {35		let mut new_path = from.clone();36		new_path.push(path);37		if new_path.exists() {38			Ok(Rc::new(new_path))39		} else {40			for library_path in self.library_paths.borrow().iter() {41				let mut cloned = library_path.clone();42				cloned.push(path);43				if cloned.exists() {44					return Ok(Rc::new(cloned));45				}46			}47			create_error_result(Error::ImportFileNotFound(from.clone(), path.clone()))48		}49	}50	fn load_file_contents(&self, id: &PathBuf) -> Result<Rc<str>> {51		let mut file =52			File::open(id).map_err(|_e| create_error(Error::ResolvedFileNotFound(id.clone())))?;53		let mut out = String::new();54		file.read_to_string(&mut out)55			.map_err(|_e| create_error(Error::ImportBadFileUtf8(id.clone())))?;56		Ok(out.into())57	}58	unsafe fn as_any(&self) -> &dyn Any {59		self60	}61}6263#[no_mangle]64pub extern "C" fn jsonnet_make() -> Box<EvaluationState> {65	let state = EvaluationState::default();66	state.with_stdlib();67	state.set_import_resolver(Box::new(NativeImportResolver::default()));68	Box::new(state)69}7071#[no_mangle]72pub extern "C" fn jsonnet_max_stack(vm: &EvaluationState, v: c_uint) {73	vm.set_max_stack(v as usize);74}7576// jrsonnet currently have no GC, so these functions is no-op77#[no_mangle]78pub extern "C" fn jsonnet_gc_min_objects(_vm: &EvaluationState, _v: c_uint) {}79#[no_mangle]80pub extern "C" fn jsonnet_gc_growth_trigger(_vm: &EvaluationState, _v: c_double) {}8182// TODO83#[no_mangle]84pub extern "C" fn jsonnet_string_output(_vm: &EvaluationState, _v: c_int) {}8586#[no_mangle]87pub extern "C" fn jsonnet_json_extract_string(_vm: &EvaluationState, v: &Val) -> *mut c_char {88	match v.unwrap_if_lazy().unwrap() {89		Val::Str(s) => CString::new(&*s as &str).unwrap().into_raw(),90		_ => std::ptr::null_mut(),91	}92}93#[no_mangle]94pub extern "C" fn jsonnet_json_extract_number(95	_vm: &EvaluationState,96	v: &Val,97	out: &mut c_double,98) -> c_int {99	match v.unwrap_if_lazy().unwrap() {100		Val::Num(n) => {101			*out = n;102			1103		}104		_ => 0,105	}106}107#[no_mangle]108pub extern "C" fn jsonnet_json_extract_bool(_vm: &EvaluationState, v: &Val) -> c_int {109	match v.unwrap_if_lazy().unwrap() {110		Val::Bool(false) => 0,111		Val::Bool(true) => 1,112		_ => 2,113	}114}115#[no_mangle]116pub extern "C" fn jsonnet_json_extract_null(_vm: &EvaluationState, v: &Val) -> c_int {117	match v.unwrap_if_lazy().unwrap() {118		Val::Null => 1,119		_ => 0,120	}121}122123/// # Safety124///125/// This function is safe, if received v is a pointer to normal C string126#[no_mangle]127pub unsafe extern "C" fn jsonnet_json_make_string(128	_vm: &EvaluationState,129	v: *const c_char,130) -> Box<Val> {131	let cstr = CStr::from_ptr(v);132	let str = cstr.to_str().unwrap();133	Box::new(Val::Str(str.into()))134}135136#[no_mangle]137pub extern "C" fn jsonnet_json_make_number(_vm: &EvaluationState, v: c_double) -> Box<Val> {138	Box::new(Val::Num(v))139}140141#[no_mangle]142pub extern "C" fn jsonnet_json_make_bool(_vm: &EvaluationState, v: c_int) -> Box<Val> {143	assert!(v == 0 || v == 1);144	Box::new(Val::Bool(v == 1))145}146147#[no_mangle]148pub extern "C" fn jsonnet_json_make_null(_vm: &EvaluationState) -> Box<Val> {149	Box::new(Val::Null)150}151152#[no_mangle]153pub extern "C" fn jsonnet_json_make_array(_vm: &EvaluationState) -> Box<Val> {154	Box::new(Val::Arr(Rc::new(Vec::new())))155}156157#[no_mangle]158pub extern "C" fn jsonnet_json_array_append(_vm: &EvaluationState, arr: &mut Val, val: &Val) {159	match arr {160		Val::Arr(old) => {161			// TODO: Mutate array, instead of recreating them162			let mut new = Vec::new();163			new.extend(old.iter().cloned());164			new.push(val.clone());165			*arr = Val::Arr(Rc::new(new));166		}167		_ => panic!("should receive array"),168	}169}170171#[no_mangle]172pub extern "C" fn jsonnet_json_make_object(_vm: &EvaluationState) -> Box<Val> {173	Box::new(Val::Obj(ObjValue::new_empty()))174}175176/// # Safety177///178/// This function is safe if passed name is ok179#[no_mangle]180pub unsafe extern "C" fn jsonnet_json_object_append(181	_vm: &EvaluationState,182	obj: &mut Val,183	name: *const c_char,184	val: &Val,185) {186	match obj {187		Val::Obj(old) => {188			let mut new = BTreeMap::new();189			new.insert(190				CStr::from_ptr(name).to_str().unwrap().into(),191				ObjMember {192					add: false,193					visibility: Visibility::Normal,194					invoke: LazyBinding::Bound(LazyVal::new_resolved(val.clone())),195				},196			);197			let new_obj = ObjValue::new(Some(old.clone()), Rc::new(new));198			*obj = Val::Obj(new_obj);199		}200		_ => panic!("should receive array"),201	}202}203204/// # Safety205///206/// This function is most definitely broken, but it works somehow, see TODO inside207#[no_mangle]208pub unsafe extern "C" fn jsonnet_realloc(209	_vm: &EvaluationState,210	buf: *mut u8,211	sz: usize,212) -> *mut u8 {213	if buf.is_null() {214		assert!(sz != 0);215		return std::alloc::alloc(Layout::from_size_align(sz, std::mem::align_of::<u8>()).unwrap());216	}217	// TODO: Somehow store size of allocation, because its real size is probally not 16 :D218	// OR (Alternative way of fixing this TODO)219	// TODO: Standard allocator uses malloc, and it doesn't uses allocation size,220	// TODO: so it should work in normal cases. Maybe force allocator for this library?221	let old_layout = Layout::from_size_align(16, std::mem::align_of::<u8>()).unwrap();222	if sz == 0 {223		std::alloc::dealloc(buf, old_layout);224		return std::ptr::null_mut();225	}226	std::alloc::realloc(buf, old_layout, sz)227}228229#[no_mangle]230#[allow(clippy::boxed_local)]231pub extern "C" fn jsonnet_json_destroy(_vm: &EvaluationState, _v: Box<Val>) {}232233#[no_mangle]234pub extern "C" fn jsonnet_import_callback() {235	todo!()236}237#[no_mangle]238pub extern "C" fn jsonnet_native_callback() {239	todo!()240}241#[no_mangle]242pub extern "C" fn jsonnet_ext_var() {243	todo!()244}245#[no_mangle]246pub extern "C" fn jsonnet_ext_code() {247	todo!()248}249#[no_mangle]250pub extern "C" fn jsonnet_tla_var() {251	todo!()252}253#[no_mangle]254pub extern "C" fn jsonnet_tla_code() {255	todo!()256}257#[no_mangle]258pub extern "C" fn jsonnet_max_trace() {259	todo!()260}261262/// # Safety263///264/// This function is safe, if received v is a pointer to normal C string265#[no_mangle]266pub unsafe extern "C" fn jsonnet_jpath_add(vm: &EvaluationState, v: *const c_char) {267	let cstr = CStr::from_ptr(v);268	let path = PathBuf::from(cstr.to_str().unwrap());269	let any_resolver = vm.import_resolver();270	let resolver = any_resolver271		.as_any()272		.downcast_ref::<NativeImportResolver>()273		.unwrap();274	resolver.add_jpath(path);275}276277/// # Safety278///279/// This function is safe, if received v is a pointer to normal C string280#[no_mangle]281pub unsafe extern "C" fn jsonnet_evaluate_file(282	vm: &EvaluationState,283	filename: *const c_char,284	error: &mut c_int,285) -> *const c_char {286	vm.run_in_state(|| {287		use std::fmt::Write;288		let filename = CStr::from_ptr(filename);289		match vm.evaluate_file_to_json(&PathBuf::from(filename.to_str().unwrap())) {290			Ok(v) => {291				*error = 0;292				CString::new(&*v as &str).unwrap().into_raw()293			}294			Err(e) => {295				*error = 1;296				let mut out = String::new();297				writeln!(out, "{:?}", e.0).unwrap();298				for i in (e.1).0.iter() {299					writeln!(out, "{:?}", i).unwrap();300				}301				CString::new(&out as &str).unwrap().into_raw()302			}303		}304	})305}306#[no_mangle]307pub extern "C" fn jsonnet_evaluate_snippet() {308	todo!()309}310#[no_mangle]311pub extern "C" fn jsonnet_evaluate_file_multi() {312	todo!()313}314#[no_mangle]315pub extern "C" fn jsonnet_evaluate_snippet_multi() {316	todo!()317}318#[no_mangle]319pub extern "C" fn jsonnet_evaluate_file_stream() {320	todo!()321}322#[no_mangle]323pub extern "C" fn jsonnet_evaluate_snippet_stream() {324	todo!()325}326327#[no_mangle]328#[allow(clippy::boxed_local)]329pub extern "C" fn jsonnet_destroy(_vm: Box<EvaluationState>) {}