grunt-drush/tasks/drush.js

Drush task runner for grunt.

/*
 * grunt-drush
 * https://github.com/nickpack/grunt-drush
 *
 * Copyright (c) 2013 Nick Pack
 * Licensed under the MIT license.
 */

 'use strict';

 module.exports = function(grunt) {

  var _ = require('lodash'),
      fs = require('fs'),
      path = require('path'),
      spawn = require('win-spawn'),
      os = require('os'),
      async = require('async'),
      concurrencyLevel = (os.cpus().length || 1) * 2;

  grunt.registerMultiTask('drush', 'Drush task runner for grunt.', function() {
    var self = this,
        options = self.options({
          cmd: 'drush',
          cwd: false,
          options: {stdio: 'inherit'}
        }),
        args = self.data.args,
        done = this.async();

    if (grunt.option('debug')) {
        args.push('--debug');
    }
    if (grunt.option('verbose')) {
        args.push('--verbose');
    }
    if (grunt.option('no-write')) {
        args.push('--simulate');
    }

    grunt.verbose.writeflags(options, 'Options');
    grunt.verbose.writeflags(args, 'Args');

    var callDrush = function(spawnArgs, filesNext) {

      var origCwd = process.cwd(),
          drushResult = null;

      if (options.cwd) {
        grunt.file.setBase(options.cwd);
      }

      var cp = spawn(options.cmd, spawnArgs, options.options);

      cp.on('error', grunt.warn);
      cp.on('close', function (code) {

        switch (code) {
          case 127:
            grunt.fatal(
              'You need to have drush installed in your PATH\n' +
              'or set it in the configuration for this task to work.'
            );
            break;

          case 0:
            grunt.verbose.writeln('Drush completed without error.');
            break;

          default:
            grunt.warn('Drush failed: ' + code);
            break;
        }

        drushResult = code === 0 || false;

        if (!_.isUndefined(filesNext)) {
          filesNext();
        } else {
          done(drushResult);
        }

      });

      grunt.file.setBase(origCwd);

    };

    var processFiles = function() {

      async.eachSeries(self.files, function (file, next) {
        var fileArgs;

        if (_.isString(file.dest)) {
          fileArgs = args.concat([file.dest]);
        }

        callDrush(fileArgs, next);
      }, done);

    };

    if (this.files.length > 0) {
      processFiles();
    } else {
      callDrush(args);
    }
  });

};