class CancerRegistryReportingTestKit::HdeaGenerator::IGMetadataExtractor

def add_metadata_from_ig
  metadata.ig_version = "v#{ig_resources.ig.version}"
end

def add_metadata_from_resources
  metadata.groups =
    resources_in_structure_definitions.flat_map do |resource|
      supported_profile = resource.url
      GroupMetadataExtractor.new(resource, supported_profile, metadata, ig_resources).group_metadata
    end.compact
end

def add_missing_supported_profiles
  case ig_resources.ig.version
  when '3.1.1'
    # The US Core v3.1.1 Server Capability Statement does not list support for the
    # required vital signs profiles, so they need to be added
    ig_resources.capability_statement.rest.first.resource
      .find { |resource| resource.type == 'Observation' }
      .supportedProfile.concat [ 
                  'http://hl7.org/fhir/StructureDefinition/bodyheight',
                  'http://hl7.org/fhir/StructureDefinition/bodytemp',
                  'http://hl7.org/fhir/StructureDefinition/bp',
                  'http://hl7.org/fhir/StructureDefinition/bodyweight',
                  'http://hl7.org/fhir/StructureDefinition/heartrate',
                  'http://hl7.org/fhir/StructureDefinition/resprate'
      ]
                
  when '5.0.1'
    # The US Core v5.0.1 Server Capability Statement does not have supported-profile for Encounter
    ig_resources.capability_statement.rest.first.resource
      .find { |resource| resource.type == 'Encounter' }
      .supportedProfile.concat [
                  'http://hl7.org/fhir/us/core/StructureDefinition/us-core-encounter'
      ]
                
  end
end

def extract
  add_metadata_from_ig
  add_metadata_from_resources
  metadata
end

def initialize(ig_resources)
  self.ig_resources = ig_resources
  add_missing_supported_profiles
  remove_version_from_supported_profiles
  self.metadata = IGMetadata.new
end

def remove_version_from_supported_profiles
  resources_in_capability_statement.each do |resource|
    resource.supportedProfile.map! { |profile_url| profile_url.split('|').first }
  end
end

def resources_in_capability_statement
  ig_resources.capability_statement.rest.first.resource
end

def resources_in_structure_definitions
  ig_resources.resources_by_type['StructureDefinition'].map
end