Move type coercion decisions into the backends

1 files changed, 16 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/couscous/backends/python/PythonCodeGenerator.java b/src/main/java/org/zwobble/couscous/backends/python/PythonCodeGenerator.java
index 913b1b7..ada7dea 100644
--- a/src/main/java/org/zwobble/couscous/backends/python/PythonCodeGenerator.java
+++ b/src/main/java/org/zwobble/couscous/backends/python/PythonCodeGenerator.java
@@ -261,6 +261,22 @@ public class PythonCodeGenerator {
         public PythonExpressionNode visit(FieldAccessNode fieldAccess) {
             return pythonAttributeAccess(generateExpression(fieldAccess.getLeft()), fieldAccess.getFieldName());
         }
+
+        @Override
+        public PythonExpressionNode visit(TypeCoercionNode typeCoercion) {
+            PythonExpressionNode value = generateExpression(typeCoercion.getExpression());
+            if (typeCoercion.isIntegerBox()) {
+                return internalMethod("boxInt", asList(value));
+            } else if (typeCoercion.isIntegerUnbox()) {
+                return internalMethod("unboxInt", asList(value));
+            } else {
+                return value;
+            }
+        }
+
+        private PythonExpressionNode internalMethod(String name, List<PythonExpressionNode> arguments) {
+            return pythonCall(pythonAttributeAccess(pythonVariableReference("_couscous"), name), arguments);
+        }
     }
     
     private static boolean isPrimitive(ExpressionNode value) {