index.js

'use strict';

var ejs = require('ejs');
var path = require('path');

class EjsPlugin {
  constructor(config) {
    this.config = config && config.plugins && config.plugins.ejs || {};
    this.config = Object.assign(this.config, {
      watched: config && config.paths && config.paths.watched || ['app', 'test', 'vendor']
    });
  }

  _basePath(filePath) {
    var baseDir;
    this.config.watched.forEach(function(d) {
      if (filePath.indexOf(d) === 0) {
        baseDir = d;
      }
    });
    var relPath = path.relative(baseDir, filePath);
    return relPath && `${path.dirname(relPath)}/`;
  }

  // file: File => Promise[File]
  // Generates a compilation function from a EJS template string
  compile(params) {
    var options = {client: true};

    try {
      var basePath = params.path && this._basePath(params.path);

      // Generate compile function
      var fn = ejs.compile(params.data, options);

      // Return a callable function which will execute the function
      // created by the source-code, with the passed data as locals
      // Add a local `include` function which uses require to load files
      var returnedFn = function(data) {
        var include = function(includePath, includeData) {
          var _fn = require(basePath + includePath);
          return _fn(includeData);
        };
        return fn(data, null, include);
      };

      var module = 'var basePath = ' + JSON.stringify(basePath) + ';\n' +
        'var fn = ' + fn + ';\n' +
        'var __module = ' + returnedFn + ';\n' +
        'if (typeof define === \'function\' && define.amd) {\n' +
        '  define([], function() {\n' +
        '    return __module;\n' +
        '  });\n' +
        '} else if (typeof module === \'object\' && module && module.exports) {\n' +
        '  module.exports = __module;\n' +
        '} else {\n' +
        '  __module;\n' +
        '}';

      return Promise.resolve(module);
    } catch (error) {
      console.error(error);
      return Promise.reject(error);
    }
  }
}

EjsPlugin.prototype.brunchPlugin = true;
EjsPlugin.prototype.type = 'javascript';
EjsPlugin.prototype.pattern = /\.ejs$/;

module.exports = EjsPlugin;