git.delta.rocks / jrsonnet / refs/commits / 1979dbda1bc0

1use std::any::Any;23use jrsonnet_gcmodule::{Trace, TraceBox, cc_dyn};4use jrsonnet_ir::function::{FunctionSignature, ParamDefault, ParamName, ParamParse};56use super::CallLocation;7use crate::{Result, Thunk, Val};89#[macro_export]10macro_rules! params {11	(@name unnamed) => { ParamName::Unnamed };12	(@name named $name:literal) => { ParamName::Named($crate::IStr::from($name)) };13	($($(#[$meta:meta])* [$kind:ident $(($lit:literal))? => $default:expr]),* $(,)?) => {14		thread_local! {15			static PARAMS: FunctionSignature = FunctionSignature::new([16				$($(#[$meta])* ParamParse::new(params!(@name $kind $($lit)?), $default)),*17			].into());18		}19	};20}2122cc_dyn!(23	#[derive(Clone)]24	BuiltinFunc,25	Builtin,26	pub(crate) fn new() {...}27);28impl Builtin for BuiltinFunc {29	fn name(&self) -> &str {30		self.0.name()31	}3233	fn params(&self) -> FunctionSignature {34		self.0.params()35	}3637	fn call(&self, loc: CallLocation<'_>, args: &[Option<Thunk<Val>>]) -> Result<Val> {38		self.0.call(loc, args)39	}4041	fn as_any(&self) -> &dyn Any {42		self.0.as_any()43	}44}4546/// Description of function defined by native code47///48/// Prefer to use #[builtin] macro, instead of manual implementation of this trait49pub trait Builtin: Trace {50	/// Function name to be used in stack traces51	fn name(&self) -> &str;52	/// Parameter names for named calls53	fn params(&self) -> FunctionSignature;54	/// Call the builtin55	fn call(&self, loc: CallLocation<'_>, args: &[Option<Thunk<Val>>]) -> Result<Val>;5657	fn as_any(&self) -> &dyn Any;58}5960#[derive(Trace)]61pub struct NativeCallback {62	pub(crate) params: FunctionSignature,63	handler: TraceBox<dyn NativeCallbackHandler>,64}65impl NativeCallback {66	#[deprecated = "prefer using builtins directly, use this interface only for bindings"]67	pub fn new(params: Vec<String>, handler: impl NativeCallbackHandler) -> Self {68		Self {69			params: FunctionSignature::new(70				params71					.into_iter()72					.map(|n| ParamParse::new(ParamName::Named(n.into()), ParamDefault::None))73					.collect(),74			),75			handler: TraceBox(Box::new(handler)),76		}77	}78}7980impl Builtin for NativeCallback {81	fn name(&self) -> &'static str {82		// TODO: standard natives gets their names from definition83		// But builitins should already have them84		"<native>"85	}8687	fn params(&self) -> FunctionSignature {88		self.params.clone()89	}9091	fn call(&self, _loc: CallLocation<'_>, args: &[Option<Thunk<Val>>]) -> Result<Val> {92		let args = args93			.iter()94			.map(|a| a.as_ref().expect("legacy natives have no default params"))95			.map(Thunk::evaluate)96			.collect::<Result<Vec<Val>>>()?;97		self.handler.call(&args)98	}99100	fn as_any(&self) -> &dyn Any {101		self102	}103}104105pub trait NativeCallbackHandler: Trace {106	fn call(&self, args: &[Val]) -> Result<Val>;107}