1 files changed, 24 insertions, 0 deletions
diff --git a/ext/psych/lib/psych/json/tree_builder.rb b/ext/psych/lib/psych/json/tree_builder.rb
new file mode 100644
index 0000000000..91d7e6cad6
--- /dev/null
+++ b/ext/psych/lib/psych/json/tree_builder.rb
@@ -0,0 +1,24 @@
+module Psych
+  module JSON
+    ###
+    # Psych::JSON::TreeBuilder is an event based AST builder.  Events are sent
+    # to an instance of Psych::JSON::TreeBuilder and a JSON AST is constructed.
+    class TreeBuilder < Psych::TreeBuilder
+      def start_document version, tag_directives, implicit
+        super(version, tag_directives, true)
+      end
+
+      def end_document implicit_end
+        super(true)
+      end
+
+      def start_mapping anchor, tag, implicit, style
+        super(anchor, tag, implicit, Nodes::Mapping::FLOW)
+      end
+
+      def start_sequence anchor, tag, implicit, style
+        super(anchor, tag, implicit, Nodes::Sequence::FLOW)
+      end
+    end
+  end
+end