Add support for table rendering in Markdown

src/browser/markdown.zig

@@ -38,6 +38,9 @@ const State = struct {
         in_pre: bool = false,
         in_code: bool = false,
         in_blockquote: bool = false,
+        in_table: bool = false,
+        table_row_index: usize = 0,
+        table_col_count: usize = 0,
         last_char_was_newline: bool = true,
     };
     
@@ -88,6 +91,7 @@ fn renderElement(el: *Element, state: *State, writer: *std.Io.Writer, page: *Pag
         // Ensure block elements start on a new line (double newline for paragraphs etc)
         switch (tag) {
             .p, .div, .section, .article, .header, .footer, .nav, .aside, .h1, .h2, .h3, .h4, .h5, .h6, .ul, .ol, .blockquote, .pre, .table, .hr => {
+                if (!state.in_table) {
                     if (!state.last_char_was_newline) {
                         try writer.writeByte('\n');
                         state.last_char_was_newline = true;
@@ -96,6 +100,7 @@ fn renderElement(el: *Element, state: *State, writer: *std.Io.Writer, page: *Pag
                         // Add an extra newline for spacing between blocks
                         try writer.writeByte('\n');
                     }
+                }
             },
             .li, .tr => {
                 if (!state.last_char_was_newline) {
@@ -143,8 +148,22 @@ fn renderElement(el: *Element, state: *State, writer: *std.Io.Writer, page: *Pag
                 }
                 state.last_char_was_newline = false;
             },
-        .blockquote => {
+            .table => {
-            try writer.writeAll("> ");
+                state.in_table = true;
+                state.table_row_index = 0;
+                state.table_col_count = 0;
+            },
+            .tr => {
+                state.table_col_count = 0;
+                try writer.writeByte('|');
+            },
+            .td, .th => {
+                // Note: leading pipe handled by previous cell closing or tr opening
+                state.last_char_was_newline = false;
+                // Add spacing
+                try writer.writeByte(' ');
+            },
+            .blockquote => {            try writer.writeAll("> ");
             state.in_blockquote = true;
             state.last_char_was_newline = false;
         },
@@ -178,8 +197,12 @@ fn renderElement(el: *Element, state: *State, writer: *std.Io.Writer, page: *Pag
             return; // Void element
         },
         .br => {
+            if (state.in_table) {
+                try writer.writeByte(' ');
+            } else {
                 try writer.writeByte('\n');
                 state.last_char_was_newline = true;
+            }
             return; // Void element
         },
         .img => {
@@ -263,17 +286,41 @@ fn renderElement(el: *Element, state: *State, writer: *std.Io.Writer, page: *Pag
         .ul, .ol => {
             if (state.list_depth > 0) state.list_depth -= 1;
         },
+        .table => {
+            state.in_table = false;
+        },
+        .tr => {
+            try writer.writeByte('\n');
+            if (state.table_row_index == 0) {
+                try writer.writeByte('|');
+                var i: usize = 0;
+                while (i < state.table_col_count) : (i += 1) {
+                    try writer.writeAll("---|");
+                }
+                try writer.writeByte('\n');
+            }
+            state.table_row_index += 1;
+            state.last_char_was_newline = true;
+        },
+        .td, .th => {
+            try writer.writeAll(" |");
+            state.table_col_count += 1;
+            state.last_char_was_newline = false;
+        },
         else => {},
     }
 
     // Post-block newlines
     switch (tag) {
-        .p, .div, .section, .article, .header, .footer, .nav, .aside, .h1, .h2, .h3, .h4, .h5, .h6, .ul, .ol, .blockquote, .table, .tr => {
+        .p, .div, .section, .article, .header, .footer, .nav, .aside, .h1, .h2, .h3, .h4, .h5, .h6, .ul, .ol, .blockquote, .table => {
+            if (!state.in_table) {
                 if (!state.last_char_was_newline) {
                     try writer.writeByte('\n');
                     state.last_char_was_newline = true;
                 }
+            }
         },
+        .tr => {}, // Handled explicitly in closing tag logic
         else => {},
     }
 }
@@ -501,3 +548,56 @@ test "markdown: ordered list" {
 
     try testing.expectString("1. First\n2. Second\n", aw.written());
 }
+
+test "markdown: table" {
+    const testing = @import("../testing.zig");
+    const page = try testing.test_session.createPage();
+    defer testing.test_session.removePage();
+    const doc = page.window._document;
+
+    const div = try doc.createElement("div", null, page);
+
+    const table = try doc.createElement("table", null, page);
+    _ = try div.asNode().appendChild(table.asNode(), page);
+
+    const thead = try doc.createElement("thead", null, page);
+    _ = try table.asNode().appendChild(thead.asNode(), page);
+
+    const tr1 = try doc.createElement("tr", null, page);
+    _ = try thead.asNode().appendChild(tr1.asNode(), page);
+
+    const th1 = try doc.createElement("th", null, page);
+    try th1.asNode().setTextContent("Head 1", page);
+    _ = try tr1.asNode().appendChild(th1.asNode(), page);
+
+    const th2 = try doc.createElement("th", null, page);
+    try th2.asNode().setTextContent("Head 2", page);
+    _ = try tr1.asNode().appendChild(th2.asNode(), page);
+
+    const tbody = try doc.createElement("tbody", null, page);
+    _ = try table.asNode().appendChild(tbody.asNode(), page);
+
+    const tr2 = try doc.createElement("tr", null, page);
+    _ = try tbody.asNode().appendChild(tr2.asNode(), page);
+
+    const td1 = try doc.createElement("td", null, page);
+    try td1.asNode().setTextContent("Cell 1", page);
+    _ = try tr2.asNode().appendChild(td1.asNode(), page);
+
+    const td2 = try doc.createElement("td", null, page);
+    try td2.asNode().setTextContent("Cell 2", page);
+    _ = try tr2.asNode().appendChild(td2.asNode(), page);
+
+    var aw: std.Io.Writer.Allocating = .init(testing.allocator);
+    defer aw.deinit();
+    try dump(div.asNode(), .{}, &aw.writer, page);
+
+    const expected =
+        \\
+        \\| Head 1 | Head 2 |
+        \\|---|---|
+        \\| Cell 1 | Cell 2 |
+        \\
+    ;
+    try testing.expectString(expected, aw.written());
+}