From 7335b1d0a4f4747e2eb6dda5fa41f5c8b2c20494 Mon Sep 17 00:00:00 2001
From: Pierre Tachoire <pierre@lightpanda.io>
Date: Wed, 20 Aug 2025 16:02:27 +0200
Subject: [PATCH] escape incoming plain text

---
 src/browser/page.zig | 31 +++++++++++++++++++++++++++----
 1 file changed, 27 insertions(+), 4 deletions(-)

diff --git a/src/browser/page.zig b/src/browser/page.zig
index 758d4a55..0633f44c 100644
--- a/src/browser/page.zig
+++ b/src/browser/page.zig
@@ -209,12 +209,18 @@ pub const Page = struct {
             },
             .raw_done => |data| return out.writeAll(data),
             .text => {
-                // processed below, along with .html
-                // return the <pre> element from the HTML
+                // returns the <pre> element from the HTML
                 const doc = parser.documentHTMLToDocument(self.window.document);
                 const list = try parser.documentGetElementsByTagName(doc, "pre");
                 const pre = try parser.nodeListItem(list, 0) orelse return error.InvalidHTML;
-                return Dump.writeChildren(pre, .{}, out);
+                const walker = Walker{};
+                var next: ?*parser.Node = null;
+                while (true) {
+                    next = try walker.get_next(pre, next) orelse break;
+                    const v = try parser.nodeTextContent(next.?) orelse return;
+                    try out.writeAll(v);
+                }
+                return;
             },
             .html => {
                 // maybe page.wait timed-out, print what we have
@@ -656,7 +662,24 @@ pub const Page = struct {
         }
 
         switch (self.mode) {
-            .html, .text => |*p| try p.process(data),
+            .html => |*p| try p.process(data),
+            .text => |*p| {
+                // we have to escape the data...
+                var v = data;
+                while (v.len > 0) {
+                    const index = std.mem.indexOfAnyPos(u8, v, 0, &.{ '<', '>' }) orelse {
+                        try p.process(v);
+                        return;
+                    };
+                    try p.process(v[0..index]);
+                    switch (v[index]) {
+                        '<' => try p.process("&lt;"),
+                        '>' => try p.process("&gt;"),
+                        else => unreachable,
+                    }
+                    v = v[index + 1 ..];
+                }
+            },
             .raw => |*buf| try buf.appendSlice(self.arena, data),
             .pre => unreachable,
             .parsed => unreachable,