us_quality_core_test_kit 0.1.0 → 0.1.2

data/lib/us_quality_core_test_kit/generator/group_metadata_extractor.rb

@@ -1,19 +1,195 @@
 # frozen_string_literal: true
 
-require 'us_core_test_kit/generator/group_metadata_extractor'
-
 require_relative 'group_metadata'
-require_relative 'special_cases'
-require_relative 'naming'
+require_relative 'ig_metadata'
+require_relative 'must_support_metadata_extractor'
 require_relative 'search_metadata_extractor'
 require_relative 'terminology_binding_metadata_extractor'
-require_relative 'must_support_metadata_extractor'
-
-require_relative 'ig_metadata'
+require_relative 'naming'
+require_relative 'special_cases'
 
 module USQualityCoreTestKit
   class Generator
-    class GroupMetadataExtractor < USCoreTestKit::Generator::GroupMetadataExtractor
+    class GroupMetadataExtractor
+      attr_accessor :resource_capabilities, :profile_url, :ig_metadata, :ig_resources
+
+      def initialize(resource_capabilities, profile_url, ig_metadata, ig_resources)
+        self.resource_capabilities = resource_capabilities
+        self.profile_url = profile_url
+        self.ig_metadata = ig_metadata
+        self.ig_resources = ig_resources
+      end
+
+      def group_metadata_hash
+        @group_metadata_hash ||=
+          {
+            name: name,
+            class_name: class_name,
+            version: version,
+            reformatted_version: reformatted_version,
+            resource: resource,
+            profile_url: profile_url,
+            profile_name: profile_name,
+            profile_version: profile_version,
+            title: title,
+            short_description: short_description,
+            interactions: interactions,
+            operations: operations,
+            searches: searches,
+            search_definitions: search_definitions,
+            include_params: include_params,
+            revincludes: revincludes,
+            required_concepts: required_concepts,
+            must_supports: must_supports,
+            mandatory_elements: mandatory_elements,
+            bindings: bindings,
+            references: references,
+            resource_conformance_expectation: resource_conformance_expectation
+          }
+
+        mark_mandatory_and_must_support_searches
+        handle_special_cases
+
+        @group_metadata_hash
+      end
+
+      ### BEGIN SPECIAL CASES ###
+
+      def handle_special_cases
+        set_first_search
+      end
+
+      def set_first_search
+        search = searches.find { |param| param[:names] == first_search_params }
+        return if search.nil?
+
+        searches.delete(search)
+        searches.unshift(search)
+      end
+
+      ### END SPECIAL CASES ###
+
+      def profile
+        @profile ||= ig_resources.profile_by_url(profile_url)
+      end
+
+      def profile_elements
+        @profile_elements ||= profile.snapshot.element
+      end
+
+      def base_name
+        profile_url.split('StructureDefinition/').last
+      end
+
+      def name
+        base_name.tr('-', '_')
+      end
+
+      def version
+        ig_metadata.ig_version
+      end
+
+      def reformatted_version
+        ig_metadata.reformatted_version
+      end
+
+      def resource
+        resource_capabilities.type
+      end
+
+      def profile_name
+        profile.title.gsub('  ', ' ')
+      end
+
+      def profile_version
+        profile.version
+      end
+
+      def interactions
+        @interactions ||=
+          resource_capabilities.interaction.map do |interaction|
+            {
+              code: interaction.code,
+              expectation: interaction.extension.first.valueCode # TODO: fix expectation extension finding
+            }
+          end
+      end
+
+      def operations
+        @operations ||=
+          resource_capabilities.operation.map do |operation|
+            {
+              code: operation.name,
+              expectation: operation.extension.first.valueCode # TODO: fix expectation extension finding
+            }
+          end
+      end
+
+      def searches
+        @searches ||= search_metadata_extractor.searches
+      end
+
+      def search_definitions
+        @search_definitions ||= search_metadata_extractor.search_definitions
+      end
+
+      def include_params
+        resource_capabilities.searchInclude || []
+      end
+
+      def revincludes
+        resource_capabilities.searchRevInclude || []
+      end
+
+      def required_concepts
+        # The base FHIR vital signs profile has a required binding that isn't
+        # relevant for any of its child profiles
+        return [] if resource == 'Observation'
+
+        profile_elements
+          .select { |element| element.type&.any? { |type| type.code == 'CodeableConcept' } }
+          .select { |element| element.binding&.strength == 'required' }
+          .map { |element| element.path.gsub("#{resource}.", '').gsub('[x]', 'CodeableConcept') }
+          .uniq
+      end
+
+      def bindings
+        @bindings ||=
+          terminology_binding_metadata_extractor.terminology_bindings
+      end
+
+      def must_supports
+        @must_supports ||=
+          must_support_metadata_extractor.must_supports
+      end
+
+      def mandatory_elements
+        @mandatory_elements ||=
+          profile_elements
+          .select { |element| element.min.positive? }
+          .map(&:path)
+          .uniq
+      end
+
+      def references
+        @references ||=
+          profile_elements
+          .select { |element| element.type&.first&.code == 'Reference' }
+          .map do |reference_definition|
+            {
+              path: reference_definition.path,
+              profiles: reference_definition.type.first.targetProfile
+            }
+          end
+      end
+
+      def resource_conformance_expectation
+        @resource_conformance_expectation ||=
+          resource_capabilities.extension.find do |extension|
+            extension.url == 'http://hl7.org/fhir/StructureDefinition/capabilitystatement-expectation'
+          end&.valueCode
+      end
+
       def group_metadata
         @group_metadata ||=
           GroupMetadata.new(group_metadata_hash)

data/lib/us_quality_core_test_kit/generator/ig_loader.rb

@@ -1,12 +1,56 @@
 # frozen_string_literal: true
 
-require 'us_core_test_kit/generator/ig_loader'
-
+require 'active_support/all'
+require 'fhir_models'
+require 'pathname'
+require 'rubygems/package'
+require 'zlib'
 require_relative 'ig_resources'
 
 module USQualityCoreTestKit
   class Generator
-    class IGLoader < USCoreTestKit::Generator::IGLoader
+    class IGLoader
+      attr_accessor :ig_file_name
+
+      def initialize(ig_file_name)
+        self.ig_file_name = ig_file_name
+      end
+
+      def load
+        load_ig
+        load_standalone_resources
+      end
+
+      def load_ig
+        tar = Gem::Package::TarReader.new(
+          Zlib::GzipReader.open(ig_file_name)
+        )
+
+        tar.each do |entry|
+          next if entry.directory?
+
+          file_name = entry.full_name.split('/').last
+
+          next if file_name.end_with? 'openapi.json'
+
+          next unless file_name.end_with? '.json'
+
+          next unless entry.full_name.start_with? 'package/'
+
+          begin
+            resource = FHIR.from_contents(entry.read)
+            next if resource.nil?
+          rescue StandardError
+            puts "#{file_name} does not appear to be a FHIR resource."
+            next
+          end
+
+          ig_resources.add(resource)
+        end
+
+        ig_resources
+      end
+
       def ig_resources
         @ig_resources ||= IGResources.new
       end

data/lib/us_quality_core_test_kit/generator/ig_metadata.rb

@@ -1,11 +1,60 @@
 # frozen_string_literal: true
 
-require 'us_core_test_kit/generator/ig_metadata'
 require_relative 'special_cases'
 
 module USQualityCoreTestKit
   class Generator
-    class IGMetadata < USCoreTestKit::Generator::IGMetadata
+    class IGMetadata
+      attr_accessor :ig_version, :groups
+
+      def reformatted_version
+        @reformatted_version ||= ig_version.delete('.').gsub('-', '_')
+      end
+
+      def ordered_groups
+        @ordered_groups ||=
+          [patient_group] + non_delayed_groups + delayed_groups
+      end
+
+      def patient_group
+        @patient_group ||=
+          groups.find { |group| group.resource == 'Patient' }
+      end
+
+      def delayed_groups
+        @delayed_groups ||=
+          groups.select(&:delayed?)
+      end
+
+      def non_delayed_groups
+        @non_delayed_groups ||=
+          groups.reject(&:delayed?) - [patient_group]
+      end
+
+      def postprocess_groups(ig_resources)
+        groups.each do |group|
+          group.add_delayed_references(delayed_profiles, ig_resources)
+        end
+      end
+
+      def granular_scope_resource_type_groups
+        @granular_scope_resource_type_groups ||=
+          Hash.new { |hash, key| hash[key] = [] }
+      end
+
+      def granular_scope_groups
+        @granular_scope_groups ||= []
+      end
+
+      def to_hash
+        {
+          ig_version:,
+          groups: groups.map(&:to_hash),
+          granular_scope_resource_type_groups:,
+          granular_scope_groups:
+        }
+      end
+
       def delayed_profiles
         @delayed_profiles ||=
           delayed_groups.reject { |group| SpecialCases.exclude_group?(group) }.map(&:profile_url)

data/lib/us_quality_core_test_kit/generator/ig_metadata_extractor.rb

@@ -1,21 +1,40 @@
 # frozen_string_literal: true
 
-require 'us_core_test_kit/generator/ig_metadata_extractor'
-
 require_relative 'ig_metadata'
 require_relative 'group_metadata_extractor'
 require_relative 'special_cases'
 
 module USQualityCoreTestKit
   class Generator
-    class IGMetadataExtractor < USCoreTestKit::Generator::IGMetadataExtractor
-      def initialize(ig_resources)
-        super
+    class IGMetadataExtractor
+      attr_accessor :ig_resources, :metadata
 
+      def initialize(ig_resources)
+        self.ig_resources = ig_resources
+        remove_version_from_supported_profiles
+        remove_extra_supported_profiles
         self.metadata = IGMetadata.new
       end
 
-      def add_missing_supported_profiles; end
+      def extract
+        add_metadata_from_ig
+        add_metadata_from_resources
+        metadata
+      end
+
+      def add_metadata_from_ig
+        metadata.ig_version = "v#{ig_resources.ig.version}".delete('-ballot')
+      end
+
+      def resources_in_capability_statement
+        ig_resources.capability_statement.rest.first.resource
+      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 remove_extra_supported_profiles
         resources_in_capability_statement.each do |resource|

data/lib/us_quality_core_test_kit/generator/ig_resources.rb

@@ -1,10 +1,28 @@
 # frozen_string_literal: true
 
-require 'us_core_test_kit/generator/ig_resources'
-
 module USQualityCoreTestKit
   class Generator
-    class IGResources < USCoreTestKit::Generator::IGResources
+    class IGResources
+      def add(resource)
+        resources_by_type[resource.resourceType] << resource
+      end
+
+      def ig
+        resources_by_type['ImplementationGuide'].first
+      end
+
+      def inspect
+        'IGResources'
+      end
+
+      def value_set_by_url(url)
+        resources_by_type['ValueSet'].find { |profile| profile.url == url }
+      end
+
+      def code_system_by_url(url)
+        resources_by_type['CodeSystem'].find { |system| system.url == url }
+      end
+
       def capability_statement(mode = 'server')
         resources_by_type['CapabilityStatement'].reverse.find do |capability_statement_resource|
           capability_statement_resource.rest.any? { |r| r.mode == mode }
@@ -56,6 +74,12 @@ module USQualityCoreTestKit
 
         candidates.first
       end
+
+      private
+
+      def resources_by_type
+        @resources_by_type ||= Hash.new { |hash, key| hash[key] = [] }
+      end
     end
   end
 end