Current.buildTrigger*, Current.random → @dependency

Author: finestructure

Sources/App/Commands/TriggerBuilds.swift

@@ -125,12 +125,14 @@ extension TriggerBuildsCommand {
 func triggerBuilds(on database: Database,
                    client: Client,
                    mode: TriggerBuildsCommand.Mode) async throws {
+    @Dependency(\.environment) var environment
     let start = DispatchTime.now().uptimeNanoseconds
+
     switch mode {
         case .limit(let limit):
             Current.logger().info("Triggering builds (limit: \(limit)) ...")
 
-            let withLatestSwiftVersion = Current.buildTriggerCandidatesWithLatestSwiftVersion
+            let withLatestSwiftVersion = environment.buildTriggerCandidatesWithLatestSwiftVersion
             let candidates = try await fetchBuildCandidates(database,
                                                             withLatestSwiftVersion: withLatestSwiftVersion)
             AppMetrics.buildCandidatesCount?.set(candidates.count)
@@ -177,6 +179,7 @@ func triggerBuilds(on database: Database,
                    packages: [Package.Id],
                    force: Bool = false) async throws {
     @Dependency(\.environment) var environment
+
     guard environment.allowBuildTriggers() else {
         Current.logger().info("Build trigger override switch OFF - no builds are being triggered")
         return
@@ -205,9 +208,8 @@ func triggerBuilds(on database: Database,
 
     await withThrowingTaskGroup(of: Void.self) { group in
         for pkgId in packages {
-            let allowListed = Current.buildTriggerAllowList().contains(pkgId)
-            let downscalingAccepted = Current.random(0...1) < Current.buildTriggerDownscaling()
-            guard allowListed || downscalingAccepted else {
+            let allowListed = environment.buildTriggerAllowList().contains(pkgId)
+            guard allowListed || environment.buildTriggerDownscalingAccepted else {
                 Current.logger().info("Build trigger downscaling in effect - skipping builds")
                 continue
             }
@@ -328,7 +330,9 @@ func fetchBuildCandidates(_ database: Database,
 
     let expectedBuildCount = BuildPair.all.count
     let expectedBuildCountWithoutLatestSwiftVersion = BuildPair.allExceptLatestSwiftVersion.count
-    let priorityIDs = Current.buildTriggerAllowList()
+
+    @Dependency(\.environment) var environment
+    let priorityIDs = environment.buildTriggerAllowList()
 
     let query: SQLQueryString = withLatestSwiftVersion
     ? """

Sources/App/Core/AppEnvironment.swift

@@ -25,9 +25,6 @@ import FoundationNetworking
 struct AppEnvironment: Sendable {
     var apiSigningKey: @Sendable () -> String?
     var appVersion: @Sendable () -> String?
-    var buildTriggerAllowList: @Sendable () -> [Package.Id]
-    var buildTriggerDownscaling: @Sendable () -> Double
-    var buildTriggerLatestSwiftVersionDownscaling: @Sendable () -> Double
     var collectionSigningCertificateChain: @Sendable () -> [URL]
     var collectionSigningPrivateKey: @Sendable () -> Data?
     var currentReferenceCache: @Sendable () -> CurrentReferenceCache?
@@ -54,7 +51,6 @@ struct AppEnvironment: Sendable {
     var plausibleBackendReportingSiteID: @Sendable () -> String?
     var postPlausibleEvent: @Sendable (Client, Plausible.Event.Kind, Plausible.Path, User?) async throws -> Void
     var processingBuildBacklog: @Sendable () -> Bool
-    var random: @Sendable (_ range: ClosedRange<Double>) -> Double
     var runnerIds: @Sendable () -> [String]
     var setHTTPClient: @Sendable (Client) -> Void
     var setLogger: @Sendable (Logger) -> Void
@@ -78,11 +74,6 @@ struct AppEnvironment: Sendable {
 
 
 extension AppEnvironment {
-    var buildTriggerCandidatesWithLatestSwiftVersion: Bool {
-        guard buildTriggerLatestSwiftVersionDownscaling() < 1 else { return true }
-        return random(0...1) < Current.buildTriggerLatestSwiftVersionDownscaling()
-    }
-
     func postPlausibleEvent(_ event: Plausible.Event.Kind, path: Plausible.Path, user: User?) {
         Task {
             do {
@@ -102,22 +93,6 @@ extension AppEnvironment {
     static let live = AppEnvironment(
         apiSigningKey: { Environment.get("API_SIGNING_KEY") },
         appVersion: { App.appVersion },
-        buildTriggerAllowList: {
-            Environment.get("BUILD_TRIGGER_ALLOW_LIST")
-                .map { Data($0.utf8) }
-                .flatMap { try? JSONDecoder().decode([Package.Id].self, from: $0) }
-            ?? []
-        },
-        buildTriggerDownscaling: {
-            Environment.get("BUILD_TRIGGER_DOWNSCALING")
-                .flatMap(Double.init)
-                ?? 1.0
-        },
-        buildTriggerLatestSwiftVersionDownscaling: {
-            Environment.get("BUILD_TRIGGER_LATEST_SWIFT_VERSION_DOWNSCALING")
-                .flatMap(Double.init)
-                ?? 1.0
-        },
         collectionSigningCertificateChain: {
             [
                 SignedCollection.certsDir
@@ -172,7 +147,6 @@ extension AppEnvironment {
         processingBuildBacklog: {
             Environment.get("PROCESSING_BUILD_BACKLOG").flatMap(\.asBool) ?? false
         },
-        random: { range in Double.random(in: range) },
         runnerIds: {
             Environment.get("RUNNER_IDS")
                 .map { Data($0.utf8) }

Sources/App/Core/Dependencies/EnvironmentClient.swift

@@ -29,6 +29,9 @@ struct EnvironmentClient {
     var awsSecretAccessKey: @Sendable () -> String?
     var builderToken: @Sendable () -> String?
     var buildTimeout: @Sendable () -> Int = { XCTFail(#function); return 10 }
+    var buildTriggerAllowList: @Sendable () -> [Package.Id] = { XCTFail(#function); return [] }
+    var buildTriggerDownscaling: @Sendable () -> Double = { XCTFail(#function); return 1 }
+    var buildTriggerLatestSwiftVersionDownscaling: @Sendable () -> Double = { XCTFail(#function); return 1 }
     // We're not defaulting current to XCTFail, because its use is too pervasive and would require the vast
     // majority of tests to be wrapped with `withDependencies`.
     // We can do so at a later time once more tests are transitioned over for other dependencies. This is
@@ -37,6 +40,7 @@ struct EnvironmentClient {
     var current: @Sendable () -> Environment = { .development }
     var mastodonCredentials: @Sendable () -> Mastodon.Credentials?
     var mastodonPost: @Sendable (_ client: Client, _ post: String) async throws -> Void
+    var random: @Sendable (_ range: ClosedRange<Double>) -> Double = { XCTFail(#function); return Double.random(in: $0) }
 }
 
 
@@ -57,17 +61,46 @@ extension EnvironmentClient: DependencyKey {
             awsSecretAccessKey: { Environment.get("AWS_SECRET_ACCESS_KEY") },
             builderToken: { Environment.get("BUILDER_TOKEN") },
             buildTimeout: { Environment.get("BUILD_TIMEOUT").flatMap(Int.init) ?? 10 },
+            buildTriggerAllowList: {
+                Environment.get("BUILD_TRIGGER_ALLOW_LIST")
+                    .map { Data($0.utf8) }
+                    .flatMap { try? JSONDecoder().decode([Package.Id].self, from: $0) }
+                ?? []
+            },
+            buildTriggerDownscaling: {
+                Environment.get("BUILD_TRIGGER_DOWNSCALING")
+                    .flatMap(Double.init)
+                    ?? 1.0
+            },
+            buildTriggerLatestSwiftVersionDownscaling: {
+                Environment.get("BUILD_TRIGGER_LATEST_SWIFT_VERSION_DOWNSCALING")
+                    .flatMap(Double.init)
+                    ?? 1.0
+            },
             current: { (try? Environment.detect()) ?? .development },
             mastodonCredentials: {
                 Environment.get("MASTODON_ACCESS_TOKEN")
                     .map(Mastodon.Credentials.init(accessToken:))
             },
-            mastodonPost: { client, message in try await Mastodon.post(client: client, message: message) }
+            mastodonPost: { client, message in try await Mastodon.post(client: client, message: message) },
+            random: { range in Double.random(in: range) }
         )
     }
 }
 
 
+extension EnvironmentClient {
+    var buildTriggerCandidatesWithLatestSwiftVersion: Bool {
+        guard buildTriggerLatestSwiftVersionDownscaling() < 1 else { return true }
+        return random(0...1) < buildTriggerLatestSwiftVersionDownscaling()
+    }
+
+    var buildTriggerDownscalingAccepted: Bool {
+        random(0...1) < buildTriggerDownscaling()
+    }
+}
+
+
 extension EnvironmentClient: TestDependencyKey {
     static var testValue: Self { Self() }
 }