cs_code = 1347

class Report
  module Renderer
    class MciRenderer < BaseRenderer
      def trends
        @trends
      end

      def impairment_classification
        case assessment_result_color
        when 'green'
          I18n.t('report.cfp.mci.potentially_cognitively_unimpaired')
        when 'yellow'
          I18n.t('report.cfp.mci.indicative_of_subjective_decline')
        when 'orange'
          I18n.t('report.cfp.mci.indicative_of_potential_mci')
        when 'red'
          I18n.t('report.cfp.mci.indicative_of_major_neurocognitive_disorder')
        else
          ''
        end
      end
    end
  end
end

report_data = [
  { preset_code: :mci, test_ids: [21, 17, 23, 19, 14, 20, 13, 24, 22, 18, 15, 16] },
  { preset_code: :mci_short, test_ids: [21, 19, 13, 24, 22, 16] },
]

report_data.each do |data|
  test_ids = data[:test_ids]
  preset_code = data[:preset_code]

  test_titles = test_ids.map do |test_id|
    test = Test.find(test_id)
    task_code = test.task_code
    [I18n.t("report.task.#{task_code}.desc") + ' - SS', I18n.t("report.task.#{task_code}.desc") + ' - %']
  end.flatten
  titles = "Admin ID
  Patient ID
  Company Name
  Assessment Sent Date
  Assessment Completed Date
  Summary Statement
  Impairment Classification
  IQ-code score
  IADL score classification
  PHQ-9 Score
  GAD-7 Score".split("\n")
  titles = titles + test_titles

  protocols = ::Health::Protocol.where(preset_code: preset_code)
  sessions = ::Health::ProtocolSession.status_complete.where(protocol_id: protocols.ids)
  p_bar = ProgressBar.new(sessions.size)


  CSV.open("tmp/CS-#{cs_code}_#{preset_code}.csv", 'w') do |csv|
    csv << titles
    sessions.find_each do |session|
      p_bar.puts(session.id)
      p_bar.increment!

      reports = session.reports
      batch_names = reports.map(&:batch).map(&:name)
      next if batch_names.exclude?('phq9')
      next if batch_names.exclude?('gad7')
      next if batch_names.exclude?('iadl')
      next if batch_names.exclude?('iqcodesr')

      report = reports.find_by(cfp: true)
      renderer = Report::Renderer::MciRenderer.new(report, {filename: '', template: ''})

      renderer.comparative_group
      renderer.trends

      admin = report.admin_user.becomes(::Health::User).clinic_admin

      admin_id = admin.id
      patient_id = report.admin_user.client_id
      company_name = admin.trial.name
      sent_at = session.created_at.humanize_datetime
      completed_at = reports.maximum(:completed_at).humanize_datetime


      summary_statement = renderer.assessment_summary_main_result[0].html_safe + renderer.assessment_summary_main_result[1].html_safe
      summary_statement = ActionView::Base.full_sanitizer.sanitize(summary_statement)

      impairment_classification = renderer.impairment_classification
      iqcode_score = renderer.score_summary('IQCODE-SR')
      iadl_score_classification = renderer.iadl_result
      phq9_score = renderer.score_summary('PHQ-9')
      gad7_score = renderer.score_summary('GAD-7')


      row = [admin_id, patient_id,company_name,sent_at,completed_at,summary_statement,impairment_classification,iqcode_score,iadl_score_classification,phq9_score,gad7_score]
      scores = renderer.cog_report.test_scores
      test_ids.each do |test_id|
        score = scores.find{ |s| s.test_id == test_id }
        score_trends = renderer.trends.find{ |trend| trend[:task_id] == test_id }
        score_trend = score_trends[:scores].find{ |t| t[:id] == score.id }
        row += [score_trend[:adjusted_standard], score_trend[:not_rounded_adjusted_percentile].round(2)]
      end
      csv << row
    end
    nil
  end
end