wip

Author: amosr-msft

src/ml/FStarC_Parser_ParseIt.ml

@@ -527,6 +527,17 @@ let _ = FStarC_Parser_AST_Util.register_extension_lang_parser "fstar" parse_fsta
 
 type lang_opts = string option
 
+let rec insert_use_lang lang decls =
+  match decls with
+  | [] -> []
+  | { FStarC_Parser_AST.d = FStarC_Parser_AST.TopLevelModule _ } as d ::ds ->
+    let use_lang =
+      { d with FStarC_Parser_AST.d = FStarC_Parser_AST.UseLangDecls lang; }
+    in
+    d :: use_lang :: ds
+  | d::ds ->
+  d :: insert_use_lang lang ds
+
 let parse_lang lang fn =
   let s =
     match fn with
@@ -547,9 +558,11 @@ let parse_lang lang fn =
     let frag_pos = FStarC_Range.mk_pos s.frag_line s.frag_col in
     let rng = FStarC_Range.mk_range s.frag_fname frag_pos frag_pos in
     let decls = FStarC_Parser_AST_Util.parse_extension_lang lang s.frag_text rng in
+    let decls = insert_use_lang lang decls in
     let comments = FStarC_Parser_Util.flush_comments () in
     let frag = match fn with
-      | Filename _ | Toplevel _ -> FStarC_Parser_AST.as_frag decls
+      | Filename _ | Toplevel _ ->
+        FStarC_Parser_AST.as_frag decls
       | _ -> FStar_Pervasives.Inr decls
     in
     ASTFragment (frag, comments)
@@ -630,7 +643,9 @@ let parse (lang_opt:lang_opts) fn =
   | Some lang, _ -> parse_lang lang fn
   | None, Filename fn' -> begin
     match take_lang_extension fn' with
-    | Some lang -> parse_lang lang fn
+    | Some lang ->
+      print_endline ("{\"kind\":\"message\",\"level\":\"info\",\"contents\":\"got lang " ^ lang ^ "\"}");
+      parse_lang lang fn
     | None -> parse_no_lang fn
     end
   | None, Toplevel fn' -> begin
@@ -639,6 +654,11 @@ let parse (lang_opt:lang_opts) fn =
     | None -> parse_no_lang fn
     end
   | _, _ -> parse_no_lang fn
+  (*
+  ../.vscode/settings.json
+
+  {"kind":"message","query-id":"2","level":"info","contents":"Parsed 7 declarations\n"}
+  *)
 
 
 (** Parsing of command-line error/warning/silent flags. *)