fix caseCmd for RepConstr

anoma · Feb 21, 2024 · d4824d8 · d4824d8
1 parent 999f539
commit d4824d8
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/Juvix/Compiler/Nockma/Translation/FromTree.hs b/src/Juvix/Compiler/Nockma/Translation/FromTree.hs
@@ -685,8 +685,13 @@ caseCmd arg defaultBranch = \case
             OpEq
               # constructorTagToTerm tag
               # (getConstructorField ConstructorTag arg)
-      elseBr <- caseCmd arg defaultBranch bs
-      return (branch cond b elseBr)
+      case nonEmpty bs of
+        Nothing -> case defaultBranch of
+          Nothing -> return b
+          Just defbr -> return (branch cond b defbr)
+        Just ((t', b') :| bs') -> do
+          elseBr <- goRepConstr t' b' bs'
+          return (branch cond b elseBr)
 
     asNockmaMemRepListConstr :: Tree.Tag -> Sem r NockmaMemRepListConstr
     asNockmaMemRepListConstr tag = case tag of