diff options
Diffstat (limited to 'src/lang/builtin')
| -rw-r--r-- | src/lang/builtin/builtin.ts | 33 | ||||
| -rw-r--r-- | src/lang/builtin/index.ts | 35 | ||||
| -rw-r--r-- | src/lang/builtin/sbuiltin.ts | 33 |
3 files changed, 68 insertions, 33 deletions
diff --git a/src/lang/builtin/builtin.ts b/src/lang/builtin/builtin.ts new file mode 100644 index 0000000..a289867 --- /dev/null +++ b/src/lang/builtin/builtin.ts @@ -0,0 +1,33 @@ +import { FnError } from "."; +import { + AddNumbers, + AddStrings, + Multiply, + ToString, + UnarrayIfOnlyHead, +} from "../util"; + +export type BUILTIN_Arr<Args extends readonly any[]> = Args; + +export type BUILTIN_ToString<Args extends readonly any[]> = ToString< + UnarrayIfOnlyHead<{ + [Idx in keyof Args]: ToString<Args[Idx]>; + }> +>; + +export type BUILTIN_Add<Args extends readonly any[]> = + Args extends readonly string[] + ? AddStrings<Args> + : Args extends readonly number[] + ? AddNumbers<Args> + : FnError<`Cannot add operands ${ToString<Args>}`>; + +export type BUILTIN_Mul<Args extends readonly any[]> = Args extends [ + infer A, + infer B, + infer C +] + ? FnError<`Can only multiply [number, number], but got ${ToString<Args>}`> + : Args extends [infer M extends number, infer N extends number] + ? Multiply<M, N> + : FnError<`Can only multiply [number, number], but got ${ToString<Args>}`>; diff --git a/src/lang/builtin/index.ts b/src/lang/builtin/index.ts index 373f54a..de2bee3 100644 --- a/src/lang/builtin/index.ts +++ b/src/lang/builtin/index.ts @@ -1,34 +1,3 @@ -import { - AddNumbers, - AddStrings, - Multiply, - ToString, - UnarrayIfOnlyHead, -} from "../util"; - export type FnError<T extends string> = `Function execution error: ${T}`; - -export type BUILTIN_Arr<Args extends readonly any[]> = Args; - -export type BUILTIN_ToString<Args extends readonly any[]> = ToString< - UnarrayIfOnlyHead<{ - [Idx in keyof Args]: ToString<Args[Idx]>; - }> ->; - -export type BUILTIN_Add<Args extends readonly any[]> = - Args extends readonly string[] - ? AddStrings<Args> - : Args extends readonly number[] - ? AddNumbers<Args> - : FnError<`Cannot add operands ${ToString<Args>}`>; - -export type BUILTIN_Mul<Args extends readonly any[]> = Args extends [ - infer A, - infer B, - infer C -] - ? FnError<`Can only multiply [number, number], but got ${ToString<Args>}`> - : Args extends [infer M extends number, infer N extends number] - ? Multiply<M, N> - : FnError<`Can only multiply [number, number], but got ${ToString<Args>}`>; +export * from "./builtin"; +export * from "./sbuiltin"; diff --git a/src/lang/builtin/sbuiltin.ts b/src/lang/builtin/sbuiltin.ts new file mode 100644 index 0000000..704e688 --- /dev/null +++ b/src/lang/builtin/sbuiltin.ts @@ -0,0 +1,33 @@ +import { ASTNode } from "../core/common"; +import { + CallFn, + FnPrim, + GetEvaluatedChildren, + StackFrame, + EvalError, +} from "../core/eval"; +import { ToString } from "../util"; + +export type SBUILTIN_Call< + Node extends ASTNode, + Frame extends StackFrame +> = GetEvaluatedChildren<Node, Frame> extends [ + infer Fn extends FnPrim, + ...infer Values extends readonly any[] +] + ? CallFn<Fn, Values, Frame> + : EvalError<`Invalid params for function call: ${ToString< + GetEvaluatedChildren<Node, Frame> + >}`>; + +export type SBUILTIN_Map< + Node extends ASTNode, + Frame extends StackFrame +> = GetEvaluatedChildren<Node, Frame> extends [ + infer Arr extends readonly any[], + infer Fn extends FnPrim +] + ? { [Idx in keyof Arr]: CallFn<Fn, [Arr[Idx]], Frame> } + : EvalError<`Invalid params for map: ${ToString< + GetEvaluatedChildren<Node, Frame> + >}`>; |
