capricorn-cli/src/commands/module.js

摩羯系统 - 命令行工具

const { log, renderAscii } = require('../core/util')
const Inquire = require('inquirer')
const path = require('path')
const fs = require('fs')
const compose = require('koa-compose')
const getGitUser = require('../core/git-user')
const Khaos = require('khaos-patched')
const Spinner = require('../core/spinner')
const download = require('download-git-repo')
const uid = require('uid')
const rm = require('rimraf').sync
const { capricornTemplateAddress, capricornModuleAddress } = require('../data/init')
const { getTemplateTypes, getModuleTypes } = require('../server/init')

const moduleHandler = {
	init () {
		this.getGitConfig().then(() => {
			compose([
				this.getInfo.bind(this),
				this.ctrl.bind(this),
				this.generate.bind(this),
				this.downloadHtml.bind(this),
				this.generateIndex.bind(this)
			])()
		})
	},
	
	moduleInfo: {},
	capricornTemplates: [],
	capricornModules: [],
	interimTemp: './capricorn-module-' + uid(),
	
	// 获取基础html模版和模块初始化模版列表
	getGitConfig () {
		const spinner = new Spinner('拉取模版列表...')
		spinner.start()
		return Promise.all([
			getModuleTypes(),
			getTemplateTypes()
		]).then(resList => {
			spinner.stop()
			resList[0].forEach(v => {
				if (v.name !== 'master') {
					this.capricornModules.push((v.name))
				}
			})
			resList[1].forEach(v => {
				if (v.name !== 'master') {
					this.capricornTemplates.push((v.name))
				}
			})
			return resList
		}).catch(e => {
			console.log(e)
			spinner.stop()
		})
	},
	
	getInfo (ctx, next) {
		this.moduleInfo = Object.assign({}, this.moduleInfo, getGitUser())
		let prompts = []
		
		prompts.push({
			type: 'list',
			name: 'templateType',
			message: '请选择你要使用的基础html模版',
			choices: this.capricornTemplates
		})
		
		prompts.push({
			type: 'list',
			name: 'moduleType',
			message: '请选择你要生成的功能模块模板',
			choices: this.capricornModules
		})
		
		prompts.push({
			type: 'input',
			name: 'moduleName',
			message: '请输入新建功能模块的名称(格式：module-***)',
			validate (val) {
				if (val) {
					if (val.indexOf('module-') !== 0) {
						return '模块名称要以"module-"开头，请重新输入'
					}
					if (val.match('.*[A-Z]+.*')) {
						return '模块名称不能包含大写字母，请重新输入'
					}
					return true
				} else {
					return '模块名称不能为空，请输入模块名称！'
				}
			}
		})
		
		prompts.push({
			type: 'input',
			name: 'moduleDesc',
			message: '请输入新建模块的描述信息',
			validate (val) {
				if (val) {
					return true
				} else {
					return '功能模块的描述信息不能为空，请输入描述信息！'
				}
			}
		})
		
		Inquire.prompt(prompts).then(answers => {
			this.moduleInfo = Object.assign(this.moduleInfo, answers)
			next()
		})
	},
	
	ctrl (ctx, next) {
		const { moduleType } = this.moduleInfo
		const spinner = new Spinner('正在初始化模块...')
		spinner.start()
		
		download(capricornModuleAddress + '#' + moduleType, this.interimTemp, (err) => {
			spinner.stop()
			if (err) {
				log.error(err)
			}
			next()
		})
	},
	
	generate (ctx, next) {
		const { moduleName } = this.moduleInfo
		const khaos = new Khaos(this.interimTemp)
		khaos.schema(this.formatOptions())
		khaos.generate(`./${moduleName.trim()}`, (err) => {
			rm(this.interimTemp)
			if (err) {
				log.error(err)
				return
			}
			log.success('模块初始化成功！')
			next()
		})
	},
	
	// 下载html模板
	downloadHtml (ctx, next) {
		const { moduleName, templateType } = this.moduleInfo
		const spinner = new Spinner('正在下载html模版...')
		spinner.start()
		download(capricornTemplateAddress + '#' + templateType, moduleName + '/template', (err) => {
			spinner.stop()
			if (err) {
				log.error(err)
				return
			}
			log.success('初始化完成！')
			renderAscii()
			next()
		})
	},
	
	generateIndex (ctx, next) {
		const { moduleName } = this.moduleInfo
		const content = `(function(){!window.Capricorn&&(window.Capricorn={});window.Capricorn.modules = {};window.Capricorn.modules["${moduleName}"] = [];window.Capricorn.modules["${moduleName}"].push({"name":"${moduleName}"});})();`
		fs.writeFile(`${moduleName}/template/lib/index.js`, content, 'utf8', (err) => {
			if (err) throw err
			log.success('index.js生成成功')
			next()
		})
	},
	
	formatOptions () {
		const keys = ['moduleName', 'moduleDesc', 'author', 'email']
		const options = {}
		
		keys.forEach(v => {
			options[v] = {
				type: 'string',
				default: this.moduleInfo[v]
			}
		})
		return options
	}
}

exports.handler = argvs => {
	moduleHandler.init(argvs)
}