Merge pull request #1505 from cryspen/ocaml-bridge-reviewable

Rust Engine: OCaml bridge for the AST (OCaml AST -> Rust AST)

Author: W95Psp

engine/lib/concrete_ident/concrete_ident.ml

@@ -27,6 +27,9 @@ module Fresh_module : sig
 
   val to_mod_path : t -> View.ModPath.t
   (** Compute a module path for a fresh module. *)
+
+  val to_rust_ast : t -> Rust_engine_types.fresh_module
+  val from_rust_ast : Rust_engine_types.fresh_module -> t
 end = struct
   open View
 
@@ -111,6 +114,21 @@ end = struct
     |> snd |> Option.value_exn
 
   let get_path_hints { hints; _ } = hints
+
+  let to_rust_ast ({ id; hints; label } : t) : Rust_engine_types.fresh_module =
+    {
+      id = Int.to_string id;
+      hints = List.map ~f:Explicit_def_id.to_rust_ast hints;
+      label;
+    }
+
+  let from_rust_ast ({ id; hints; label } : Rust_engine_types.fresh_module) : t
+      =
+    {
+      id = Int.of_string id;
+      hints = List.map ~f:Explicit_def_id.from_rust_ast hints;
+      label;
+    }
 end
 
 type reserved_suffix = [ `Cast | `Pre | `Post ]
@@ -743,3 +761,27 @@ let matches_namespace (ns : Types.namespace) (did : t) : bool =
     | _ -> false
   in
   aux ns.chunks path
+
+let to_rust_ast ({ def_id; moved; suffix } : t) : Rust_engine_types.concrete_id
+    =
+  let moved = Option.map ~f:Fresh_module.to_rust_ast moved in
+  let suffix =
+    Option.map
+      ~f:(fun s ->
+        match s with
+        | `Cast -> Rust_engine_types.Cast
+        | `Pre -> Rust_engine_types.Pre
+        | `Post -> Rust_engine_types.Post)
+      suffix
+  in
+  { def_id = Explicit_def_id.to_rust_ast def_id; moved; suffix }
+
+let from_rust_ast ({ def_id; moved; suffix } : Rust_engine_types.concrete_id) :
+    t =
+  let moved = Option.map ~f:Fresh_module.from_rust_ast moved in
+  let suffix =
+    Option.map
+      ~f:(fun s -> match s with Cast -> `Cast | Pre -> `Pre | Post -> `Post)
+      suffix
+  in
+  { def_id = Explicit_def_id.from_rust_ast def_id; moved; suffix }

engine/lib/concrete_ident/concrete_ident.mli

@@ -104,3 +104,5 @@ module ImplInfoStore : sig
 end
 
 val matches_namespace : Types.namespace -> t -> bool
+val to_rust_ast : t -> Rust_engine_types.concrete_id
+val from_rust_ast : Rust_engine_types.concrete_id -> t

engine/lib/concrete_ident/explicit_def_id.ml

@@ -125,3 +125,46 @@ module ImplInfoStore = struct
   let lookup_raw (impl_def_id : t) : Types.impl_infos option =
     find (to_def_id impl_def_id)
 end
+
+module ToRustAST = struct
+  module A = Types
+  module B = Rust_engine_types
+
+  let rec def_id_contents_to_rust_ast
+      ({ krate; path; parent; kind; _ } : A.def_id_contents) : B.def_id =
+    let f (o : A.def_id) = def_id_contents_to_rust_ast o.contents.value in
+    let parent = Option.map ~f parent in
+    { krate; path; parent; kind }
+
+  let to_rust_ast ({ is_constructor; def_id } : t) : B.explicit_def_id =
+    { is_constructor; def_id = def_id_contents_to_rust_ast def_id }
+end
+
+module FromRustAST = struct
+  module A = Rust_engine_types
+  module B = Types
+
+  let rec def_id_contents_to_rust_ast
+      ({ krate; path; parent; kind; _ } : A.def_id) : B.def_id_contents =
+    let f (o : A.def_id) : B.def_id =
+      let contents : B.node_for__def_id_contents =
+        { value = def_id_contents_to_rust_ast o; id = Int64.of_int (-1) }
+      in
+      { contents }
+    in
+    let parent = Option.map ~f parent in
+    {
+      krate;
+      path;
+      parent;
+      kind;
+      index = (Int64.of_int (-1), Int64.of_int (-1), None);
+      is_local = false;
+    }
+
+  let to_rust_ast ({ is_constructor; def_id } : A.explicit_def_id) : t =
+    { is_constructor; def_id = def_id_contents_to_rust_ast def_id }
+end
+
+let to_rust_ast = ToRustAST.to_rust_ast
+let from_rust_ast = FromRustAST.to_rust_ast

engine/lib/concrete_ident/explicit_def_id.mli

@@ -81,3 +81,6 @@ module ImplInfoStore : sig
   engine, this function may return [None] even though the supplied
   identifier points to an [Impl] block. *)
 end
+
+val to_rust_ast : t -> Rust_engine_types.explicit_def_id
+val from_rust_ast : Rust_engine_types.explicit_def_id -> t