From 765114fccd08440a2676da74ee76229f571d8330 Mon Sep 17 00:00:00 2001
From: Andy Hanson <anhans@microsoft.com>
Date: Fri, 13 Jan 2017 07:57:01 -0800
Subject: [PATCH] Refactor to move code into checker

---
 src/compiler/checker.ts     | 11 ++++++++++-
 src/compiler/types.ts       |  3 ++-
 src/services/completions.ts | 18 ++++++------------
 3 files changed, 18 insertions(+), 14 deletions(-)

diff --git a/src/compiler/checker.ts b/src/compiler/checker.ts
index 46a381a561e..a5d1e439879 100644
--- a/src/compiler/checker.ts
+++ b/src/compiler/checker.ts
@@ -108,7 +108,7 @@ namespace ts {
             getAliasedSymbol: resolveAlias,
             getEmitResolver,
             getExportsOfModule: getExportsOfModuleAsArray,
-            resolveExternalModuleSymbol,
+            getExportsAndPropertiesOfModule,
             getAmbientModules,
             getJsxElementAttributesType,
             getJsxIntrinsicTagNames,
@@ -1528,6 +1528,15 @@ namespace ts {
             return symbolsToArray(getExportsOfModule(moduleSymbol));
         }
 
+        function getExportsAndPropertiesOfModule(moduleSymbol: Symbol): Symbol[] {
+            const exports = getExportsOfModuleAsArray(moduleSymbol);
+            const exportEquals = resolveExternalModuleSymbol(moduleSymbol);
+            if (exportEquals !== moduleSymbol) {
+                addRange(exports, getPropertiesOfType(getTypeOfSymbol(exportEquals)));
+            }
+            return exports;
+        }
+
         function tryGetMemberInModuleExports(memberName: string, moduleSymbol: Symbol): Symbol | undefined {
             const symbolTable = getExportsOfModule(moduleSymbol);
             if (symbolTable) {
diff --git a/src/compiler/types.ts b/src/compiler/types.ts
index 8b3b821a25b..0906488ea33 100644
--- a/src/compiler/types.ts
+++ b/src/compiler/types.ts
@@ -2370,7 +2370,8 @@ namespace ts {
         isValidPropertyAccess(node: PropertyAccessExpression | QualifiedName, propertyName: string): boolean;
         getAliasedSymbol(symbol: Symbol): Symbol;
         getExportsOfModule(moduleSymbol: Symbol): Symbol[];
-        /* @internal */ resolveExternalModuleSymbol(moduleSymbol: Symbol): Symbol;
+        /** Unlike `getExportsOfModule`, this includes properties of an `export =` value. */
+        /* @internal */ getExportsAndPropertiesOfModule(moduleSymbol: Symbol): Symbol[];
 
         getJsxElementAttributesType(elementNode: JsxOpeningLikeElement): Type;
         getJsxIntrinsicTagNames(): Symbol[];
diff --git a/src/services/completions.ts b/src/services/completions.ts
index 27cd78bbe17..bb5922af9ad 100644
--- a/src/services/completions.ts
+++ b/src/services/completions.ts
@@ -1318,20 +1318,14 @@ namespace ts.Completions {
             isMemberCompletion = true;
             isNewIdentifierLocation = false;
 
-            let exports: Symbol[];
-            const moduleSpecifierSymbol = typeChecker.getSymbolAtLocation(importOrExportDeclaration.moduleSpecifier);
-            if (moduleSpecifierSymbol) {
-                exports = typeChecker.getExportsOfModule(moduleSpecifierSymbol);
-                const exportEquals = typeChecker.resolveExternalModuleSymbol(moduleSpecifierSymbol);
-                if (exportEquals !== moduleSpecifierSymbol) {
-                    // Location doesn't matter so long as it's not an identifier.
-                    const exportEqualsType = typeChecker.getTypeOfSymbolAtLocation(exportEquals, moduleSpecifier);
-                    exports = ts.concatenate(exports, typeChecker.getPropertiesOfType(exportEqualsType));
-                }
+            const moduleSpecifierSymbol = typeChecker.getSymbolAtLocation(moduleSpecifier);
+            if (!moduleSpecifierSymbol) {
+                symbols = emptyArray;
+                return true;
             }
 
-            symbols = exports ? filterNamedImportOrExportCompletionItems(exports, namedImportsOrExports.elements) : emptyArray;
-
+            const exports = typeChecker.getExportsAndPropertiesOfModule(moduleSpecifierSymbol);
+            symbols = filterNamedImportOrExportCompletionItems(exports, namedImportsOrExports.elements);
             return true;
         }