sanger/sequencescape

# frozen_string_literal: true
module Heron
  module Factories
    #
    # Factory class to generate sample tubes inside a Heron rack
    class Sample
      include ActiveModel::Model

      validates_presence_of :study, unless: :sample_already_present?
      validate :check_no_other_params_when_uuid, if: :sample_already_present?
      validate :all_fields_are_existing_columns

      def initialize(params)
        @params = params
      end

      def check_no_other_params_when_uuid
        return if sample_keys.empty?

        sample_keys.each { |key| errors.add(key, 'No other params can be added when sample uuid specified') }
      end

      def sample_keys
        (@params.keys.map(&:to_sym) - %i[sample_uuid study study_uuid aliquot uuid])
      end

      ##
      # Persists the material including the associated container
      def create
        return unless valid?
        return @sample if @sample

        @sample = create_sample!
      end

      def create_aliquot_at(container)
        return unless create

        new_aliquot = container&.aliquots&.create(params_for_aliquot_creation)

        container.register_stock! if new_aliquot&.persisted?

        new_aliquot
      end

      def study
        @study ||= @params[:study] || Study.with_uuid(@params[:study_uuid]).first
      end

      def create_sanger_sample_id!
        SangerSampleId.generate_sanger_sample_id!(study.abbreviation)
      end

      def sanger_sample_id
        @sanger_sample_id ||= @params[:sanger_sample_id] || create_sanger_sample_id!
      end

      def sample
        return @sample if @sample

        @sample = ::Sample.with_uuid(@params[:sample_uuid]).first if @params[:sample_uuid]
      end

      def sample_already_present?
        sample.present?
      end

      def create_sample!
        return sample if sample

        @sample =
          ::Sample.create!(params_for_sample_creation) do |sample|
            replace_uuid(sample) if @params[:uuid]
            sample.sample_metadata.update!(params_for_sample_metadata_table)
            sample.studies << study
          end
      end

      def replace_uuid(sample)
        uuid = @params[:uuid]
        handle_uuid_duplication(uuid) if Uuid.with_external_id(uuid).count.positive?
        sample.lazy_uuid_generation = true
        sample.uuid_object = Uuid.new
        sample.uuid_object.update!(resource: sample, external_id: uuid) if uuid
      end

      def handle_uuid_duplication(uuid)
        msg = "Sample with uuid #{uuid} already exists"
        errors.add(:uuid, msg)
        raise StandardError, msg
      end

      def unexisting_column_keys
        (sample_keys - [params_for_sample_table.keys, params_for_sample_metadata_table.keys].flatten.map(&:to_sym))
      end

      def all_fields_are_existing_columns
        return if unexisting_column_keys.empty?

        unexisting_column_keys.each { |col| errors.add(col, 'Unexisting field for sample or sample_metadata') }
      end

      def params_for_sample_creation
        { name: sanger_sample_id, sanger_sample_id: sanger_sample_id }.merge(params_for_sample_table)
      end

      def params_for_aliquot_creation
        { sample: sample, study: study }.merge(@params.dig(:aliquot) || {})
      end

      def params_for_sample_table
        @params.select { |k, _v| ::Sample.column_names.include?(k.to_s) }
      end

      def params_for_sample_metadata_table
        @params.select { |k, _v| ::Sample::Metadata.column_names.include?(k.to_s) }
      end
    end
  end
end