# frozen_string_literal: true

require 'caxlsx'

module RepositoryXlsxExport
  def self.to_empty_xlsx(repository, column_ids)
    package = Axlsx::Package.new
    workbook = package.workbook
    workbook.add_worksheet(name: 'Data Export') do |sheet|
      sheet.add_row prepare_header(repository, column_ids, false)
    end

    add_instruction(workbook)

    package.to_stream.read
  end

  def self.to_xlsx(rows, column_ids, user, repository, handle_file_name_func, in_module)
    package = Axlsx::Package.new
    workbook = package.workbook
    datetime_style = workbook.styles.add_style format_code: 'dd-mmm-yyyy hh:mm:ss'
    date_style = workbook.styles.add_style format_code: 'dd-mmm-yyyy'

    add_consumption = in_module && !repository.is_a?(RepositorySnapshot) && repository.has_stock_management?

    workbook.add_worksheet(name: 'Data Export') do |sheet|
      sheet.add_row prepare_header(repository, column_ids, add_consumption)

      rows.each do |row|
        row_data = []
        column_ids.each do |c_id|
          case c_id
          when -1, -2
            next
          when -3
            row_data << (repository.is_a?(RepositorySnapshot) ? row.parent.code : row.code)
          when -4
            row_data << row.name
          when -5
            row_data << row.created_by.full_name
          when -6
            row_data << row.created_at
          when -7
            row_data << row.updated_at
          when -8
            row_data << row.last_modified_by.full_name
          when -9
            row_data << (row.archived? && row.archived_by.present? ? row.archived_by.full_name : '')
          when -10
            row_data << row.archived_on
          when -11
            row_data << row.parent_repository_rows.map(&:code).join(' | ')
            row_data << row.child_repository_rows.map(&:code).join(' | ')
          else
            cell = row.repository_cells.find_by(repository_column_id: c_id)
            row_data << if cell
                          if cell.value_type == 'RepositoryAssetValue' && handle_file_name_func
                            handle_file_name_func.call(cell.value.asset)
                          elsif cell.value.is_a?(RepositoryDateTimeValue) || cell.value.is_a?(RepositoryDateValue)
                            cell.value.data
                          else
                            cell.value.export_formatted
                          end
                        end
          end
        end
        row_data << row.row_consumption(row.stock_consumption) if add_consumption

        style = row_data.map do |c|
          case c
          when ActiveSupport::TimeWithZone
            datetime_style
          when Time # Date values are of class Time for some reason
            date_style
          end
        end

        sheet.add_row(
          row_data,
          style: style
        )
      end
    end

    add_instruction(workbook)

    package.to_stream.read
  end

  def self.add_instruction(workbook)
    workbook.add_worksheet(name: 'Instruction') do |sheet|
      image = File.expand_path('app/assets/images/import_instruction.png')
      sheet.add_image(image_src: image, start_at: 'A1', width: 1260, height: 994)
    end
  end

  def self.prepare_header(repository, column_ids, add_consumption)
    header = []
    column_ids.each do |c_id|
      case c_id
      when -1, -2
        next
      when -3
        header << I18n.t('repositories.table.id')
      when -4
        header << I18n.t('repositories.table.row_name')
      when -5
        header << I18n.t('repositories.table.added_by')
      when -6
        header << I18n.t('repositories.table.added_on')
      when -7
        header << I18n.t('repositories.table.updated_on')
      when -8
        header << I18n.t('repositories.table.updated_by')
      when -9
        header << I18n.t('repositories.table.archived_by')
      when -10
        header << I18n.t('repositories.table.archived_on')
      when -11
        header << I18n.t('repositories.table.parents')
        header << I18n.t('repositories.table.children')
      else
        header << repository.repository_columns.find_by(id: c_id)&.name
      end
    end
    header << I18n.t('repositories.table.row_consumption') if add_consumption

    header
  end
end