Add support for flow type imports

src/findExports.ts

@@ -239,6 +239,7 @@ function findNamedExports(
     let defaultName = null;
     nodes.forEach((node) => {
         result.push(...findESNamedExports(node));
+        result.push(...findNamedTypeExports(node).map(name => `type ${name}`));
         const named = findCommonJSExports(node, {
             definedNames,
             absolutePathToFile,
@@ -332,6 +333,13 @@ function findRootNodes(ast) {
     return realRootNodes;
 }
 
+function findNamedTypeExports(node) {
+    if (node.type !== 'ExportNamedDeclaration' || !node.declaration || node.declaration.type !== 'TypeAlias') {
+        return [];
+    }
+    return [node.declaration.id.name];
+}
+
 export default function findExports(data, absolutePathToFile) {
     if (/\.json$/.test(absolutePathToFile)) {
         return {

src/importCompletion.ts

@@ -34,16 +34,22 @@ export class ImportCompletion implements vscode.CompletionItemProvider {
                 const handlers = [];
                 items.forEach(item => {
                     handlers.push({
-                        label: item.importObj.module.name,
+                        label: item.importObj.module.name.replace('type ', ''),
                         kind: vscode.CompletionItemKind.Reference,
                         detail: item.label,
                         documentation: '',
-                        command: autofix ? { title: 'Autocomplete', command: 'extension.fixImport', arguments: [item.importObj, item.doc, item.range] } : null
+                        command: autofix
+                            ? {
+                                  title: 'Autocomplete',
+                                  command: 'extension.fixImport',
+                                  arguments: [item.importObj, item.doc, item.range]
+                              }
+                            : null
                     });
                 });
                 return resolve(handlers);
             }
             return resolve([]);
-        })
+        });
     }
 }

src/interpreter.ts

@@ -24,7 +24,7 @@ export default class Interpreter {
         exports.default\\s*=\\s*(\\w+).default
         |module.exports\\s*=\\s*(\\w+)
         |exports\\[[\\'\\"]default[\\'\\"]\\]\\s*=\\s*(\\w+)
-        |export\\s+(default\\s+){0,1}(?:const|let|var|interface|enum|async\\s+function|function|function\\*|class|abstract\\sclass)\\s+([\\w]+)
+        |export\\s+(default\\s+){0,1}(?:const|let|var|interface|type|enum|async\\s+function|function|function\\*|class|abstract\\sclass)\\s+([\\w]+)
         |exports\\.([\\w]+)\\s*=
         |exports\\[\\"([\\w]+)\\"\\]\\s*=
         |Object.defineProperty\\(\\s*exports\\s*,\\s*[\\'|\\"]([\\w]+)[\\'|\\"]
@@ -72,14 +72,18 @@ export default class Interpreter {
             if (res[4] != null) {
                 resultList.push({
                     default: true,
-                    name: res[5],
-                })
+                    name: res[0] && res[0].startsWith('export default type') ? `type ${res[5]}` : res[5]
+                });
                 continue;
             }
-            for (i = 5; i <= 8; i+=1) {
+            for (i = 5; i <= 8; i += 1) {
                 if (res[i] != null) {
                     if (!this.isUnwantedName(res[i]) && !nameList.includes(res[i])) {
-                        nameList.push(res[i]);
+                        if (res[0] && res[0].startsWith('export type')) {
+                            nameList.push(`type ${res[i]}`);
+                        } else {
+                            nameList.push(res[i]);
+                        }
                     }
                     break;
                 }

src/rootScanner.ts

@@ -25,6 +25,7 @@ export interface ImportObj {
          * should add member to import statement like import 'file.less'
          */
         isNotMember?: boolean;
+        type?: boolean;
     };
     isNodeModule: boolean;
 }
@@ -224,11 +225,13 @@ export default class RootScanner {
     }
 
     private cacheModulesFromMain(moduleName, modulePath, packageJson) {
-        if (!packageJson.hasOwnProperty('main'))
-            return;
-        let mainFilePath = path.join(modulePath, packageJson.main);
+        let mainFilePath = path.join(modulePath, 'src/index.js');
         if (!fs.existsSync(mainFilePath)) {
-            mainFilePath += '.js';
+            if (!packageJson.hasOwnProperty('main')) return;
+            mainFilePath = path.join(modulePath, packageJson.main);
+            if (!fs.existsSync(mainFilePath)) {
+                mainFilePath += '.js';
+            }
         }
         if (fs.existsSync(mainFilePath)) {
             fs.readFile(mainFilePath, 'utf-8', (err, data) => {

test/importFixer.test.ts

@@ -92,7 +92,6 @@ suite("extractImportFromRoot", () => {
                     isPlainFile: true
                 }
             });
-        console.log(obj)
         assert.equal('./index.less', importFixer.extractImportFromRoot(obj, path.join(vscode.workspace.rootPath, 'src/app.js')))
     });