private static CallablesEnvironment InitDefaultEnvironment()
{
var environment = new CallablesEnvironment();
AddBuiltins(environment);
AddCode(environment);
return(environment);
}
private static void AddCode(CallablesEnvironment environment)
{
var stdlib = ReadStdLibCode();
var(ast, _) = new Parser().Parse(stdlib);
foreach (FunctionDefinition def in ExtractFunctionDefinitions(ast))
{
environment.SetCallable(def.Name, def.Arguments, def.Body);
}
}
private static void StringFunctions(CallablesEnvironment environment)
{
environment.SetCallable(
new FunctionName("startswith", 1),
StringBuiltins.Function_2(
"startswith",
(str1, str2) => Json.Bool(str1.StartsWith(str2))));
environment.SetCallable(
new FunctionName("endswith", 1),
StringBuiltins.Function_2(
"endswith",
(str1, str2) => Json.Bool(str1.EndsWith(str2))));
environment.SetCallable(
new FunctionName("utf8bytelength", 0),
StringBuiltins.Function_1(
"utf8bytelength",
(str1) => Json.Number(Encoding.UTF8.GetBytes(str1).Length)));
environment.SetCallable(
new FunctionName("explode", 0),
StringBuiltins.Function_1(
"explode",
(str1) => Explode(str1)));
environment.SetCallable(new FunctionName("implode", 0), Implode.Builtin);
environment.SetCallable(
new FunctionName("ltrimstr", 1),
StringBuiltins.Function_2(
"ltrimstr",
(str1, str2) => {
while (str1.StartsWith(str2))
{
str1 = str1.Substring(str2.Length);
}
return(Json.String(str1));
}));
environment.SetCallable(
new FunctionName("rtrimstr", 1),
StringBuiltins.Function_2(
"rtrimstr",
(str1, str2) => {
while (str1.EndsWith(str2))
{
str1 = str1.Substring(0, str1.Length - str2.Length);
}
return(Json.String(str1));
}));
environment.SetCallable(
new FunctionName("fromjson", 0),
StringBuiltins.Function_1(
"fromjson",
(str1) => str1.AsJson()));
}
private static void AddBuiltins(CallablesEnvironment environment)
{
environment.SetCallable(new FunctionName("not", 0), Not.Builtin);
environment.SetCallable(new FunctionName("empty", 0), Empty.Builtin);
environment.SetCallable(new FunctionName("range", 1), Builtins.Range.Builtin_1);
environment.SetCallable(new FunctionName("range", 2), Builtins.Range.Builtin_2);
environment.SetCallable(new FunctionName("range", 3), Builtins.Range.Builtin_3);
environment.SetCallable(new FunctionName("length", 0), Length.Builtin);
environment.SetCallable(new FunctionName("limit", 2), Limit.Builtin);
environment.SetCallable(new FunctionName("keys", 0), Keys.Builtin);
environment.SetCallable(new FunctionName("keys_unsorted", 0), Keys.Builtin);
environment.SetCallable(new FunctionName("debug", 0), Debug.Builtin);
environment.SetCallable(new FunctionName("path", 1), Builtins.Path.Builtin);
environment.SetCallable(new FunctionName("sort", 0), Sort.Builtin_0);
environment.SetCallable(new FunctionName("_sort_by_impl", 1), Sort.Builtin_1);
environment.SetCallable(new FunctionName("has", 1), Has.Builtin);
environment.SetCallable(new FunctionName("getpath", 1), GetPath.Builtin);
environment.SetCallable(new FunctionName("setpath", 2), SetPath.Builtin);
environment.SetCallable(new FunctionName("delpaths", 1), DelPaths.Builtin);
environment.SetCallable(new FunctionName("type", 0), JsonType.Builtin);
environment.SetCallable(new FunctionName("isinfinite", 0), IsInfinite.Builtin);
environment.SetCallable(new FunctionName("isnan", 0), IsNan.Builtin);
environment.SetCallable(new FunctionName("isnormal", 0), IsNormal.Builtin);
environment.SetCallable(new FunctionName("tostring", 0), Tostring.Builtin);
environment.SetCallable(new FunctionName("tonumber", 0), Tonumber.Builtin);
environment.SetCallable(new FunctionName("error", 0), Error.Builtin);
environment.SetCallable(new FunctionName("first", 1), First.Builtin);
environment.SetCallable(new FunctionName("_group_by_impl", 1), GroupBy.Builtin);
environment.SetCallable(new FunctionName("_min_by_impl", 1), MinBy.Builtin);
environment.SetCallable(new FunctionName("_max_by_impl", 1), MaxBy.Builtin);
environment.SetCallable(new FunctionName("_strindices", 1), StrIndices.Builtin);
environment.SetCallable(new FunctionName("contains", 1), Contains.Builtin);
environment.SetCallable(new FunctionName("tojson", 0), ToJson.Builtin);
environment.SetCallable(new FunctionName("fromdateiso8601", 0), DateFunctions.FromDate);
environment.SetCallable(new FunctionName("todateiso8601", 0), DateFunctions.ToDate);
environment.SetCallable(new FunctionName("now", 0), DateFunctions.Now);
environment.SetCallable(new FunctionName("strftime", 1), DateFunctions.Strftime);
environment.SetCallable(new FunctionName("strptime", 1), DateFunctions.Strptime);
MathFunctions(environment);
StringFunctions(environment);
environment.SetCallable(new FunctionName("builtins", 0), new Builtin((mashers, json, context) => {
var builtins = environment
.GetFunctionNames()
.OrderBy(fn => fn.Name)
.Select(fn => Json.String($"{fn.Name}/{fn.Arity}"));
return(Json.Array(builtins).AsEnumerable());
}, 0));
environment.SetCallable(new FunctionName("_match_impl", 3), RegexMatch.Builtin);
}
private static void MathFunctions(CallablesEnvironment environment)
{
void arity1(string name, Func <double, double> function)
{
environment.SetCallable(
new FunctionName(name, 0), MathBuiltins.Function_1(name, function));
}
arity1("floor", Math.Floor);
arity1("sqrt", Math.Sqrt);
arity1("fabs", Math.Abs);
arity1("round", Math.Round);
arity1("ceil", Math.Ceiling);
arity1("trunc", Math.Truncate);
arity1("acos", Math.Acos);
arity1("acosh", Math.Acosh);
arity1("asin", Math.Asin);
arity1("asinh", Math.Asinh);
arity1("atan", Math.Atan);
arity1("atanh", Math.Atanh);
arity1("cbrt", Math.Cbrt);
arity1("cos", Math.Cos);
arity1("cosh", Math.Cosh);
arity1("sin", Math.Sin);
arity1("sinh", Math.Sinh);
arity1("tan", Math.Tan);
arity1("tanh", Math.Tanh);
arity1("exp", Math.Exp);
arity1("exp10", x => Math.Exp(Math.Log(10) * x));
arity1("exp2", x => Math.Exp(Math.Log(2) * x));
arity1("expm1", x => Math.Exp(x) - 1);
arity1("log", Math.Log);
arity1("log10", Math.Log10);
arity1("log2", Math.Log2);
arity1("log1p", x => Math.Log(x + 1));
arity1("pow10", x => Math.Pow(10, x));
void arity1Generator(string name, Func <double> function)
{
environment.SetCallable(
new FunctionName(name, 0),
new Builtin((_, _, _) => Json.Number(function()).AsEnumerable(), 0));
}
arity1Generator("nan", () => double.NaN);
arity1Generator("infinite", () => double.PositiveInfinity);
void arity2(string name, Func <double, double, double> function)
{
environment.SetCallable(
new FunctionName(name, 2), MathBuiltins.Function_2(name, function));
}
arity2("pow", Math.Pow);
arity2("atan2", Math.Atan2);
arity2("copysign", Math.CopySign);
arity2("hypot", (x, y) => Math.Sqrt(x * x + y * y));
arity2("fmin", Math.Min);
arity2("fmax", Math.Max);
arity2("fmod", (x, y) => x % y);
arity2("fdim", (x, y) => x > y ? x - y : 0);
void arity3(string name, Func <double, double, double, double> function)
{
environment.SetCallable(
new FunctionName(name, 3), MathBuiltins.Function_3(name, function));
}
arity3("fma", Math.FusedMultiplyAdd);
}