diff options
Diffstat (limited to 'src/lang/ts-lang')
| -rw-r--r-- | src/lang/ts-lang/builtin/builtin.ts | 15 | ||||
| -rw-r--r-- | src/lang/ts-lang/core/eval.ts | 3 |
2 files changed, 18 insertions, 0 deletions
diff --git a/src/lang/ts-lang/builtin/builtin.ts b/src/lang/ts-lang/builtin/builtin.ts index a289867..ed6b0a3 100644 --- a/src/lang/ts-lang/builtin/builtin.ts +++ b/src/lang/ts-lang/builtin/builtin.ts @@ -31,3 +31,18 @@ export type BUILTIN_Mul<Args extends readonly any[]> = Args extends [ : Args extends [infer M extends number, infer N extends number] ? Multiply<M, N> : FnError<`Can only multiply [number, number], but got ${ToString<Args>}`>; + +export type BUILTIN_IfElse<Args extends readonly any[]> = Args extends [ + infer A, + infer B, + infer C, + infer D +] + ? FnError<`Invalid args for "if": ${ToString<Args>}`> + : Args extends [infer Cond, infer TrueVal, infer FalseVal] + ? Cond extends true + ? TrueVal + : Cond extends false + ? FalseVal + : FnError<`Condition value ${ToString<Cond>} is not a boolean`> + : FnError<`Invalid args for "if": ${ToString<Args>}`>; diff --git a/src/lang/ts-lang/core/eval.ts b/src/lang/ts-lang/core/eval.ts index 5f22299..b2fecaa 100644 --- a/src/lang/ts-lang/core/eval.ts +++ b/src/lang/ts-lang/core/eval.ts @@ -1,6 +1,7 @@ import { BUILTIN_Add, BUILTIN_Arr, + BUILTIN_IfElse, BUILTIN_Mul, BUILTIN_ToString, SBUILTIN_Call, @@ -26,6 +27,8 @@ export type MapBuiltins< ? BUILTIN_Add<Args> : Node["name"] extends "mul" ? BUILTIN_Mul<Args> + : Node["name"] extends "?" + ? BUILTIN_IfElse<Args> : SENTINEL_NO_BUILTIN : never; |