support custom complete message

Author: yyx990803

bin/vue-init

@@ -152,8 +152,8 @@ function downloadAndGenerate (template) {
     spinner.stop()
     if (err) logger.fatal(err)
     generate(name, tmp, to, function (err) {
-      if (err) logger.fatal(err)
       rm(tmp)
+      if (err) logger.fatal(err)
       console.log()
       logger.success('Generated "%s".', name)
     })

lib/eval.js

@@ -1,3 +1,5 @@
+var chalk = require('chalk')
+
 /**
  * Evaluate an expression in meta.json in the context of
  * prompt answers data.

lib/filter.js

@@ -1,5 +1,4 @@
 var match = require('minimatch')
-var chalk = require('chalk')
 var evaluate = require('./eval')
 
 module.exports = function (files, filters, data, done) {

lib/generate.js

@@ -17,16 +17,25 @@ var filter = require('./filter')
 
 module.exports = function generate (name, src, dest, done) {
   var opts = getOptions(name, src)
-  Metalsmith(path.join(src, 'template'))
+  var metalsmith = Metalsmith(path.join(src, 'template'))
+  var data = metalsmith.metadata()
+  // avoid handlebars escaping HTML
+  data.noEscape = true
+  metalsmith
     .use(askQuestions(opts.schema))
     .use(filterFiles(opts.filters))
     .use(renderTemplateFiles)
     .clean(false)
     .source('.') // start from template root instead of `./src` which is Metalsmith's default for `source`
     .destination(dest)
     .build(function (err) {
-      if (err) throw err
-      done()
+      done(err)
+      if (opts.completeMessage) {
+        formatMessage(opts.completeMessage, data, function (err, message) {
+          if (err) return done(err)
+          console.log(message)
+        })
+      }
     })
 }
 
@@ -67,7 +76,6 @@ function filterFiles (filters) {
 function renderTemplateFiles (files, metalsmith, done) {
   var keys = Object.keys(files)
   var metalsmithMetadata = metalsmith.metadata()
-  metalsmithMetadata.noEscape = true
   async.each(keys, function (file, next) {
     var str = files[file].contents.toString()
     // do not attempt to render files that do not have mustaches
@@ -81,3 +89,20 @@ function renderTemplateFiles (files, metalsmith, done) {
     })
   }, done)
 }
+
+/**
+ * Format complete message.
+ *
+ * @param {String} message
+ * @param {Object} data
+ * @param {Function} cb
+ */
+
+function formatMessage (message, data, cb) {
+  render(message, data, function (err, res) {
+    if (err) return cb(err)
+    cb(null, '\n' + res.split(/\r?\n/g).map(function (line) {
+      return '   ' + line
+    }).join('\n'))
+  })
+}

lib/logger.js

@@ -41,5 +41,4 @@ exports.fatal = function (message) {
 exports.success = function () {
   var msg = format.apply(format, arguments)
   console.log(chalk.white(prefix), sep, msg)
-  process.exit(0)
 }