# frozen_string_literal: true module ActionText module PlainTextConversion extend self def node_to_plain_text(node) remove_trailing_newlines(plain_text_for_node(node)) end private def plain_text_for_node(node, index = 0) if respond_to?(plain_text_method_for_node(node), true) send(plain_text_method_for_node(node), node, index) else plain_text_for_node_children(node) end end def plain_text_for_node_children(node) texts = [] node.children.each_with_index do |child, index| texts << plain_text_for_node(child, index) end texts.join end def plain_text_method_for_node(node) :"plain_text_for_#{node.name}_node" end def plain_text_for_block(node, index = 0) "#{remove_trailing_newlines(plain_text_for_node_children(node))}\n\n" end %i[ h1 p ].each do |element| alias_method :"plain_text_for_#{element}_node", :plain_text_for_block end def plain_text_for_list(node, index) "#{break_if_nested_list(node, plain_text_for_block(node))}" end %i[ ul ol ].each do |element| alias_method :"plain_text_for_#{element}_node", :plain_text_for_list end def plain_text_for_br_node(node, index) "\n" end def plain_text_for_text_node(node, index) remove_trailing_newlines(node.text) end def plain_text_for_div_node(node, index) "#{remove_trailing_newlines(plain_text_for_node_children(node))}\n" end def plain_text_for_figcaption_node(node, index) "[#{remove_trailing_newlines(plain_text_for_node_children(node))}]" end def plain_text_for_blockquote_node(node, index) text = plain_text_for_block(node) text.sub(/\A(\s*)(.+?)(\s*)\Z/m, '\1“\2”\3') end def plain_text_for_li_node(node, index) bullet = bullet_for_li_node(node, index) text = remove_trailing_newlines(plain_text_for_node_children(node)) indentation = indentation_for_li_node(node) "#{indentation}#{bullet} #{text}\n" end def remove_trailing_newlines(text) text.chomp("") end def bullet_for_li_node(node, index) if list_node_name_for_li_node(node) == "ol" "#{index + 1}." else "•" end end def list_node_name_for_li_node(node) node.ancestors.lazy.map(&:name).grep(/^[uo]l$/).first end def indentation_for_li_node(node) depth = list_node_depth_for_node(node) if depth > 1 " " * (depth - 1) end end def list_node_depth_for_node(node) node.ancestors.map(&:name).grep(/^[uo]l$/).count end def break_if_nested_list(node, text) if list_node_depth_for_node(node) > 0 "\n#{text}" else text end end end end