unikernel/duniverse/dune_/otherlibs/dune-rpc/private/menu.ml

@@ -0,0 +1,47 @@
+open Import
+open Types
+
+(* This list of methods and generations is exactly the list of existing methods
+   as of commit [3cd240e], which the initial versioning implementation was based
+   on. *)
+let compatibility_menu =
+  Method.Name.Map.of_list_exn
+    [ "ping", 1
+    ; "diagnostics", 1
+    ; "shutdown", 1
+    ; "subscribe", 1
+    ; "unsubscribe", 1
+    ; "build", 1
+    ; "status", 1
+    ; "notify/abort", 1
+    ; "notify/diagnostic", 1
+    ; "notify/log", 1
+    ; "notify/progress", 1
+    ]
+;;
+
+type t = Method.Version.t Method.Name.Map.t
+
+let default = compatibility_menu
+let find = Method.Name.Map.find
+
+let select_common ~local_versions ~remote_versions =
+  let selected_versions =
+    List.filter_map remote_versions ~f:(fun (method_, remote_versions) ->
+      let remote_versions = Method.Version.Set.of_list remote_versions in
+      let open Option.O in
+      let* local_versions = Method.Name.Map.find local_versions method_ in
+      let+ greatest_common_version =
+        Method.Version.Set.max_elt
+          (Method.Version.Set.inter remote_versions local_versions)
+      in
+      method_, greatest_common_version)
+  in
+  match selected_versions with
+  | [] -> None
+  | _ :: _ -> Some (Method.Name.Map.of_list_exn selected_versions)
+;;
+
+let of_list = Method.Name.Map.of_list
+let to_list = Method.Name.Map.to_list
+let to_dyn = Method.Name.Map.to_dyn Int.to_dyn