Add dependencies cache

Author: mcmath

lib/dependencies.js

@@ -0,0 +1,28 @@
+'use strict';
+
+var cache = {};
+
+// Add dependencies from cache in case of error
+exports.addFromCache = function(context) {
+  add(context, get(context));
+};
+
+// Add dependencies and from compiled template and save to cache
+exports.addAndSave = function(context, deps) {
+  add(context, save(context, deps));
+};
+
+function get(context) {
+  return cache[context.resourcePath] || [];
+}
+
+function save(context, deps) {
+  cache[context.resourcePath] = deps;
+  return get(context);
+}
+
+function add(context, deps) {
+  deps.forEach(function(dep) {
+    context.addDependency(dep);
+  });
+}

lib/ejs-html-loader.js

@@ -0,0 +1,37 @@
+'use strict';
+
+var parseQuery = require('loader-utils').parseQuery;
+var compile = require('ejs').compile;
+var dependencies = require('./dependencies');
+
+module.exports = function ejsHtmlLoader(source) {
+  this.cacheable();
+
+  var options = this.options;
+  var query = parseQuery(this.query);
+  var data = options.ejsHtmlLoader || options.ejsHtml || query;
+
+  var template;
+  var rendered;
+
+  try {
+    template = compile(source, {
+      filename: this.resourcePath,
+      delimiter: data.delimiter,
+      context: data.context
+    });
+    rendered = template(data);
+  } catch (e) {
+    dependencies.addFromCache(this);
+    throwError(e.message);
+  }
+
+  dependencies.addAndSave(this, template.dependencies);
+  return rendered;
+};
+
+function throwError(message) {
+  var err = new Error('ejs-html-loader\n' + message);
+  Error.captureStackTrace(err, message);
+  throw err;
+}