sanger/sequencescape

# frozen_string_literal: true

namespace :auto_imported_samples do
  desc 'Send messages to the stock_resource table in MLWH for samples that were imported before the bug fix in GPL-557'
  task register_as_stock: :environment do
    # retrieve the relevant receptacles
    # Any imported by Lighthouse or Wrangler
    # Find these by any created from start of May
    # Add extra checks that sample_manifest is nil
    # And purpose is one of expected ones

    relevant_study_names = [
      'Heron Project',
      'Heron Project R & D',
      'Heron Project ONT',
      'Heron Project PacBio',
      'Project Kestrel'
    ]
    relevant_study_ids = Study.where(name: relevant_study_names).map(&:id).join(',')

    # Some samples will be in more than one of these, but we think that's ok.
    relevant_purpose_names = ['Stock Plate', 'LHR Stock', 'TR Stock 96', 'Heron Lysed Plate', 'Heron Lysed Tube Rack']
    relevant_purpose_ids = Purpose.where(name: relevant_purpose_names).map(&:id).join(',')

    labware_samples =
      Labware
        .joins(:samples)
        .where(
          "
      labware.created_at > '2020-05-01 00:00:00' AND
      labware.sti_type IN ('Plate', 'TubeRack') AND
      labware.plate_purpose_id IN (#{relevant_purpose_ids}) AND
      samples.sample_manifest_id IS NULL
      "
        ) # 78,507 in training 2020-07-07
    puts "labware_samples count: #{labware_samples.count}"

    labware = labware_samples.uniq # 890 in training 2020-07-07
    puts "labware count: #{labware.count}"

    receptacles =
      Receptacle.where(labware: labware).joins(:aliquots).where("aliquots.study_id IN (#{relevant_study_ids})")

    # 78,507 in training 2020-07-08 (85,440 before join with aliquots)
    puts "receptacles count: #{receptacles.count}"

    existing_stock_resource_messengers =
      Messenger.where(target: receptacles, root: 'stock_resource', target_type: 'Receptacle')

    # 0 in training 2020-07-08
    puts "existing_stock_resource_messengers count: #{existing_stock_resource_messengers.count}"
    receptacle_ids_with_messengers = existing_stock_resource_messengers.filter_map(&:target_id).uniq
    puts "receptacle_ids_with_messengers count: #{receptacle_ids_with_messengers.count}"

    puts 'registering as stock...'

    # call register_stock! on each of them
    receptacles.each { |r| r.register_stock! unless receptacle_ids_with_messengers.include? r.id }
    puts 'Done'
  end
end