src/main/java/org/zwobble/couscous/interpreter/JavaProject.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
43

package org.zwobble.couscous.interpreter;

import com.google.common.collect.Iterables;
import org.zwobble.couscous.ast.TypeNode;
import org.zwobble.couscous.ast.visitors.NodeTransformer;
import org.zwobble.couscous.interpreter.types.InterpreterType;
import org.zwobble.couscous.interpreter.types.IntrinsicInterpreterType;
import org.zwobble.couscous.interpreter.types.UserDefinedInterpreterType;
import org.zwobble.couscous.interpreter.values.BoxedIntegerInterpreterValue;
import org.zwobble.couscous.interpreter.values.InternalCouscousInterpreterValue;
import org.zwobble.couscous.transforms.DesugarForToWhile;
import org.zwobble.couscous.transforms.DesugarSwitchToIfElse;
import org.zwobble.couscous.transforms.HoistNestedTypes;

import java.util.List;

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

public class JavaProject {
    private static final InterpreterType OBJECT_TYPE = IntrinsicInterpreterType.classBuilder("java.lang.Object")
        .build();
    
    public static MapBackedProject.Builder builder() {
        return MapBackedProject.builder()
            .addClass(InternalCouscousInterpreterValue.TYPE)
            .addClass(OBJECT_TYPE)
            .addClass(BoxedIntegerInterpreterValue.TYPE);
    }

    public static Project of(List<TypeNode> classNodes) {
        NodeTransformer switchTransformer = DesugarSwitchToIfElse.transformer();
        NodeTransformer forTransformer = DesugarForToWhile.transformer();
        Iterable<InterpreterType> concreteTypes = Iterables.transform(
            HoistNestedTypes.hoist(
                NodeTransformer.applyAll(list(switchTransformer, forTransformer), classNodes)
            ),
            UserDefinedInterpreterType::new
        );
        return builder()
                .addClasses(concreteTypes)
                .build();
    }
}