fix: missing related entity imports

dzhelezov · Jul 2, 2020 · c879782 · c879782
1 parent d45ab9d
commit c879782
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/substrate-query-framework/cli/src/generate/ModelRenderer.ts b/substrate-query-framework/cli/src/generate/ModelRenderer.ts
@@ -99,12 +99,13 @@ export class ModelRenderer extends AbstractRenderer {
   withImportProps(): GeneratorContext {
     const relatedEntityImports: string[] = [];
     this.objType.relatedEntityImports.forEach(entityName => {
-      const import_ = path.join(
-        `import { ${entityName} } from  '..`,
-        utils.kebabCase(entityName),
-        `${utils.kebabCase(entityName)}.model'`
+      relatedEntityImports.push(
+        path.join(
+          `import { ${entityName} } from  '..`,
+          utils.kebabCase(entityName),
+          `${utils.kebabCase(entityName)}.model'`
+        )
       );
-      relatedEntityImports.push(import_);
     });
     return {
       relatedEntityImports,

diff --git a/substrate-query-framework/cli/src/templates/entities/model.ts.mst b/substrate-query-framework/cli/src/templates/entities/model.ts.mst
@@ -23,8 +23,9 @@ import { InterfaceType } from 'type-graphql';
 
 {{#has.mtm}}import { JoinTable } from 'typeorm';{{/has.mtm}}
 
+
 {{#relatedEntityImports}}
-  {{{ entityName }}}
+  {{{ . }}}
 {{/relatedEntityImports}}
 
 {{#enums}}