index.js

/* jshint node: true */
'use strict';

var path = require('path');
var Funnel = require('broccoli-funnel');
var mergeTrees = require('broccoli-merge-trees');
var coverageMiddleware = require('./lib/coverage-middleware');
var bodyParser = require('body-parser');

function logErrors(err, req, res, next) {
  console.error(err.stack);
  next(err);
}

module.exports = {
  name: 'Ember CLI Blanket',

  validEnv: function() {
    return this.app.env !== 'production';
  },

  blueprintsPath: function() {
    return path.join(__dirname, 'blueprints');
  },

  included: function included(app) {
    this._super.included(app);

    if (app.tests) {
      var fileAssets = [
        path.join(app.bowerDirectory, 'blanket', 'dist', 'qunit', 'blanket.js'),
      ];

      fileAssets.forEach(function(file) {
        app.import(file, {
          type: 'test'
        });
      });
    }
  },

  middleware: function(app, options) {
    app.post('/write-blanket-coverage',
             bodyParser.json({ limit: '50mb' }),
             coverageMiddleware(options),
             logErrors);
  },
  testemMiddleware: function(app) {
    this.middleware(app, { root: this.project.root });
  },
  serverMiddleware: function(options) {
    this.app = options.app;
    if(!this.validEnv()) { return; }
    this.middleware(options.app, { root: this.project.root });
  },
  postprocessTree: function(type, tree) {
    this._requireBuildPackages();

    if (type === 'all' && this.app.tests) {
      var treeTestLoader = new Funnel(tree, {
        files: ['test-loader.js'],
        srcDir: 'assets',
        destDir: 'app'
      });

      var tests = this.treeGenerator(path.join(this.project.root, 'tests'));

      var blanketOptions = new Funnel(tests, {
        files: ['blanket-options.js'],
        srcDir: '/',
        destDir: '/assets'
      });

      var vendor = this.treeGenerator(path.join(__dirname, 'vendor'));

      var blanketLoader = new Funnel(vendor, {
        files: ['blanket-reporter.js', 'blanket-require.js'],
        srcDir: '/',
        destDir: '/'
      });

      blanketLoader = this.concatFiles(blanketLoader, {
        inputFiles: ['blanket-reporter.js', 'blanket-require.js'], // Order here is important
        outputFile: '/assets/blanket-loader.js'
      });

      var start = new Funnel(vendor, {
        files: ['start.js'],
        srcDir: '/',
        destDir: '/'
      });

      var testLoaderTree = this.concatFiles(mergeTrees([treeTestLoader, start]), {
        inputFiles: ['**/*.js'],
        outputFile: '/assets/test-loader.js'
      });

      return mergeTrees([tree, blanketOptions, blanketLoader, testLoaderTree], {
        overwrite: true
      });
    } else {
      return tree;
    }
  }
};