Fred/Generator Fix

Give choice to add ThemeProvider and headers
calblueprint · Dec 3, 2019 · af511ba · af511ba
1 parent f3625d3
commit af511ba
Show file tree

Hide file tree

Showing 4 changed files with 56 additions and 5 deletions.
diff --git a/internals/generators/class/index.js b/internals/generators/class/index.js
@@ -34,6 +34,12 @@ module.exports = {
       default: true,
       message: 'Do you want styles?',
     },
+    {
+      type: 'confirm',
+      name: 'wantTheme',
+      default: false,
+      message: 'Do you want to wrap your component with ThemeProvider?',
+    },
     {
       type: 'confirm',
       name: 'wantLoadable',

diff --git a/internals/generators/class/index.js.hbs b/internals/generators/class/index.js.hbs
@@ -7,25 +7,34 @@
 import React from 'react';
 {{#if wantStyles}}
 import PropTypes from 'prop-types';
-import theme from 'utils/theme';
 {{else}}
 // import PropTypes from 'prop-types';
 {{/if}}
 {{#if wantHeaders}}
 import { Helmet } from 'react-helmet';
 {{/if}}
 {{#if wantStyles}}
-import { withStyles, ThemeProvider } from '@material-ui/core/styles';
-import styles from './styles';
+import { withStyles, {{#if wantTheme }}ThemeProvider{{/if}} } from '@material-ui/core/styles';
 {{else}}
+{{#if wantTheme}}
 import { ThemeProvider } from '@material-ui/core/styles';
 {{/if}}
+{{/if}}
+{{#if wantTheme}}
+import theme from 'utils/theme';
+{{/if}}
+
+{{#if wantStyles}}
+import styles from './styles';
+{{/if}}
 
 class {{ properCase name }} extends React.Component {
 render() {
 {{#if wantStyles}}const { classes } = this.props;{{/if}}
 return (
+{{#if wantTheme}}
 <ThemeProvider theme={theme}>
+{{/if}}
 <div{{#if wantStyles}} className={ classes.root }{{/if}}>
   {{#if wantHeaders}}
   <Helmet>
@@ -34,7 +43,9 @@ return (
   </Helmet>
   {{/if}}
   </div>
+{{#if wantTheme}}
   </ThemeProvider>
+{{/if}}
   );
 }
 }

diff --git a/internals/generators/functional/index.js b/internals/generators/functional/index.js
@@ -26,6 +26,12 @@ module.exports = {
         return 'The name is required';
       },
     },
+    {
+      type: 'confirm',
+      name: 'wantHeaders',
+      default: false,
+      message: 'Do you want headers?',
+    },
     {
       type: 'confirm',
       name: 'memo',
@@ -38,6 +44,12 @@ module.exports = {
       default: true,
       message: 'Do you want styles?',
     },
+    {
+      type: 'confirm',
+      name: 'wantTheme',
+      default: false,
+      message: 'Do you want to wrap your component with ThemeProvider?',
+    },
     {
       type: 'confirm',
       name: 'wantLoadable',

diff --git a/internals/generators/functional/index.js.hbs b/internals/generators/functional/index.js.hbs
@@ -14,8 +14,18 @@ import PropTypes from 'prop-types';
 {{else}}
 // import PropTypes from 'prop-types';
 {{/if}}
+{{#if wantHeaders}}
+import { Helmet } from 'react-helmet';
+{{/if}}
 {{#if wantStyles}}
-import { withStyles } from '@material-ui/core/styles';
+import { withStyles, {{#if wantTheme }}ThemeProvider{{/if}} } from '@material-ui/core/styles';
+{{else}}
+{{#if wantTheme}}
+import { ThemeProvider } from '@material-ui/core/styles';
+{{/if}}
+{{/if}}
+{{#if wantTheme}}
+import theme from 'utils/theme';
 {{/if}}
 
 {{#if wantStyles}}
@@ -24,8 +34,20 @@ import styles from './styles';
 
 function {{ properCase name }}({{#if wantStyles}}{ classes }{{/if}}) {
 return (
-<div{{#if wantStyles}} className={ classes.root }{{/if}}>
+{{#if wantTheme}}
+<ThemeProvider theme={theme}>
+{{/if}}
+  <div{{#if wantStyles}} className={ classes.root }{{/if}}>
+  {{#if wantHeaders}}
+  <Helmet>
+    <title>{{properCase name}}</title>
+    <meta name="description" content="Description of {{properCase name}}" />
+  </Helmet>
+  {{/if}}
   </div>
+{{#if wantTheme}}
+</ThemeProvider>
+{{/if}}
   );
   }