angular-cn/tools/broccoli/traceur/index.ts

var fs = require('fs');
var fse = require('fs-extra');
var path = require('path');
var traceur = require('../../transpiler');
var walkSync = require('walk-sync');
var Writer = require('broccoli-writer');
var xtend = require('xtend');

class TraceurFilter extends Writer {
  constructor(private inputTree, private destExtension: string = '.js',
              private options = {}, private hackSourceMapExtension: boolean = false) {}
  static RUNTIME_PATH = traceur.RUNTIME_PATH;
  write(readTree, destDir) {
    return readTree(this.inputTree)
        .then(srcDir => {
          walkSync(srcDir)
              .filter(filepath =>
                      {
                        var extension = path.extname(filepath).toLowerCase();
                        return extension === '.js' || extension === '.es6' || extension === '.cjs';
                      })
              .map(filepath => {
                var options = xtend({filename: filepath}, this.options);

                var fsOpts = {encoding: 'utf-8'};
                var sourcecode = fs.readFileSync(path.join(srcDir, filepath), fsOpts);

                var result = traceur.compile(options, filepath, sourcecode);

                // TODO: we should fix the sourceMappingURL written by Traceur instead of overriding
                // (but we might switch to typescript first)
                var url = path.basename(filepath).replace(/\.es6$/, '') +
                  (this.destExtension === '.js' ? '.js.map' : '.map');
                if (this.hackSourceMapExtension) {
                  url = path.basename(filepath).replace(/\.\w+$/, '') + '.map';
                }
                result.js = result.js + `\n//# sourceMappingURL=./${url}`;

                var destFilepath = filepath.replace(/\.\w+$/, this.destExtension);
                var destFile = path.join(destDir, destFilepath);
                fse.mkdirsSync(path.dirname(destFile));
                var destMap = path.join(destDir, destFilepath + '.map');


                fs.writeFileSync(destFile, result.js, fsOpts);

                result.sourceMap.file = destFilepath;
                fs.writeFileSync(destMap, JSON.stringify(result.sourceMap), fsOpts);
              });
        });
  }
}

module.exports = TraceurFilter;