Set program as affected if emitting/diagnostics for whole program

sheetalkamat · sheetalkamat · commit ffa64e8c4f1c · 2017-11-22T11:49:08.000-08:00
diff --git a/src/compiler/builder.ts b/src/compiler/builder.ts
@@ -334,8 +334,8 @@ namespace ts {
         /**
          * Returns the result with affected file
          */
-        function toAffectedFileResult<T>(result: T, affectedFile?: SourceFile): AffectedFileResult<T> {
-            return { result, affectedFile };
+        function toAffectedFileResult<T>(result: T, affected: SourceFile | Program): AffectedFileResult<T> {
+            return { result, affected };
         }
 
         /**
@@ -350,7 +350,10 @@ namespace ts {
             }
             else if (affectedFile === programOfThisState) {
                 // When whole program is affected, do emit only once (eg when --out or --outFile is specified)
-                return toAffectedFileResult(programOfThisState.emit(/*targetSourceFile*/ undefined, writeFileCallback, cancellationToken, /*emitOnlyDtsFiles*/ false, customTransformers));
+                return toAffectedFileResult(
+                    programOfThisState.emit(/*targetSourceFile*/ undefined, writeFileCallback, cancellationToken, /*emitOnlyDtsFiles*/ false, customTransformers),
+                    programOfThisState
+                );
             }
 
             // Emit the affected file
@@ -374,7 +377,10 @@ namespace ts {
                 }
                 else if (affectedFile === programOfThisState) {
                     // When whole program is affected, get all semantic diagnostics (eg when --out or --outFile is specified)
-                    return toAffectedFileResult(programOfThisState.getSemanticDiagnostics(/*targetSourceFile*/ undefined, cancellationToken));
+                    return toAffectedFileResult(
+                        programOfThisState.getSemanticDiagnostics(/*targetSourceFile*/ undefined, cancellationToken),
+                        programOfThisState
+                    );
                 }
 
                 // Get diagnostics for the affected file if its not ignored
@@ -693,7 +699,7 @@ namespace ts {
         text: string;
     }
 
-    export type AffectedFileResult<T> = { result: T; affectedFile?: SourceFile; } | undefined;
+    export type AffectedFileResult<T> = { result: T; affected: SourceFile | Program; } | undefined;
 
     export interface BuilderOptions {
         getCanonicalFileName: (fileName: string) => string;
diff --git a/tests/baselines/reference/api/tsserverlibrary.d.ts b/tests/baselines/reference/api/tsserverlibrary.d.ts
@@ -3773,7 +3773,7 @@ declare namespace ts {
     }
     type AffectedFileResult<T> = {
         result: T;
-        affectedFile?: SourceFile;
+        affected: SourceFile | Program;
     } | undefined;
     interface BuilderOptions {
         getCanonicalFileName: (fileName: string) => string;
diff --git a/tests/baselines/reference/api/typescript.d.ts b/tests/baselines/reference/api/typescript.d.ts
@@ -3720,7 +3720,7 @@ declare namespace ts {
     }
     type AffectedFileResult<T> = {
         result: T;
-        affectedFile?: SourceFile;
+        affected: SourceFile | Program;
     } | undefined;
     interface BuilderOptions {
         getCanonicalFileName: (fileName: string) => string;

Original file line number	Diff line number	Diff line change
`@@ -3773,7 +3773,7 @@ declare namespace ts {`
`3773`	`3773`	`}`
`3774`	`3774`	`type AffectedFileResult<T> = {`
`3775`	`3775`	`result: T;`
`3776`		`- affectedFile?: SourceFile;`
	`3776`	`+ affected: SourceFile \| Program;`
`3777`	`3777`	`} \| undefined;`
`3778`	`3778`	`interface BuilderOptions {`
`3779`	`3779`	`getCanonicalFileName: (fileName: string) => string;`
Original file line number	Diff line number	Diff line change
`@@ -3720,7 +3720,7 @@ declare namespace ts {`
`3720`	`3720`	`}`
`3721`	`3721`	`type AffectedFileResult<T> = {`
`3722`	`3722`	`result: T;`
`3723`		`- affectedFile?: SourceFile;`
	`3723`	`+ affected: SourceFile \| Program;`
`3724`	`3724`	`} \| undefined;`
`3725`	`3725`	`interface BuilderOptions {`
`3726`	`3726`	`getCanonicalFileName: (fileName: string) => string;`