src/test/java/org/zwobble/couscous/tests/MethodRunner.java


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

package org.zwobble.couscous.tests;

import com.google.common.collect.ImmutableList;
import com.google.common.collect.Iterables;
import org.zwobble.couscous.ast.*;
import org.zwobble.couscous.values.PrimitiveValue;

import java.util.List;

import static org.zwobble.couscous.util.ExtraLists.list;

public interface MethodRunner {
    PrimitiveValue runMethod(
        List<ClassNode> classNodes,
        TypeName className,
        String methodName,
        List<PrimitiveValue> arguments, TypeName returnType);


    default PrimitiveValue runMethod(List<ClassNode> classNodes, TypeName className, String methodName, List<PrimitiveValue> arguments) {
        ClassNode classNode = Iterables.find(classNodes, c -> c.getName().equals(className));
        MethodNode method = Iterables.find(classNode.getMethods(), m -> m.getName().equals(methodName));
        return runMethod(classNodes, className, methodName, arguments, method.getReturnType());
    }

    default PrimitiveValue evalExpression(List<ClassNode> classes, ExpressionNode expression) {
        ClassNode programNode = ClassNode.builder("Program")
            .method(MethodNode.staticMethod("run")
                .returns(expression.getType())
                .statement(ReturnNode.returns(expression))
                .build())
            .build();
        return runMethod(
            ImmutableList.copyOf(Iterables.concat(
                classes,
                list(programNode))),
            programNode.getName(),
            "run",
            list(),
            expression.getType());
    }
}