index.js

/*
 **  grunt-traceur-simple -- Grunt Task for ECMAScript 6 to ECMAScript 5 Transpiling
 **  Copyright (c) 2014 Ralf S. Engelschall <rse@engelschall.com>
 **
 **  Permission is hereby granted, free of charge, to any person obtaining
 **  a copy of this software and associated documentation files (the
 **  "Software"), to deal in the Software without restriction, including
 **  without limitation the rights to use, copy, modify, merge, publish,
 **  distribute, sublicense, and/or sell copies of the Software, and to
 **  permit persons to whom the Software is furnished to do so, subject to
 **  the following conditions:
 **
 **  The above copyright notice and this permission notice shall be included
 **  in all copies or substantial portions of the Software.
 **
 **  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 **  EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 **  MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
 **  IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
 **  CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
 **  TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
 **  SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 */

/* global require:   false */
/* global module:    false */
/* global process:   false */
/* global __dirname: false */

/*  Grunt plugin information  */
var NAME = "traceur";
var DESC = "Transpiles ECMAScript 6 to ECMAScript 5 with Traceur";

/*  external requirements  */
var exec  = require("child_process").exec;
var os    = require("os");
var chalk = require("chalk");
var traceurCompiler = require("./xing-traceur/traceurCompiler")
var fs = require("fs");
var og = require("grunt-legacy-log").Log;
var Log = require('grunt-legacy-log').Log;
var objectAssign = require('object-assign');
var gruntStyleLog = new Log();

/*  external paths to Traceur  */
var traceurRuntimePath = traceurCompiler.RUNTIME_PATH

var defaultOptions = {
  traceurRuntime: traceurRuntimePath,
  traceurOptions: {},
  moduleMaps: {},
  includeRuntime: false
}

function runTraceur(log) {
  function makeDone(promise, done){
    return promise.then(function(){
      //console.log("tasks/traceur.js:61", "yes");
      done(true);}, function(rej){
      //console.log("tasks/traceur.js:61", "no");
        //grunt.warn(rej);
        done(false); });
  }

  function injectRuntime(options){
    return function(f) {
      log.writeln("injecting:   " + chalk.green(f.dest + " <- " + options.traceurRuntime));
      var rt = fs.readFileSync(options.traceurRuntime, { encoding: "utf8" });
      var txt = fs.readFileSync(f.dest, { encoding: "utf8" });
      txt = rt + "\n" + txt;
      fs.writeFileSync(f.dest, txt, { encoding: "utf8" });
    }
  }

  function logCompileErrors(f) {
    return function(err){
      var errors;
      if (Array.isArray(err)) {
        errors = err;
      } else {
        errors = err.errors || [err];
      }
      log.writeln("error while transpiling: " + chalk.red(f.dest + " <- " + f.src.join(" ")));
      errors.forEach(function(err) {
        log.error(err.stack || err);
      });
      throw new Error(errors[0]);
    }
  }

  function compileSingleFile(f, commandOptions) {
    out = f.dest;
    rootSources = f.src.map(function (name) { return {name: name, type: 'module'}})

    var promise = traceurCompiler.recursiveCompile(out, rootSources, commandOptions)
    .then(function() {
      log.writeln("transpiled: " + chalk.green(f.dest + " <- " + f.src.join(" ")));
      return f;
    }, logCompileErrors(f));

    return promise;
  }


  /*  export the Grunt task  */
  return function (options, files, done) {
    /*  provide default options  */
    log.verbose.writeflags(options, "Options");

    var tasksCur = 0;
    var tasksMax = files.length;
    if (options.includeRuntime && tasksMax !== 1) {
      throw "including the runtime into more than one output file rejected";
    }

    if (options.moduleMaps) {
      Object.keys(options.moduleMaps).forEach(function(key) {
        System.map[key] = options.moduleMaps[key];
      });
    }

    commandOptions = traceurCompiler.buildOptions(options.traceurOptions);

    if (options.srcDir && options.destDir) {
      var f = { src: [options.srcDir], dest: options.destDir };
      return makeDone( traceurCompiler.compileAllJsFilesInDir(options.srcDir, options.destDir, commandOptions).then(function(pairs){
        pairs.forEach(function(pair){
          log.verbose.writeln("  transpiled: " + chalk.green(pair.in + " <- " + pair.out));
        });
        log.writeln(chalk.green("Transpiled " + pairs.length + " files"));
        return pairs;
      }, logCompileErrors(f)), done );
    } else {
      return makeDone(Promise.all(files.map(function (f) {
        var promise = compileSingleFile(f, commandOptions);
        if (options.includeRuntime) {
          promise.then(injectRuntime(options));
        }
        return promise;
      })), done);
    }
  };
}

var rawTask = function(options, files, done) {
  var mergedOptions = objectAssign(defaultOptions, options);
  return runTraceur(gruntStyleLog)(mergedOptions, files, done);

}

var gruntTask = function(grunt) {
  traceurTask = runTraceur(grunt.log);
  var withCatch = function() {
    try {
      var options = this.options(defaultOptions);
      traceurTask(options, this.files, this.async())
    }
    catch (e) {
      grunt.fail.fatal(e)
    }
  }
  grunt.registerMultiTask(NAME, DESC, withCatch);
}

module.exports = {
  rawTask: rawTask,
  gruntTask: gruntTask
}