New tags: table, tr, td

Author: blocknotes

lib/prawn_html/tag.rb

@@ -9,7 +9,7 @@ class Tag
       'StrikeThrough' => Callbacks::StrikeThrough
     }.freeze
 
-    TAG_CLASSES = %w[A B Blockquote Body Br Code Del Div H Hr I Img Li Mark Ol P Pre Small Span Sub Sup U Ul].freeze
+    TAG_CLASSES = %w[A B Blockquote Body Br Code Del Div H Hr I Img Li Mark Ol P Pre Small Span Sub Sup Table Td Tr U Ul].freeze
 
     def_delegators :@attrs, :styles, :update_styles
 

lib/prawn_html/tags/table.rb

@@ -0,0 +1,38 @@
+# frozen_string_literal: true
+
+module PrawnHtml
+  module Tags
+    class Table < Tag
+      ELEMENTS = [:table].freeze
+
+      attr_reader :table_data
+
+      def block?
+        true
+      end
+
+      def new_cell
+        @col_index += 1
+        @table_data[@row_index] << ''
+      end
+
+      def new_row
+        @row_index += 1
+        @col_index = -1
+        @table_data << []
+      end
+
+      def update_content(content)
+        @table_data[@row_index][@col_index] = content
+      end
+
+      def tag_opening(context: nil)
+        super.tap do
+          context.current_table = self
+          @row_index = -1
+          @table_data = []
+        end
+      end
+    end
+  end
+end

lib/prawn_html/tags/td.rb

@@ -0,0 +1,19 @@
+# frozen_string_literal: true
+
+module PrawnHtml
+  module Tags
+    class Td < Tag
+      ELEMENTS = [:td].freeze
+
+      def block?
+        true
+      end
+
+      def tag_opening(context: nil)
+        super.tap do
+          context.current_table&.new_cell
+        end
+      end
+    end
+  end
+end

lib/prawn_html/tags/tr.rb

@@ -0,0 +1,19 @@
+# frozen_string_literal: true
+
+module PrawnHtml
+  module Tags
+    class Tr < Tag
+      ELEMENTS = [:tr].freeze
+
+      def block?
+        true
+      end
+
+      def tag_opening(context: nil)
+        super.tap do
+          context.current_table&.new_row
+        end
+      end
+    end
+  end
+end