add themeColor && Partial import Element

components.js

@@ -0,0 +1,79 @@
+module.exports = [
+  'Pagination',
+  'Dialog',
+  'Autocomplete',
+  'Dropdown',
+  'DropdownMenu',
+  'DropdownItem',
+  'Menu',
+  'Submenu',
+  'MenuItem',
+  'MenuItemGroup',
+  'Input',
+  'InputNumber',
+  'Radio',
+  'RadioGroup',
+  'RadioButton',
+  'Checkbox',
+  'CheckboxButton',
+  'CheckboxGroup',
+  'Switch',
+  'Select',
+  'Option',
+  'OptionGroup',
+  'Button',
+  'ButtonGroup',
+  'Table',
+  'TableColumn',
+  'DatePicker',
+  'TimeSelect',
+  'TimePicker',
+  'Popover',
+  'Tooltip',
+  'Breadcrumb',
+  'BreadcrumbItem',
+  'Form',
+  'FormItem',
+  'Tabs',
+  'TabPane',
+  'Tag',
+  'Tree',
+  'Alert',
+  'Slider',
+  'Icon',
+  'Row',
+  'Col',
+  'Upload',
+  'Progress',
+  'Spinner',
+  'Badge',
+  'Card',
+  'Rate',
+  'Steps',
+  'Step',
+  'Carousel',
+  'Scrollbar',
+  'CarouselItem',
+  'Collapse',
+  'CollapseItem',
+  'Cascader',
+  'ColorPicker',
+  'Transfer',
+  'Container',
+  'Header',
+  'Aside',
+  'Main',
+  'Footer',
+  'Timeline',
+  'TimelineItem',
+  'Link',
+  'Divider',
+  'Image',
+  'Calendar',
+  'Backtop',
+  'PageHeader',
+  'CascaderPanel',
+  'Avatar',
+  'Drawer',
+  'CollapseTransition'
+]

generator/templates/src/element-variables.scss

@@ -3,7 +3,7 @@ Write your variables here. All available variables can be
 found in element-ui/packages/theme-chalk/src/common/var.scss.
 For example, to overwrite the theme color:
 */
-$--color-primary: teal;
+$--color-primary: <%= options.themeColor %>;
 
 /* icon font path, required */
 $--font-path: '~element-ui/lib/theme-chalk/fonts';

generator/templates/src/plugins/element.js

@@ -15,13 +15,18 @@ Vue.use(Element, { locale })
 Vue.use(Element)
 <%_ } _%>
 <%_ } else { _%>
-import { Button } from 'element-ui'
+import { 
+<%_ options.components.forEach(function(item){ _%>
+  <%= item %>,
+<%_ }) _%>
+} from 'element-ui'
 <%_ if (options.lang !== 'zh-CN') { _%>
 import lang from 'element-ui/lib/locale/lang/<%= options.lang %>'
 import locale from 'element-ui/lib/locale'
 
 locale.use(lang)
 <%_ } _%>
-
-Vue.use(Button)
+<%_ options.components.forEach(function(item){ _%>
+Vue.use(<%= item %>);
+<%_ }) _%>
 <%_ } _%>

prompts.js

@@ -1,6 +1,17 @@
 const localeList = require('./lang.js')
+const componentList = require('./components.js')
 
 module.exports = [
+  {
+    type: 'list',
+    name: 'lang',
+    message: 'Choose the locale you want to load',
+    choices: localeList.map(locale => ({
+      name: locale,
+      value: locale
+    })),
+    default: 'zh-CN'
+  },
   {
     type: 'list',
     name: 'import',
@@ -11,6 +22,17 @@ module.exports = [
     ],
     default: 'full',
   },
+  {
+    when: answers => answers.import === 'partial',
+    type: 'checkbox',
+    name: 'components',
+    message: 'Choose the components you want to load',
+    choices: componentList.map(component => ({
+      name: component,
+      value: component,
+      checked: true
+    }))
+  },
   {
     when: answers => answers.import === 'full',
     type: 'confirm',
@@ -19,13 +41,11 @@ module.exports = [
     default: false,
   },
   {
-    type: 'list',
-    name: 'lang',
-    message: 'Choose the locale you want to load',
-    choices: localeList.map(locale => ({
-      name: locale,
-      value: locale
-    })),
-    default: 'zh-CN'
+    when: answers => answers.customTheme === true,
+    name: 'themeColor',
+    type: 'color',
+    message: 'Theme color',
+    description: 'This is used to change the system UI color around the app',
+    default: '#0073e8'
   }
 ]