summaryrefslogtreecommitdiff
path: root/src/lang/ts-lang/builtin
diff options
context:
space:
mode:
Diffstat (limited to 'src/lang/ts-lang/builtin')
-rw-r--r--src/lang/ts-lang/builtin/sbuiltin.ts57
1 files changed, 57 insertions, 0 deletions
diff --git a/src/lang/ts-lang/builtin/sbuiltin.ts b/src/lang/ts-lang/builtin/sbuiltin.ts
index c92fd6a..f089aea 100644
--- a/src/lang/ts-lang/builtin/sbuiltin.ts
+++ b/src/lang/ts-lang/builtin/sbuiltin.ts
@@ -42,6 +42,63 @@ export type SBUILTIN_Map<
GetEvaluatedChildren<Node, Frame, Callstack>
>}`>;
+type Reduce<
+ Arr extends readonly any[],
+ Fn extends FnPrim,
+ Acc,
+ IdxLen extends readonly any[] = readonly []
+> = Arr extends [infer Head, ...infer Tail]
+ ? Reduce<
+ Tail,
+ Fn,
+ CallFn<Fn, [Acc, Head, IdxLen["length"]]>,
+ [...IdxLen, any]
+ >
+ : Acc;
+
+export type SBUILTIN_Reduce<
+ Node extends ASTNode,
+ Frame extends StackFrame,
+ Callstack extends readonly string[]
+> = GetEvaluatedChildren<Node, Frame, Callstack> extends [
+ infer Arr extends readonly any[],
+ infer Fn extends FnPrim,
+ infer Acc
+]
+ ? Reduce<Arr, Fn, Acc>
+ : EvalError<`Invalid params for reduce: ${ToString<
+ GetEvaluatedChildren<Node, Frame, Callstack>
+ >}`>;
+
+type Filter<
+ Arr extends readonly any[],
+ Fn extends FnPrim,
+ Collected extends readonly any[] = [],
+ IdxLen extends readonly any[] = readonly []
+> = Arr extends [infer Head, ...infer Tail]
+ ? Filter<
+ Tail,
+ Fn,
+ CallFn<Fn, [Head, IdxLen["length"]]> extends true
+ ? [...Collected, Head]
+ : Collected,
+ [...IdxLen, any]
+ >
+ : Collected;
+
+export type SBUILTIN_Filter<
+ Node extends ASTNode,
+ Frame extends StackFrame,
+ Callstack extends readonly string[]
+> = GetEvaluatedChildren<Node, Frame, Callstack> extends [
+ infer Arr extends readonly any[],
+ infer Fn extends FnPrim
+]
+ ? Filter<Arr, Fn>
+ : EvalError<`Invalid params for filter: ${ToString<
+ GetEvaluatedChildren<Node, Frame, Callstack>
+ >}`>;
+
export type SBUILTIN_IfElse<
Node extends ASTNode,
Frame extends StackFrame,