aboutsummaryrefslogtreecommitdiff
path: root/src/main/java/org/zwobble/couscous/backends/csharp/CsharpCodeGenerator.java
blob: 293673532e7d91aa22024c082696a6f47237044f (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
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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
package org.zwobble.couscous.backends.csharp;

import com.google.common.collect.ImmutableMap;
import org.zwobble.couscous.ast.TypeNode;
import org.zwobble.couscous.ast.visitors.NodeTransformer;
import org.zwobble.couscous.backends.csharp.primitives.CsharpPrimitiveMethods;
import org.zwobble.couscous.backends.naming.Naming;
import org.zwobble.couscous.transforms.*;
import org.zwobble.couscous.types.*;

import java.util.List;
import java.util.Map;

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

public class CsharpCodeGenerator {
    public static final Naming NAMING = Naming.noMangling();

    private final static Map<ScalarType, ScalarType> PRIMITIVES = ImmutableMap.<ScalarType, ScalarType>builder()
        .put(Types.INT, ScalarType.topLevel("int"))
        .put(Types.BOXED_INT, ScalarType.topLevel("int"))
        .put(Types.STRING, ScalarType.topLevel("string"))
        .put(Types.CHAR, ScalarType.topLevel("char"))
        .put(Types.BOOLEAN, ScalarType.topLevel("bool"))
        .put(Types.BOXED_BOOLEAN, ScalarType.topLevel("bool"))
        .put(Types.CLASS, ScalarType.topLevel("System.Type"))
        .put(Types.VOID, ScalarType.topLevel("void"))
        .put(Types.ARRAY, Types.ARRAY)
        .build();

    private final String namespace;
    private final NodeTransformer nodeTransformer;

    public static List<TypeNode> generateCode(List<TypeNode> types, String namespace) {
        return new CsharpCodeGenerator(NAMING, namespace).generateCode(types);
    }

    private CsharpCodeGenerator(Naming naming, String namespace) {
        this.namespace = namespace;
        nodeTransformer = NodeTransformer.builder()
            .transformType(this::transformType)
            .transformMethodName(naming::methodName)
            .transformFieldName(name -> "_" + name)
            .build();
    }

    private List<TypeNode> generateCode(List<TypeNode> types) {
        return SplitStaticsFromInterface.transform(
            NodeTransformer.applyAll(
                list(
                    DesugarForEachToFor.transformer(),
                    DesugarForToWhile.transformer(),
                    CsharpPrimitiveMethods.TRANSFORMER,
                    nodeTransformer
                ),
                HoistNestedTypes.hoist(eagerMap(types, AnonymousClassToInnerClass::transform))
            )
        );
    }

    private Type transformType(Type type) {
        return type.accept(new Type.Visitor<Type>() {
            @Override
            public Type visit(ScalarType type) {
                if (PRIMITIVES.containsKey(type)) {
                    return PRIMITIVES.get(type);
                } else {
                    return addPrefix(type, namespace);
                }
            }

            @Override
            public TypeParameter visit(TypeParameter parameter) {
                return parameter;
            }

            @Override
            public Type visit(ParameterizedType type) {
                if (type.getRawType().equals(Types.CLASS)) {
                    return ScalarType.topLevel("System.Type");
                } else {
                    return new ParameterizedType(
                        (ScalarType) transformType(type.getRawType()),
                        eagerMap(type.getParameters(), parameter -> transformType(parameter)));
                }
            }

            @Override
            public Type visit(BoundTypeParameter type) {
                return new BoundTypeParameter(
                    visit(type.getParameter()),
                    transformType(type.getValue()));
            }
        });
    }

    private Type addPrefix(Type type, String namespace) {
        return type.accept(new Type.Visitor<Type>() {
            @Override
            public ScalarType visit(ScalarType type) {
                return ScalarType.topLevel(namespace + "." + type.getQualifiedName());
            }

            @Override
            public TypeParameter visit(TypeParameter parameter) {
                return parameter;
            }

            @Override
            public Type visit(ParameterizedType type) {
                return new ParameterizedType(
                    visit(type.getRawType()),
                    eagerMap(type.getParameters(), parameter -> addPrefix(parameter, namespace)));
            }

            @Override
            public Type visit(BoundTypeParameter type) {
                return new BoundTypeParameter(
                    visit(type.getParameter()),
                    addPrefix(type.getValue(), namespace));
            }
        });
    }
}