1 files changed, 32 insertions, 5 deletions
diff --git a/LibMatrix/Helpers/MessageBuilder.cs b/LibMatrix/Helpers/MessageBuilder.cs
index 0753aca..d897078 100644
--- a/LibMatrix/Helpers/MessageBuilder.cs
+++ b/LibMatrix/Helpers/MessageBuilder.cs
@@ -1,4 +1,3 @@
-using ArcaneLibs;
 using LibMatrix.EventTypes.Spec;
 
 namespace LibMatrix.Helpers;
@@ -50,7 +49,7 @@ public class MessageBuilder(string msgType = "m.text", string format = "org.matr
         Content.FormattedBody += "</font>";
         return this;
     }
-    
+
     public MessageBuilder WithCustomEmoji(string mxcUri, string name) {
         Content.Body += $"{{{name}}}";
         Content.FormattedBody += $"<img data-mx-emoticon height=\"32\" src=\"{mxcUri}\" alt=\"{name}\" title=\"{name}\" />";
@@ -72,23 +71,51 @@ public class MessageBuilder(string msgType = "m.text", string format = "org.matr
         // }
         return this;
     }
-    
+
     public MessageBuilder WithCodeBlock(string code, string language = "plaintext") {
         Content.Body += code;
         Content.FormattedBody += $"<pre><code class=\"language-{language}\">{code}</code></pre>";
         return this;
     }
-    
+
     public MessageBuilder WithCollapsibleSection(string title, string body) {
         Content.Body += body;
         Content.FormattedBody += $"<details><summary>{title}</summary>{body}</details>";
         return this;
     }
-    
+
     public MessageBuilder WithCollapsibleSection(string title, Action<MessageBuilder> bodyBuilder) {
         Content.FormattedBody += $"<details><summary>{title}</summary>";
         bodyBuilder(this);
         Content.FormattedBody += "</details>";
         return this;
     }
+
+    public MessageBuilder WithTable(Action<TableBuilder> tableBuilder) {
+        var tb = new TableBuilder(this);
+        this.WithHtmlTag("table", msb => tableBuilder(tb));
+        return this;
+    }
+
+    public class TableBuilder(MessageBuilder msb) {
+        public TableBuilder WithTitle(string title, int colspan) {
+            msb.Content.Body += title + "\n";
+            msb.Content.FormattedBody += $"<thead><tr><th colspan=\"{colspan}\">{title}</th></tr></thead>";
+            return this;
+        }
+
+        public TableBuilder WithRow(Action<RowBuilder> rowBuilder) {
+            var rb = new RowBuilder(msb);
+            msb.WithHtmlTag("tr", msb => rowBuilder(rb)).WithBody("\n");
+            return this;
+        }
+
+        public class RowBuilder(MessageBuilder msb) {
+            public RowBuilder WithCell(string content, Dictionary<string, string>? attributes = null) {
+                msb.Content.Body += content + "\n";
+                msb.Content.FormattedBody += $"<td>{content}</td>\t";
+                return this;
+            }
+        }
+    }
 }
\ No newline at end of file