added option.alias(dir path) support for define file load path

Akimyou · Mar 7, 2018 · b13a009 · b13a009
1 parent 704d289
commit b13a009
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 3 deletions.
diff --git a/lib/preprocess.js b/lib/preprocess.js
@@ -61,7 +61,8 @@ function preprocess(src, context, typeOrOptions) {
     fileNotFoundSilentFail: false,
     srcDir: process.cwd(),
     srcEol: getEolType(src),
-    type: delim['html']
+    type: delim['html'],
+    alias: {}
   };
 
   // needed for backward compatibility with 2.x.x series
@@ -82,6 +83,7 @@ function preprocess(src, context, typeOrOptions) {
     options.fileNotFoundSilentFail = typeOrOptions.fileNotFoundSilentFail || options.fileNotFoundSilentFail;
     options.srcEol = typeOrOptions.srcEol || options.srcEol;
     options.type = delim[typeOrOptions.type] || options.type;
+    options.alias = typeOrOptions.alias || options.alias;
   }
 
   context = copy(context);
@@ -327,7 +329,24 @@ function processIncludeDirective(isStatic, context, opts, match, linePrefix, fil
   var indent = linePrefix.replace(/\S/g, ' ');
   var includedContext = copy(context);
   var includedOpts = copy(opts);
-  includedContext.src = path.join(opts.srcDir,file);
+
+  var aliasKeys = Object.keys(opts.alias);
+  for (var i = 0; i < aliasKeys.length; i++) {
+    var aliasKey = aliasKeys[i];
+    var aliasKeyReg = new RegExp(aliasKey);
+    var aliasVal = opts.alias[aliasKey];
+
+    if (file.match(aliasKeyReg)) {
+      file = file.replace(aliasKeyReg, aliasVal);
+      break;
+    }
+  }
+
+  if (file[0] === '/' || file[0] === '\\') {
+    includedContext.src = file;
+  } else {
+    includedContext.src = path.join(opts.srcDir,file);
+  }
   includedOpts.srcDir = path.dirname(includedContext.src);
 
   var fileContents = getFileContents(includedContext.src, opts.fileNotFoundSilentFail, context.src);

diff --git a/test/options.spec.js b/test/options.spec.js
@@ -64,4 +64,11 @@ describe('shall support multiple call signatures', function () {
       pp.preprocess(input, {}, {srcEol: '\r\n', srcDir: 'test/fixtures/include'}).should.equal("a\r\n!bazqux!\r\nc");
     });
   });
-});
+
+  describe('and support alias(dir path) options', function () {
+    it('and use alias option', function () {
+      input = "a<!--@include @static.txt-->c";
+      pp.preprocess(input, {}, { alias: { '@': 'test/fixtures/include/' } }).should.equal("a!bazqux!c");
+    });
+  });
+});