@gx-design-vue/create-gx-cli 0.0.2 → 0.0.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@gx-design-vue/create-gx-cli",
-  "version": "0.0.2",
+  "version": "0.0.3",
   "license": "MIT",
   "description": "a cli to bootstrap gx project",
   "main": "src/index.js",

package/src/cli.js

@@ -1,24 +1,28 @@
+import fs from 'fs'
+import path from 'path'
 import minimist from 'minimist'
 import prompts from 'prompts'
-import { blue, reset, yellow } from 'kolorist'
+import { yellow, reset } from 'kolorist'
 import { createProject } from './main'
 
 const defaultTargetDir = 'gx-cli-project'
 
-const FRAMEWORKS = [
-  {
-    name: 'vue',
-    color: yellow
-  },
-  {
-    name: 'mobile',
-    color: blue
-  }
-]
+function getTemplate() {
+  const templateList = []
+  const files = fs.readdirSync(path.join(process.cwd(), '../'))
 
-const TEMPLATES = FRAMEWORKS.map(
-  (f) => (f.variants && f.variants.map((v) => v.name)) || [f.name]
-).reduce((a, b) => a.concat(b), [])
+  for (const file of files) {
+    if (file.indexOf('template') >= 0) {
+      templateList.push(file)
+    }
+  }
+  return templateList.map((el) => {
+    return {
+      name: el.split('-')[1],
+      color: yellow
+    }
+  })
+}
 
 function formatTargetDir(targetDir) {
   return targetDir ? targetDir.trim().replace(/\/+$/g, '') : ''
@@ -35,6 +39,11 @@ function parseArgumentsIntoOptions() {
 }
 
 async function promptForMissingOptions(options) {
+  const FRAMEWORKS = getTemplate()
+
+  const TEMPLATES = FRAMEWORKS.map(
+    (f) => (f.variants && f.variants.map((v) => v.name)) || [f.name]
+  ).reduce((a, b) => a.concat(b), [])
   const answers = await prompts([
     {
       type: options.projectName ? null : 'text',