tec: Provide a Dom::html method

flusio · Jul 29, 2023 · a2bd429 · a2bd429
1 parent 72f0d71
commit a2bd429
Show file tree

Hide file tree

Showing 2 changed files with 52 additions and 0 deletions.
diff --git a/lib/SpiderBits/src/Dom.php b/lib/SpiderBits/src/Dom.php
@@ -109,6 +109,28 @@ public function list(): mixed
         }
     }
 
+    /**
+     * Return the HTML of the current selected node(s) as a string
+     */
+    public function html(): string
+    {
+        if ($this->nodes_selected) {
+            $htmls = [];
+            foreach ($this->nodes_selected as $node) {
+                $htmls[] = $this->dom->saveHTML($node);
+            }
+            return implode("\n", $htmls);
+        } else {
+            $html = $this->dom->saveHTML();
+
+            if ($html === false) {
+                $html = '';
+            }
+
+            return $html;
+        }
+    }
+
     /**
      * Return the content of the current selected node(s) as a string
      */

diff --git a/tests/lib/SpiderBits/DomTest.php b/tests/lib/SpiderBits/DomTest.php
@@ -26,6 +26,36 @@ public function testTextWithMixOfHtmlEntitiesAndUtf8(): void
         $this->assertSame("Site d'information français", $text);
     }
 
+    public function testHtml(): void
+    {
+        $dom = Dom::fromText(<<<HTML
+            <div>Hello World!</div>
+        HTML, LIBXML_HTML_NOIMPLIED | LIBXML_HTML_NODEFDTD);
+
+        $html = $dom->html();
+
+        $this->assertSame("<div>Hello World!</div>\n", $html);
+    }
+
+    public function testHtmlWithSelectedNodes(): void
+    {
+        $dom = Dom::fromText(<<<HTML
+            <div>
+                <p>Hello</p>
+            </div>
+            <div>
+                <p>World!</p>
+            </div>
+        HTML);
+        $p_nodes = $dom->select('//p');
+
+        $this->assertNotNull($p_nodes);
+
+        $html = $p_nodes->html();
+
+        $this->assertSame("<p>Hello</p>\n<p>World!</p>", $html);
+    }
+
     public function testSelect(): void
     {
         $dom = Dom::fromText(<<<HTML