18F/identity-idp

# frozen_string_literal: true

module DocAuth
  module Mock
    module YmlLoaderConcern
      extend ActiveSupport::Concern
      # @param [String] uploaded_file: string content
      def parse_yaml(uploaded_file)
        data = YAML.safe_load(uploaded_file, permitted_classes: [Date])
        if data.is_a?(Hash)
          ['dob', 'state_id_expiration'].each do |date_key|
            if (date_s = data.dig('document', date_key))
              data['document'][date_key] = DateParser.parse_legacy(date_s)
            end
          end

          if data.dig('document', 'zipcode')
            data['document']['zipcode'] = data.dig('document', 'zipcode').to_s
          end

          JSON.parse(data.to_json) # converts Dates back to strings
        else
          { general: ["YAML data should have been a hash, got #{data.class}"] }
        end
      rescue Psych::SyntaxError
        if uploaded_file.ascii_only? # don't want this error for images
          { general: ['invalid YAML file'] }
        else
          {}
        end
      end
    end
  end
end