From c9a17f325b96f812604b529ce2ba36d21e3441b8 Mon Sep 17 00:00:00 2001
From: Sheetal Nandi <shkamat@microsoft.com>
Date: Thu, 9 Nov 2017 13:35:56 -0800
Subject: [PATCH] Add api to get the dependencies of the file

---
 src/compiler/builder.ts                       | 54 ++++++++++++++++++-
 .../reference/api/tsserverlibrary.d.ts        |  4 ++
 tests/baselines/reference/api/typescript.d.ts |  4 ++
 3 files changed, 61 insertions(+), 1 deletion(-)

diff --git a/src/compiler/builder.ts b/src/compiler/builder.ts
index 0881f68800c..efa0f0a0372 100644
--- a/src/compiler/builder.ts
+++ b/src/compiler/builder.ts
@@ -93,6 +93,11 @@ namespace ts {
          * Note that it is assumed that the when asked about semantic diagnostics, the file has been taken out of affected files
          */
         getSemanticDiagnostics(programOfThisState: Program, sourceFile?: SourceFile, cancellationToken?: CancellationToken): ReadonlyArray<Diagnostic>;
+
+        /**
+         * Get all the dependencies of the file
+         */
+        getAllDependencies(programOfThisState: Program, sourceFile: SourceFile): string[];
     }
 
     /**
@@ -190,7 +195,8 @@ namespace ts {
             canCreateNewStateFrom,
             getFilesAffectedBy,
             emitNextAffectedFile,
-            getSemanticDiagnostics
+            getSemanticDiagnostics,
+            getAllDependencies
         };
 
         /**
@@ -306,6 +312,52 @@ namespace ts {
             return diagnostics;
         }
 
+        /**
+         * Get all the dependencies of the sourceFile
+         */
+        function getAllDependencies(programOfThisState: Program, sourceFile: SourceFile): string[] {
+            const compilerOptions = programOfThisState.getCompilerOptions();
+            // With --out or --outFile all outputs go into single file, all files depend on each other
+            if (compilerOptions.outFile || compilerOptions.out) {
+                return programOfThisState.getSourceFiles().map(getFileName);
+            }
+
+            // If this is non module emit, or its a global file, it depends on all the source files
+            if (!isModuleEmit || (!isExternalModule(sourceFile) && !containsOnlyAmbientModules(sourceFile))) {
+                return programOfThisState.getSourceFiles().map(getFileName);
+            }
+
+            // Get the references, traversing deep from the referenceMap
+            Debug.assert(!!referencedMap);
+            const seenMap = createMap<true>();
+            const queue = [sourceFile.path];
+            while (queue.length) {
+                const path = queue.pop();
+                if (!seenMap.has(path)) {
+                    seenMap.set(path, true);
+                    const references = referencedMap.get(path);
+                    if (references) {
+                        const iterator = references.keys();
+                        for (let { value, done } = iterator.next(); !done; { value, done } = iterator.next()) {
+                            queue.push(value as Path);
+                        }
+                    }
+                }
+            }
+
+            return flatMapIter(seenMap.keys(), path => {
+                const file = programOfThisState.getSourceFileByPath(path as Path);
+                if (file) {
+                    return file.fileName;
+                }
+                return path;
+            });
+        }
+
+        function getFileName(sourceFile: SourceFile) {
+            return sourceFile.fileName;
+        }
+
         /**
          * For script files that contains only ambient external modules, although they are not actually external module files,
          * they can only be consumed via importing elements from them. Regular script files cannot consume them. Therefore,
diff --git a/tests/baselines/reference/api/tsserverlibrary.d.ts b/tests/baselines/reference/api/tsserverlibrary.d.ts
index 3ec16254780..5c306474533 100644
--- a/tests/baselines/reference/api/tsserverlibrary.d.ts
+++ b/tests/baselines/reference/api/tsserverlibrary.d.ts
@@ -3815,6 +3815,10 @@ declare namespace ts {
          * Note that it is assumed that the when asked about semantic diagnostics, the file has been taken out of affected files
          */
         getSemanticDiagnostics(programOfThisState: Program, sourceFile?: SourceFile, cancellationToken?: CancellationToken): ReadonlyArray<Diagnostic>;
+        /**
+         * Get all the dependencies of the file
+         */
+        getAllDependencies(programOfThisState: Program, sourceFile: SourceFile): string[];
     }
     /**
      * Information about the source file: Its version and optional signature from last emit
diff --git a/tests/baselines/reference/api/typescript.d.ts b/tests/baselines/reference/api/typescript.d.ts
index 63e2d453b18..2212a859c0a 100644
--- a/tests/baselines/reference/api/typescript.d.ts
+++ b/tests/baselines/reference/api/typescript.d.ts
@@ -3762,6 +3762,10 @@ declare namespace ts {
          * Note that it is assumed that the when asked about semantic diagnostics, the file has been taken out of affected files
          */
         getSemanticDiagnostics(programOfThisState: Program, sourceFile?: SourceFile, cancellationToken?: CancellationToken): ReadonlyArray<Diagnostic>;
+        /**
+         * Get all the dependencies of the file
+         */
+        getAllDependencies(programOfThisState: Program, sourceFile: SourceFile): string[];
     }
     /**
      * Information about the source file: Its version and optional signature from last emit