From 39f262406608e509463ffffd2813b7176544d493 Mon Sep 17 00:00:00 2001
From: Jeroen Engels <jfm.engels@gmail.com>
Date: Sun, 3 Sep 2023 11:44:39 +0200
Subject: [PATCH] Extract function

---
 src/CognitiveComplexity.elm | 21 ++++++++++++++-------
 1 file changed, 14 insertions(+), 7 deletions(-)

diff --git a/src/CognitiveComplexity.elm b/src/CognitiveComplexity.elm
index 0ea4ef4..7f52ca3 100644
--- a/src/CognitiveComplexity.elm
+++ b/src/CognitiveComplexity.elm
@@ -751,7 +751,7 @@ processDFSTree graph stack visited =
         [] ->
             { recursiveCalls = Dict.empty, visited = visited, stack = [] }
 
-        head :: _ ->
+        head :: restOfStack ->
             let
                 vertices : List String
                 vertices =
@@ -781,12 +781,19 @@ processDFSTree graph stack visited =
                 )
                 { recursiveCalls = Dict.empty, visited = visited }
                 vertices
-                |> (\res ->
-                        { recursiveCalls = res.recursiveCalls
-                        , visited = Dict.insert head Done res.visited
-                        , stack = List.drop 1 stack
-                        }
-                   )
+                |> updateStack head restOfStack
+
+
+updateStack :
+    String
+    -> List String
+    -> { recursiveCalls : RecursiveCalls, visited : Visited }
+    -> { recursiveCalls : RecursiveCalls, visited : Visited, stack : List String }
+updateStack head stack res =
+    { recursiveCalls = res.recursiveCalls
+    , visited = Dict.insert head Done res.visited
+    , stack = stack
+    }
 
 
 dataExtractor : ProjectContext -> Encode.Value