emberjs/emberjs-build

'use strict';

var merge            = require('lodash/object/merge');
var util             = require('util');
var Funnel           = require('broccoli-funnel');
var concat           = require('broccoli-concat');
var writeFile        = require('broccoli-file-creator');
var mergeTrees       = require('broccoli-merge-trees');
var transpileES6     = require('./utils/transpile-es6');
var replaceVersion   = require('./utils/replace-version');

var debug               = require('./utils/debug-tree');
var getVendoredPackages = require('./get-vendored-packages');

var defaultOptions = {
  inputFiles: ['**/*.js'],
  generators: 'generators',
  wrapInIIFE: true
};

/*
  Responsible for concatenating ES6 modules together wrapped in loader and iife
  (immediately-invoked function expression)
*/
module.exports = function concatenateES6Modules(inputTrees, _options) {
  var options = merge({}, defaultOptions, _options || {});
  var vendoredPackages  = options.vendoredPackages || getVendoredPackages();
  var loader        = _options.loader || vendoredPackages['loader'];
  var externalHelpers = _options.externalHelpers || vendoredPackages['external-helpers'];
  var inputFiles    = options.inputFiles;
  var headerFiles   = [];
  var footerFiles   = [];
  var destFile      = options.destFile;

  var sourceTrees;

  // if given an array of trees merge into single tree
  if (util.isArray(inputTrees)) {
    sourceTrees = mergeTrees(inputTrees, {
      overwrite: true,
      annotation: destFile
    });
  } else {
    sourceTrees = inputTrees;
  }

  var includeDevHelpers = options.babel ? options.babel.includeDevHelpers : false;

  sourceTrees = transpileES6(sourceTrees, destFile, {
    plugins: options.babel ? options.babel.plugins : [],
    stripRuntimeChecks: !includeDevHelpers
  });

  if (!loader) {
    throw new TypeError('vendoredPackages.loader not found, it is not optional');
  }
  var concatTrees = [
    replaceVersion(options.generators, {
      version: options.version,
      annotation: 'generators'
    }),
    sourceTrees
  ];

  if (!options.includeLicense) { headerFiles.push('license.js'); }
  if (options.includeLoader)   {
    concatTrees.push(loader);

    var helpersFile;
    headerFiles.push('loader/index.js');

    if (includeDevHelpers) {
      helpersFile = 'external-helpers/external-helpers-dev.js';
    } else {
      helpersFile = 'external-helpers/external-helpers-prod.js';
    }

    headerFiles.push(helpersFile);
    concatTrees.push(new Funnel(externalHelpers, {
      include: [helpersFile]
    }));
  }

  if (options.bootstrapModule) {
    options.bootstrapModules = [options.bootstrapModule];
  }

  if (options.moduleExport || options.moduleExportDefault) {
    options.bootstrapModules = [];
  }

  if (options.bootstrapModules) {
    var annotation = [];
    var contents = options.bootstrapModules.map(function(module) {
      annotation.push(module);
      return 'requireModule("' + module + '");';
    })
    .join('\n');

    if (options.moduleExport) {
      annotation.push(options.moduleExport);
      contents += '(function (m) { if (typeof module === "object" && module.exports) { module.exports = m } }(requireModule("'+options.moduleExport+'")));\n';
    } else if (options.moduleExportDefault) {
      annotation.push(options.moduleExportDefault);
      contents += '(function (m) { if (typeof module === "object" && module.exports) { module.exports = m } }(requireModule("'+options.moduleExportDefault+'").default));\n';
    }

    var bootstrapTree = writeFile('bootstrap', contents + '\n', {
      annotation: 'require ' + annotation.join(' ')
    });

    concatTrees.push(bootstrapTree);
    footerFiles.push('bootstrap');
  }

  if (options.vendorTrees) {
    concatTrees.push(options.vendorTrees);
  }

  var footer = '';

  if (options.footer) {
    footer += options.footer;
  }

  if (options.wrapInIIFE) {
    footer += '}());';
  }

  return concat(debug(mergeTrees(concatTrees, { annotation: destFile }), 'concatted-es6' + destFile), {
    sourceMapConfig: {
      enabled: !!options.enableSourceMaps
    },
    header: options.wrapInIIFE && ';(function() {',
    headerFiles: headerFiles,
    inputFiles: inputFiles,
    footerFiles: footerFiles,
    footer: footer,
    outputFile: destFile,
    allowNone: true,
    annotation: destFile
  });
};