feat: add --ecs-config option for AWS Batch Forge CE (#558)

georgi-seqera · web-flow · commit c7acffcff77c · 2025-12-01T14:01:08.000+02:00
diff --git a/src/main/java/io/seqera/tower/cli/commands/computeenvs/platforms/AwsBatchForgePlatform.java b/src/main/java/io/seqera/tower/cli/commands/computeenvs/platforms/AwsBatchForgePlatform.java
@@ -19,6 +19,7 @@
 
 import io.seqera.tower.ApiException;
 import io.seqera.tower.cli.exceptions.TowerException;
+import io.seqera.tower.cli.utils.FilesHelper;
 import io.seqera.tower.model.AwsBatchConfig;
 import io.seqera.tower.model.ComputeEnvComputeConfig.PlatformEnum;
 import io.seqera.tower.model.ForgeConfig;
@@ -28,6 +29,7 @@
 import picocli.CommandLine.Option;
 
 import java.io.IOException;
+import java.nio.file.Path;
 import java.util.List;
 
 public class AwsBatchForgePlatform extends AbstractPlatform<AwsBatchConfig> {
@@ -74,6 +76,9 @@ public class AwsBatchForgePlatform extends AbstractPlatform<AwsBatchConfig> {
     @Option(names = "--preserve-resources", description = "Enable this if you want to preserve the Batch compute resources created by Tower independently from the lifecycle of this compute environment.")
     public boolean preserveResources;
 
+    @Option(names = {"--ecs-config"}, description = "Path to ECS agent configuration file - custom configuration for the ECS agent parameters used by AWS Batch. This is appended to the /etc/ecs/ecs.config file in each cluster node.")
+    public Path ecsConfig;
+
     @ArgGroup(heading = "%nEFS filesystem options:%n", validate = false)
     public EfsFileSystem efs;
     @ArgGroup(heading = "%nFSX filesytem options:%n", validate = false)
@@ -130,7 +135,7 @@ private Boolean isFusionV2Enabled() throws TowerException {
         return fusionV2;
     }
 
-    private ForgeConfig buildForge() throws TowerException {
+    private ForgeConfig buildForge() throws TowerException, IOException {
 
         // TODO: delete this once fusion v1 is completely removed
         if (fusion) {
@@ -155,7 +160,8 @@ private ForgeConfig buildForge() throws TowerException {
                 .ebsBlockSize(adv().ebsBlockSize)
                 .ebsBootSize(adv().bootDiskSizeGb)
                 .bidPercentage(adv().bidPercentage)
-                .fargateHeadEnabled(fargate);
+                .fargateHeadEnabled(fargate)
+                .ecsConfig(FilesHelper.readString(ecsConfig));
 
 
         if (efs != null) {
