1 files changed, 69 insertions, 0 deletions
diff --git a/node_modules/@11ty/eleventy-plugin-bundle/src/bundlePlucker.js b/node_modules/@11ty/eleventy-plugin-bundle/src/bundlePlucker.js
new file mode 100644
index 0000000..5dcad46
--- /dev/null
+++ b/node_modules/@11ty/eleventy-plugin-bundle/src/bundlePlucker.js
@@ -0,0 +1,69 @@
+import debugUtil from "debug";
+import matchHelper from "posthtml-match-helper";
+
+const debug = debugUtil("Eleventy:Bundle");
+
+const ATTRS = {
+	ignore: "eleventy:ignore",
+	bucket: "eleventy:bucket",
+};
+
+const POSTHTML_PLUGIN_NAME = "11ty/eleventy/html-bundle-plucker";
+
+function hasAttribute(node, name) {
+	return node?.attrs?.[name] !== undefined;
+}
+
+function addHtmlPlucker(eleventyConfig, bundleManager) {
+	let matchSelector = bundleManager.getPluckedSelector();
+
+	if(!matchSelector) {
+		throw new Error("Internal error: missing plucked selector on bundle manager.");
+	}
+
+	eleventyConfig.htmlTransformer.addPosthtmlPlugin(
+		"html",
+		function (context = {}) {
+			let pageUrl = context?.url;
+			if(!pageUrl) {
+				throw new Error("Internal error: missing `url` property from context.");
+			}
+
+			return function (tree, ...args) {
+				tree.match(matchHelper(matchSelector), function (node) {
+					try {
+						// ignore
+						if(hasAttribute(node, ATTRS.ignore)) {
+							delete node.attrs[ATTRS.ignore];
+							return node;
+						}
+
+						if(Array.isArray(node?.content) && node.content.length > 0) {
+							// TODO make this better decoupled
+							if(node?.content.find(entry => entry.includes(`/*__EleventyBundle:`))) {
+								// preserve {% getBundle %} calls as-is
+								return node;
+							}
+
+							let bucketName = node?.attrs?.[ATTRS.bucket];
+							bundleManager.addToPage(pageUrl, [ ...node.content ], bucketName);
+
+							return { attrs: [], content: [], tag: false };
+						}
+					} catch(e) {
+						debug(`Bundle plucker: error adding content to bundle in HTML Assets: %o`, e);
+						return node;
+					}
+
+					return node;
+				});
+			};
+		},
+		{
+			// pluginOptions
+			name: POSTHTML_PLUGIN_NAME,
+		},
+	);
+}
+
+export { addHtmlPlucker };