Apply plugins to tree, tested with include

Author: thewebartisan7

package-lock.json

@@ -45,6 +45,7 @@
     "markdown-it-toc-done-right": "^4.2.0",
     "posthtml": "^0.16.6",
     "posthtml-beautify": "^0.7.0",
+    "posthtml-include": "^1.7.4",
     "posthtml-markdownit": "^1.3.0",
     "xo": "^0.37.1"
   }

package.json

@@ -65,9 +65,6 @@ module.exports = (options = {}) => tree => {
   }
 
   options.roots = Array.isArray(options.folders) ? options.folders : [options.folders];
-  // options.roots.forEach((root, index) => {
-  //   options.roots[index] = path.resolve(options.root, root);
-  // });
   options.namespaces = Array.isArray(options.namespaces) ? options.namespaces : [options.namespaces];
   options.namespaces.forEach((namespace, index) => {
     options.namespaces[index].root = path.resolve(namespace.root);
@@ -109,6 +106,10 @@ function processTree(options) {
   let processCounter = 0;
 
   return function (tree) {
+    if (options.plugins.length > 0) {
+      tree = applyPluginsToTree(tree, options.plugins);
+    }
+
     match.call(tree, options.matcher, currentNode => {
       if (!currentNode.attrs) {
         currentNode.attrs = {};
@@ -149,6 +150,10 @@ function processTree(options) {
       // const plugins = [...options.plugins, expressions(options.expressions)];
       nextNode = expressions(options.expressions)(nextNode);
 
+      if (options.plugins.length > 0) {
+        nextNode = applyPluginsToTree(nextNode, options.plugins);
+      }
+
       // Process <yield> tag
       const content = match.call(nextNode, {tag: options.yield}, nextNode => {
         // Fill <yield> with current node content or default <yield> content or empty
@@ -183,3 +188,10 @@ function processTree(options) {
     return tree;
   };
 }
+
+function applyPluginsToTree(tree, plugins) {
+  return plugins.reduce((tree, plugin) => {
+    tree = plugin(tree);
+    return tree;
+  }, tree);
+}

src/index.js

@@ -0,0 +1 @@
+<div><include src="components/included-file.html"></include></div>

test/templates/components/component-with-include.html

@@ -0,0 +1 @@
+<module href="/components/module.html"><p>Module file</p></module>

test/templates/components/component-with-module.html

@@ -0,0 +1 @@
+<p>Included file</p>

test/templates/components/included-file.html

@@ -1,9 +1,9 @@
 'use strict';
 
-// const test = require('ava');
-// const plugin = require('../src');
-// const posthtml = require('posthtml');
-// const clean = html => html.replace(/(\n|\t)/g, '').trim();
+const test = require('ava');
+const plugin = require('../src');
+const posthtml = require('posthtml');
+const clean = html => html.replace(/(\n|\t)/g, '').trim();
 
 // test('Must work with posthtml-extend syntax', async t => {
 //   const actual = `<extends src="layouts/extend.html"><block name="content">My Content</block></extends>`;
@@ -31,3 +31,22 @@
 //
 //   t.is(html, expected);
 // });
+
+test('Must include file using posthtml-include', async t => {
+  const actual = `<component src="components/component-with-include.html"></component>`;
+  const expected = `<div><p>Included file</p></div>`;
+
+  const html = await posthtml([plugin({root: './test/templates', tag: 'component', attribute: 'src', plugins: [require('posthtml-include')({root: './test/templates', encoding: 'utf8'})]})]).process(actual).then(result => clean(result.html));
+
+  t.is(html, expected);
+});
+
+// Unfortunately it's not working
+// test('Must include file using posthtml-modules', async t => {
+//   const actual = `<component src="components/component-with-module.html"></component>`;
+//   const expected = `<div><p>Module file</p></div>`;
+//
+//   const html = await posthtml([plugin({root: './test/templates', tag: 'component', attribute: 'src', plugins: [require('posthtml-modules')({root: './test/templates'})]})]).process(actual).then(result => clean(result.html));
+//
+//   t.is(html, expected);
+// });