Feat#Provide a quickfix for non-exported types

src/compiler/diagnosticMessages.json

@@ -5185,6 +5185,14 @@
         "category": "Message",
         "code": 90053
     },
+    "Export '{0}' from module '{1}'": {
+        "category": "Message",
+        "code": 90054
+    },
+    "Export all non-exported member": {
+        "category": "Message",
+        "code": 90055
+    },
     "Convert function to an ES2015 class": {
         "category": "Message",
         "code": 95001

src/services/codefixes/fixImportNonExportedMember.ts

@@ -0,0 +1,62 @@
+/* @internal */
+namespace ts.codefix {
+    const fixId = "importNonExportedMember";
+    const errorCodes = [
+        Diagnostics.Module_0_declares_1_locally_but_it_is_not_exported.code
+    ];
+    registerCodeFix({
+        errorCodes,
+        getCodeActions(context) {
+            const { sourceFile } = context;
+            const info = getInfo(sourceFile, context, context.span.start);
+            if (!info) {
+                return undefined;
+            }
+            const changes = textChanges.ChangeTracker.with(context, t => doChange(t, info.originSourceFile, info.node));
+            return [createCodeFixAction(fixId, changes, [Diagnostics.Export_0_from_module_1, info.node.text, showModuleSpecifier(info.importDecl)], fixId, Diagnostics.Export_all_non_exported_member)];
+        },
+        fixIds: [fixId],
+        getAllCodeActions: context => codeFixAll(context, errorCodes, (changes, diag) => {
+            const info = getInfo(diag.file, context, diag.start);
+            if (info) doChange(changes, info.originSourceFile, info.node);
+        }),
+    });
+
+    interface Info {
+        readonly node: Identifier;
+        readonly importDecl: ImportDeclaration;
+        readonly originSourceFile: SourceFile
+    }
+
+    function getInfo(sourceFile: SourceFile, context: CodeFixContext | CodeFixAllContext, pos: number): Info | undefined {
+        const node = getTokenAtPosition(sourceFile, pos);
+        if (node && isIdentifier(node)) {
+            const importStart = getLineStartPositionForPosition(pos, sourceFile);
+            const importDecl = tryGetImportFrom(getTokenAtPosition(sourceFile, importStart));
+            if (!importDecl || !isStringLiteralLike(importDecl.moduleSpecifier)) {
+                return undefined;
+            }
+            const resolvedModule = getResolvedModule(sourceFile, importDecl.moduleSpecifier.text);
+            const originSourceFile = resolvedModule && context.program.getSourceFile(resolvedModule.resolvedFileName);
+            if (!originSourceFile) {
+                return undefined;
+            }
+            return { node, importDecl, originSourceFile };
+        }
+    }
+
+    function tryGetImportFrom(token: Node): ImportDeclaration | undefined {
+        return token.kind === SyntaxKind.ImportKeyword ? tryCast(token.parent, isImportDeclaration) : undefined;
+    }
+
+    function doChange(changes: textChanges.ChangeTracker, sourceFile: SourceFile, node: Identifier): void {
+        const exportSpecifier = createExportSpecifier(/*propertyName*/ undefined, node);
+        const exportDeclaration = createExportDeclaration(
+            /*decorators*/ undefined,
+            /*modifiers*/ undefined,
+            createNamedExports([exportSpecifier]),
+            /*moduleSpecifier*/ undefined,
+            /*isTypeOnly*/ false);
+        changes.insertNodeAtEndOfScope(sourceFile, sourceFile, exportDeclaration);
+    }
+}

src/services/tsconfig.json

@@ -77,6 +77,7 @@
         "codefixes/fixModuleAndTargetOptions.ts",
         "codefixes/fixExtendsInterfaceBecomesImplements.ts",
         "codefixes/fixForgottenThisPropertyAccess.ts",
+        "codefixes/fixImportNonExportedMember.ts",
         "codefixes/fixInvalidJsxCharacters.ts",
         "codefixes/fixUnusedIdentifier.ts",
         "codefixes/fixUnreachableCode.ts",

tests/cases/fourslash/codeFixImportNonExportedMember.ts

@@ -0,0 +1,32 @@
+/// <reference path='fourslash.ts' />
+// @Filename: /a.ts
+////declare function zoo(): any;
+////export { zoo };
+
+// @Filename: /b.ts
+////declare function foo(): any;
+////declare function bar(): any;
+////export { foo };
+
+// @Filename: /c.ts
+////import { zoo } from "./a";
+////import { bar } from "./b";
+
+goTo.file("/c.ts");
+verify.codeFixAvailable([
+    { description: `Export 'bar' from module './b'` },
+    { description: `Remove import from './a'` },
+    { description: `Remove import from './b'` },
+]);
+verify.codeFix({
+    index: 0,
+    description: `Export 'bar' from module './b'`,
+    newFileContent: {
+        '/b.ts': `declare function foo(): any;
+declare function bar(): any;
+export { foo };
+
+export { bar };
+`
+    }
+});

tests/cases/fourslash/codeFixImportNonExportedMember_all.ts

@@ -0,0 +1,28 @@
+/// <reference path='fourslash.ts' />
+
+// @Filename: /a.ts
+////declare function foo(): any;
+////declare function bar(): any;
+////declare function zoo(): any;
+////export { zoo }
+
+// @Filename: /b.ts
+////import { foo, bar } from "./a";
+
+goTo.file("/b.ts");
+verify.codeFixAll({
+    fixId: "importNonExportedMember",
+    fixAllDescription: "Export all non-exported member",
+    newFileContent: {
+        '/a.ts': `declare function foo(): any;
+declare function bar(): any;
+declare function zoo(): any;
+export { zoo }
+
+export { foo };
+
+
+export { bar };
+`
+    }
+});