fix caseCmd for RepConstr

anoma · Feb 23, 2024 · 3b53204 · 3b53204
1 parent be2cb5c
commit 3b53204
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
@@ -753,8 +753,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