Fix tree traversal breaking on errors

2020-07-10 12:45:49 +04:00 · 2020-07-10 12:45:49 +04:00 · 5737cd5686
commit 5737cd5686
parent e5ec32f46a
1 changed files with 3 additions and 1 deletions
--- a/tools/lsp/squirrel/src/Tree.hs
+++ b/tools/lsp/squirrel/src/Tree.hs
@ -213,7 +213,9 @@ traverseMany visitors orElse = go
          case tree of
            Tree (Right (r, union)) -> do
              union' <- traverse go union
-              return $ Tree (Right (orElse r, union'))
+              return $ Tree $ Right (orElse r, union')
+            Tree (Left err) -> do
+              return $ Tree $ Left $ fmap orElse err

 -- | Make a tree out of a layer and an info.
 mk :: (Functor f, Element f fs) => info -> f (Tree fs info) -> Tree fs info