class PrawnHtml::Tags::Td

def find_parent_of_type(klass)
  p = parent
  p = p.parent while p && !p.is_a?(klass)
  p
end

def on_context_add(_context)
  @text_content = +''
end

def on_context_remove(_context)
  row_parent = find_parent_of_type(PrawnHtml::Tags::Tr)
  return unless row_parent
  cell_data = { content: @text_content.strip }
  cell_data[:colspan] = attrs[:colspan].to_i if attrs[:colspan]
  cell_data[:rowspan] = attrs[:rowspan].to_i if attrs[:rowspan]
  if attrs[:width]
    cell_data[:width] = attrs[:width].to_i
  end
  row_parent.add_cell(cell_data)
  @text_content = +''
end

def on_text_node(content)
  @text_content ||= +''
  @text_content << content
end