core: save loaded infras to s3 when available

Signed-off-by: Eloi Charpentier <eloi.charpentier.42@gmail.com>

Author: eckter

core/src/main/java/fr/sncf/osrd/api/InfraManager.kt

@@ -4,6 +4,7 @@ import fr.sncf.osrd.parseRJSInfra
 import fr.sncf.osrd.railjson.schema.infra.RJSInfra
 import fr.sncf.osrd.reporting.exceptions.ErrorType
 import fr.sncf.osrd.reporting.exceptions.OSRDError
+import fr.sncf.osrd.utils.compressToZip
 import fr.sncf.osrd.utils.jacoco.ExcludeFromGeneratedCodeCoverage
 import io.opentelemetry.api.trace.SpanKind
 import io.opentelemetry.instrumentation.annotations.WithSpan
@@ -14,8 +15,12 @@ import okhttp3.OkHttpClient
 import org.slf4j.Logger
 import org.slf4j.LoggerFactory
 
-class InfraManager(baseUrl: String, authorizationToken: String?, httpClient: OkHttpClient) :
-    APIClient(baseUrl, authorizationToken, httpClient), InfraProvider {
+class InfraManager(
+    baseUrl: String,
+    authorizationToken: String?,
+    httpClient: OkHttpClient,
+    val s3Context: S3Context? = null,
+) : APIClient(baseUrl, authorizationToken, httpClient), InfraProvider {
     private val infraCache = ConcurrentHashMap<String, InfraCacheEntry>()
     private val signalingSimulator = makeSignalingSimulator()
 
@@ -118,6 +123,16 @@ class InfraManager(baseUrl: String, authorizationToken: String?, httpClient: OkH
                 cacheEntry.version = version
                 checkNotNull(response.body) { "missing body in railjson response" }
                 rjsInfra = RJSInfra.adapter.fromJson(response.body.source())!!
+
+                // Save railjson to s3 if available, for better reproducibility.
+                // This is done on a different thread while the infra is parsed (on a single
+                // thread), it should not take any extra time.
+                s3Context?.writeFileIfMissing("stdcm/infras/$infraId-$version.railjson.zip") {
+                    RJSInfra.adapter
+                        .toJson(rjsInfra)
+                        .encodeToByteArray()
+                        .compressToZip("$infraId-$version.railjson")
+                }
             }
 
             // Parse railjson into a proper infra

core/src/main/java/fr/sncf/osrd/cli/WorkerCommand.kt

@@ -122,7 +122,7 @@ class WorkerCommand : CliCommand {
 
         val infraId = WORKER_KEY.split("-").first()
         val timetableId = WORKER_KEY.split("-").getOrNull(1)?.toInt()
-        val infraManager = InfraManager(editoastUrl!!, editoastAuthorization, httpClient)
+        val infraManager = InfraManager(editoastUrl!!, editoastAuthorization, httpClient, s3Context)
         val timetableCache =
             TimetableCacheManager(
                 TimetableDownloader(