Assign node types automatically

2 files changed, 8 insertions, 4 deletions

diff --git a/src/main/java/org/zwobble/couscous/backends/csharp/primitives/PrimitiveInstanceMethodCall.java b/src/main/java/org/zwobble/couscous/backends/csharp/primitives/PrimitiveInstanceMethodCall.java
index 25555d8..4eeb22c 100644
--- a/src/main/java/org/zwobble/couscous/backends/csharp/primitives/PrimitiveInstanceMethodCall.java
+++ b/src/main/java/org/zwobble/couscous/backends/csharp/primitives/PrimitiveInstanceMethodCall.java
@@ -2,8 +2,8 @@ package org.zwobble.couscous.backends.csharp.primitives;
 
 import org.zwobble.couscous.ast.ExpressionNode;
 import org.zwobble.couscous.ast.Node;
+import org.zwobble.couscous.ast.NodeTypes;
 import org.zwobble.couscous.ast.visitors.NodeTransformer;
-import org.zwobble.couscous.backends.csharp.CsharpNodeTypes;
 import org.zwobble.couscous.types.Type;
 
 import java.util.List;
@@ -11,6 +11,8 @@ import java.util.List;
 import static org.zwobble.couscous.util.ExtraIterables.lazyCons;
 
 public class PrimitiveInstanceMethodCall implements ExpressionNode {
+    private final static int NODE_TYPE = NodeTypes.register(PrimitiveInstanceMethodCall.class);
+
     private final PrimitiveInstanceMethod method;
     private final ExpressionNode receiver;
     private final List<ExpressionNode> arguments;
@@ -28,7 +30,7 @@ public class PrimitiveInstanceMethodCall implements ExpressionNode {
 
     @Override
     public int type() {
-        return CsharpNodeTypes.PRIMITIVE_INSTANCE_METHOD_CALL;
+        return NODE_TYPE;
     }
 
     @Override
diff --git a/src/main/java/org/zwobble/couscous/backends/csharp/primitives/PrimitiveStaticMethodCall.java b/src/main/java/org/zwobble/couscous/backends/csharp/primitives/PrimitiveStaticMethodCall.java
index 94b26da..6ce3660 100644
--- a/src/main/java/org/zwobble/couscous/backends/csharp/primitives/PrimitiveStaticMethodCall.java
+++ b/src/main/java/org/zwobble/couscous/backends/csharp/primitives/PrimitiveStaticMethodCall.java
@@ -2,13 +2,15 @@ package org.zwobble.couscous.backends.csharp.primitives;
 
 import org.zwobble.couscous.ast.ExpressionNode;
 import org.zwobble.couscous.ast.Node;
+import org.zwobble.couscous.ast.NodeTypes;
 import org.zwobble.couscous.ast.visitors.NodeTransformer;
-import org.zwobble.couscous.backends.csharp.CsharpNodeTypes;
 import org.zwobble.couscous.types.Type;
 
 import java.util.List;
 
 public class PrimitiveStaticMethodCall implements ExpressionNode {
+    private final static int NODE_TYPE = NodeTypes.register(PrimitiveStaticMethodCall.class);
+
     private final PrimitiveStaticMethod method;
     private final List<ExpressionNode> arguments;
 
@@ -24,7 +26,7 @@ public class PrimitiveStaticMethodCall implements ExpressionNode {
 
     @Override
     public int type() {
-        return CsharpNodeTypes.PRIMITIVE_STATIC_METHOD_CALL;
+        return NODE_TYPE;
     }
 
     @Override