require 'csv'

class Seeder
  def seed_academic_years(*academic_year_ranges)
    academic_year_ranges.each do |range|
      AcademicYear.find_or_create_by! range: range
    end
  end

  def seed_districts_and_schools(csv_file)
    dese_ids = []
    CSV.parse(File.read(csv_file), headers: true) do |row|
      district_name = row['District'].strip
      district_code = row['District Code'].try(:strip)
      dese_id = row['DESE School ID'].strip
      dese_ids << dese_id
      school_name = row['School Name'].strip
      school_code = row['School Code'].try(:strip)
      hs = row['HS?']

      district = District.find_or_create_by! name: district_name
      district.update! slug: district_name.parameterize, qualtrics_code: district_code

      school = School.find_or_initialize_by dese_id: dese_id, district: district
      school.name = school_name
      school.qualtrics_code = school_code
      school.is_hs = is_hs? hs
      school.save!
    end

    School.where.not(dese_id: dese_ids).destroy_all
  end

  def seed_respondents(csv_file)
    schools = []
    year = '2020-21'
    academic_year = AcademicYear.find_by_range year
    CSV.parse(File.read(csv_file), headers: true) do |row|
      dese_id = row['DESE School ID'].strip.to_i
      total_students = row["Total Students for Response Rate (#{year})"]
      total_teachers = row["Total Teachers for Response Rate (#{year})"]

      district_name = row['District'].strip
      district = District.find_or_create_by! name: district_name

      school = School.find_by dese_id: dese_id, district: district
      schools << school
      respondent = Respondent.find_or_initialize_by school: school
      respondent.total_students = total_students
      respondent.total_teachers = total_teachers
      respondent.academic_year = academic_year
      respondent.save
    end

    Respondent.where.not(school: schools).destroy_all
  end

  def seed_sqm_framework(csv_file)
    CSV.parse(File.read(csv_file), headers: true) do |row|
      category_id = row['Category ID'].strip
      category = Category.find_or_create_by!(category_id:)
      category_slugs = {
        '1' => 'teachers-and-leadership',
        '2' => 'school-culture',
        '3' => 'resources',
        '4' => 'academic-learning',
        '5' => 'community-and-wellbeing'
      }
      category.update! name: row['Category'].strip, description: row['Category Description'].strip,
                       short_description: row['Category Short Description'], slug: category_slugs[category_id], sort_index: category_slugs.keys.index(category_id)

      subcategory_id = row['Subcategory ID'].strip
      subcategory = Subcategory.find_or_create_by! subcategory_id: subcategory_id, category: category
      subcategory.update! name: row['Subcategory'].strip, description: row['Subcategory Description'].strip

      measure_id = row['Measure ID'].strip
      measure_name = row['Measures'].try(:strip)
      watch_low = row['Item Watch Low'].try(:strip)
      growth_low = row['Item Growth Low'].try(:strip)
      approval_low = row['Item Approval Low'].try(:strip)
      ideal_low = row['Item Ideal Low'].try(:strip)
      measure_description = row['Measure Description'].try(:strip)

      next if row['Source'] == 'No source'

      measure = Measure.find_or_create_by! measure_id: measure_id, subcategory: subcategory
      measure.name = measure_name
      measure.description = measure_description
      measure.save!

      data_item_id = row['Survey Item ID'].strip
      if %w[Teachers Students].include? row['Source']
        survey_item = SurveyItem.find_or_create_by! survey_item_id: data_item_id, measure: measure
        survey_item.watch_low_benchmark = watch_low if watch_low
        survey_item.growth_low_benchmark = growth_low if growth_low
        survey_item.approval_low_benchmark = approval_low if approval_low
        survey_item.ideal_low_benchmark = ideal_low if ideal_low
        survey_item.update! prompt: row['Question/item (20-21)'].strip
      end

      if row['Source'] == 'Admin Data'
        admin_data_item = AdminDataItem.find_or_create_by! admin_data_item_id: data_item_id, measure: measure
        admin_data_item.watch_low_benchmark = watch_low if watch_low
        admin_data_item.growth_low_benchmark = growth_low if growth_low
        admin_data_item.approval_low_benchmark = approval_low if approval_low
        admin_data_item.ideal_low_benchmark = ideal_low if ideal_low
        admin_data_item.description = row['Question/item (20-21)'].strip
        admin_data_item.hs_only_item = is_hs? row['HS only admin item?']
        admin_data_item.save!
      end
    end
  end

  private

  def is_hs?(is_hs)
    is_hs.present? ? is_hs.strip == 'X' : false
  end
end