From ad39ec68f3edba05581c6dbfc4007727877a814b Mon Sep 17 00:00:00 2001
From: Florian Verdonck <florian.verdonck@outlook.com>
Date: Wed, 25 Oct 2023 11:47:21 +0200
Subject: [PATCH] Add GetDependentFiles to slim service. (#14)

---
 fcs/service_slim.fs | 48 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)

diff --git a/fcs/service_slim.fs b/fcs/service_slim.fs
index 7cccc9f4168..dd15de27d58 100644
--- a/fcs/service_slim.fs
+++ b/fcs/service_slim.fs
@@ -40,6 +40,7 @@ open FSharp.Compiler.TypedTree
 open FSharp.Compiler.TypedTreeBasics
 open FSharp.Compiler.TypedTreeOps
 open FSharp.Compiler.BuildGraph
+open FSharp.Compiler.GraphChecking
 
 //-------------------------------------------------------------------------
 // InteractiveChecker
@@ -403,3 +404,50 @@ type InteractiveChecker internal (compilerStateCache) =
 
         return (parseFileResults, checkFileResults, projectResults)
     }
+
+    /// Find the dependent files of the current file based on the untyped syntax tree
+    member _.GetDependentFiles(currentFileName: string, fileNames: string[], sourceReader: string -> int * Lazy<string>) = async {
+        // parse files
+        let! ct = Async.CancellationToken
+        let! compilerState = compilerStateCache.Get()
+        // parse files
+        let parsingOptions = FSharpParsingOptions.FromTcConfig(compilerState.tcConfig, fileNames, false)
+        let parseFile fileName =
+            let sourceHash, source = sourceReader fileName
+            ParseFile (fileName, sourceHash, source, parsingOptions, compilerState, ct)
+
+        // TODO: not sure if parse cache issues still applies
+        let parseResults = fileNames |> Array.Parallel.map parseFile
+        let sourceFiles: FileInProject array =
+            parseResults
+            |> Array.mapi (fun idx (parseResults: FSharpParseFileResults) ->
+                let input = parseResults.ParseTree
+                {
+                    Idx = idx
+                    FileName = input.FileName
+                    ParsedInput = input
+                })
+
+        let currentFileIdx = Array.IndexOf(fileNames, currentFileName)
+
+        let filePairs = FilePairMap(sourceFiles)
+        let graph, _trie = DependencyResolution.mkGraph filePairs sourceFiles
+        
+        let dependentFiles =
+            graph.Keys
+            |> Seq.choose (fun fileIndex ->
+                let isDependency =
+                    // Only files that came after the current file are relevant
+                    fileIndex > currentFileIdx
+                    &&
+                    // The current file is listed as a dependency 
+                    Array.contains currentFileIdx graph.[fileIndex]
+                if not isDependency then
+                    None
+                else
+                    Some(Array.item fileIndex fileNames)
+                )
+            |> Seq.toArray
+
+        return dependentFiles
+    }
\ No newline at end of file