pulibrary / pdc_describe / 4e4e59fc-9df4-4838-9fd4-6c7ea33cdb7c

class WorkUploadsEditService

  attr_reader :work, :s3_service, :current_user

  def initialize(work, current_user)

    @work = work

    @s3_service = work.s3_query_service

    @current_user = current_user

  end

  def update_precurated_file_list(added_files, deleted_files)

    delete_uploads(deleted_files)

    add_uploads(added_files)

    file_list_changed = deleted_files.count > 0 || added_files.count > 0

    if file_list_changed

      s3_service.client_s3_files(reload: true)

      work.reload

      work.reload_snapshots(user_id: current_user.id)

    end

    work

  end

  def snapshot_uppy_and_delete_files(deleted_files)

    deleted_files.each do |filename|

      s3_service.delete_s3_object(filename)

    end

    work.reload_snapshots(user_id: current_user.id)

    work

  end

  def find_post_curation_uploads(upload_keys: [])

    return [] unless work.approved? && !upload_keys.empty?

    work.post_curation_uploads.select { |upload| upload_keys.include?(upload.key) }

  end

  private

    def delete_uploads(deleted_files)

      return if deleted_files.empty?

      deleted_files.each do |filename|

        s3_service.delete_s3_object(filename)

      end

    end

    def add_uploads(added_files)

      return if added_files.empty?

      last_snapshot = work.upload_snapshots.first

      snapshot = BackgroundUploadSnapshot.new(work:)

      snapshot.store_files(added_files, pre_existing_files: last_snapshot&.files, current_user: @current_user)

      snapshot.save

      added_files.map do |file|

        if upload_file(file)

          snapshot.mark_complete(file.original_filename, work.s3_query_service.last_response.etag.delete('"'))

          snapshot.save!

          File.delete(file.path) # delete the local copy

        end

      end

    end

    def upload_file(file)

      key = work.s3_query_service.upload_file(io: file.to_io, filename: file.original_filename, size: file.size)

      if key.blank?

        Rails.logger.error("Error uploading #{file.original_filename} to work #{work.id}")

        Honeybadger.notify("Error uploading #{file.original_filename} to work #{work.id}")

        false

      else

        true

      end

    end

end