add test and spit commandLineParser changes to another PR

Author: zhengbli

src/compiler/commandLineParser.ts

@@ -727,22 +727,10 @@ namespace ts {
                 const supportedExtensions = getSupportedExtensions(options);
                 Debug.assert(indexOf(supportedExtensions, ".ts") < indexOf(supportedExtensions, ".d.ts"), "Changed priority of extensions to pick");
 
-                const potentialFiles: string[] = [];
-                if (host.readDirectoryWithMultipleExtensions) {
-                    addRange(potentialFiles, host.readDirectoryWithMultipleExtensions(basePath, supportedExtensions, exclude));
-                }
-                else {
-                    for (const extension of supportedExtensions) {
-                        addRange(potentialFiles, host.readDirectory(basePath, extension, exclude));
-                    }
-                }
                 // Get files of supported extensions in their order of resolution
                 for (const extension of supportedExtensions) {
-                    for (const fileName of potentialFiles) {
-                        if (!fileExtensionIs(fileName, extension)) {
-                            continue;
-                        }
-
+                    const filesInDirWithExtension = host.readDirectory(basePath, extension, exclude);
+                    for (const fileName of filesInDirWithExtension) {
                         // .ts extension would read the .d.ts extension files too but since .d.ts is lower priority extension,
                         // lets pick them when its turn comes up
                         if (extension === ".ts" && fileExtensionIs(fileName, ".d.ts")) {

src/compiler/sys.ts

@@ -27,7 +27,6 @@ namespace ts {
         getCurrentDirectory(): string;
         getDirectories(path: string): string[];
         readDirectory(path: string, extension?: string, exclude?: string[]): string[];
-        readDirectoryWithMultipleExtensions?(path: string, extensions: string[], exclude?: string[]): string[];
         getModifiedTime?(path: string): Date;
         createHash?(data: string): string;
         getMemoryUsage?(): number;
@@ -416,64 +415,39 @@ namespace ts {
                 return filter<string>(_fs.readdirSync(path), p => fileSystemEntryExists(combinePaths(path, p), FileSystemEntryKind.Directory));
             }
 
-            function visitDirectory(path: string, result: string[], extension: string | string[], exclude: string[]) {
-                const files = _fs.readdirSync(path || ".").sort();
-                const directories: string[] = [];
-                for (const current of files) {
+            function readDirectory(path: string, extension?: string, exclude?: string[]): string[] {
+                const result: string[] = [];
+                exclude = map(exclude, s => getCanonicalPath(combinePaths(path, s)));
+                visitDirectory(path);
+                return result;
+                function visitDirectory(path: string) {
+                    const files = _fs.readdirSync(path || ".").sort();
+                    const directories: string[] = [];
+                    for (const current of files) {
                         // This is necessary because on some file system node fails to exclude
                         // "." and "..". See https://github.com/nodejs/node/issues/4002
                         if (current === "." || current === "..") {
                             continue;
                         }
-                    const name = combinePaths(path, current);
-                    if (!contains(exclude, getCanonicalPath(name))) {
-                        // fs.statSync would throw an exception if the file is a symlink
-                        // whose linked file doesn't exist.
-                        try {
+                        const name = combinePaths(path, current);
+                        if (!contains(exclude, getCanonicalPath(name))) {
                             const stat = _fs.statSync(name);
                             if (stat.isFile()) {
-                                if (checkExtension(name)) {
+                                if (!extension || fileExtensionIs(name, extension)) {
                                     result.push(name);
                                 }
                             }
                             else if (stat.isDirectory()) {
                                 directories.push(name);
                             }
                         }
-                        catch (e) { }
-                    }
-                }
-                for (const current of directories) {
-                    visitDirectory(current, result, extension, exclude);
-                }
-
-                function checkExtension(name: string) {
-                    if (!extension) {
-                        return true;
                     }
-                    if (typeof extension === "string") {
-                        return fileExtensionIs(name, extension);
-                    }
-                    else {
-                        return forEach(extension, ext => fileExtensionIs(name, ext));
+                    for (const current of directories) {
+                        visitDirectory(current);
                     }
                 }
             }
 
-            function readDirectoryWithMultipleExtensions(path: string, extensions: string[], exclude?: string[]): string[] {
-                const result: string[] = [];
-                exclude = map(exclude, s => getCanonicalPath(combinePaths(path, s)));
-                visitDirectory(path, result, extensions, exclude);
-                return result;
-            }
-
-            function readDirectory(path: string, extension?: string, exclude?: string[]): string[] {
-                const result: string[] = [];
-                exclude = map(exclude, s => getCanonicalPath(combinePaths(path, s)));
-                visitDirectory(path, result, extension, exclude);
-                return result;
-            }
-
             return {
                 args: process.argv.slice(2),
                 newLine: _os.EOL,
@@ -548,7 +522,6 @@ namespace ts {
                 },
                 getDirectories,
                 readDirectory,
-                readDirectoryWithMultipleExtensions,
                 getModifiedTime(path) {
                     try {
                         return _fs.statSync(path).mtime;

tests/cases/unittests/tsserverProjectSystem.ts

@@ -25,6 +25,7 @@ namespace ts {
     interface FileOrFolder {
         path: string;
         content?: string;
+        fileSize?: number;
     }
 
     interface FSEntry {
@@ -34,6 +35,7 @@ namespace ts {
 
     interface File extends FSEntry {
         content: string;
+        fileSize?: number;
     }
 
     interface Folder extends FSEntry {
@@ -157,7 +159,7 @@ namespace ts {
                 const path = this.toPath(fileOrFolder.path);
                 const fullPath = getNormalizedAbsolutePath(fileOrFolder.path, this.currentDirectory);
                 if (typeof fileOrFolder.content === "string") {
-                    const entry = { path, content: fileOrFolder.content, fullPath };
+                    const entry = { path, content: fileOrFolder.content, fullPath, fileSize: fileOrFolder.fileSize };
                     this.fs.set(path, entry);
                     addFolder(getDirectoryPath(fullPath), this.toPath, this.fs).entries.push(entry);
                 }
@@ -172,6 +174,17 @@ namespace ts {
             return this.fs.contains(path) && isFile(this.fs.get(path));
         };
 
+        getFileSize?(s: string) {
+            const path = this.toPath(s);
+            if (this.fs.contains(path)) {
+                const entry = this.fs.get(path);
+                if (isFile(entry)) {
+                    return entry.fileSize ? entry.fileSize : entry.content.length;
+                }
+            }
+            return undefined;
+        }
+
         directoryExists(s: string) {
             const path = this.toPath(s);
             return this.fs.contains(path) && isFolder(this.fs.get(path));
@@ -567,5 +580,30 @@ namespace ts {
             checkConfiguredProjectActualFiles(project, [file1.path, classicModuleFile.path]);
             checkNumberOfInferredProjects(projectService, 1);
         });
+
+        it("should disable language service for project with too many non-ts files", () => {
+            const files: FileOrFolder [] = [];
+            for (let i = 0; i < 1000; i++) {
+                files.push({
+                    path: `/a/b/file${i}.js`,
+                    content: "",
+                    fileSize: 50000
+                });
+            }
+            const configFile: FileOrFolder = {
+                path: `/a/b/jsconfig.json`,
+                content: "{}"
+            };
+            files.push(configFile);
+            const host = new TestServerHost(/*useCaseSensitiveFileNames*/ false, getExecutingFilePathFromLibFile(libFile), "/", files);
+            const projectService = new server.ProjectService(host, nullLogger);
+            projectService.openClientFile(files[0].path);
+            projectService.openClientFile(files[1].path);
+            checkNumberOfConfiguredProjects(projectService, 1);
+            checkNumberOfInferredProjects(projectService, 0);
+
+            const project = projectService.configuredProjects[0];
+            assert(project.languageServiceDiabled, "the project's language service is expected to be disabled");
+        });
     });
 }