implicit start and end document

3 files changed, 22 insertions, 2 deletions

diff --git a/ext/psych/parser.c b/ext/psych/parser.c
index a9b0667..2efdb09 100644
--- a/ext/psych/parser.c
+++ b/ext/psych/parser.c
@@ -60,6 +60,11 @@ static VALUE parse_string(VALUE self, VALUE string)
           );
         }
         break;
+      case YAML_DOCUMENT_END_EVENT:
+        rb_funcall(handler, rb_intern("end_document"), 1,
+            event.data.document_end.implicit == 1 ? Qtrue : Qfalse
+        );
+        break;
       case YAML_STREAM_END_EVENT:
         rb_funcall(handler, rb_intern("end_stream"), 0);
         done = 1;
diff --git a/lib/psych/parser/handler.rb b/lib/psych/parser/handler.rb
index 211ebf1..5b330a0 100644
--- a/lib/psych/parser/handler.rb
+++ b/lib/psych/parser/handler.rb
@@ -15,6 +15,11 @@ module Psych
       end
 
       ###
+      # Called with the document ends.
+      def end_document implicit = true
+      end
+
+      ###
       # Called when the YAML stream ends
       def end_stream
       end
diff --git a/test/psych/test_parser.rb b/test/psych/test_parser.rb
index 187bace..13d4376 100644
--- a/test/psych/test_parser.rb
+++ b/test/psych/test_parser.rb
@@ -33,9 +33,19 @@ module Psych
       assert_called :start_stream
     end
 
-    def test_start_document_implicit
+    def test_end_document_implicit
+      @parser.parse("\"foo\"\n")
+      assert_called :end_document, [true]
+    end
+
+    def test_end_document_explicit
       @parser.parse("\"foo\"\n")
-      assert_called :start_document, [[], [], true]
+      assert_called :end_document, [true]
+    end
+
+    def test_start_document_implicit
+      @parser.parse("\"foo\"\n...")
+      assert_called :end_document, [false]
     end
 
     def test_start_document_version