diff options
Diffstat (limited to 'src/lang/js-lang/builtin/sbuiltin.ts')
| -rw-r--r-- | src/lang/js-lang/builtin/sbuiltin.ts | 19 |
1 files changed, 18 insertions, 1 deletions
diff --git a/src/lang/js-lang/builtin/sbuiltin.ts b/src/lang/js-lang/builtin/sbuiltin.ts index e5bc6ab..663c2f9 100644 --- a/src/lang/js-lang/builtin/sbuiltin.ts +++ b/src/lang/js-lang/builtin/sbuiltin.ts @@ -1,4 +1,4 @@ -import { callFn, getEvaluatedChildren } from "../core/eval"; +import { _evaluate, callFn, getEvaluatedChildren } from "../core/eval"; import { ASTNode, FnPrim, StackFrame } from "../../ts-lang"; type SBUILTIN = (node: ASTNode, frame: StackFrame) => any; @@ -40,7 +40,24 @@ export const V_SBUILTIN_Map: SBUILTIN = (node, frame) => { return values.map((v, i) => callFn(fn, [v, i], frame)); }; +export const V_SBUILTIN_IfElse: SBUILTIN = (node, frame) => { + const children = node.children; + + if (children.length !== 3) { + throw new Error(`Invalid args for "if": ${JSON.stringify(children)}`); + } + + const cond = _evaluate(children[0], frame); + + if (typeof cond !== "boolean") { + throw new Error(`Condition value ${JSON.stringify(cond)} is not a boolean`); + } + + return cond ? _evaluate(children[1], frame) : _evaluate(children[2], frame); +}; + export const nameToSBUILTIN: Record<string, SBUILTIN> = { call: V_SBUILTIN_Call, map: V_SBUILTIN_Map, + "?": V_SBUILTIN_IfElse, }; |
