From d42986efefc661dfa4f9d3090be17eb2d406829d Mon Sep 17 00:00:00 2001
From: henriquecesp <henriquecesps4@gmail.com>
Date: Sun, 24 Oct 2021 15:29:06 -0300
Subject: [PATCH] feat(app): implements react choices on cli

---
 app/index.js | 25 ++++++++++++++++++++++++-
 1 file changed, 24 insertions(+), 1 deletion(-)

diff --git a/app/index.js b/app/index.js
index c36fb83..d0310c7 100644
--- a/app/index.js
+++ b/app/index.js
@@ -45,13 +45,36 @@ class SecGenerator extends Generator {
           { name: 'NestJS TypeORM', value: 'nestjsTypeOrm' },
           { name: 'NestJS Mongoose', value: 'nestjsMongoose' },
           { name: 'NestJS PrismaORM', value: 'nestjsPrismaOrm' },
+          { name: 'ReactJS', value: 'reactjs' },
+        ],
+      },
+      {
+        when: answers => answers.framework === 'reactjs',
+        type: 'list',
+        name: 'reactjs',
+        message: 'Select your component type:',
+        choices: [
+          { name: 'ReactJS Component', value: 'reactjsComponent' },
+          { name: 'ReactJS Material-UI', value: 'reactjsMui' },
+          { name: 'ReactJS StyledComponents', value: 'reactjsStyled' },
+          { name: 'ReactJS ContextAPI', value: 'reactjsContext' },
+          { name: 'ReactJS Hook', value: 'reactjsHook' },
         ],
       },
     ])
   }
 
+  frameworkChoice(framework) {
+    switch (framework) {
+      case 'reactjs':
+        return this.answers.reactjs
+      default:
+        return this.answers.framework
+    }
+  }
+
   async writing() {
-    const framework = this.answers.framework
+    const framework = this.frameworkChoice(this.answers.framework)
     const dir = path.join(__dirname, `templates/${framework}/`)
 
     for await (const f of makeFileTemplate(dir, this.variables.name)) {