summaryrefslogtreecommitdiff
path: root/src/lang/ts-lang/builtin/builtin.ts
blob: cdd89918284e9d69a29b68246d892fabe0c332fb (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
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>}`>;