# frozen_string_literal: true

module Abbreviato
  DEFAULT_OPTIONS = {
    max_length: 30,
    tail: '…',
    fragment: true
  }.freeze

  # Truncates the source XML string and returns the truncated XML and a boolean flag indicating
  # whether any truncation took place. It will keep a valid XML structure
  # 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]
      parsed_results = [truncated_sax_document.truncated_string.strip, truncated_sax_document.truncated]

      html_fragment = Nokogiri::HTML.fragment(truncated_sax_document.truncated_string.strip)
      return parsed_results if html_fragment.nil?

      last_table_in_doc = html_fragment.xpath('.//table').last
      return parsed_results unless last_table_in_doc

      first_row = last_table_in_doc.xpath('.//tr').first
      return parsed_results unless first_row

      cols_in_first_row = first_row.xpath('.//td').length
      return parsed_results unless cols_in_first_row.positive?

      last_table_in_doc.xpath('.//tr').each do |row|
        row.remove if row.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