Add ml printing

Author: nojaf

compiler/ml/pprintast.ml

@@ -794,17 +794,54 @@ and simple_expr ctxt f x =
       let expression = expression ctxt in
       pp f fmt (pattern ctxt) s expression e1 direction_flag df expression e2
         expression e3
+      (* TODO: what should this really be?
+        I miss some context why this print form even exists.
+        Can/should we improve the output here?
+        *)
     | Pexp_jsx_fragment (_, xs, _) ->
       pp f "<>%a</>" (list (simple_expr ctxt)) (collect_jsx_children xs)
-    | Pexp_jsx_unary_element _ -> failwith "TODO: Pexp_jsx_unary_element 2"
-    | Pexp_jsx_container_element _ ->
-      failwith "TODO: Pexp_jsx_container_element 2"
+    | Pexp_jsx_unary_element
+        {jsx_unary_element_tag_name = tag_name; jsx_unary_element_props = props}
+      -> (
+      let name = Longident.flatten tag_name.txt |> String.concat "." in
+      match props with
+      | [] -> pp f "<%s />" name
+      | _ -> pp f "<%s %a />" name (print_jsx_props ctxt) props)
+    | Pexp_jsx_container_element
+        {
+          jsx_container_element_tag_name_start = tag_name;
+          jsx_container_element_props = props;
+          jsx_container_element_children = children;
+        } -> (
+      let name = Longident.flatten tag_name.txt |> String.concat "." in
+      match props with
+      | [] ->
+        pp f "<%s>%a</%s>" name
+          (list (simple_expr ctxt))
+          (collect_jsx_children children)
+          name
+      | _ ->
+        pp f "<%s %a>%a</%s>" name (print_jsx_props ctxt) props
+          (list (simple_expr ctxt))
+          (collect_jsx_children children)
+          name)
     | _ -> paren true (expression ctxt) f x
 
 and collect_jsx_children = function
   | JSXChildrenSpreading e -> [e]
   | JSXChildrenItems xs -> xs
 
+and print_jsx_prop ctxt f = function
+  | JSXPropPunning (is_optional, name) ->
+    pp f "%s" (if is_optional then "?" ^ name.txt else name.txt)
+  | JSXPropValue (name, is_optional, value) ->
+    pp f "%s=%s%a" name.txt
+      (if is_optional then "?" else "")
+      (simple_expr ctxt) value
+  | JSXPropSpreading (_, expr) -> pp f "{...%a}" (simple_expr ctxt) expr
+
+and print_jsx_props ctxt f = list ~sep:" " (print_jsx_prop ctxt) f
+
 and attributes ctxt f l = List.iter (attribute ctxt f) l
 
 and item_attributes ctxt f l = List.iter (item_attribute ctxt f) l