Implement negation in C# backend

6 files changed, 59 insertions, 22 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/OperationNode.java b/src/main/java/org/zwobble/couscous/ast/OperationNode.java
index 0840044..6be841f 100644
--- a/src/main/java/org/zwobble/couscous/ast/OperationNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/OperationNode.java
@@ -28,6 +28,10 @@ public class OperationNode implements ExpressionNode {
             type);
     }
 
+    public Operator getOperator() {
+        return operator;
+    }
+
     public List<ExpressionNode> getArguments() {
         return arguments;
     }
diff --git a/src/main/java/org/zwobble/couscous/ast/Operations.java b/src/main/java/org/zwobble/couscous/ast/Operations.java
index 5af2f8b..0ae4923 100644
--- a/src/main/java/org/zwobble/couscous/ast/Operations.java
+++ b/src/main/java/org/zwobble/couscous/ast/Operations.java
@@ -5,9 +5,6 @@ import org.zwobble.couscous.values.IntegerValue;
 import org.zwobble.couscous.values.InternalCouscousValue;
 import org.zwobble.couscous.values.ObjectValues;
 
-import java.util.Collections;
-
-import static org.zwobble.couscous.ast.MethodCallNode.methodCall;
 import static org.zwobble.couscous.ast.MethodCallNode.staticMethodCall;
 import static org.zwobble.couscous.ast.TypeCoercionNode.typeCoercion;
 import static org.zwobble.couscous.util.ExtraLists.list;
@@ -19,7 +16,9 @@ public class Operations {
 
     public static ExpressionNode not(ExpressionNode value) {
         if (value.getType().equals(BooleanValue.REF)) {
-            return methodCall(value, "negate", Collections.emptyList(), BooleanValue.REF);
+            return OperationNode.operation(
+                Operator.BOOLEAN_NOT, list(value),
+                BooleanValue.REF);
         } else {
             throw new IllegalArgumentException("Can only negate booleans");
         }
diff --git a/src/main/java/org/zwobble/couscous/ast/Operator.java b/src/main/java/org/zwobble/couscous/ast/Operator.java
index a215138..a51a01b 100644
--- a/src/main/java/org/zwobble/couscous/ast/Operator.java
+++ b/src/main/java/org/zwobble/couscous/ast/Operator.java
@@ -1,30 +1,37 @@
 package org.zwobble.couscous.ast;
 
 public enum Operator {
-    BOOLEAN_AND("and", true),
-    BOOLEAN_OR("or", true),
-
-    ADD("add", false),
-    SUBTRACT("subtract", false),
-    MULTIPLY("multiply", false),
-    DIVIDE("divide", false),
-    MOD("mod", false),
-
-    EQUALS("equals", true),
-    NOT_EQUALS("notEquals", true),
-    GREATER_THAN("greaterThan", true),
-    GREATER_THAN_OR_EQUAL("greaterThanOrEqual", true),
-    LESS_THAN("lessThan", true),
-    LESS_THAN_OR_EQUAL("lessThanOrEqual", true);
+    BOOLEAN_AND("&&", "and", true),
+    BOOLEAN_OR("||", "or", true),
+    BOOLEAN_NOT("!", "negate", true),
 
+    ADD("+", "add", false),
+    SUBTRACT("-", "subtract", false),
+    MULTIPLY("*", "multiply", false),
+    DIVIDE("/", "divide", false),
+    MOD("%", "mod", false),
+
+    EQUALS("==", "equals", true),
+    NOT_EQUALS("!=", "notEquals", true),
+    GREATER_THAN(">", "greaterThan", true),
+    GREATER_THAN_OR_EQUAL(">=", "greaterThanOrEqual", true),
+    LESS_THAN("<", "lessThan", true),
+    LESS_THAN_OR_EQUAL("<=", "lessThanOrEqual", true);
+
+    private final String symbol;
     private final String methodName;
     private final boolean isBoolean;
 
-    Operator(String methodName, boolean isBoolean) {
+    Operator(String symbol, String methodName, boolean isBoolean) {
+        this.symbol = symbol;
         this.methodName = methodName;
         this.isBoolean = isBoolean;
     }
 
+    public String getSymbol() {
+        return symbol;
+    }
+
     public String getMethodName() {
         return methodName;
     }
diff --git a/src/main/java/org/zwobble/couscous/backends/csharp/CsharpCodeGenerator.java b/src/main/java/org/zwobble/couscous/backends/csharp/CsharpCodeGenerator.java
index 9e48e9a..ab55e0d 100644
--- a/src/main/java/org/zwobble/couscous/backends/csharp/CsharpCodeGenerator.java
+++ b/src/main/java/org/zwobble/couscous/backends/csharp/CsharpCodeGenerator.java
@@ -50,7 +50,8 @@ public class CsharpCodeGenerator extends NodeTransformer {
         return call.getReceiver().accept(new Receiver.Mapper<Optional<ExpressionNode>>() {
             @Override
             public Optional<ExpressionNode> visit(ExpressionNode receiver) {
-                return Optional.empty();
+                return CsharpPrimitiveMethods.getPrimitiveMethod(receiver.getType(), call.getMethodName())
+                    .map(generator -> generator.generate(receiver, call.getArguments()));
             }
 
             @Override
diff --git a/src/main/java/org/zwobble/couscous/backends/csharp/CsharpPrimitiveMethods.java b/src/main/java/org/zwobble/couscous/backends/csharp/CsharpPrimitiveMethods.java
index 9424b2d..e094258 100644
--- a/src/main/java/org/zwobble/couscous/backends/csharp/CsharpPrimitiveMethods.java
+++ b/src/main/java/org/zwobble/couscous/backends/csharp/CsharpPrimitiveMethods.java
@@ -3,6 +3,8 @@ package org.zwobble.couscous.backends.csharp;
 import com.google.common.collect.ImmutableMap;
 import org.zwobble.couscous.ast.ExpressionNode;
 import org.zwobble.couscous.ast.TypeName;
+import org.zwobble.couscous.backends.python.ast.PythonExpressionNode;
+import org.zwobble.couscous.values.BooleanValue;
 import org.zwobble.couscous.values.ObjectValues;
 
 import java.util.List;
@@ -13,6 +15,18 @@ import static org.zwobble.couscous.ast.MethodCallNode.staticMethodCall;
 import static org.zwobble.couscous.util.ExtraLists.list;
 
 public class CsharpPrimitiveMethods {
+    private static final Map<String, PrimitiveMethodGenerator> BOOLEAN_METHODS;
+
+    static {
+        ImmutableMap.Builder<String, PrimitiveMethodGenerator> methods = ImmutableMap.builder();
+        BOOLEAN_METHODS = methods.build();
+    }
+
+    private static final Map<TypeName, Map<String, PrimitiveMethodGenerator>> METHODS =
+        ImmutableMap.<TypeName, Map<String, PrimitiveMethodGenerator>>builder()
+            .put(BooleanValue.REF, BOOLEAN_METHODS)
+            .build();
+
     private static final Map<String, PrimitiveStaticMethodGenerator> STATIC_INT_METHODS =
         ImmutableMap.<String, PrimitiveStaticMethodGenerator>builder()
 
@@ -29,6 +43,16 @@ public class CsharpPrimitiveMethods {
             .put(ObjectValues.BOXED_INT, STATIC_INT_METHODS)
             .build();
 
+    public static Optional<PrimitiveMethodGenerator> getPrimitiveMethod(TypeName type, String methodName) {
+        return Optional.ofNullable(METHODS.get(type))
+            .flatMap(methodsForType -> Optional.ofNullable(methodsForType.get(methodName)));
+    }
+
+    @FunctionalInterface
+    public interface PrimitiveMethodGenerator {
+        ExpressionNode generate(ExpressionNode receiver, List<ExpressionNode> arguments);
+    }
+
     public static Optional<PrimitiveStaticMethodGenerator> getPrimitiveStaticMethod(TypeName type, String methodName) {
         return Optional.ofNullable(STATIC_METHODS.get(type))
             .flatMap(methodsForType -> Optional.ofNullable(methodsForType.get(methodName)));
diff --git a/src/main/java/org/zwobble/couscous/backends/csharp/CsharpSerializer.java b/src/main/java/org/zwobble/couscous/backends/csharp/CsharpSerializer.java
index 06dafbb..6f20310 100644
--- a/src/main/java/org/zwobble/couscous/backends/csharp/CsharpSerializer.java
+++ b/src/main/java/org/zwobble/couscous/backends/csharp/CsharpSerializer.java
@@ -121,7 +121,9 @@ public class CsharpSerializer implements NodeVisitor {
 
     @Override
     public void visit(OperationNode operation) {
-        write(operation.desugar());
+        // TODO: distinguish operator types (prefix, infix)
+        writer.writeSymbol(operation.getOperator().getSymbol());
+        write(operation.getArguments().get(0));
     }
 
     private void writeArguments(List<? extends ExpressionNode> arguments) {