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

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.ast.types.ScalarType;
import org.zwobble.couscous.ast.types.Type;
import org.zwobble.couscous.values.PrimitiveValue;

import java.util.List;

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

public interface MethodRunner {
    PrimitiveValue runMethod(
        List<TypeNode> classNodes,
        ScalarType className,
        String methodName,
        List<PrimitiveValue> arguments,
        Type returnType);

    default PrimitiveValue evalExpression(List<TypeNode> 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());
    }
}