Allow any int type in Patio switch

1 files changed, 5 insertions, 3 deletions

diff --git a/src/main/java/org/shedlang/compiler/backends/wasm/PatioToWasm.java b/src/main/java/org/shedlang/compiler/backends/wasm/PatioToWasm.java
index 63ab8d3a..bedfe7e7 100644
--- a/src/main/java/org/shedlang/compiler/backends/wasm/PatioToWasm.java
+++ b/src/main/java/org/shedlang/compiler/backends/wasm/PatioToWasm.java
@@ -380,11 +380,13 @@ public class PatioToWasm {
 
                 for (var patioSwitchCase : patioSwitch.cases()) {
                     context = generateOperand(patioSwitch.value(), context);
-                    // TODO: should probably allow any int type in switches
-                    context = context.addInstruction(new WasmI32Const(
+                    context = context.addInstruction(wasmIntConst(
+                        patioOperandToWasmValueType(patioSwitch.value(), context),
                         patioSwitchCase.value()
                     ));
-                    context = context.addInstruction(new WasmI32Eq());
+                    context = context.addInstruction(wasmIntEq(
+                        patioOperandToWasmValueType(patioSwitch.value(), context)
+                    ));
                     context = context.enterIfInstruction(wasmBlockType);
                     context = generateBlockBody(patioSwitchCase.body(), context);
                     context = context.enterElse();