class OasRails::Spec::MediaType

def clean_example_object(obj:)
  obj.reject { |key, _| OasRails.config.send("excluded_columns_#{@context}").include?(key.to_sym) }
end

def fetch_factory_bot_examples(klass:)
  klass_sym = Utils.class_to_symbol(klass)
  begin
    @factory_examples[klass_sym] = FactoryBot.build_stubbed_list(klass_sym, 1) if @factory_examples[klass_sym].nil?
    @factory_examples[klass_sym].each_with_index.to_h do |obj, index|
      ["#{klass_sym}#{index + 1}", { value: { klass_sym => clean_example_object(obj: obj.as_json) } }]
    end
  rescue KeyError
    {}
  end
end

def fetch_fixture_examples(klass:)
  fixture_file = Rails.root.join('test', 'fixtures', "#{klass.to_s.pluralize.downcase}.yml")
  begin
    fixture_data = YAML.load_file(fixture_file).with_indifferent_access
  rescue Errno::ENOENT
    return {}
  end
  fixture_data.transform_values { |attributes| { value: { klass.to_s.downcase => clean_example_object(obj: attributes) } } }
end

def initialize(specification)
  @specification = specification
  @schema = {}
  @example =  {}
  @examples = {}
end

def oas_fields
  [:schema, :example, :examples, :encoding]
end

def search_for_examples_in_tests(klass, context: :incoming, utils: Utils)
  @context = context
  case utils.detect_test_framework
  when :factory_bot
    fetch_factory_bot_examples(klass:)
  when :fixtures
    fetch_fixture_examples(klass:)
  else
    {}
  end
end