# frozen_string_literal: true

# This is a module that can be included in any class that needs to use the methods defined here.
module MalawiHivProgramReports
  module Utils
    module CommonSqlQueryUtils
      def process_occupation(start_date:, end_date:, occupation:, location:, definition: 'moh')
        return if occupation.blank?

        MalawiHivProgramReports::Moh::CohortBuilder.new(outcomes_definition: definition, location:).init_temporary_tables(start_date, end_date,
                                                                                                                          occupation)
      end

      def occupation_filter(occupation:, field_name:, table_name: '', include_clause: true)
        clause = 'WHERE' if include_clause
        table_name = "#{table_name}." unless table_name.blank?
        return '' if occupation.blank?
        return '' if occupation == 'All'
        if occupation == 'Military'
          return "#{clause} #{table_name}#{field_name} IN ('#{occupation}', 'MDF Reserve', 'MDF Retired', 'Soldier', 'Soldier/Police')"
        end
        return unless occupation == 'Civilian'

        "#{clause} #{table_name}#{field_name} NOT IN ('Military', 'MDF Reserve', 'MDF Retired', 'Soldier', 'Soldier/Police')"
      end

      def external_client_query(end_date:)
        end_date = ActiveRecord::Base.connection.quote(end_date)
        <<~SQL
          SELECT obs.person_id FROM obs,
          (SELECT person_id, Max(obs_datetime) AS obs_datetime, concept_id FROM obs
          WHERE concept_id IN (SELECT concept_id FROM concept_name WHERE name = 'Type of patient' AND voided = 0)
          AND DATE(obs_datetime) <= #{end_date}
          AND voided = 0
          GROUP BY person_id,concept_id) latest_record
          WHERE obs.person_id = latest_record.person_id
          AND obs.concept_id = latest_record.concept_id
          AND obs.obs_datetime = latest_record.obs_datetime
          AND obs.value_coded IN (SELECT concept_id FROM concept_name WHERE name = 'Drug refill' || name = 'External consultation')
          AND obs.voided = 0
          #{site_manager(operator: 'AND', column: 'obs.site_id', location: @location)}
        SQL
      end

      def current_occupation_query
        ActiveRecord::Base.connection.adapter_name.downcase
        <<~SQL
          SELECT a.person_id, a.value, a.site_id
          FROM person_attribute #{current_partition} a
          LEFT OUTER JOIN person_attribute #{current_partition} b
          ON a.person_attribute_id = b.person_attribute_id
          AND a.date_created < b.date_created
          AND b.voided = 0
          WHERE b.person_attribute_id IS NULL AND a.person_attribute_type_id = 13 AND a.voided = 0
        SQL
      end

      def partition_by_site
        @partition_by_site ||= Location.all.map do |location|
          "PARTITION p#{location.id} VALUES IN (#{location.id}) ENGINE = InnoDB"
        end.join(', ')
      end
    end
  end
end