class AppSessionVaccinationsComponent

# File app/components/app_session_vaccinations_component.rb, line 32
def initialize(session)
  @session = session
end

# File app/components/app_session_vaccinations_component.rb, line 94
def compute_total_tallies
  totals = Array.new(session_column_names.length, 0)

  session_dates_rows.each do |row|
    row[:tallies].each_with_index { |tally, index| totals[index] += tally }
  end

  totals
end

# File app/components/app_session_vaccinations_component.rb, line 87
def count_by_vaccine_method(vaccination_records)
  nasal_spray_count =
    vaccination_records.count(&:delivery_method_nasal_spray?)
  injection_count = vaccination_records.count(&:delivery_method_injection?)
  [nasal_spray_count, injection_count]
end

# File app/components/app_session_vaccinations_component.rb, line 114
def patients_for_programme(programme)
  @patients_for_programme ||= {}
  @patients_for_programme[
    programme.type
  ] ||= session.patients.appear_in_programmes([programme], session:)
end

# File app/components/app_session_vaccinations_component.rb, line 36
def render? = session.started?

private

attr_reader :session

delegate :programmes, to: :session

delegate :govuk_table, to: :helpers

def session_column_names
  @session_column_names ||=
    programmes.flat_map do |programme|
      if programme.has_multiple_vaccine_methods?
        base_name = programme.name_in_sentence.titlecase
        ["#{base_name} (nasal spray)", "#{base_name} (injection)"]
      else
        programme.name_in_sentence
      end
    end
end

def rows = session_dates_rows + [total_row]

def session_dates_rows
  @session_dates_rows ||=
    session.dates.map do |date|
      {
        label: date.strftime("%e %B %Y"),
        tallies: tally_vaccination_counts_for_date(date)
      }
    end
end

def total_row
  { label: "Total", tallies: compute_total_tallies }
end

def tally_vaccination_counts_for_date(date)
  programmes.flat_map do |programme|
    vaccination_records_for_date =
      vaccination_records_by_date(programme).fetch(date, [])

    if programme.has_multiple_vaccine_methods?
      count_by_vaccine_method(vaccination_records_for_date)
    else
      [vaccination_records_for_date.count]
    end
  end
end

def count_by_vaccine_method(vaccination_records)
  nasal_spray_count =
    vaccination_records.count(&:delivery_method_nasal_spray?)
  injection_count = vaccination_records.count(&:delivery_method_injection?)
  [nasal_spray_count, injection_count]
end

def compute_total_tallies
  totals = Array.new(session_column_names.length, 0)

  session_dates_rows.each do |row|
    row[:tallies].each_with_index { |tally, index| totals[index] += tally }
  end

  totals
end

def vaccination_records_by_date(programme)
  @vaccination_records_by_date ||= {}
  @vaccination_records_by_date[programme.type] ||= VaccinationRecord
    .administered
    .kept
    .for_programme(programme)
    .where(session:, patient_id: patients_for_programme(programme))
    .group_by { |record| record.performed_at.to_date }
end

def patients_for_programme(programme)
  @patients_for_programme ||= {}
  @patients_for_programme[
    programme.type
  ] ||= session.patients.appear_in_programmes([programme], session:)
end

# File app/components/app_session_vaccinations_component.rb, line 58
  def rows = session_dates_rows + [total_row]

  def session_dates_rows
    @session_dates_rows ||=
      session.dates.map do |date|
        {
          label: date.strftime("%e %B %Y"),
          tallies: tally_vaccination_counts_for_date(date)
        }
      end
  end

  def total_row
    { label: "Total", tallies: compute_total_tallies }
  end

  def tally_vaccination_counts_for_date(date)
    programmes.flat_map do |programme|
      vaccination_records_for_date =
        vaccination_records_by_date(programme).fetch(date, [])

      if programme.has_multiple_vaccine_methods?
        count_by_vaccine_method(vaccination_records_for_date)
      else
        [vaccination_records_for_date.count]
      end
    end
  end

  def count_by_vaccine_method(vaccination_records)
    nasal_spray_count =
      vaccination_records.count(&:delivery_method_nasal_spray?)
    injection_count = vaccination_records.count(&:delivery_method_injection?)
    [nasal_spray_count, injection_count]
  end

  def compute_total_tallies
    totals = Array.new(session_column_names.length, 0)

    session_dates_rows.each do |row|
      row[:tallies].each_with_index { |tally, index| totals[index] += tally }
    end

    totals
  end

  def vaccination_records_by_date(programme)
    @vaccination_records_by_date ||= {}
    @vaccination_records_by_date[programme.type] ||= VaccinationRecord
      .administered
      .kept
      .for_programme(programme)
      .where(session:, patient_id: patients_for_programme(programme))
      .group_by { |record| record.performed_at.to_date }
  end

  def patients_for_programme(programme)
    @patients_for_programme ||= {}
    @patients_for_programme[
      programme.type
    ] ||= session.patients.appear_in_programmes([programme], session:)
  end
end

# File app/components/app_session_vaccinations_component.rb, line 46
def session_column_names
  @session_column_names ||=
    programmes.flat_map do |programme|
      if programme.has_multiple_vaccine_methods?
        base_name = programme.name_in_sentence.titlecase
        ["#{base_name} (nasal spray)", "#{base_name} (injection)"]
      else
        programme.name_in_sentence
      end
    end
end

# File app/components/app_session_vaccinations_component.rb, line 60
def session_dates_rows
  @session_dates_rows ||=
    session.dates.map do |date|
      {
        label: date.strftime("%e %B %Y"),
        tallies: tally_vaccination_counts_for_date(date)
      }
    end
end

# File app/components/app_session_vaccinations_component.rb, line 74
def tally_vaccination_counts_for_date(date)
  programmes.flat_map do |programme|
    vaccination_records_for_date =
      vaccination_records_by_date(programme).fetch(date, [])

    if programme.has_multiple_vaccine_methods?
      count_by_vaccine_method(vaccination_records_for_date)
    else
      [vaccination_records_for_date.count]
    end
  end
end

# File app/components/app_session_vaccinations_component.rb, line 70
def total_row
  { label: "Total", tallies: compute_total_tallies }
end

# File app/components/app_session_vaccinations_component.rb, line 104
def vaccination_records_by_date(programme)
  @vaccination_records_by_date ||= {}
  @vaccination_records_by_date[programme.type] ||= VaccinationRecord
    .administered
    .kept
    .for_programme(programme)
    .where(session:, patient_id: patients_for_programme(programme))
    .group_by { |record| record.performed_at.to_date }
end