abbreviato.rb in abbreviato-0.10.0

- old
+ new
@@ -10,18 +10,33 @@
   # and insert a _tail_ text indicating the position where content was removed (...).
   #
   # @param [String] source the XML source to truncate
   # @param [Hash] user_options truncation options
   # @option user_options [Integer] :max_length Maximum length
+  # @option user_options [Boolean] :truncate_incomplete_row Indicates whether or
+  #     not to truncate the last row in a table if truncation due to max_length
+  #     occurs in the middle of a row.
   # @option user_options [String] :tail Text to append when the truncation happens
   # @option user_options [Boolean] :fragment Indicates whether the document to be truncated is an HTML fragment
   #     or an entire document (with `HTML`, `HEAD` & `BODY` tags). Setting to true prevents automatic addition of
   #     these tags if they are missing. Defaults to `true`.
   # @return [[String] the truncated string, [boolean] whether the string was truncated]
   def self.truncate(source = '', user_options = {})
     return [nil, false] if source.nil?
     truncated_sax_document = TruncatedSaxDocument.new(DEFAULT_OPTIONS.merge(user_options))
     parser = Nokogiri::HTML::SAX::Parser.new(truncated_sax_document)
     parser.parse(source) { |context| context.replace_entities = false }
+
+    if truncated_sax_document.truncated && user_options[:truncate_incomplete_row]
+      html_fragment = Nokogiri::HTML.fragment(truncated_sax_document.truncated_string.strip)
+      last_table_in_doc = html_fragment.xpath('.//table').last
+      cols_in_first_row = last_table_in_doc.xpath('.//tr').first.xpath('.//td').length
+      last_table_in_doc.xpath('.//tr').reverse.each do |element|
+        element.remove if element.xpath('.//td').length != cols_in_first_row
+      end
+
+      return [html_fragment.to_html, truncated_sax_document.truncated]
+    end
+
     [truncated_sax_document.truncated_string.strip, truncated_sax_document.truncated]
   end
 end