Apply mirrors to root dependencies

We weren't applying mirros to root dependencies so far.

rdar://110869499

Author: neonichu

Sources/PackageGraph/PackageGraphRoot.swift

@@ -47,14 +47,39 @@ public struct PackageGraphRoot {
     }
 
     /// The top level dependencies.
-    public let dependencies: [PackageDependency]
+    private let _dependencies: [PackageDependency]
+
+    public var dependencies: [PackageDependency] {
+        guard let mirrors = mirrors else {
+            return self._dependencies
+        }
+
+        return self._dependencies.map {
+            switch $0 {
+            case .fileSystem: return $0
+            case .registry: return $0
+            case .sourceControl(let settings):
+                let mappedLocation: PackageDependency.SourceControl.Location
+                switch settings.location {
+                case .local:
+                    mappedLocation = settings.location
+                case .remote(let url):
+                    let effectiveLocation = mirrors.effective(for: url.absoluteString)
+                    mappedLocation = .remote(SourceControlURL(effectiveLocation))
+                }
+                return .sourceControl(identity: settings.identity, nameForTargetDependencyResolutionOnly: settings.nameForTargetDependencyResolutionOnly, location: mappedLocation, requirement: settings.requirement, productFilter: settings.productFilter)
+            }
+        }
+    }
+
+    private let mirrors: DependencyMirrors?
 
     /// Create a package graph root.
     /// Note this quietly skip inputs for which manifests are not found. this could be because the manifest  failed to load or for some other reasons
     // FIXME: This API behavior wrt to non-found manifests is fragile, but required by IDEs
     // it may lead to incorrect assumption in downstream code which may expect an error if a manifest was not found
     // we should refactor this API to more clearly return errors for inputs that do not have a corresponding manifest
-    public init(input: PackageGraphRootInput, manifests: [AbsolutePath: Manifest], explicitProduct: String? = nil) {
+    public init(input: PackageGraphRootInput, manifests: [AbsolutePath: Manifest], explicitProduct: String? = nil, mirrors: DependencyMirrors? = nil) {
         self.packages = input.packages.reduce(into: .init(), { partial, inputPath in
             if let manifest = manifests[inputPath]  {
                 let packagePath = manifest.path.parentDirectory
@@ -77,7 +102,8 @@ public struct PackageGraphRoot {
             }
         }
 
-        self.dependencies = adjustedDependencies
+        self._dependencies = adjustedDependencies
+        self.mirrors = mirrors
     }
 
     /// Returns the constraints imposed by root manifests + dependencies.

Sources/Workspace/Workspace.swift

@@ -1051,7 +1051,7 @@ extension Workspace {
         let resolvedFileOriginHash = try self.computeResolvedFileOriginHash(root: root)
 
         // Load the current manifests.
-        let graphRoot = PackageGraphRoot(input: root, manifests: rootManifests)
+        let graphRoot = PackageGraphRoot(input: root, manifests: rootManifests, mirrors: self.mirrors)
         let currentManifests = try self.loadDependencyManifests(root: graphRoot, observabilityScope: observabilityScope)
 
         // Abort if we're unable to load the pinsStore or have any diagnostics.
@@ -2582,7 +2582,7 @@ extension Workspace {
 
         // FIXME: this should not block
         let rootManifests = try temp_await { self.loadRootManifests(packages: root.packages, observabilityScope: observabilityScope, completion: $0) }
-        let graphRoot = PackageGraphRoot(input: root, manifests: rootManifests, explicitProduct: explicitProduct)
+        let graphRoot = PackageGraphRoot(input: root, manifests: rootManifests, explicitProduct: explicitProduct, mirrors: self.mirrors)
 
         // Load the pins store or abort now.
         guard let pinsStore = observabilityScope.trap({ try self.pinsStore.load() }), !observabilityScope.errorsReported else {
@@ -2697,7 +2697,7 @@ extension Workspace {
         let resolvedFileOriginHash = try self.computeResolvedFileOriginHash(root: root)
 
         // Load the current manifests.
-        let graphRoot = PackageGraphRoot(input: root, manifests: rootManifests, explicitProduct: explicitProduct)
+        let graphRoot = PackageGraphRoot(input: root, manifests: rootManifests, explicitProduct: explicitProduct, mirrors: self.mirrors)
         let currentManifests = try self.loadDependencyManifests(root: graphRoot, observabilityScope: observabilityScope)
         guard !observabilityScope.errorsReported else {
             return currentManifests