summaryrefslogtreecommitdiff
path: root/src/lang/ts-lang/builtin/builtin.ts
blob: 853207273a481a4f2dbb7288ee8aa3d062165240 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
import { FnError } from ".";
import {
  AddNumbers,
  AddStrings,
  ArrayEqual,
  Multiply,
  SubNumbers,
  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_Sub<Args extends readonly any[]> = Args extends [
  infer A,
  infer B,
  infer C
]
  ? FnError<`Can only sub [number, number], but got ${ToString<Args>}`>
  : Args extends [infer M extends number, infer N extends number]
  ? SubNumbers<M, N>
  : FnError<`Can only sub [number, number], but got ${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 type BUILTIN_Eq<Args extends readonly any[]> = Args extends
  | readonly number[]
  | readonly string[]
  | readonly boolean[]
  ? ArrayEqual<Args>
  : FnError<`Can only check equality of numbers or string or boolean, 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>}`>;