module Measures
  # Utility class for loading measure definitions into the database
  class Loader

    SOURCE_PATH = File.join(".", "db", "measures")
    VALUE_SET_PATH = File.join(".", "db", "value_sets")
    HQMF_VS_OID_CACHE = File.join(".", "db", "hqmf_vs_oid_cache")
    PARSERS = [HQMF::Parser::V2CQLParser, HQMF::Parser::V2Parser,HQMF::Parser::V1Parser,SimpleXml::Parser::V1Parser]

    def self.parse_hqmf_model(xml_path)
      xml_contents = Nokogiri::XML(File.new xml_path)
      parser = get_parser(xml_contents)
      parser.parse(xml_contents)
    end

    def self.load_hqmf_model_json(json, user, measure_oids, measure_details=nil)

      measure = Measure.new
      measure.user = user if user
      measure.bundle = user.bundle if (user && user.respond_to?(:bundle) )
      measure.id = json["hqmf_id"]
      measure.measure_id = json["id"]
      measure.hqmf_id = json["hqmf_id"]
      measure.hqmf_set_id = json["hqmf_set_id"]
      measure.hqmf_version_number = json["hqmf_version_number"]
      measure.cms_id = json["cms_id"]
      measure.title = json["title"]
      measure.description = json["description"]
      measure.measure_attributes = json["attributes"]
      measure.populations = json['populations']
      measure.value_set_oids = measure_oids

      metadata = measure_details
      if metadata
        measure.measure_id = metadata["nqf_id"] || json["id"]
        measure.type = metadata["type"]
        measure.category = metadata["category"]
        measure.episode_of_care = metadata["episode_of_care"]
        measure.continuous_variable = metadata["continuous_variable"]
        measure.episode_ids = metadata["episode_ids"]
        puts "\tWARNING: Episode of care does not align with episode ids existance" if ((!measure.episode_ids.nil? && measure.episode_ids.length > 0) ^ measure.episode_of_care)
        if (measure.populations.count > 1)
          sub_ids = ('a'..'az').to_a
          measure.populations.each_with_index do |population, population_index|
            sub_id = sub_ids[population_index]
            population_title = metadata['subtitles'][sub_id] if metadata['subtitles']
            measure.populations[population_index]['title'] = population_title if population_title
          end
        end
        measure.custom_functions = metadata["custom_functions"]
        measure.force_sources = metadata["force_sources"]
      else
        measure.type = "unknown"
        measure.category = "Miscellaneous"
        measure.episode_of_care = false
        measure.continuous_variable = false
        puts "\tWARNING: Could not find metadata for measure: #{measure.hqmf_set_id}"
      end

      measure.population_criteria = json["population_criteria"]
      measure.data_criteria = json["data_criteria"]
      measure.source_data_criteria = json["source_data_criteria"]
      puts "\tCould not find episode ids #{measure.episode_ids} in measure #{measure.cms_id || measure.measure_id}" if (measure.episode_ids && measure.episode_of_care && (measure.episode_ids - measure.source_data_criteria.keys).length > 0)
      measure.measure_period = json["measure_period"]
      measure
    end

    def self.load_hqmf_cql_model_json(json, user, measure_oids, main_cql_library, cql_definition_dependency_structure, elm, elm_annotations, cql, measure_details, value_set_oid_version_objects)
      measure = CqlMeasure.new
      measure.user = user if user
      measure.bundle = user.bundle if (user && user.respond_to?(:bundle))
      measure.measure_id = json["id"]
      measure.cql = cql
      measure.elm = elm
      measure.elm_annotations = elm_annotations
      # Add metadata
      measure.hqmf_id = json["hqmf_id"]
      measure.hqmf_set_id = json["hqmf_set_id"]
      measure.hqmf_version_number = json["hqmf_version_number"]
      measure.cms_id = json["cms_id"]
      measure.title = json["title"]
      measure.description = json["description"]
      measure.measure_attributes = json["attributes"]
      measure.populations = json['populations']
      measure.value_set_oids = measure_oids
      measure.value_set_oid_version_objects = value_set_oid_version_objects

      # Set CQL specific information
      measure.cql = cql
      measure.elm = elm
      measure.main_cql_library = main_cql_library
      measure.cql_statement_dependencies = cql_definition_dependency_structure

      # Add metadata
      metadata = measure_details
      measure.measure_id = metadata["nqf_id"] || json["id"]
      measure.type = metadata["type"]
      measure.category = metadata["category"] || "Miscellaneous"
      measure.episode_of_care = metadata["episode_of_care"]
      measure.calculate_sdes = metadata['calculate_sdes']
      measure.continuous_variable = metadata["continuous_variable"] || false
      measure.composite = metadata["composite"]
      measure.episode_ids = metadata["episode_ids"]
      puts "\tWARNING: Episode of care does not align with episode ids existance" if ((!measure.episode_ids.nil? && measure.episode_ids.length > 0) ^ measure.episode_of_care)
      if (measure.populations.count > 1)
        sub_ids = ('a'..'az').to_a
        measure.populations.each_with_index do |population, population_index|
          sub_id = sub_ids[population_index]
          population_title = metadata['subtitles'][sub_id] if metadata['subtitles']
          measure.populations[population_index]['title'] = population_title if population_title
        end
      end

      # Set measure population information
      measure.population_criteria = json["population_criteria"]
      measure.data_criteria = json["data_criteria"]
      measure.source_data_criteria = json["source_data_criteria"]
      puts "\tCould not find episode ids #{measure.episode_ids} in measure #{measure.cms_id || measure.measure_id}" if (measure.episode_ids && measure.episode_of_care && (measure.episode_ids - measure.source_data_criteria.keys).length > 0)
      measure.measure_period = json["measure_period"]
      measure.population_criteria = json["population_criteria"]
      measure.populations_cql_map = json["populations_cql_map"]
      measure.observations = json["observations"]

      measure
    end

    def self.save_sources(measure, hqmf_path, html_path, xls_path=nil)
      # Save original files
      if (html_path)
        html_out_path = File.join(SOURCE_PATH, "html")
        FileUtils.mkdir_p html_out_path
        FileUtils.cp(html_path, File.join(html_out_path,"#{measure.hqmf_id}.html"))
      end

      if (xls_path)
        value_set_out_path = File.join(SOURCE_PATH, "value_sets")
        FileUtils.mkdir_p value_set_out_path
        FileUtils.cp(xls_path, File.join(value_set_out_path,"#{measure.hqmf_id}.xls"))
      end

      hqmf_out_path = File.join(SOURCE_PATH, "hqmf")
      FileUtils.mkdir_p hqmf_out_path
      FileUtils.cp(hqmf_path, File.join(hqmf_out_path, "#{measure.hqmf_id}.xml"))
    end

    def self.parse(xml_contents)
      doc = xml_contents.kind_of?(Nokogiri::XML::Document) ? xml_contents : Nokogiri::XML(xml_contents)
      doc
    end

    def self.clear_sources
      FileUtils.rm_r File.join(SOURCE_PATH, "html") if File.exist?(File.join(SOURCE_PATH, "html"))
      FileUtils.rm_r File.join(SOURCE_PATH, "value_sets") if File.exist?(File.join(SOURCE_PATH, "value_sets"))
      FileUtils.rm_r File.join(SOURCE_PATH, "hqmf") if File.exist?(File.join(SOURCE_PATH, "hqmf"))
    end

    def self.parse_measures_yml(measures_yml)
      YAML.load_file(measures_yml)['measures']
    end

    def self.get_parser(xml_contents)
      doc = self.parse(xml_contents)
      PARSERS.each do |p|
        if p.valid? doc
          return p.new
        end
      end
      raise "unknown document type"
    end

  end
end