# frozen_string_literal: true

module HealthcarePhony
  # Public: Creates discharge information for a patient visit.
  class VisitDischarge
    attr_accessor :disposition,
                  :location,
                  :datetime

    # Public: Initializes an EthnicGroup. Pass in hash of different parameters, currently this includes:
    # visit_type - VisitType of this patient's visit
    # discharge_disposition - Array of discharge disposition codes (PV1.36) to randomly choose from.  Specified as comma
    # separated String or Ruby array. Otherwise default HL7 v2.5.1 Table 0112 values are used.
    # discharge_location - Array of discharge locations to randomly choose from.  Specified as comma separated String or
    # Ruby array. Otherwise a string of data is generated with Faker::Lorem.sentence
    # admit_datetime - The admit date/time associated with this visit.  If not specified the current date/time is used.
    def initialize(init_args = {})
      if init_args[:visit_type] == HealthcarePhony::VisitType::DISCHARGE # init_args[:event_type] == 'A03'
        @disposition = define_discharge_disposition(init_args)
        @location = define_discharge_location(init_args)
        @datetime = define_discharge_datetime(init_args)
      else
        @disposition = ''
        @location = ''
        @datetime = nil
      end
    end

    private

    def define_discharge_disposition(init_args = {})
      dd_choices = Helper.get_array(init_args[:discharge_disposition])
      if init_args[:visit_type] != HealthcarePhony::VisitType::DISCHARGE # init_args[:event_type] != 'A03'
        ''
      elsif !dd_choices.empty?
        dd_choices.sample
      else
        data_file = "#{::File.expand_path(::File.join("..", "data_files"), __FILE__)}/discharge_disposition.yml"
        file_based_choices = Psych.load_file(data_file)
        file_based_choices.nil? ? '' : file_based_choices.sample
      end
    end

    def define_discharge_location(init_args = {})
      dl_choices = Helper.get_array(init_args[:discharge_location])
      if !dl_choices.empty?
        dl_choices.sample
      else
        Faker::Lorem.sentence
      end
    end

    def define_discharge_datetime(init_args = {})
      from_datetime = if init_args[:admit_datetime].nil?
                        Time.now
                      else
                        init_args[:admit_datetime]
                      end
      Faker::Time.between(from: from_datetime, to: DateTime.now)
    end
  end
end