Add "wrap" argument

Author: vs-cristian

src/bin.ts

@@ -24,16 +24,20 @@ yargs
   .command({
     command: '*',
     handler: (args: IArgs) => {
-      const { directory } = args;
+      const { directory, wrap } = args;
 
-      runGenerator({ directory })
+      runGenerator({ directory, wrap })
         .then(() => process.exit(0))
         .catch(err => {
           Logger.error(err);
           process.exit(1);
         });
     },
   })
+  .option('wrap', {
+    type: 'boolean',
+    description: 'Wrap generated files in folder',
+  })
   .option('directory', {
     alias: 'd',
     type: 'string',

src/index.ts

@@ -8,11 +8,12 @@ import { FileGenerateManager } from './core/FileGenerateManager';
 import { CONFIG } from './constants';
 
 export interface IArgs {
+  wrap?: boolean;
   directory?: string;
 }
 
 export async function runGenerator(args: IArgs) {
-  const { directory } = args;
+  const { directory, wrap } = args;
   const { type } = await inquirer.prompt(typeQuestion as any);
 
   process.stdout.write('\n');
@@ -21,6 +22,10 @@ export async function runGenerator(args: IArgs) {
     CONFIG.path = path.resolve(process.cwd(), directory);
   }
 
+  if (wrap != null) {
+    CONFIG.wrapFolder = wrap;
+  }
+
   switch (type) {
     case 'component': {
       const answers = await inquirer.prompt(questionTypes.getComponentQuestions());