From b0fc9e0468133a956c28a5ee89bf7cb5d00e7eed Mon Sep 17 00:00:00 2001
From: zhengbli <zhengbli@microsoft.com>
Date: Tue, 15 Mar 2016 15:49:42 -0700
Subject: [PATCH] Address PR

---
 src/compiler/program.ts   | 23 +++++++++--------------
 src/compiler/utilities.ts |  2 --
 2 files changed, 9 insertions(+), 16 deletions(-)

diff --git a/src/compiler/program.ts b/src/compiler/program.ts
index 626d970e8ff..df4fb5fbba7 100644
--- a/src/compiler/program.ts
+++ b/src/compiler/program.ts
@@ -7,6 +7,7 @@ namespace ts {
     /* @internal */ export let emitTime = 0;
     /* @internal */ export let ioReadTime = 0;
     /* @internal */ export let ioWriteTime = 0;
+    /* @internal */ export const maxProgramSizeForNonTsFiles = 20 * 1024 * 1024;
 
     /** The version of the TypeScript compiler release */
 
@@ -1038,10 +1039,8 @@ namespace ts {
                     diagnosticArgument = [fileName, "'" + supportedExtensions.join("', '") + "'"];
                 }
                 else if (!findSourceFile(fileName, toPath(fileName, currentDirectory, getCanonicalFileName), isDefaultLib, refFile, refPos, refEnd)) {
-                    if (hasTypeScriptFileExtension(fileName) || !exceedProgramSizeLimit()) {
-                        diagnostic = Diagnostics.File_0_not_found;
-                        diagnosticArgument = [fileName];
-                    }
+                    diagnostic = Diagnostics.File_0_not_found;
+                    diagnosticArgument = [fileName];
                 }
                 else if (refFile && host.getCanonicalFileName(fileName) === host.getCanonicalFileName(refFile.fileName)) {
                     diagnostic = Diagnostics.A_file_cannot_have_a_reference_to_itself;
@@ -1052,22 +1051,18 @@ namespace ts {
                 const nonTsFile: SourceFile = options.allowNonTsExtensions && findSourceFile(fileName, toPath(fileName, currentDirectory, getCanonicalFileName), isDefaultLib, refFile, refPos, refEnd);
                 if (!nonTsFile) {
                     if (options.allowNonTsExtensions) {
-                        if (!exceedProgramSizeLimit()) {
-                            diagnostic = Diagnostics.File_0_not_found;
-                            diagnosticArgument = [fileName];
-                        }
+                        diagnostic = Diagnostics.File_0_not_found;
+                        diagnosticArgument = [fileName];
                     }
                     else if (!forEach(supportedExtensions, extension => findSourceFile(fileName + extension, toPath(fileName + extension, currentDirectory, getCanonicalFileName), isDefaultLib, refFile, refPos, refEnd))) {
-                        if (!exceedProgramSizeLimit()) {
-                            diagnostic = Diagnostics.File_0_not_found;
-                            fileName += ".ts";
-                            diagnosticArgument = [fileName];
-                        }
+                        diagnostic = Diagnostics.File_0_not_found;
+                        fileName += ".ts";
+                        diagnosticArgument = [fileName];
                     }
                 }
             }
 
-            if (diagnostic) {
+            if (diagnostic && !exceedProgramSizeLimit()) {
                 if (refFile !== undefined && refEnd !== undefined && refPos !== undefined) {
                     fileProcessingDiagnostics.add(createFileDiagnostic(refFile, refPos, refEnd - refPos, diagnostic, ...diagnosticArgument));
                 }
diff --git a/src/compiler/utilities.ts b/src/compiler/utilities.ts
index c8e4143a601..9d1aa877cf0 100644
--- a/src/compiler/utilities.ts
+++ b/src/compiler/utilities.ts
@@ -2862,6 +2862,4 @@ namespace ts {
     export function isParameterPropertyDeclaration(node: ParameterDeclaration): boolean {
         return node.flags & NodeFlags.AccessibilityModifier && node.parent.kind === SyntaxKind.Constructor && isClassLike(node.parent.parent);
     }
-
-    export const maxProgramSizeForNonTsFiles = 20 * 1024 * 1024;
 }