module CSVImportable

# File app/models/concerns/csv_importable.rb, line 200
def aggregate_row_level_errors
  row_offset = csv_data_object.has_instruction_row? ? 3 : 2

  rows.each.with_index do |row, index|
    next if row.errors.empty?

    # The first row is the header and the index is 0-based, so we add two
    # to match what the user sees in the spreadsheet

    formatted_errors =
      row.errors.map do |error|
        if error.attribute == :base
          error.message
        else
          "<code>#{error.attribute}</code>: #{error.message}"
        end
      end

    errors.add("row_#{index + row_offset}".to_sym, formatted_errors)
  end
end

# File app/models/concerns/csv_importable.rb, line 167
def count_columns
  %i[
    new_record_count
    changed_record_count
    exact_duplicate_record_count
  ].freeze
end

# File app/models/concerns/csv_importable.rb, line 130
  def csv = csv_data

  def csv_data_object
    @csv_data_object ||= Import::CSVData.new(csv_data)
  end

  def csv_removed?
    csv_removed_at != nil
  end

  def parse_rows!
    return if invalid?

    self.rows = csv_data_object.records.map { |row_data| parse_row(row_data) }

    if invalid?(:parse_rows)
      self.serialized_errors = errors.to_hash
      self.status = :rows_are_invalid
      save!(validate: false)
    end
  end

  def remove!
    return if csv_removed?
    update!(csv_data: nil, csv_removed_at: Time.zone.now)
  end

  def load_serialized_errors!(limit: nil)
    return if serialized_errors.blank?

    serialized_errors
      .then { limit ? it.first(limit) : it }
      .each do |attribute, messages|
        messages.each { errors.add(attribute, _1) }
      end
  end

  def count_columns
    %i[
      new_record_count
      changed_record_count
      exact_duplicate_record_count
    ].freeze
  end

  def ensure_processed_with_count_statistics
    if processed_at? && count_columns.any? { |column| send(column).nil? }
      raise "Count statistics must be set for a processed import."
    end
  end

  private

  def csv_is_valid
    errors.add(:csv, :invalid) unless csv_data_object.well_formed?
  end

  def csv_is_not_too_large
    if rows_count > MAX_CSV_ROWS
      errors.add(:csv, :too_many_rows, count: MAX_CSV_ROWS)
    end
  end

  def csv_has_records
    csv_has_no_records =
      csv_data_object.empty? ||
        (csv_data_object.count == 1 && csv_data_object.has_instruction_row?)
    errors.add(:csv, :empty) if csv_has_no_records
  end

  def aggregate_row_level_errors
    row_offset = csv_data_object.has_instruction_row? ? 3 : 2

    rows.each.with_index do |row, index|
      next if row.errors.empty?

      # The first row is the header and the index is 0-based, so we add two
      # to match what the user sees in the spreadsheet

      formatted_errors =
        row.errors.map do |error|
          if error.attribute == :base
            error.message
          else
            "<code>#{error.attribute}</code>: #{error.message}"
          end
        end

      errors.add("row_#{index + row_offset}".to_sym, formatted_errors)
    end
  end
end

# File app/models/concerns/csv_importable.rb, line 110
def csv=(source)
  if persisted?
    raise "Cannot change the CSV file for an existing import. " \
            "Create a new import instead."
  end

  if source.is_a?(ActionDispatch::Http::UploadedFile)
    # CSV files exported from Excel may have a BOM.
    # https://en.wikipedia.org/wiki/Byte_order_mark
    # e.g. if you create a new class import from scratch in Excel on Mac v16,
    # save the file as CSV, and upload it.
    self.csv_data = source.to_io.tap(&:set_encoding_by_bom).read
    self.csv_filename = source&.original_filename
    self.rows_count = csv_data_object&.count
  else
    raise ArgumentError, "Expected an uploaded file, got #{source}"
  end
end

# File app/models/concerns/csv_importable.rb, line 132
def csv_data_object
  @csv_data_object ||= Import::CSVData.new(csv_data)
end

# File app/models/concerns/csv_importable.rb, line 193
def csv_has_records
  csv_has_no_records =
    csv_data_object.empty? ||
      (csv_data_object.count == 1 && csv_data_object.has_instruction_row?)
  errors.add(:csv, :empty) if csv_has_no_records
end

# File app/models/concerns/csv_importable.rb, line 187
def csv_is_not_too_large
  if rows_count > MAX_CSV_ROWS
    errors.add(:csv, :too_many_rows, count: MAX_CSV_ROWS)
  end
end

# File app/models/concerns/csv_importable.rb, line 183
def csv_is_valid
  errors.add(:csv, :invalid) unless csv_data_object.well_formed?
end

# File app/models/concerns/csv_importable.rb, line 136
def csv_removed?
  csv_removed_at != nil
end

# File app/models/concerns/csv_importable.rb, line 175
def ensure_processed_with_count_statistics
  if processed_at? && count_columns.any? { |column| send(column).nil? }
    raise "Count statistics must be set for a processed import."
  end
end

# File app/models/concerns/csv_importable.rb, line 157
def load_serialized_errors!(limit: nil)
  return if serialized_errors.blank?

  serialized_errors
    .then { limit ? it.first(limit) : it }
    .each do |attribute, messages|
      messages.each { errors.add(attribute, _1) }
    end
end

# File app/models/concerns/csv_importable.rb, line 140
def parse_rows!
  return if invalid?

  self.rows = csv_data_object.records.map { |row_data| parse_row(row_data) }

  if invalid?(:parse_rows)
    self.serialized_errors = errors.to_hash
    self.status = :rows_are_invalid
    save!(validate: false)
  end
end

# File app/models/concerns/csv_importable.rb, line 152
def remove!
  return if csv_removed?
  update!(csv_data: nil, csv_removed_at: Time.zone.now)
end