refactor(pkg): consolidate repository handling

Signed-off-by: Ali Caglayan <[email protected]>

Author: Alizter

bin/pkg/pkg_common.ml

@@ -120,28 +120,7 @@ let unset_solver_vars_of_workspace workspace ~lock_dir_path =
 ;;
 
 let get_repos repos ~repositories =
-  let module Repository = Dune_pkg.Pkg_workspace.Repository in
-  repositories
-  |> Fiber.parallel_map ~f:(fun (loc, name) ->
-    match Repository.Name.Map.find repos name with
-    | None ->
-      User_error.raise
-        ~loc
-        [ Pp.textf "Repository '%s' is not a known repository"
-          @@ Repository.Name.to_string name
-        ]
-    | Some repo ->
-      let loc, opam_url = Repository.opam_url repo in
-      let module Opam_repo = Dune_pkg.Opam_repo in
-      (match Dune_pkg.OpamUrl.classify opam_url loc with
-       | `Git -> Opam_repo.of_git_repo loc opam_url
-       | `Path path -> Fiber.return @@ Opam_repo.of_opam_repo_dir_path loc path
-       | `Archive ->
-         User_error.raise
-           ~loc
-           [ Pp.textf "Repositories stored in archives (%s) are currently unsupported"
-             @@ OpamUrl.to_string opam_url
-           ]))
+  Dune_pkg.Opam_repo.resolve_repositories ~available_repos:repos ~repositories
 ;;
 
 let find_local_packages =

src/dune_pkg/opam_repo.ml

@@ -122,6 +122,31 @@ let of_git_repo loc url =
   { source = Repo at_rev; serializable; loc }
 ;;
 
+let resolve_repositories ~available_repos ~repositories =
+  repositories
+  |> Fiber.parallel_map ~f:(fun (loc, name) ->
+    match Workspace.Repository.Name.Map.find available_repos name with
+    | None ->
+      User_error.raise
+        ~loc
+        [ Pp.textf
+            "Repository '%s' is not a known repository"
+            (Workspace.Repository.Name.to_string name)
+        ]
+    | Some repo ->
+      let loc, opam_url = Workspace.Repository.opam_url repo in
+      (match OpamUrl.classify opam_url loc with
+       | `Git -> of_git_repo loc opam_url
+       | `Path path -> Fiber.return @@ of_opam_repo_dir_path loc path
+       | `Archive ->
+         User_error.raise
+           ~loc
+           [ Pp.textf
+               "Repositories stored in archives (%s) are currently unsupported"
+               (OpamUrl.to_string opam_url)
+           ]))
+;;
+
 let revision t =
   match t.source with
   | Repo r -> r

src/dune_pkg/opam_repo.mli

@@ -23,6 +23,16 @@ val of_opam_repo_dir_path : Loc.t -> Path.t -> t
     supports. *)
 val of_git_repo : Loc.t -> OpamUrl.t -> t Fiber.t
 
+(** [resolve_repositories ~available_repos ~repositories] resolves a list of
+    repository references by looking them up in [available_repos] and creating
+    appropriate [t] instances based on their URL types (git, local path, or
+    archive). Raises [User_error] if a repository is not found or if an archive
+    URL is encountered (not supported). *)
+val resolve_repositories
+  :  available_repos:Workspace.Repository.t Workspace.Repository.Name.Map.t
+  -> repositories:(Loc.t * Workspace.Repository.Name.t) list
+  -> t list Fiber.t
+
 val revision : t -> Rev_store.At_rev.t
 val serializable : t -> Serializable.t option
 

src/dune_rules/lock_rules.ml

@@ -312,9 +312,6 @@ let setup_lock_rules ~dir ~lock_dir : Gen_rules.result =
          >>| Dune_lang.Package.Name.Map.map ~f:Local_package.of_package
          |> Action_builder.of_memo
        and+ repos =
-         (* CR-soon Alizter: This repository handling logic is duplicated in
-            bin/pkg/pkg_common.ml:get_repos. The OpamUrl.classify pattern
-            matching and repository resolution could be shared. *)
          Action_builder.of_memo
            (Memo.of_thunk (fun () ->
               let repositories =
@@ -334,30 +331,7 @@ let setup_lock_rules ~dir ~lock_dir : Gen_rules.result =
                   Pkg_workspace.Repository.name repo, repo)
                 |> Pkg_workspace.Repository.Name.Map.of_list_exn
               in
-              let module Repository = Pkg_workspace.Repository in
-              repositories
-              |> Fiber.parallel_map ~f:(fun (loc, name) ->
-                match Repository.Name.Map.find available_repos name with
-                | None ->
-                  User_error.raise
-                    ~loc
-                    [ Pp.textf "Repository '%s' is not a known repository"
-                      @@ Repository.Name.to_string name
-                    ]
-                | Some repo ->
-                  let loc, opam_url = Repository.opam_url repo in
-                  (match OpamUrl.classify opam_url loc with
-                   | `Git -> Opam_repo.of_git_repo loc opam_url
-                   | `Path path ->
-                     Fiber.return @@ Opam_repo.of_opam_repo_dir_path loc path
-                   | `Archive ->
-                     User_error.raise
-                       ~loc
-                       [ Pp.textf
-                           "Repositories stored in archives (%s) are currently \
-                            unsupported"
-                         @@ OpamUrl.to_string opam_url
-                       ]))
+              Opam_repo.resolve_repositories ~available_repos ~repositories
               |> Memo.of_non_reproducible_fiber))
        and+ pins =
          (* CR-soon Alizter: This pin logic (extracting workspace pins,