Read import and another declarations in the svelte code

Author: andrelmlins

package.json

@@ -29,11 +29,13 @@
   ],
   "dependencies": {
     "commander": "^7.1.0",
-    "recursive-readdir": "^2.2.2"
+    "recursive-readdir": "^2.2.2",
+    "tmp": "^0.2.1"
   },
   "devDependencies": {
     "@types/estree": "^0.0.46",
     "@types/node": "^14.14.30",
+    "@types/tmp": "^0.2.0",
     "estree-walker": "^3.0.0",
     "locate-character": "^2.0.5",
     "magic-string": "^0.25.7",

src/transformer/svelte.ts

@@ -2,6 +2,7 @@ import * as ts from 'typescript';
 import path from 'path';
 import { promises as fs } from 'fs';
 import { Ast, TemplateNode } from 'svelte/types/compiler/interfaces';
+import tmp from 'tmp';
 import { Prop, Event, SlotProp } from '../types';
 import ITransformer from './transformer';
 
@@ -16,6 +17,9 @@ class SvelteTransformer implements ITransformer {
   private subdir: string;
   private moduleName: string;
   private isDefault: boolean;
+  private typesForSearch: ts.TypeReferenceNode[];
+  private declarationNode: string[];
+  private declarationImport: string[];
 
   constructor(content: string, fileName: string, ast: Ast, dir: string, moduleName: string, isDefault: boolean) {
     this.sourceFile = ts.createSourceFile(fileName, content, ts.ScriptTarget.Latest);
@@ -28,6 +32,9 @@ class SvelteTransformer implements ITransformer {
     this.subdir = path.dirname(this.fileName).replace(this.dir, '');
     this.moduleName = moduleName;
     this.isDefault = isDefault;
+    this.typesForSearch = [];
+    this.declarationNode = [];
+    this.declarationImport = [];
   }
 
   private containExportModifier = (node: ts.VariableStatement): boolean => {
@@ -58,6 +65,10 @@ class SvelteTransformer implements ITransformer {
       if (declaration.type) {
         type = declaration.type.getText(this.sourceFile);
 
+        if (ts.isTypeReferenceNode(declaration.type)) {
+          this.typesForSearch.push(declaration.type);
+        }
+
         if (ts.isUnionTypeNode(declaration.type)) {
           const nameValidTypes = declaration.type.types.reduce((acc, type) => {
             if (type.kind === ts.SyntaxKind.NullKeyword || type.kind === ts.SyntaxKind.UndefinedKeyword) {
@@ -109,6 +120,21 @@ class SvelteTransformer implements ITransformer {
     }
   }
 
+  private verifyImportDeclaration(node: ts.ImportDeclaration, name: string): void {
+    if (node.importClause && node.importClause.namedBindings && ts.isNamedImports(node.importClause.namedBindings)) {
+      const elements = node.importClause.namedBindings.elements;
+      const newElements = elements.filter((element) => element.name.getText(this.sourceFile) === name);
+
+      if (newElements.length > 0) {
+        const importString = newElements.map((item) => item.name.getText(this.sourceFile)).join(', ');
+
+        this.declarationImport.push(
+          `import { ${importString} } from ${node.moduleSpecifier.getText(this.sourceFile)};`
+        );
+      }
+    }
+  }
+
   exec(): void {
     ts.forEachChild(this.sourceFile, (node: ts.Node) => {
       if (ts.isVariableStatement(node)) {
@@ -120,10 +146,47 @@ class SvelteTransformer implements ITransformer {
       }
     });
 
+    this.typesForSearch.forEach((item) => {
+      const name = item.typeName.getText(this.sourceFile);
+      ts.forEachChild(this.sourceFile, (node: ts.Node) => {
+        if (ts.isInterfaceDeclaration(node) || ts.isClassDeclaration(node) || ts.isTypeAliasDeclaration(node)) {
+          if (node.name?.getText(this.sourceFile) === name) {
+            this.declarationNode.push(node.getText(this.sourceFile));
+          }
+        } else if (ts.isImportDeclaration(node)) {
+          this.verifyImportDeclaration(node, name);
+        }
+      });
+    });
+
     this.execSlotProperty(this.ast.html);
   }
 
-  toString(): string {
+  private async toStringDeclarations(): Promise<string> {
+    const tempFile = tmp.fileSync({ postfix: '.ts' });
+    const content = this.declarationNode.reduce((acc, item) => `${acc}${item}\n\n`, '');
+    let declaration = '';
+
+    await fs.writeFile(tempFile.name, content);
+
+    const options = { declaration: true, emitDeclarationOnly: true };
+    const host = ts.createCompilerHost(options);
+    host.writeFile = (_, contents: string) => (declaration = contents);
+    const program = ts.createProgram([tempFile.name], options, host);
+    program.emit();
+
+    tempFile.removeCallback();
+
+    declaration = declaration
+      .replace(/declare /g, '')
+      .split('\n')
+      .map((item) => `\t${item}`)
+      .join('\n');
+
+    return `${declaration}\n`;
+  }
+
+  async toString(): Promise<string> {
     const pathParse = path.parse(this.fileName);
     const propsString = this.props.reduce(
       (acc, prop) => `${acc}\n\t\t${prop.name}${prop.isOptional ? '?' : ''}: ${prop.type};`,
@@ -138,6 +201,15 @@ class SvelteTransformer implements ITransformer {
       string = `declare module '${this.moduleName}' {\n`;
     }
 
+    if (this.declarationImport.length > 0) {
+      string += this.declarationImport.reduce((acc, item) => `${acc}\t${item}\n`, '');
+      string += '\n';
+    }
+
+    if (this.declarationNode.length > 0) {
+      string += await this.toStringDeclarations();
+    }
+
     string += `\tinterface ${pathParse.name}Props {${propsString}\n\t}\n\n`;
     string += `\tclass ${pathParse.name} extends SvelteComponentTyped<\n`;
     string += `\t\t${pathParse.name}Props,\n\t\t{ ${eventsString} },\n\t\t{ ${slotPropsString} }\n\t> {}`;
@@ -148,7 +220,7 @@ class SvelteTransformer implements ITransformer {
 
   async appendFile(path: string): Promise<void> {
     this.exec();
-    await fs.appendFile(path, this.toString());
+    await fs.appendFile(path, await this.toString());
   }
 }
 

src/transformer/transformer.ts

@@ -1,6 +1,6 @@
 interface ITransformer {
   exec(): void;
-  toString(): string;
+  toString(): string | Promise<string>;
   appendFile(path: string): Promise<void>;
 }
 

src/transformer/typescript.ts

@@ -39,6 +39,7 @@ class TypescriptTransformer implements ITransformer {
     }
 
     string += this.declaration
+      .replace(/declare /g, '')
       .split('\n')
       .map((item) => (item !== '' ? `\t${item}` : undefined))
       .filter((item) => !!item)

yarn.lock

@@ -21,6 +21,11 @@
   resolved "https://registry.yarnpkg.com/@types/node/-/node-14.14.30.tgz#7d5162eec085ba34f8cb9011e9ba12119f76f961"
   integrity sha512-gUWhy8s45fQp4PqqKecsnOkdW0kt1IaKjgOIR3HPokkzTmQj9ji2wWFID5THu1MKrtO+d4s2lVrlEhXUsPXSvg==
 
+"@types/tmp@^0.2.0":
+  version "0.2.0"
+  resolved "https://registry.yarnpkg.com/@types/tmp/-/tmp-0.2.0.tgz#e3f52b4d7397eaa9193592ef3fdd44dc0af4298c"
+  integrity sha512-flgpHJjntpBAdJD43ShRosQvNC0ME97DCfGvZEDlAThQmnerRXrLbX6YgzRBQCZTthET9eAWFAMaYP0m0Y4HzQ==
+
 ansi-styles@^3.2.1:
   version "3.2.1"
   resolved "https://registry.yarnpkg.com/ansi-styles/-/ansi-styles-3.2.1.tgz#41fbb20243e50b12be0f04b8dedbf07520ce841d"
@@ -616,7 +621,7 @@ [email protected]:
   resolved "https://registry.yarnpkg.com/text-hex/-/text-hex-1.0.0.tgz#69dc9c1b17446ee79a92bf5b884bb4b9127506f5"
   integrity sha512-uuVGNWzgJ4yhRaNSiubPY7OjISw4sw4E5Uv0wbjp+OzcbmVU/rsT8ujgcXJhn9ypzsgr5vlzpPqP+MBBKcGvbg==
 
-[email protected]:
+[email protected], tmp@^0.2.1:
   version "0.2.1"
   resolved "https://registry.yarnpkg.com/tmp/-/tmp-0.2.1.tgz#8457fc3037dcf4719c251367a1af6500ee1ccf14"
   integrity sha512-76SUhtfqR2Ijn+xllcI5P1oyannHNHByD80W1q447gU3mp9G9PSpGdWmjUOHRDPiHYacIk66W7ubDTuPF3BEtQ==