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
68
69
70
71
72
73
74
75
76
77
|
import { Type } from 'ts-morph'
import { match } from './util'
import { Ctx } from './types'
import * as builtins from './eval-env/builtins'
type EffDefn = {
default: (ctx: Ctx, args: Type[]) => Record<string, () => Promise<string[]>>,
cleanup: () => void,
}
const mergeEffDefns = (a: EffDefn, b: EffDefn): EffDefn => ({
default: (ctx: Ctx, args: Type[]) => ({
...a.default(ctx, args),
...b.default(ctx, args),
}),
cleanup: () => {
a.cleanup()
b.cleanup()
},
})
const cleanupActions = new Set<() => void>()
export const cleanup = () => cleanupActions.forEach(f => f())
let prevEnv: string
export const evaluateType = async (
ctx: Ctx,
effTyp: Type
): Promise<string[]> => {
const name = effTyp.getSymbol()?.getName()
const args = effTyp.getTypeArguments()
// console.log(ctx.typeToString(effTyp))
// console.log(name, args.map(ctx.typeToString))
const envDefns = await match(ctx.currentEnv, {
node: () => import('./eval-env/node') as Promise<EffDefn>,
'test.node': async () => mergeEffDefns(
await import('./eval-env/test'),
await import('./eval-env/node'),
),
_: async () => {
throw new Error(`Invalid env: ${ctx.currentEnv}`)
},
})
const { default: envEffects, cleanup } = mergeEffDefns(builtins as unknown as EffDefn, envDefns)
// Update cleanup if env has changed
if (prevEnv !== ctx.currentEnv) {
cleanupActions.add(cleanup)
prevEnv = ctx.currentEnv
}
return match(name, {
// TODO: Allow overriding effects
...envEffects(ctx, args),
_: async () => {
if (name && ctx.hasCustomEffect(name)) {
return ctx.runCustomEffect(name, args)
}
console.log(ctx.typeToString(effTyp))
throw new Error(`${name} effect is not handled`)
},
})
}
export const evalList = async (ctx: Ctx, effectTyps: Type[]) => {
const effectResults: string[] = []
for (const item of effectTyps ?? []) {
effectResults.push(...(await evaluateType(ctx, item)))
}
return effectResults
}
|
