From 3977aab9567196913d1682f00a4ac86906ceb859 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Felix=20B=C3=B6hm?= <me@feedic.com>
Date: Thu, 1 Apr 2021 10:38:55 -0700
Subject: [PATCH] fix: All elements should have `type: tag` in `xmlMode`

---
 src/__fixtures__/23-dom-lvl1.json            |  2 +-
 src/__fixtures__/27-xml-no-special-tags.json | 29 ++++++++++++++++++++
 src/index.ts                                 | 10 ++++++-
 src/node.ts                                  | 20 +++++++-------
 4 files changed, 49 insertions(+), 12 deletions(-)
 create mode 100644 src/__fixtures__/27-xml-no-special-tags.json

diff --git a/src/__fixtures__/23-dom-lvl1.json b/src/__fixtures__/23-dom-lvl1.json
index 366fbf9d..4e38f411 100644
--- a/src/__fixtures__/23-dom-lvl1.json
+++ b/src/__fixtures__/23-dom-lvl1.json
@@ -1,6 +1,6 @@
 {
     "name": "DOM level 1",
-    "options": { "withDomLvl1": true },
+    "options": {},
     "html": "<div>some stray text<h1>Hello, world.</h1><!-- comment node -->more stray text</div>",
     "expected": [
         {
diff --git a/src/__fixtures__/27-xml-no-special-tags.json b/src/__fixtures__/27-xml-no-special-tags.json
new file mode 100644
index 00000000..d998b654
--- /dev/null
+++ b/src/__fixtures__/27-xml-no-special-tags.json
@@ -0,0 +1,29 @@
+{
+    "name": "XML mode: All tags should have `type: tag`",
+    "options": {
+        "xmlMode": true
+    },
+    "html": "<script><style><div>",
+    "expected": [
+        {
+            "type": "tag",
+            "name": "script",
+            "attribs": {},
+            "children": [
+                {
+                    "type": "tag",
+                    "name": "style",
+                    "attribs": {},
+                    "children": [
+                        {
+                            "type": "tag",
+                            "name": "div",
+                            "attribs": {},
+                            "children": []
+                        }
+                    ]
+                }
+            ]
+        }
+    ]
+}
diff --git a/src/index.ts b/src/index.ts
index ca2f1868..8ea0f6c9 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -42,6 +42,13 @@ export interface DomHandlerOptions {
      * @deprecated
      */
     normalizeWhitespace?: boolean;
+
+    /**
+     * Treat the markup as XML.
+     *
+     * @default false
+     */
+    xmlMode?: boolean;
 }
 
 // Default options
@@ -151,7 +158,8 @@ export class DomHandler {
     }
 
     public onopentag(name: string, attribs: { [key: string]: string }): void {
-        const element = new Element(name, attribs);
+        const type = this.options.xmlMode ? ElementType.Tag : undefined;
+        const element = new Element(name, attribs, undefined, type);
         this.addNode(element);
         this.tagStack.push(element);
     }
diff --git a/src/node.ts b/src/node.ts
index c1c37f54..9e1c22a6 100644
--- a/src/node.ts
+++ b/src/node.ts
@@ -184,17 +184,17 @@ export class Element extends NodeWithChildren {
     constructor(
         public name: string,
         public attribs: { [name: string]: string },
-        children: Node[] = []
+        children: Node[] = [],
+        type:
+            | ElementType.Tag
+            | ElementType.Script
+            | ElementType.Style = name === "script"
+            ? ElementType.Script
+            : name === "style"
+            ? ElementType.Style
+            : ElementType.Tag
     ) {
-        super(
-            name === "script"
-                ? ElementType.Script
-                : name === "style"
-                ? ElementType.Style
-                : ElementType.Tag,
-            children
-        );
-        this.attribs = attribs;
+        super(type, children);
     }
 
     // DOM Level 1 aliases