mirror of
https://github.com/xuxiaobo-bobo/boda_jsEnv.git
synced 2025-04-20 10:35:01 +08:00
336 lines
11 KiB
JavaScript
336 lines
11 KiB
JavaScript
|
|
var spawn = require('child_process').spawn;
|
|
var exec = require('child_process').exec;
|
|
var chalk = require('chalk');
|
|
var util = require('util');
|
|
var fmt = require('../tools/fmt.js');
|
|
var fs = require('fs');
|
|
var path = require('path');
|
|
var cst = require('../../constants.js');
|
|
var Promise = require('../tools/promise.min.js');
|
|
|
|
function pspawn(cmd) {
|
|
return new Promise(function(resolve, reject) {
|
|
var p_cmd = cmd.split(' ');
|
|
|
|
var install_instance = spawn(p_cmd[0], p_cmd.splice(1, cmd.length), {
|
|
stdio : 'inherit',
|
|
env : process.env,
|
|
shell : true
|
|
});
|
|
|
|
install_instance.on('close', function(code) {
|
|
if (code != 0) {
|
|
console.log(chalk.bold.red('Command failed'));
|
|
return reject(new Error('Bad cmd return'));
|
|
}
|
|
return resolve();
|
|
});
|
|
|
|
install_instance.on('error', function (err) {
|
|
return reject(err);
|
|
});
|
|
});
|
|
}
|
|
|
|
function checkDockerSetup() {
|
|
return new Promise(function(resolve, reject) {
|
|
exec("docker version -f '{{.Client.Version}}'", function(err, stdout, stderr) {
|
|
if (err) {
|
|
console.error(chalk.red.bold('[Docker access] Error while trying to use docker command'));
|
|
if (err.message && err.message.indexOf('Cannot connect to the Docker') > -1) {
|
|
console.log();
|
|
console.log(chalk.blue.bold('[Solution] Setup Docker to be able to be used without sudo rights:'));
|
|
console.log(chalk.bold('$ sudo groupadd docker'));
|
|
console.log(chalk.bold('$ sudo usermod -aG docker $USER'));
|
|
console.log(chalk.bold('Then LOGOUT and LOGIN your Linux session'));
|
|
console.log('Read more: http://bit.ly/29JGdCE');
|
|
}
|
|
return reject(err);
|
|
}
|
|
return resolve();
|
|
});
|
|
});
|
|
}
|
|
|
|
/**
|
|
* Switch Dockerfile mode
|
|
* check test/programmatic/containerizer.mocha.js
|
|
*/
|
|
function parseAndSwitch(file_content, main_file, opts) {
|
|
var lines = file_content.split('\n');
|
|
var mode = opts.mode;
|
|
|
|
lines[0] = 'FROM keymetrics/pm2:' + opts.node_version;
|
|
|
|
for (var i = 0; i < lines.length; i++) {
|
|
var line = lines[i];
|
|
|
|
if (['## DISTRIBUTION MODE', '## DEVELOPMENT MODE'].indexOf(line) > -1 ||
|
|
i == lines.length - 1) {
|
|
lines.splice(i, lines.length);
|
|
lines[i] = '## ' + mode.toUpperCase() + ' MODE';
|
|
lines[i + 1] = 'ENV NODE_ENV=' + (mode == 'distribution' ? 'production' : mode);
|
|
|
|
if (mode == 'distribution') {
|
|
lines[i + 2] = 'COPY . /var/app';
|
|
lines[i + 3] = 'CMD ["pm2-docker", "' + main_file + '", "--env", "production"]';
|
|
}
|
|
if (mode == 'development') {
|
|
lines[i + 2] = 'CMD ["pm2-dev", "' + main_file + '", "--env", "development"]';
|
|
}
|
|
break;
|
|
}
|
|
};
|
|
lines = lines.join('\n');
|
|
return lines;
|
|
};
|
|
|
|
/**
|
|
* Replace ENV, COPY and CMD depending on the mode
|
|
* @param {String} docker_filepath Dockerfile absolute path
|
|
* @param {String} main_file Main file to start in container
|
|
* @param {String} mode Mode to switch the Dockerfile
|
|
*/
|
|
function switchDockerFile(docker_filepath, main_file, opts) {
|
|
return new Promise(function(resolve, reject) {
|
|
var data = fs.readFileSync(docker_filepath, 'utf8').toString();
|
|
|
|
if (['distribution', 'development'].indexOf(opts.mode) == -1)
|
|
return reject(new Error('Unknown mode'));
|
|
|
|
var lines = parseAndSwitch(data, main_file, opts)
|
|
fs.writeFile(docker_filepath, lines, function(err) {
|
|
if (err) return reject(err);
|
|
resolve({
|
|
Dockerfile_path : docker_filepath,
|
|
Dockerfile : lines,
|
|
CMD : ''
|
|
});
|
|
})
|
|
});
|
|
}
|
|
|
|
/**
|
|
* Generate sample Dockerfile (lib/templates/Dockerfiles)
|
|
* @param {String} docker_filepath Dockerfile absolute path
|
|
* @param {String} main_file Main file to start in container
|
|
* @param {String} mode Mode to switch the Dockerfile
|
|
*/
|
|
function generateDockerfile(docker_filepath, main_file, opts) {
|
|
return new Promise(function(resolve, reject) {
|
|
var tpl_file = path.join(cst.TEMPLATE_FOLDER, cst.DOCKERFILE_NODEJS);
|
|
var template = fs.readFileSync(tpl_file, {encoding: 'utf8'});
|
|
var CMD;
|
|
|
|
template = parseAndSwitch(template, main_file, opts);
|
|
|
|
fs.writeFile(docker_filepath, template, function(err) {
|
|
if (err) return reject(err);
|
|
resolve({
|
|
Dockerfile_path : docker_filepath,
|
|
Dockerfile : template,
|
|
CMD : CMD
|
|
});
|
|
});
|
|
});
|
|
}
|
|
|
|
function handleExit(CLI, opts, mode) {
|
|
process.on('SIGINT', function() {
|
|
CLI.disconnect();
|
|
|
|
if (mode != 'distribution')
|
|
return false;
|
|
|
|
exec('docker ps -lq', function(err, stdout, stderr) {
|
|
if (err) {
|
|
console.error(err);
|
|
}
|
|
require('vizion').analyze({folder : process.cwd()}, function recur_path(err, meta){
|
|
if (!err && meta.revision) {
|
|
var commit_id = util.format('#%s(%s) %s',
|
|
meta.branch,
|
|
meta.revision.slice(0, 5),
|
|
meta.comment);
|
|
|
|
console.log(chalk.bold.magenta('$ docker commit -m "%s" %s %s'),
|
|
commit_id,
|
|
stdout.replace('\n', ''),
|
|
opts.imageName);
|
|
}
|
|
else
|
|
console.log(chalk.bold.magenta('$ docker commit %s %s'), stdout.replace('\n', ''), opts.imageName);
|
|
|
|
console.log(chalk.bold.magenta('$ docker push %s'), opts.imageName);
|
|
});
|
|
});
|
|
});
|
|
}
|
|
|
|
module.exports = function(CLI) {
|
|
CLI.prototype.generateDockerfile = function(script, opts) {
|
|
var docker_filepath = path.join(process.cwd(), 'Dockerfile');
|
|
var that = this;
|
|
|
|
fs.stat(docker_filepath, function(err, stat) {
|
|
if (err || opts.force == true) {
|
|
generateDockerfile(docker_filepath, script, {
|
|
mode : 'development'
|
|
})
|
|
.then(function() {
|
|
console.log(chalk.bold('New Dockerfile generated in current folder'));
|
|
console.log(chalk.bold('You can now run\n$ pm2 docker:dev <file|config>'));
|
|
return that.exitCli(cst.SUCCESS_EXIT);
|
|
});
|
|
return false;
|
|
}
|
|
console.log(chalk.red.bold('Dockerfile already exists in this folder, use --force if you want to replace it'));
|
|
that.exitCli(cst.ERROR_EXIT);
|
|
});
|
|
};
|
|
|
|
CLI.prototype.dockerMode = function(script, opts, mode) {
|
|
var promptly = require('promptly');
|
|
var self = this;
|
|
handleExit(self, opts, mode);
|
|
|
|
if (mode == 'distribution' && !opts.imageName) {
|
|
console.error(chalk.bold.red('--image-name [name] option is missing'));
|
|
return self.exitCli(cst.ERROR_EXIT);
|
|
}
|
|
|
|
var template;
|
|
var app_path, main_script;
|
|
var image_name;
|
|
var node_version = opts.nodeVersion ? opts.nodeVersion.split('.')[0] : 'latest';
|
|
|
|
image_name = opts.imageName || require('crypto').randomBytes(6).toString('hex');
|
|
|
|
if (script.indexOf('/') > -1) {
|
|
app_path = path.join(process.cwd(), path.dirname(script));
|
|
main_script = path.basename(script);
|
|
}
|
|
else {
|
|
app_path = process.cwd();
|
|
main_script = script;
|
|
}
|
|
|
|
checkDockerSetup()
|
|
.then(function() {
|
|
/////////////////////////
|
|
// Generate Dockerfile //
|
|
/////////////////////////
|
|
return new Promise(function(resolve, reject) {
|
|
var docker_filepath = path.join(process.cwd(), 'Dockerfile');
|
|
|
|
fs.stat(docker_filepath, function(err, stat) {
|
|
if (err) {
|
|
// Dockerfile does not exist, generate one
|
|
// console.log(chalk.blue.bold('Generating new Dockerfile'));
|
|
if (opts.force == true) {
|
|
return resolve(generateDockerfile(docker_filepath, main_script, {
|
|
node_version : node_version,
|
|
mode : mode
|
|
}));
|
|
}
|
|
if (opts.dockerdaemon)
|
|
return resolve(generateDockerfile(docker_filepath, main_script, {
|
|
node_version : node_version,
|
|
mode : mode
|
|
}));
|
|
promptly.prompt('No Dockerfile in current directory, ok to generate a new one? (y/n)', function(err, value) {
|
|
if (value == 'y')
|
|
return resolve(generateDockerfile(docker_filepath, main_script, {
|
|
node_version : node_version,
|
|
mode : mode
|
|
}));
|
|
else
|
|
return self.exitCli(cst.SUCCESS_EXIT);
|
|
});
|
|
return false;
|
|
}
|
|
return resolve(switchDockerFile(docker_filepath, main_script, {
|
|
node_version : node_version,
|
|
mode : mode
|
|
}));
|
|
});
|
|
});
|
|
})
|
|
.then(function(_template) {
|
|
template = _template;
|
|
return Promise.resolve();
|
|
})
|
|
.then(function() {
|
|
//////////////////
|
|
// Docker build //
|
|
//////////////////
|
|
|
|
var docker_build = util.format('docker build -t %s -f %s',
|
|
image_name,
|
|
template.Dockerfile_path);
|
|
|
|
if (opts.fresh == true)
|
|
docker_build += ' --no-cache';
|
|
docker_build += ' .';
|
|
|
|
console.log();
|
|
fmt.sep();
|
|
fmt.title('Building Boot System');
|
|
fmt.field('Type', chalk.cyan.bold('Docker'));
|
|
fmt.field('Mode', mode);
|
|
fmt.field('Image name', image_name);
|
|
fmt.field('Docker build command', docker_build);
|
|
fmt.field('Dockerfile path', template.Dockerfile_path);
|
|
fmt.sep();
|
|
|
|
return pspawn(docker_build);
|
|
})
|
|
.then(function() {
|
|
////////////////
|
|
// Docker run //
|
|
////////////////
|
|
|
|
var docker_run = 'docker run --net host';
|
|
|
|
if (opts.dockerdaemon == true)
|
|
docker_run += ' -d';
|
|
if (mode != 'distribution')
|
|
docker_run += util.format(' -v %s:/var/app -v /var/app/node_modules', app_path);
|
|
docker_run += ' ' + image_name;
|
|
var dockerfile_parsed = template.Dockerfile.split('\n');
|
|
var base_image = dockerfile_parsed[0];
|
|
var run_cmd = dockerfile_parsed[dockerfile_parsed.length - 1];
|
|
|
|
console.log();
|
|
fmt.sep();
|
|
fmt.title('Booting');
|
|
fmt.field('Type', chalk.cyan.bold('Docker'));
|
|
fmt.field('Mode', mode);
|
|
fmt.field('Base Image', base_image);
|
|
fmt.field('Image Name', image_name);
|
|
fmt.field('Docker Command', docker_run);
|
|
fmt.field('RUN Command', run_cmd);
|
|
fmt.field('CWD', app_path);
|
|
fmt.sep();
|
|
return pspawn(docker_run);
|
|
})
|
|
.then(function() {
|
|
console.log(chalk.blue.bold('>>> Leaving Docker instance uuid=%s'), image_name);
|
|
self.disconnect();
|
|
return Promise.resolve();
|
|
})
|
|
.catch(function(err) {
|
|
console.log();
|
|
console.log(chalk.grey('Raw error=', err.message));
|
|
self.disconnect();
|
|
});
|
|
|
|
};
|
|
|
|
};
|
|
|
|
module.exports.generateDockerfile = generateDockerfile;
|
|
module.exports.parseAndSwitch = parseAndSwitch;
|
|
module.exports.switchDockerFile = switchDockerFile;
|