hask to get open elements in all cases

src/html_parser.ml

@@ -1015,7 +1015,8 @@ end
 
 
 
-let parse requested_context report (tokens, set_tokenizer_state, set_foreign) =
+let parse ?(get_opens=ref None) requested_context report
+      (tokens, set_tokenizer_state, set_foreign) =
   let context = Context.uninitialized () in
 
   let throw = ref (fun _ -> ()) in
@@ -1036,14 +1037,14 @@ let parse requested_context report (tokens, set_tokenizer_state, set_foreign) =
   set_foreign (fun () ->
     Stack.current_element_is_foreign context open_elements);
 
-  let report l error throw k =
-    let elts = List.map
+  let opens () = List.map
                  (fun e ->
                    let (ns, s) = e.element_name in
                    ((Ns.to_string ns, s), e.location, e.attributes))
                  !open_elements
-    in
-    report elts l error throw k in
+  in
+  get_opens := Some opens;
+  let report l error throw k = report (opens ()) l error throw k in
   let report_if = Error.report_if report in
   let unmatched_end_tag l name k =
     report l (`Unmatched_end_tag name) !throw k in

src/html_parser.mli

@@ -4,6 +4,7 @@
 open Common
 
 val parse :
+  ?get_opens:((unit -> open_elements) option ref) ->
   [< `Document | `Fragment of string ] option ->
   (open_elements -> Error.parse_handler) ->
   (location * Html_tokenizer.token) Kstream.t *

src/markup.ml

@@ -119,13 +119,15 @@ struct
     |> Utility.strings_to_bytes
 
   let parse_html report ?encoding context source =
-    let report' = report [] in
+    let get_opens = ref None in
+    let parse = Html_parser.parse ~get_opens in
+    let report' = report (match !get_opens with None -> [] | Some f -> f ()) in
     let with_encoding (encoding : Encoding.t) k =
       source
       |> encoding ~report:report'
       |> Input.preprocess Common.is_valid_html_char report'
       |> Html_tokenizer.tokenize report'
-      |> Html_parser.parse context report
+      |> parse context report
       |> k
     in