feat(core-component): add option to create barrel file for component

Author: vs-cristian

src/core/FileGenerateManager.ts

@@ -1,5 +1,6 @@
 import { config } from '@/config';
 import { IComponentVariables, IHOCVariables, IHookVariables } from '@/types';
+import { BarrelTemplate } from '@/core/temlpates/components/BarrelTemplate';
 import * as utils from '../utils';
 import { FileService } from './FileService';
 import { TemplateGenerator } from './TemplateGenerator';
@@ -51,6 +52,11 @@ export class FileGenerateManager {
       fileService.genTest(testTemplate);
     }
 
+    if (variables.barrel) {
+      const barrelTemplate = templateGenerator.generateTemplate(BarrelTemplate);
+      fileService.genBarrel(barrelTemplate);
+    }
+
     const styleTemplate = config.cssModules ? '.root {}' : '';
     fileService.genStyle(styleTemplate);
   }

src/core/FileService.ts

@@ -37,6 +37,11 @@ export class FileService {
     fs.mkdirpSync(this.dirPath);
   }
 
+  genBarrel(template: string) {
+    const filePath = `${this.dirPath}/index.${config.ext.component}`;
+    this.genFile(filePath, template, 'barrel');
+  }
+
   genJs(template: string) {
     const filePath = this.getFilePath(config.ext.component);
     this.genFile(filePath, template, 'component');

src/core/temlpates/components/BarrelTemplate.ts

@@ -0,0 +1,30 @@
+import * as t from '@babel/types';
+import { TemplateBase } from '@/core/TemplateBase';
+import { Template } from '@/core/TemplateGenerator';
+import { config } from '@/config';
+import * as c from '../shared';
+
+export class BarrelTemplate extends TemplateBase implements Template {
+  generateAST(): t.File {
+    const body: t.Statement[] = [];
+
+    body.push(
+      t.exportNamedDeclaration(
+        null,
+        [
+          t.exportSpecifier(
+            t.identifier(
+              config.exportType === 'named'
+                ? this.vars.componentName
+                : 'default'
+            ),
+            t.identifier(this.vars.componentName)
+          ),
+        ],
+        t.stringLiteral(`./${this.vars.fileName}`)
+      )
+    );
+
+    return c.program(body);
+  }
+}

src/questions/questionTypes.ts

@@ -18,17 +18,18 @@ export function getComponentQuestions() {
     mods.push('propTypes');
   }
 
-  const val = [
+  const questionList = [
     questions.name('Component'),
     questions.test(),
+    questions.barrel(),
     questions.hooks(reactHooks),
   ];
 
   if (mods.length) {
-    val.push(questions.mods(mods));
+    questionList.push(questions.mods(mods));
   }
 
-  return val;
+  return questionList;
 }
 
 export function getHOCQuestions() {

src/questions/questions.ts

@@ -25,6 +25,12 @@ export const name = (genType: string, endStr = ':') => ({
   validate: (str: string) => str.length > 0,
 });
 
+export const barrel = () => ({
+  type: 'confirm',
+  name: 'barrel',
+  message: 'Create a barrel file?',
+});
+
 export const test = () => ({
   type: 'confirm',
   name: 'test',

src/types/types.ts

@@ -35,6 +35,7 @@ export interface IComponentVariables extends IBaseVariables {
   type: 'component';
   name: string;
   test: boolean;
+  barrel: boolean;
   hooks: ReactHooks;
   mods: string[];
 }