9f/lib/update.js

'use strict'
/**
 * 更新模板
 */

const co = require('co')
const prompt = require('co-prompt')
const chalk = require('chalk')
const tplJson = require('../tpl')
const fs = require('fs')

module.exports = () => {
  co(function *() {
    // 模板名称
    let tplName = process.argv[3]

    if(!tplName) {
      tplName = yield prompt('Template name: ')
      if (!tplName) {
        console.log(chalk.red('Template name not empty!'))
        process.exit()
      }
    }

    if (!tplJson.tpl[tplName]) {
      console.log(chalk.red('Template name not existed!'))
      process.exit()
    }

  	let gitUrl = yield prompt('Repository address: ')      // 仓库地址
    if(gitUrl == "") {
      console.log(chalk.red('Repository address not empty!'))
      process.exit()
    }

  	let branch = yield prompt('Branch(master): ')          // 分支

    tplJson.tpl[tplName]['url'] = gitUrl.replace(/[\u0000-\u0019]/g, '')
    tplJson.tpl[tplName]['branch'] = branch || 'master'

    // 写入配置文件
		fs.writeFile(__dirname + '/../tpl.json', JSON.stringify(tplJson), 'utf-8', (err) => {
			if (err) console.log(err)
      console.log()
			console.log(chalk.green('New template added!\n'))
      console.log(chalk.grey('The last template list is: \n'))
      console.log(tplJson)
      console.log('\n')
			process.exit()
		})
  })
}