Raise exception on duplicate keys

ruby · May 13, 2022 · 469c7e3 · 469c7e3
1 parent 40743b3
commit 469c7e3
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/lib/psych/handlers/document_stream.rb b/lib/psych/handlers/document_stream.rb
@@ -18,6 +18,18 @@ def end_document implicit_end = !streaming?
         @last.implicit_end = implicit_end
         @block.call pop
       end
+
+      def end_mapping
+        mapping = pop
+        keys = {}
+        mapping.children.each_slice(2) do |(key_scalar, _)|
+          next if key_scalar.is_a?(Psych::Nodes::Sequence) or key_scalar.is_a?(Psych::Nodes::Alias) or key_scalar.is_a?(Psych::Nodes::Mapping)
+          key = key_scalar.value
+          raise Psych::Exception, "Duplicate key #{key} exists on this level" if keys.key? key
+          keys[key] = nil
+        end
+        mapping
+      end
     end
   end
 end