meteorup/lib/controller.js

Meteorup is a meteor enhancement tools

var nodemiral = require( 'nodemiral' );
var path = require( 'path' );
var fs = require( 'fs' );
// var exec = require( 'child_process' ).exec;
var exec = require( 'executive' );
var spawn = require( 'child_process' ).spawn;
var uuid = require( 'uuid' );
var rimraf = require( 'rimraf' );
var format = require( 'util' ).format;
var extend = require( 'util' )._extend;
var _ = require( 'underscore' );
var async = require( 'async' );
var cjson = require( 'cjson' );
var Config = require( './config' );
var buildApp = require( './build.js' );
require( 'colors' );
module.exports = Actions;
global.notification = "finished";

function Actions( pwd, options ) {
    this.pwd = pwd;
    this.options = options;
}
// ================================================
// 创建一个 ssh 连接  ===============================
// ================================================
Actions.prototype._createPublicSession = function() {
    var host = "meteorup.cn";
    var auth = {
        username: "www",
        pem: fs.readFileSync( resolveHome( "~/.ssh/id_rsa" ), 'utf8' )
    };
    var nodemiralOptions = {
        keepAlive: true
    };
    // console.log(auth);
    return nodemiral.session( host, auth, nodemiralOptions );
};
// ================================================
// 创建一个 ssh 连接
// ================================================
Actions.prototype._createPrivateSession = function( server ) {
    var host = server.host;
    var auth = {
        username: server.username,
    };
    if ( server.pem ) {
        auth.pem = fs.readFileSync( server.pem, 'utf8' );
    } else {
        auth.password = server.password;
    }
    var nodemiralOptions = {
        ssh: server.sshOptions,
        keepAlive: true
    };
    // console.log(auth);
    return nodemiral.session( host, auth, nodemiralOptions );
};
// ================================================
//  deploy
// ================================================
Actions.prototype.deploy = function( appName , privateKey) {
    var self = this;
    var buildLocation = path.resolve( '/tmp', uuid.v4() );
    var bundlePath = path.resolve( buildLocation, 'bundle.tar.gz' );

    var appPath = this.pwd;
    var buildOptions = this.options;
    var config = {
        appName: appName,
        privateKey: privateKey
    };
    console.log( ( 'Started building ' + appName + " application." ).bold.blue );

    buildApp( appPath, buildLocation, buildOptions, function( err ) {
        if ( err ) {
            process.exit( 1 );
        } else {
            async.mapSeries(
                [ self._createPublicSession() ],
                function( session, callback ) {
                    var taskListsBuilder = require( './actions/deploy.js' );
                    var env = {};
                    if ( buildOptions.env ) {
                        var envPath = path.resolve( self.pwd, buildOptions.env );
                        if ( fs.existsSync( envPath ) ) {
                            var setttingsJson = require( envPath );
                            env = _.extend( env, setttingsJson );
                        }
                    }
                    var taskList = taskListsBuilder.deploy( bundlePath, env, config );
                    taskList.run( session, function( summaryMap ) {
                        callback( null, summaryMap );
                    } );
                },
                // function(){process.exit(0);}
                whenAfterDeployed( buildLocation ) )
        }
    } );
};
// ================================================
// setup
// ================================================
Actions.prototype.setup = function() {
    var self = this;
    var packagePath = path.resolve( self.pwd, 'package.json' );
    if ( fs.existsSync( packagePath ) ) {
        var config = Config.setup( packagePath );
        // console.log(config);
        var appPath = this.pwd;
        // console.log(('Started installation server').bold.blue);
        async.mapSeries(
            [ self._createPrivateSession( config.server ) ],
            function( session, callback ) {
                var taskListsBuilder = require( './actions/setup.js' );
                var taskList = taskListsBuilder.setup( config );
                taskList.run( session, function( summaryMap ) {
                    callback( null, summaryMap );
                } );
            }, whenAfterCompleted )
    } else {
        console.log( ( 'configuration file does not exist! ' + packagePath ).bold.red );
    }
};
// ================================================
//   push
// ================================================
Actions.prototype.push = function( appName ) {
    var self = this;
    var buildLocation = path.resolve( '/tmp', uuid.v4() );
    var bundlePath = path.resolve( buildLocation, 'bundle.tar.gz' );
    process.env.BUILD_LOCATION = buildLocation;

    var packagePath = path.resolve( self.pwd, 'package.json' );
    if ( fs.existsSync( packagePath ) ) {
        var config = Config.setup( packagePath );
        var appPath = this.pwd;
        var buildOptions = this.options;
        // console.log(buildOptions);
        console.log( ( 'Started building ' + config.deploy.appName + " application." ).bold.blue );
        buildApp( appPath, buildLocation, buildOptions, function( err ) {
            if ( err ) {
                process.exit( 1 );
            } else {
                async.mapSeries(
                    [ self._createPrivateSession( config.server ) ],
                    function( session, callback ) {
                        var taskListsBuilder = require( './actions/push.js' );
                        var taskList = taskListsBuilder.main( bundlePath, config, buildOptions);
                        taskList.run( session, function( summaryMap ) {
                            callback( null, summaryMap );
                        } );
                    }, whenAfterDeployed( buildLocation ) );
            }
        } );
    } else {
        console.log( ( 'configuration file does not exist! ' + packagePath ).bold.red );
    }
};
// ================================================
//   logs
// ================================================
Actions.prototype.logs = function( appName ) {
    var self = this;
    var packagePath = path.resolve( self.pwd, 'package.json' );
    if ( fs.existsSync( packagePath ) ) {
        var config = Config.setup( packagePath );
        async.mapSeries(
            [ self._createPrivateSession( config.server ) ],
            function( session, callback ) {
                console.log( ( '[' + config.server.host + ']' ).magenta + ( ' - Print the log of ' + config.deploy.appName + ' project.' ).bold.blue );

                var SCRIPT_DIR = path.resolve(__dirname, '../scripts/logs');
                var hostPrefix = '[' + session._host + '] ';
                var options = {
                    onStdout: function( data ) {
                        process.stdout.write(data.toString() );
                    },
                    onStderr: function( data ) {
                        process.stderr.write( data.toString() );
                    },
                    vars: {
                        appName: config.deploy.appName
                    }
                };
                var command = path.resolve(SCRIPT_DIR, 'logs.sh');
                session.executeScript( command, options, callback );
            }, whenAfterCompleted );
    } else {
        console.log( ( 'configuration file does not exist! ' + packagePath ).bold.red );
    }
};
// ================================================
//   mongo
// ================================================
Actions.prototype.mongo = function( appName ) {
    var self = this;
    var packagePath = path.resolve( self.pwd, 'package.json' );
    if ( fs.existsSync( packagePath ) ) {
        var config = Config.setup( packagePath );
        var mongourl = config.deploy.env.MONGO_URL;
        var port = /:(\d+)\//.exec( mongourl )[ 1 ];
        var db = mongourl.substring( mongourl.lastIndexOf( '/' ) + 1 );
        var host = /(2[0-4][0-9]|25[0-5] |1[0-9][0-9]|[1-9]?[0-9])(\.(2[0-4][0-9]|25[0-5] |1[0-9][0-9]|[1-9]?[0-9])){3}/.exec( mongourl )[ 0 ];
        host = ( host == 'localhost' || host == '127.0.0.1' ) ? config.server.host : host;
        var executable = [ "mongo", "--host", host, '--port', port, db ];
        console.info( ( '[' + config.server.host + ']' ).magenta + ( ' - Connection to a remote mongo database.' ).bold.blue );
        console.log( executable.join( ' ' ).bold.blue );
        exec.interactive( executable.join( ' ' ), function( err ) {
            whenAfterCompleted(err);
        } );
    } else {
        console.log( ( 'configuration file does not exist! ' + packagePath ).bold.red );
    }
};
// ================================================
//   test
// ================================================
Actions.prototype.test = function( appName ) {
    var self = this;
    var packagePath = path.resolve( self.pwd, 'package.json' );
    if ( fs.existsSync( packagePath ) ) {
        var config = Config.setup( packagePath );
        async.series( [function( callback ) {
            var SCRIPT_DIR = path.resolve(__dirname, '../scripts/push');
            var session = self._createPrivateSession( config.server );
            var hostPrefix = '[' + session._host + '] ';
            var options = {
                onStdout: function( data ) {
                    process.stdout.write( hostPrefix + data.toString() );
                },
                onStderr: function( data ) {
                    process.stderr.write( hostPrefix + data.toString() );
                }
            };
            // var command = "curl -o- https://raw.githubusercontent.com/creationix/nvm/v0.31.1/install.sh | bash && . ~/.bashrc && env && nvm --version";
            // session.execute( command, options, callback );

            var command = path.resolve(SCRIPT_DIR, 'verify.sh');
            console.log(command);
            session.executeScript( command, options, callback );

        }], whenAfterCompleted );
    } else {
        console.log( ( 'configuration file does not exist! ' + packagePath ).bold.red );
    }
};
// 找到用户目录 ~
function resolveHome( filepath ) {
    if ( filepath[ 0 ] === '~' ) {
        return path.join( process.env.HOME, filepath.slice( 1 ) );
    }
    return path;
}
// 任务完成时，删除临时文件
function whenAfterDeployed( buildLocation ) {
    return function( error, summaryMaps ) {
        rimraf.sync( buildLocation );
        whenAfterCompleted( error, summaryMaps );
    };
}

function whenAfterCompleted( error, summaryMaps ) {
    var errorCode = error || haveSummaryMapsErrors( summaryMaps ) ? 1 : 0;
    say( global.notification );
    process.exit( errorCode );
}

function haveSummaryMapsErrors( summaryMaps ) {
    return _.some( summaryMaps, hasSummaryMapErrors );
}

function hasSummaryMapErrors( summaryMap ) {
    return _.some( summaryMap, function( summary ) {
        return summary.error;
    } )
}

function escapeshell( cmd ) {
    //http://stackoverflow.com/a/7685469
    return '"' + cmd.replace( /(["\s'$`\\])/g, '\\$1' ) + '"';
};

function say( string ) {
    var os = require( 'os' );
    if ( os.type() == 'Darwin' ) {
        exec( "say " + escapeshell( string ) );
    }
}