nci-ansible-ui/resources/projects.js

Simple web interface for run ansible playbooks

'use strict';

var Steppy = require('twostep').Steppy,
	_ = require('underscore'),
	helpers = require('./helpers');

var stringifyArgValue = function(value) {
	return '"' + value.replace(/"/g, '\\"') + '"';
};

var makeProject = function(project, buildParams) {
	var newProject = _(project).clone();

	var playbookName = buildParams.playbook && buildParams.playbook.name;
	if (playbookName) {

		if (!project.playbooks) {
			throw new Error(
				'No playbooks in the project ' + project.name + ' but ' +
				'playbookName is specified'
			);
		}

		var playbook = _(project.playbooks).findWhere({name: playbookName});

		if (!playbook) {
			throw new Error(
				'No playbook ' + playbookName + ' in ' +
				project.name + ' project'
			);
		}

		var inventoryNames = buildParams.playbook.inventoryNames,
			limit = buildParams.playbook.limit,
			extraVars = buildParams.playbook.extraVars;

		if (!inventoryNames || !inventoryNames.length) {
			throw new Error(
				'Inventory not specified for playbook ' + playbook.name +
				' (project ' + project.name + ')'
			);
		}

		var inventories = _(inventoryNames).map(function(inventoryName) {
			var inventory = _(playbook.inventories).findWhere({
				name: inventoryName
			});

			if (!inventory) {
				throw new Error(
					'No Inventory ' + inventoryName + ' in ' + playbook.name +
					' (project ' + project.name + ')'
				);

			}

			return inventory;
		});

		var playbookSteps = _(inventories).map(function(inventory) {
			var args = [
				project.playbookCommand,
				playbook.path,
				'--inventory-file=' + inventory.path
			];

			if (limit) {
				args.push('--limit=' + limit);
			}

			if (extraVars) {
				args.push('--extra-vars=' + stringifyArgValue(extraVars));
			}

			var stepName = (
				'run playbook ' + playbook.name + ' with ' + inventory.name +
				' inventory'
			);

			var yellow ='\\033[0;33m',
				noColor='\\033[0m';

			var echoCommand = (
				'echo "******************";' +
				'echo -e "********* ' + yellow + stepName.toUpperCase() +
				noColor + ' *********";' +
				'echo "******************";'
			);

			return {
				type: 'shell',
				name: stepName,
				cmd: echoCommand + args.join(' ')
			};
		});

		newProject.steps = newProject.steps.concat(playbookSteps);
	}

	return newProject;
};

var patchDirstributor = function(distributor) {
	var originalMakeProjet = distributor._makeProject;
	distributor._makeProject = function(project, buildParams) {
		var newProject = originalMakeProjet(project, buildParams);
		newProject = makeProject(newProject, buildParams);
		return newProject;
	};
};

var extendProject = function(project) {
	if (project.playbooks) {
		_(project).defaults({
			playbookCommand: 'ANSIBLE_FORCE_COLOR=true ansible-playbook'
		});
	}

	return project;
};

module.exports = function(app) {
	var logger = app.lib.logger('projects resource'),
		resource = app.dataio.resource('projects');

	patchDirstributor(app.builds.distributor);

	app.projects.on('projectLoaded', function(project) {
		extendProject(project);
	});

	resource.use('createBuildDataResource', function(req, res) {
		helpers.createBuildDataResource(app, req.data.buildId);
		res.send();
	});

	resource.use('readAll', function(req, res) {
		var filteredProjects = app.projects.getAll(),
			nameQuery = req.data && req.data.nameQuery;

		filteredProjects = app.projects.filter(function(project) {
			return !project.archived;
		});

		if (nameQuery) {
			filteredProjects = _(filteredProjects).filter(function(project) {
				return project.name.indexOf(nameQuery) !== -1;
			});
		}

		filteredProjects = _(filteredProjects).sortBy('name');

		res.send(filteredProjects);
	});

	// get project with additional fields
	var getProject = function(name, callback) {
		var project;
		Steppy(
			function() {
				project = _(app.projects.get(name)).clone();

				app.builds.getRecent({
					projectName: project.name,
					status: 'done',
					limit: 10
				}, this.slot());

				app.builds.getDoneStreak({projectName: project.name}, this.slot());
			},
			function(err, doneBuilds, doneBuildsStreak) {
				project.avgBuildDuration = app.builds.getAvgBuildDuration(doneBuilds);
				project.lastDoneBuild = doneBuilds[0];
				project.doneBuildsStreak = doneBuildsStreak;

				this.pass(project);
			},
			callback
		);
	};

	// resource custom method which finds project by name
	// and emits event about it change to clients
	resource.clientEmitSyncChange = function(name) {
		Steppy(
			function() {
				getProject(name, this.slot());
			},
			function(err, project) {
				resource.clientEmitSync('change', {project: project});
			},
			function(err) {
				console.error(
					'Error during sync project change occurred:',
					err.stack || err
				);
			}
		);
	};

	resource.use('read', function(req, res) {
		Steppy(
			function() {
				getProject(req.data.name, this.slot());
			},
			function(err, project) {
				res.send(project);
			}
		);
	});

	resource.use('run', function(req, res) {
		var projectName = req.data.projectName,
			buildParams = req.data.buildParams;

		logger.log(
			'Run the project: "%s" with params: %j',
			projectName,
			buildParams
		);

		app.builds.create({
			projectName: projectName,
			initiator: {type: 'user'},
			queueQueued: true,
			buildParams: buildParams
		});

		res.send();
	});

	return resource;
};