[#32] Added a check for empty and unused ctrl operators

ohua-dev · Feb 21, 2019 · 3a432fa · 3a432fa
1 parent 4faf6a2
commit 3a432fa
Showing 1 changed file with 17 additions and 12 deletions.
diff --git a/core/src/Ohua/ALang/Passes/Control.hs b/core/src/Ohua/ALang/Passes/Control.hs
@@ -177,15 +177,20 @@ liftIntoCtrlCtxt ::
        (Monad m, MonadGenBnd m) => Binding -> Expression -> m Expression
 liftIntoCtrlCtxt ctrlIn e = do
     (lam', actuals) <- lambdaLifting e
-    let (originalFormals, _) = lambdaArgsAndBody e
-    let (allFormals, e) = lambdaArgsAndBody lam'
-    ctrlOut <- generateBindingWith "ctrl"
-    let formals = reverse $ take (length actuals) $ reverse allFormals
-    let actuals' = [Var ctrlIn] ++ actuals
-    let ie = mkDestructured formals ctrlOut e
-    return $
-        mkLambda originalFormals $
-        Let
-            ctrlOut
-            (fromListToApply (FunRef "ohua.lang/ctrl" Nothing) actuals')
-            ie
+    if null actuals
+        then do
+            dAssertM $ lam' == e
+            pure lam'
+        else do
+            let (originalFormals, _) = lambdaArgsAndBody e
+            let (allFormals, e) = lambdaArgsAndBody lam'
+            ctrlOut <- generateBindingWith "ctrl"
+            let formals = reverse $ take (length actuals) $ reverse allFormals
+            let actuals' = [Var ctrlIn] ++ actuals
+            let ie = mkDestructured formals ctrlOut e
+            return $
+                mkLambda originalFormals $
+                Let
+                    ctrlOut
+                    (fromListToApply (FunRef "ohua.lang/ctrl" Nothing) actuals')
+                    ie