QutBioacoustics/baw-server

# frozen_string_literal: true

require 'csv'

module BawWorkers
  class ReadCsv
    class << self
      # Enumerate through rows in an audio recording csv file using a block.
      # @return [void]
      def read_audio_recording_csv(csv_file)
        index_to_key_map = {
          id: 0,
          uuid: 1,
          recorded_date: 2,
          duration_seconds: 3,
          sample_rate_hertz: 4,
          channels: 5,
          bit_rate_bps: 6,
          media_type: 7,
          data_length_bytes: 8,
          file_hash: 9,
          original_file_name: 10
        }

        #audio_params_array = []

        # load csv file
        CSV.foreach(csv_file, **{ headers: true, return_headers: false }) do |row|
          # get values from row, put into hash that matches what check action expects
          audio_params = index_to_key_map.inject({}) { |hash, (k, _v)|
            hash.merge(k.to_sym => row[k.to_s])
          }

          # try a few ways to get original_format
          original_format = audio_params[:original_format]

          if original_format.blank?
            original_file_name = audio_params[:original_file_name]
            original_extension = if original_file_name.blank?
                                   ''
                                 else
                                   File.extname(original_file_name).trim('.',
                                                                         '').downcase
                                 end
            original_format = original_extension
          end

          original_format = Mime::Type.lookup(audio_params[:media_type].downcase).to_sym.to_s if original_format.blank?

          audio_params[:original_format] = original_format

          #audio_params_array.push(audio_params)

          # provide the audio parameters to yield
          yield audio_params if block_given?
        end

        #audio_params_array
      end
    end
  end
end