add support of jsx extension

Kingwl · Kingwl · commit 4d57419faa2e · 2018-06-23T01:08:36.000+08:00
diff --git a/src/compiler/moduleSpecifiers.ts b/src/compiler/moduleSpecifiers.ts
@@ -3,7 +3,7 @@
 namespace ts.moduleSpecifiers {
     export interface ModuleSpecifierPreferences {
         importModuleSpecifierPreference?: "relative" | "non-relative";
-        includingJsExtensionOnAutoImports?: boolean;
+        includeExtensionInImports?: Extension.Js | Extension.Jsx;
     }
 
     // Note: fromSourceFile is just for usesJsExtensionOnImports
@@ -38,39 +38,39 @@ namespace ts.moduleSpecifiers {
 
     interface Info {
         readonly moduleResolutionKind: ModuleResolutionKind;
-        readonly addJsExtension: boolean;
+        readonly addExtension: Extension.Js | Extension.Jsx | undefined;
         readonly getCanonicalFileName: GetCanonicalFileName;
         readonly sourceDirectory: string;
     }
     // importingSourceFileName is separate because getEditsForFileRename may need to specify an updated path
     function getInfo(compilerOptions: CompilerOptions, importingSourceFile: SourceFile, importingSourceFileName: string, host: ModuleSpecifierResolutionHost, preferences: ModuleSpecifierPreferences): Info {
         const moduleResolutionKind = getEmitModuleResolutionKind(compilerOptions);
-        const addJsExtension = usesJsExtensionOnImports(importingSourceFile, preferences);
+        const addExtension = usesExtensionOnImports(importingSourceFile, preferences);
         const getCanonicalFileName = createGetCanonicalFileName(host.useCaseSensitiveFileNames ? host.useCaseSensitiveFileNames() : true);
         const sourceDirectory = getDirectoryPath(importingSourceFileName);
-        return { moduleResolutionKind, addJsExtension, getCanonicalFileName, sourceDirectory };
+        return { moduleResolutionKind, addExtension, getCanonicalFileName, sourceDirectory };
     }
 
     function getGlobalModuleSpecifier(
         moduleFileName: string,
-        { addJsExtension, getCanonicalFileName, sourceDirectory }: Info,
+        { addExtension, getCanonicalFileName, sourceDirectory }: Info,
         host: ModuleSpecifierResolutionHost,
         compilerOptions: CompilerOptions,
     ) {
-        return tryGetModuleNameFromTypeRoots(compilerOptions, host, getCanonicalFileName, moduleFileName, addJsExtension)
+        return tryGetModuleNameFromTypeRoots(compilerOptions, host, getCanonicalFileName, moduleFileName, addExtension)
             || tryGetModuleNameAsNodeModule(compilerOptions, moduleFileName, host, getCanonicalFileName, sourceDirectory)
             || compilerOptions.rootDirs && tryGetModuleNameFromRootDirs(compilerOptions.rootDirs, moduleFileName, sourceDirectory, getCanonicalFileName);
     }
 
     function getLocalModuleSpecifiers(
         moduleFileName: string,
-        { moduleResolutionKind, addJsExtension, getCanonicalFileName, sourceDirectory }: Info,
+        { moduleResolutionKind, addExtension, getCanonicalFileName, sourceDirectory }: Info,
         compilerOptions: CompilerOptions,
         preferences: ModuleSpecifierPreferences,
     ) {
         const { baseUrl, paths } = compilerOptions;
 
-        const relativePath = removeExtensionAndIndexPostFix(ensurePathIsNonModuleName(getRelativePathFromDirectory(sourceDirectory, moduleFileName, getCanonicalFileName)), moduleResolutionKind, addJsExtension);
+        const relativePath = removeExtensionAndIndexPostFix(ensurePathIsNonModuleName(getRelativePathFromDirectory(sourceDirectory, moduleFileName, getCanonicalFileName)), moduleResolutionKind, addExtension);
         if (!baseUrl || preferences.importModuleSpecifierPreference === "relative") {
             return [relativePath];
         }
@@ -80,7 +80,7 @@ namespace ts.moduleSpecifiers {
             return [relativePath];
         }
 
-        const importRelativeToBaseUrl = removeExtensionAndIndexPostFix(relativeToBaseUrl, moduleResolutionKind, addJsExtension);
+        const importRelativeToBaseUrl = removeExtensionAndIndexPostFix(relativeToBaseUrl, moduleResolutionKind, addExtension);
         if (paths) {
             const fromPaths = tryGetModuleNameFromPaths(removeFileExtension(relativeToBaseUrl), importRelativeToBaseUrl, paths);
             if (fromPaths) {
@@ -130,8 +130,14 @@ namespace ts.moduleSpecifiers {
         return relativeFirst ? [relativePath, importRelativeToBaseUrl] : [importRelativeToBaseUrl, relativePath];
     }
 
-    function usesJsExtensionOnImports({ imports }: SourceFile, preferences: ModuleSpecifierPreferences): boolean {
-        return firstDefined(imports, ({ text }) => pathIsRelative(text) ? fileExtensionIs(text, Extension.Js) : undefined) || !!preferences.includingJsExtensionOnAutoImports;
+    function tryGetFileExtension (text: string) {
+        return pathIsRelative(text) && fileExtensionIsOneOf(text, [Extension.Js, Extension.Jsx]) ? (
+            fileExtensionIs(text, Extension.Js) ? Extension.Js : Extension.Jsx
+        ) : undefined;
+    }
+
+    function usesExtensionOnImports({ imports }: SourceFile, preferences: ModuleSpecifierPreferences): Extension.Js | Extension.Jsx | undefined {
+        return preferences.includeExtensionInImports === undefined ? firstDefined(imports, ({ text }) => tryGetFileExtension(text)) : preferences.includeExtensionInImports;
     }
 
     function discoverProbableSymlinks(files: ReadonlyArray<SourceFile>, getCanonicalFileName: (file: string) => string, host: ModuleSpecifierResolutionHost) {
@@ -245,14 +251,14 @@ namespace ts.moduleSpecifiers {
         host: GetEffectiveTypeRootsHost,
         getCanonicalFileName: (file: string) => string,
         moduleFileName: string,
-        addJsExtension: boolean,
+        addExtension: Extension.Js | Extension.Jsx | undefined,
     ): string | undefined {
         const roots = getEffectiveTypeRoots(options, host);
         return firstDefined(roots, unNormalizedTypeRoot => {
             const typeRoot = toPath(unNormalizedTypeRoot, /*basePath*/ undefined, getCanonicalFileName);
             if (startsWith(moduleFileName, typeRoot)) {
                 // For a type definition, we can strip `/index` even with classic resolution.
-                return removeExtensionAndIndexPostFix(moduleFileName.substring(typeRoot.length + 1), ModuleResolutionKind.NodeJs, addJsExtension);
+                return removeExtensionAndIndexPostFix(moduleFileName.substring(typeRoot.length + 1), ModuleResolutionKind.NodeJs, addExtension);
             }
         });
     }
@@ -397,10 +403,10 @@ namespace ts.moduleSpecifiers {
         });
     }
 
-    function removeExtensionAndIndexPostFix(fileName: string, moduleResolutionKind: ModuleResolutionKind, addJsExtension: boolean): string {
+    function removeExtensionAndIndexPostFix(fileName: string, moduleResolutionKind: ModuleResolutionKind, addExtension: Extension.Js | Extension.Jsx | undefined): string {
         const noExtension = removeFileExtension(fileName);
-        return addJsExtension
-            ? noExtension + ".js"
+        return addExtension
+            ? noExtension + addExtension
             : moduleResolutionKind === ModuleResolutionKind.NodeJs
                 ? removeSuffix(noExtension, "/index")
                 : noExtension;
diff --git a/src/server/protocol.ts b/src/server/protocol.ts
@@ -2733,7 +2733,7 @@ namespace ts.server.protocol {
         readonly includeCompletionsWithInsertText?: boolean;
         readonly importModuleSpecifierPreference?: "relative" | "non-relative";
         readonly allowTextChangesInNewFiles?: boolean;
-        readonly includingJsExtensionOnAutoImports?: boolean;
+        readonly includeExtensionInImports?: Extension.Js | Extension.Jsx;
     }
 
     export interface CompilerOptions {
diff --git a/src/services/types.ts b/src/services/types.ts
@@ -240,7 +240,7 @@ namespace ts {
         readonly includeCompletionsWithInsertText?: boolean;
         readonly importModuleSpecifierPreference?: "relative" | "non-relative";
         readonly allowTextChangesInNewFiles?: boolean;
-        readonly includingJsExtensionOnAutoImports?: boolean;
+        readonly includeExtensionInImports?: Extension.Js | Extension.Jsx;
     }
     /* @internal */
     export const defaultPreferences: UserPreferences = {};
diff --git a/tests/baselines/reference/api/tsserverlibrary.d.ts b/tests/baselines/reference/api/tsserverlibrary.d.ts
@@ -9252,6 +9252,7 @@ declare namespace ts {
 declare namespace ts.moduleSpecifiers {
     interface ModuleSpecifierPreferences {
         importModuleSpecifierPreference?: "relative" | "non-relative";
+        includeExtensionInImports?: Extension.Js | Extension.Jsx;
     }
     function getModuleSpecifier(compilerOptions: CompilerOptions, fromSourceFile: SourceFile, fromSourceFileName: string, toFileName: string, host: ModuleSpecifierResolutionHost, preferences?: ModuleSpecifierPreferences): string;
     function getModuleSpecifiers(moduleSymbol: Symbol, compilerOptions: CompilerOptions, importingSourceFile: SourceFile, host: ModuleSpecifierResolutionHost, files: ReadonlyArray<SourceFile>, preferences: ModuleSpecifierPreferences): ReadonlyArray<ReadonlyArray<string>>;
@@ -9941,6 +9942,7 @@ declare namespace ts {
         readonly includeCompletionsWithInsertText?: boolean;
         readonly importModuleSpecifierPreference?: "relative" | "non-relative";
         readonly allowTextChangesInNewFiles?: boolean;
+        readonly includeExtensionInImports?: Extension.Js | Extension.Jsx;
     }
     const defaultPreferences: UserPreferences;
     interface LanguageService {
@@ -13203,6 +13205,7 @@ declare namespace ts.server.protocol {
         readonly includeCompletionsWithInsertText?: boolean;
         readonly importModuleSpecifierPreference?: "relative" | "non-relative";
         readonly allowTextChangesInNewFiles?: boolean;
+        readonly includeExtensionInImports?: Extension.Js | Extension.Jsx;
     }
     interface CompilerOptions {
         allowJs?: boolean;
diff --git a/tests/baselines/reference/api/typescript.d.ts b/tests/baselines/reference/api/typescript.d.ts
@@ -4731,6 +4731,7 @@ declare namespace ts {
         readonly includeCompletionsWithInsertText?: boolean;
         readonly importModuleSpecifierPreference?: "relative" | "non-relative";
         readonly allowTextChangesInNewFiles?: boolean;
+        readonly includeExtensionInImports?: Extension.Js | Extension.Jsx;
     }
     interface LanguageService {
         cleanupSemanticCache(): void;
diff --git a/tests/cases/fourslash/fourslash.ts b/tests/cases/fourslash/fourslash.ts
@@ -531,7 +531,7 @@ declare namespace FourSlashInterface {
         includeCompletionsForModuleExports?: boolean;
         includeInsertTextCompletions?: boolean;
         importModuleSpecifierPreference?: "relative" | "non-relative";
-        includingJsExtensionOnAutoImports?: boolean
+        includeExtensionInImports?: '.js' | '.jsx'
     }
     interface CompletionsAtOptions extends UserPreferences {
         triggerCharacter?: string;
diff --git a/tests/cases/fourslash/importNameCodeFix_jsExtensionPreference.ts b/tests/cases/fourslash/importNameCodeFix_jsExtensionPreference.ts
@@ -32,6 +32,15 @@
 // @Filename: /i.ts
 ////h;
 
+// @Filename: /j.jsx
+////export function j() {}
+
+// @Filename: /k.js
+////j;
+
+// @Filename: /l.js
+////j;
+
 goTo.file("/b.js");
 verify.importFixAtPosition([
 `import { a } from "./a";
@@ -43,7 +52,7 @@ verify.importFixAtPosition([
 `import { a } from "./a.js";
 
 a;`], /* errorCode */ undefined, {
-    includingJsExtensionOnAutoImports: true
+    includeExtensionInImports: '.js'
 });
 
 goTo.file("/d.ts");
@@ -57,21 +66,37 @@ verify.importFixAtPosition([
 `import { a } from "./a.js";
 
 a;`], /* errorCode */ undefined, {
-    includingJsExtensionOnAutoImports: true
+    includeExtensionInImports: '.js'
 });
 
 goTo.file("/g.ts");
 verify.importFixAtPosition([
 `import { f } from "./f.js";
 
 f;`], /* errorCode */ undefined, {
-    includingJsExtensionOnAutoImports: true
+    includeExtensionInImports: '.js'
 });
 
 goTo.file("/i.ts");
 verify.importFixAtPosition([
 `import h from "./h.js";
 
 h;`], /* errorCode */ undefined, {
-    includingJsExtensionOnAutoImports: true
+    includeExtensionInImports: '.js'
+});
+
+goTo.file("/k.js");
+verify.importFixAtPosition([
+`import { j } from "./j.js";
+
+j;`], /* errorCode */ undefined, {
+    includeExtensionInImports: '.js'
+});
+
+goTo.file("/l.js");
+verify.importFixAtPosition([
+`import { j } from "./j.jsx";
+
+j;`], /* errorCode */ undefined, {
+    includeExtensionInImports: '.jsx'
 });

Original file line number	Diff line number	Diff line change
`@@ -2733,7 +2733,7 @@ namespace ts.server.protocol {`
`2733`	`2733`	`readonly includeCompletionsWithInsertText?: boolean;`
`2734`	`2734`	`readonly importModuleSpecifierPreference?: "relative" \| "non-relative";`
`2735`	`2735`	`readonly allowTextChangesInNewFiles?: boolean;`
`2736`		`- readonly includingJsExtensionOnAutoImports?: boolean;`
	`2736`	`+ readonly includeExtensionInImports?: Extension.Js \| Extension.Jsx;`
`2737`	`2737`	`}`
`2738`	`2738`
`2739`	`2739`	`export interface CompilerOptions {`
Original file line number	Diff line number	Diff line change
`@@ -240,7 +240,7 @@ namespace ts {`
`240`	`240`	`readonly includeCompletionsWithInsertText?: boolean;`
`241`	`241`	`readonly importModuleSpecifierPreference?: "relative" \| "non-relative";`
`242`	`242`	`readonly allowTextChangesInNewFiles?: boolean;`
`243`		`- readonly includingJsExtensionOnAutoImports?: boolean;`
	`243`	`+ readonly includeExtensionInImports?: Extension.Js \| Extension.Jsx;`
`244`	`244`	`}`
`245`	`245`	`/* @internal */`
`246`	`246`	`export const defaultPreferences: UserPreferences = {};`
Original file line number	Diff line number	Diff line change
`@@ -4731,6 +4731,7 @@ declare namespace ts {`
`4731`	`4731`	`readonly includeCompletionsWithInsertText?: boolean;`
`4732`	`4732`	`readonly importModuleSpecifierPreference?: "relative" \| "non-relative";`
`4733`	`4733`	`readonly allowTextChangesInNewFiles?: boolean;`
	`4734`	`+ readonly includeExtensionInImports?: Extension.Js \| Extension.Jsx;`
`4734`	`4735`	`}`
`4735`	`4736`	`interface LanguageService {`
`4736`	`4737`	`cleanupSemanticCache(): void;`
Original file line number	Diff line number	Diff line change
`@@ -531,7 +531,7 @@ declare namespace FourSlashInterface {`
`531`	`531`	`includeCompletionsForModuleExports?: boolean;`
`532`	`532`	`includeInsertTextCompletions?: boolean;`
`533`	`533`	`importModuleSpecifierPreference?: "relative" \| "non-relative";`
`534`		`- includingJsExtensionOnAutoImports?: boolean`
	`534`	`+ includeExtensionInImports?: '.js' \| '.jsx'`
`535`	`535`	`}`
`536`	`536`	`interface CompletionsAtOptions extends UserPreferences {`
`537`	`537`	`triggerCharacter?: string;`