Fixed Missing HTML Instructions #32

soranoo · Mar 16, 2024 · c2e71a4 · c2e71a4
1 parent e456f02
commit c2e71a4
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 1 deletion.
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "next-css-obfuscator",
-  "version": "2.2.14",
+  "version": "2.2.15",
   "description": "A package deeply inspired by PostCSS-Obfuscator but for Next.js.",
   "main": "dist/index.js",
   "type": "commonjs",

diff --git a/src/handlers/html.test.ts b/src/handlers/html.test.ts
@@ -201,4 +201,17 @@ describe("obfuscateHtmlClassNames", () => {
     expect(result.obfuscatedContent).toEqual(`<div class="key a b c"></div>`);
     expect(result.usedKeys).to.deep.equal([".foo", ".bar", ".baz"]);
   });
+
+  it("should handle HTML instruction", () => {
+    // Arrange
+    const html = `<!DOCTYPE html><div class="foo"></div>`;
+    const selectorConversion: SelectorConversion = { ".foo": ".a" };
+
+    // Act
+    const result = obfuscateHtmlClassNames({ html, selectorConversion });
+
+    // Assert
+    expect(result.obfuscatedContent).toEqual(`<!DOCTYPE html><div class="a"></div>`);
+    expect(result.usedKeys).to.deep.equal([".foo"]);
+  });
 });
diff --git a/src/handlers/html.ts b/src/handlers/html.ts
@@ -132,6 +132,9 @@ function obfuscateHtmlClassNames({
   const usedKeys: string[] = [];
 
   const parser = new htmlparser2.Parser({
+    onprocessinginstruction(name, data) {
+      modifiedHtml += `<${data}>`;
+    },
     onopentag(tagName, attribs) {
       if (tagName === "script") {
         isScriptTag = true;