gulp-file-include/index.js

'use strict';

var concat = require('concat-stream'),
  es = require('event-stream'),
  gutil = require('gulp-util'),
  path = require('path'),
  fs = require('fs');

module.exports = function(options) {
  var prefix, basepath, filters, context;

  if (typeof options === 'object') {
    basepath = options.basepath || '@file';
    prefix = options.prefix || '@@';
    context = options.context || {};
    filters = options.filters;
  } else {
    prefix = options || '@@';
    basepath = '@file';
    context = {};
  }

  var includeRegExp = new RegExp(prefix + 'include\\s*\\([^)]*["\'](.*?)["\'](,\\s*({[\\s\\S]*?})){0,1}\\s*\\)+');

  function fileInclude(file) {
    var self = this;

    if (file.isNull()) {
      self.emit('data', file);
    } else if (file.isStream()) {
      file.contents.pipe(concat(function(data) {
        var text = String(data);
        text = stripCommentedIncludes(text);
        text = parseConditionalIncludes(text);

        try {
          self.emit('data', include(file, text));
        } catch (e) {
          self.emit('error', new gutil.PluginError('gulp-file-include', e.message));
        }
      }));
    } else if (file.isBuffer()) {
      try {
        var text = String(file.contents);
        text = stripCommentedIncludes(text);
        text = parseConditionalIncludes(text);

        self.emit('data', include(file, text));
      } catch (e) {
        self.emit('error', new gutil.PluginError('gulp-file-include', e.message));
      }
    }
  }

  return es.through(fileInclude);

  /**
   * utils
   */
  function stripCommentedIncludes(content) {
    // remove single line html comments that use the format: <!-- @@include() -->
    var regex = new RegExp('<\!--(.*)' + prefix + 'include([\\s\\S]*?)-->', 'g');
    return content.replace(regex, '');
  }

  function parseConditionalIncludes(content) {
    // parse @@if (something) { include('...') }
    var regexp = new RegExp(prefix + 'if.*\\{[^{}]*\\}\\s*'),
      matches = regexp.exec(content),
      included = false;

    context.content = content;

    while (matches) {
      var match = matches[0],
        includeExps = /\{([^{}]*)\}/.exec(match)[1];

      // jshint ignore: start
      var exp = /if(.*)\{/.exec(match)[1];
      included = new Function('var context = this; return ' + exp + ';').call(context);
      // jshint ignore: end

      if (included) {
        content = content.replace(match, includeExps);
      } else {
        content = content.replace(match, '');
      }

      matches = regexp.exec(content);
    }

    return content;
  }

  function include(file, text) {
    var filebase = basepath === "@file" ? path.dirname(file.path) : basepath === "@root" ? process.cwd() : basepath;
    var matches = includeRegExp.exec(text);

    filebase = path.resolve(process.cwd(), filebase);

    // for checking if we are not including the current file again
    var currentFilename = path.resolve(file.base, file.path);

    while (matches) {
      var match = matches[0];
      var includePath = path.resolve(filebase, matches[1]);

      if (currentFilename.toLowerCase() === includePath.toLowerCase()) {
        throw new Error('recursion detected in file: ' + currentFilename);
      }

      var includeContent = fs.readFileSync(includePath);

      // strip utf-8 BOM  https://github.com/joyent/node/issues/1918
      includeContent = includeContent.toString('utf-8').replace(/\uFEFF/, '');

      // need to double each `$` to escape it in the `replace` function
      includeContent = includeContent.replace(/\$/gi, '$$$$');

      // apply filters on include content
      if (typeof filters === 'object') {
        includeContent = applyFilters(includeContent, match);
      }

      var recMatches = includeRegExp.exec(includeContent);
      if (recMatches && basepath == "@file") {
        var recFile = new gutil.File({
          cwd: process.cwd(),
          base: file.base,
          path: includePath,
          contents: new Buffer(includeContent)
        });
        recFile = include(recFile, includeContent);
        includeContent = String(recFile.contents);
      }

      text = text.replace(match, includeContent);

      if (matches[3]) {
        // replace variables
        var data = JSON.parse(matches[3]);
        for (var k in data) {
          text = text.replace(new RegExp(prefix + k, 'g'), data[k]);
        }
      }

      matches = includeRegExp.exec(text);
    }

    file.contents = new Buffer(text);
    return file;
  }

  function applyFilters(includeContent, match) {
    if (match.match(/\)+$/)[0].length === 1) {
      // nothing to filter return unchanged
      return includeContent;
    }

    // now get the ordered list of filters
    var filterlist = match.split('(').slice(1, -1);
    filterlist = filterlist.map(function(str) {
      return filters[str.trim()];
    });

    // compose them together into one function
    var filter = filterlist.reduce(compose);

    // and apply the composed function to the stringified content
    return filter(String(includeContent));
  }
};

function compose(f, g) {
  return function(x) {
    return f(g(x));
  };
}