add command init database

Author: lisong

README.md

@@ -32,14 +32,20 @@ $ code-push login http://codepush.19910225.com:8080 #登录
 
 [CodePushDemo](https://github.com/lisong/code-push-demo-app)
 
-## INSTALL
+## INSTALL FROM NPM PACKAGE
 
 ```shell
-$ cd /path/to/code-push-server
-$ mysql -uroot -e"create database codepush default charset utf8;"
-$ mysql -uroot codepush < ./sql/codepush.sql
-$ mysql -uroot codepush < ./sql/codepush-v0.1.1.sql
-$ mysql -uroot codepush < ./sql/codepush-v0.1.5.sql
+$ npm install code-push-server -g
+$ code-push-server-db init --dbhost localhost --dbuser root --dbpassword #初始化数据库
+$ code-push-server #启动服务
+```
+
+## INSTALL FROM SOURCE CODE
+
+```shell
+$ git clone https://github.com/lisong/code-push-server.git
+$ cd code-push-server
+$ ./bin/db init --dbhost localhost --dbuser root --dbpassword
 $ npm install
 ```
 

bin/db

@@ -0,0 +1,70 @@
+#!/usr/bin/env node
+
+/**
+ * Module dependencies.
+ */
+var fs = require('fs');
+var path   = require('path');
+var yargs = require('yargs')
+    .usage('Usage: $0 <command> [options]')
+    .command('init', '初始化数据库')
+    .example('$0 init --dbname codepush --dbhost localhost --dbuser root --dbpassword 123456', '初始化code-push-server数据库')
+    .default({dbname: 'codepush', dbhost: 'localhost', dbuser: 'root', dbpassword: null})
+    .help('h')
+    .alias('h', 'help');
+var argv = yargs
+    .argv;
+var command = argv._[0];
+var mysql = require('mysql');
+var Promise = require("bluebird");
+var dbname = argv.dbname ? argv.dbname : 'codepush';
+var dbhost = argv.dbhost ? argv.dbhost : 'localhost';
+var dbuser = argv.dbuser ? argv.dbuser : 'root';
+var dbpassword = argv.dbpassword ? argv.dbpassword : null;
+if (command === 'init') {
+    var connection2;
+    var connection = mysql.createConnection({
+        host: dbhost,
+        user: dbuser,
+        password: dbpassword
+    });
+    Promise.promisifyAll(connection);
+    connection.connect();
+    connection.queryAsync(`CREATE DATABASE ${dbname}`)
+    .then(function(){
+        connection2 = mysql.createConnection({
+            host: dbhost,
+            user: dbuser,
+            password: dbpassword,
+            database: dbname,
+            multipleStatements: true
+        });
+        connection2.connect();
+        Promise.promisifyAll(connection2);
+        return connection2;
+    })
+    .then(function(connection2){
+        var sql = fs.readFileSync(path.resolve(__dirname, '../sql/codepush.sql'), 'utf-8');
+        var sql2 = fs.readFileSync(path.resolve(__dirname, '../sql/codepush-v0.1.1.sql'), 'utf-8');
+        var sql3 = fs.readFileSync(path.resolve(__dirname, '../sql/codepush-v0.1.5.sql'), 'utf-8');
+        return connection2.queryAsync(sql)
+        .then(function(){
+            return connection2.queryAsync(sql2);
+        })
+        .then(function(){
+            return connection2.queryAsync(sql3);
+        });
+    })
+    .then(function(){
+        console.log('success.');
+    })
+    .catch(function(e){
+        console.log(e);
+    })
+    .finally(function(){
+        if(connection) connection.end();
+        if(connection2) connection2.end()
+    });
+} else {
+    yargs.showHelp();
+}

package.json

@@ -13,7 +13,8 @@
     "url": "https://github.com/lisong/"
   },
   "bin": {
-    "code-push-server": "./bin/www"
+    "code-push-server": "./bin/www",
+    "code-push-server-db": "./bin/db"
   },
   "engines": {
     "node": ">= 5.0",
@@ -55,6 +56,7 @@
     "serve-favicon": "~2.3.0",
     "slash": "^1.0.0",
     "validator": "^5.0.0",
+    "yargs": "^6.2.0",
     "yazl": "^2.3.0"
   },
   "devDependencies": {