Implement getPath in terms of unfoldr

koalaman · Dec 31, 2023 · e1ad063 · e1ad063
1 parent 980e7d3
commit e1ad063
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/ShellCheck/ASTLib.hs b/src/ShellCheck/ASTLib.hs
@@ -897,10 +897,9 @@ getUnmodifiedParameterExpansion t =
         _ -> Nothing
 
 --- A list of the element and all its parents up to the root node.
-getPath tree t = t :
-    case Map.lookup (getId t) tree of
-        Nothing     -> []
-        Just parent -> getPath tree parent
+getPath tree t = t : unfoldr go t
+  where
+    go s = (\x -> (x,x)) <$> Map.lookup (getId s) tree
 
 isClosingFileOp op =
     case op of