WIP: Fix unary tag comments handling

Author: shulhi

compiler/syntax/src/res_comments_table.ml

@@ -24,7 +24,7 @@ let copy tbl =
 
 let empty = make ()
 
-let print_loc (k : Warnings.loc) =
+let print_location (k : Warnings.loc) =
   Doc.concat
     [
       Doc.lbracket;
@@ -41,7 +41,7 @@ let print_loc (k : Warnings.loc) =
 let print_entries tbl =
   Hashtbl.fold
     (fun (k : Location.t) (v : Comment.t list) acc ->
-      let loc = print_loc k in
+      let loc = print_location k in
       let doc =
         Doc.breakable_group ~force_break:true
           (Doc.concat
@@ -1472,9 +1472,22 @@ and walk_expression expr t comments =
     in
     let xs = exprs |> List.map (fun e -> Expression e) in
     walk_list xs t rest
-  | Pexp_jsx_unary_element _ ->
-    (* TODO: save me shulhi, not sure what needs to be done here *)
-    ()
+  | Pexp_jsx_unary_element {jsx_unary_element_props = props} ->
+      let xs = List.filter_map (fun prop -> 
+      match prop with 
+      | Parsetree.JSXPropPunning (_, _) -> None
+      | Parsetree.JSXPropValue ({txt; loc}, _, expr) -> 
+          let () = print_endline txt in
+          let () = print_location loc |> Doc.to_string ~width:80 |> print_endline in
+          let () = print_location expr.pexp_loc|> Doc.to_string ~width:80 |> print_endline in
+          let () = log t in
+          let (_leading, _inside, trailing) = partition_by_loc comments expr.pexp_loc in
+          let after_expr, _ = partition_adjacent_trailing expr.pexp_loc trailing in
+          attach t.trailing expr.pexp_loc after_expr;
+          Some (Expression expr)
+      | Parsetree.JSXPropSpreading (_loc, expr) -> Some (Expression expr)
+      ) props in
+      walk_list xs t []
   | Pexp_jsx_container_element
       {
         jsx_container_element_opening_tag_end = opening_greater_than;