require 'fhir_models'

require_relative 'ext/fhir_models'
require_relative 'generator/ig_loader'
require_relative 'generator/ig_metadata_extractor'
require_relative 'generator/group_generator'
require_relative 'generator/must_support_test_generator'
require_relative 'generator/provenance_revinclude_search_test_generator'
require_relative 'generator/read_test_generator'
require_relative 'generator/reference_resolution_test_generator'
require_relative 'generator/resource_list_generator'
require_relative 'generator/search_test_generator'
require_relative 'generator/suite_generator'
require_relative 'generator/validation_test_generator'

module USCoreTestKit
  class Generator
    attr_accessor :ig_resources, :ig_metadata

    def generate
      load_ig_package
      extract_metadata
      generate_resource_list
      generate_search_tests
      generate_read_tests
      # TODO: generate_vread_tests
      # TODO: generate_history_tests
      generate_provenance_revinclude_search_tests
      generate_validation_tests
      generate_must_support_tests
      generate_reference_resolution_tests

      generate_groups

      generate_suites
    end

    def extract_metadata
      self.ig_metadata = IGMetadataExtractor.new(ig_resources).extract
      File.open(File.join(__dir__, 'generated', 'metadata.yml'), 'w') do |file|
        file.write(YAML.dump(ig_metadata.to_hash))
      end
    end

    def load_ig_package
      FHIR.logger = Logger.new('/dev/null')
      self.ig_resources = IGLoader.new.load
    end

    def generate_resource_list
      ResourceListGenerator.generate(ig_metadata)
    end

    def generate_reference_resolution_tests
      ReferenceResolutionTestGenerator.generate(ig_metadata)
    end

    def generate_must_support_tests
      MustSupportTestGenerator.generate(ig_metadata)
    end

    def generate_validation_tests
      ValidationTestGenerator.generate(ig_metadata)
    end

    def generate_read_tests
      ReadTestGenerator.generate(ig_metadata)
    end

    def generate_search_tests
      SearchTestGenerator.generate(ig_metadata)
    end

    def generate_provenance_revinclude_search_tests
      ProvenanceRevincludeSearchTestGenerator.generate(ig_metadata)
    end

    def generate_groups
      GroupGenerator.generate(ig_metadata)
    end

    def generate_suites
      SuiteGenerator.generate(ig_metadata)
    end
  end
end