From 0554394870c207ede907a13fb64db44447d7089f Mon Sep 17 00:00:00 2001
From: "georgi.hristov" <georgi.hristov@seqera.io>
Date: Mon, 22 Jun 2026 20:50:33 +0100
Subject: [PATCH] feat(studios): extend studio commands with support for
 configuration from git repository

---
 .../commands/studios/AbstractStudiosCmd.java  |   4 +
 .../tower/cli/commands/studios/AddCmd.java    |  36 +-
 .../tower/cli/commands/studios/StartCmd.java  |  12 +
 .../studios/StudioConfigurationOptions.java   |   5 +
 .../studios/StudioRemoteConfigOptions.java    |  45 +++
 .../studios/StudioTemplateOptions.java        |   5 +-
 .../tower/cli/commands/studios/UpdateCmd.java |  58 +++-
 .../cli/responses/studios/StudiosView.java    |  18 +
 .../tower/cli/studios/StudiosCmdTest.java     | 327 ++++++++++++++++++
 9 files changed, 503 insertions(+), 7 deletions(-)
 create mode 100644 src/main/java/io/seqera/tower/cli/commands/studios/StudioRemoteConfigOptions.java

diff --git a/src/main/java/io/seqera/tower/cli/commands/studios/AbstractStudiosCmd.java b/src/main/java/io/seqera/tower/cli/commands/studios/AbstractStudiosCmd.java
index 73cfe3626..37e42fcc9 100644
--- a/src/main/java/io/seqera/tower/cli/commands/studios/AbstractStudiosCmd.java
+++ b/src/main/java/io/seqera/tower/cli/commands/studios/AbstractStudiosCmd.java
@@ -146,6 +146,10 @@ protected DataStudioConfiguration studioConfigurationFrom(Long wspId, DataStudio
 
         studioConfiguration.setMountData(getMountDataIds(configOptions, studioConfiguration, wspId));
 
+        if (configOptions.environment != null) {
+            studioConfiguration.setEnvironment(configOptions.environment);
+        }
+
 
         if (condaEnvOverride != null && !condaEnvOverride.isEmpty()) {
             studioConfiguration.setCondaEnvironment(condaEnvOverride);
diff --git a/src/main/java/io/seqera/tower/cli/commands/studios/AddCmd.java b/src/main/java/io/seqera/tower/cli/commands/studios/AddCmd.java
index a7c358607..80cbc68ed 100644
--- a/src/main/java/io/seqera/tower/cli/commands/studios/AddCmd.java
+++ b/src/main/java/io/seqera/tower/cli/commands/studios/AddCmd.java
@@ -67,12 +67,21 @@ public class AddCmd extends AbstractStudiosCmd{
     @CommandLine.Mixin
     public StudioConfigurationOptions studioConfigOptions;
 
+    @CommandLine.Mixin
+    public StudioRemoteConfigOptions remoteConfigOptions;
+
     @CommandLine.Option(names = {"-a", "--auto-start"}, description = "Create studio and start it immediately (default: false)", defaultValue = "false")
     public Boolean autoStart;
 
     @CommandLine.Option(names = {"--private"}, description = "Create a private studio that only you can access or manage (default: false)", defaultValue = "false")
     public Boolean isPrivate;
 
+    @CommandLine.Option(names = {"--spot"}, description = "Launch the studio on spot instances (default: provider/compute environment default).")
+    public Boolean spot;
+
+    @CommandLine.Option(names = {"--ssh"}, description = "Enable SSH connectivity to the studio (default: false).")
+    public Boolean ssh;
+
     @CommandLine.Option(names = {"--labels"}, description = "Comma-separated list of labels", split = ",", converter = Label.StudioResourceLabelsConverter.class)
     public List<Label> labels;
 
@@ -83,7 +92,7 @@ public class AddCmd extends AbstractStudiosCmd{
     protected Response exec() throws ApiException {
         Long wspId = workspaceId(workspace.workspace);
 
-        templateValidation(templateOptions, condaEnv, wspId);
+        templateValidation(templateOptions, remoteConfigOptions, condaEnv, wspId);
         DataStudioCreateRequest request = prepareRequest(wspId);
         DataStudioCreateResponse response = studiosApi().createDataStudio(request, wspId, autoStart);
         DataStudioDto studioDto = response.getStudio();
@@ -91,9 +100,16 @@ protected Response exec() throws ApiException {
         return new StudiosCreated(studioDto.getSessionId(), wspId, workspaceRef(wspId), baseWorkspaceUrl(wspId), autoStart);
     }
 
-    private void templateValidation(StudioTemplateOptions templateOptions, Path condaEnv, Long wspId) throws ApiException {
-        if (templateOptions.template.standardTemplate != null) {
-            checkIfTemplateIsAvailable(templateOptions.template.standardTemplate, wspId);
+    private void templateValidation(StudioTemplateOptions templateOptions, StudioRemoteConfigOptions remoteConfigOptions, Path condaEnv, Long wspId) throws ApiException {
+        String standardTemplate = templateOptions.template == null ? null : templateOptions.template.standardTemplate;
+
+        // A template is mandatory unless a Git repository is provided, since the remote repository may define it.
+        if (templateOptions.getTemplate() == null && remoteConfigOptions.isEmpty()) {
+            throw new TowerException("A studio template is required: provide -t/--template or -ct/--custom-template, or a Git repository via -u/--url");
+        }
+
+        if (standardTemplate != null) {
+            checkIfTemplateIsAvailable(standardTemplate, wspId);
         } else if (condaEnv != null) {
             throw new StudiosCustomTemplateWithCondaException();
         }
@@ -115,6 +131,15 @@ DataStudioCreateRequest prepareRequest(Long wspId) throws ApiException {
         if (description != null && !description.isEmpty()) {request.description(description);}
         request.setLabelIds(getLabelIds(labels, wspId));
         request.setIsPrivate(isPrivate);
+        if (spot != null) {
+            request.setSpot(spot);
+        }
+        if (remoteConfigOptions.revision != null && remoteConfigOptions.isEmpty()) {
+            throw new TowerException("--revision requires --url to be set");
+        }
+        if (!remoteConfigOptions.isEmpty()) {
+            request.setRemoteConfig(remoteConfigOptions.toRemoteConfiguration());
+        }
         request.setDataStudioToolUrl(templateOptions.getTemplate());
         ComputeEnvResponseDto ceResponse = computeEnvByRef(wspId, computeEnv);
         request.setComputeEnvId(ceResponse.getId());
@@ -130,6 +155,9 @@ DataStudioCreateRequest prepareRequest(Long wspId) throws ApiException {
 
 
         DataStudioConfiguration newConfig = studioConfigurationFrom(workspaceId(workspace.workspace), studioConfigOptions, condaEnvString);
+        if (ssh != null) {
+            newConfig.setSshEnabled(ssh);
+        }
         request.setConfiguration(setDefaults(newConfig));
         return request;
     }
diff --git a/src/main/java/io/seqera/tower/cli/commands/studios/StartCmd.java b/src/main/java/io/seqera/tower/cli/commands/studios/StartCmd.java
index 581ac0109..dcc2bb3ee 100644
--- a/src/main/java/io/seqera/tower/cli/commands/studios/StartCmd.java
+++ b/src/main/java/io/seqera/tower/cli/commands/studios/StartCmd.java
@@ -58,6 +58,12 @@ public class StartCmd extends AbstractStudiosCmd {
     @CommandLine.Option(names = {"--description"}, description = "Optional configuration override for 'description'.")
     public String description;
 
+    @CommandLine.Option(names = {"--spot"}, description = "Optional override to launch the studio on spot instances.")
+    public Boolean spot;
+
+    @CommandLine.Option(names = {"--ssh"}, description = "Optional override to enable SSH connectivity to the studio.")
+    public Boolean ssh;
+
     @Override
     protected Response exec() throws ApiException {
         Long wspId = workspaceId(workspace.workspace);
@@ -97,6 +103,9 @@ protected Integer onBeforeExit(int exitCode, Response response) {
 
     private DataStudioStartRequest getStartRequestWithOverridesApplied(DataStudioDto studioDto) throws ApiException {
         DataStudioConfiguration newConfig = studioConfigurationFrom(studioDto.getWorkspaceId(), studioDto, studioConfigOptions);
+        if (ssh != null) {
+            newConfig.setSshEnabled(ssh);
+        }
         String appliedDescription = description == null
                 ? studioDto.getDescription()
                 : description;
@@ -106,6 +115,9 @@ private DataStudioStartRequest getStartRequestWithOverridesApplied(DataStudioDto
         request.setConfiguration(newConfig);
         request.setDescription(appliedDescription);
         request.setLabelIds(getLabelIds(labels, studioDto.getWorkspaceId()));
+        if (spot != null) {
+            request.setSpot(spot);
+        }
 
         return request;
     }
diff --git a/src/main/java/io/seqera/tower/cli/commands/studios/StudioConfigurationOptions.java b/src/main/java/io/seqera/tower/cli/commands/studios/StudioConfigurationOptions.java
index 73e2d4acb..7723159de 100644
--- a/src/main/java/io/seqera/tower/cli/commands/studios/StudioConfigurationOptions.java
+++ b/src/main/java/io/seqera/tower/cli/commands/studios/StudioConfigurationOptions.java
@@ -16,6 +16,8 @@
 
 package io.seqera.tower.cli.commands.studios;
 
+import java.util.Map;
+
 import picocli.CommandLine;
 
 public class StudioConfigurationOptions {
@@ -32,6 +34,9 @@ public class StudioConfigurationOptions {
     @CommandLine.Option(names = {"--lifespan"}, description = "Optional configuration override for 'lifespan' setting (integer representing hours). Defaults to workspace lifespan setting.")
     public Integer lifespan;
 
+    @CommandLine.Option(names = {"-e", "--env"}, description = "Add environment variables to the studio as key=value pairs. Can be specified multiple times (e.g. -e KEY1=value1 -e KEY2=value2).")
+    public Map<String, String> environment;
+
     @CommandLine.Mixin
     public DataLinkRefOptions dataLinkRefOptions;
 
diff --git a/src/main/java/io/seqera/tower/cli/commands/studios/StudioRemoteConfigOptions.java b/src/main/java/io/seqera/tower/cli/commands/studios/StudioRemoteConfigOptions.java
new file mode 100644
index 000000000..37a2d2922
--- /dev/null
+++ b/src/main/java/io/seqera/tower/cli/commands/studios/StudioRemoteConfigOptions.java
@@ -0,0 +1,45 @@
+/*
+ * Copyright 2021-2026, Seqera.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package io.seqera.tower.cli.commands.studios;
+
+import io.seqera.tower.model.StudioRemoteConfiguration;
+import picocli.CommandLine;
+
+public class StudioRemoteConfigOptions {
+
+    @CommandLine.Option(names = {"-u", "--url"}, description = "Git repository URL to import studio configuration from.")
+    public String url;
+
+    @CommandLine.Option(names = {"--revision"}, description = "Optional branch, tag or commit of the Git repository to check out. Requires --url.")
+    public String revision;
+
+    public boolean isEmpty() {
+        return url == null || url.isEmpty();
+    }
+
+    public StudioRemoteConfiguration toRemoteConfiguration() {
+        if (isEmpty()) {
+            return null;
+        }
+        StudioRemoteConfiguration remoteConfig = new StudioRemoteConfiguration();
+        remoteConfig.setRepository(url);
+        if (revision != null && !revision.isEmpty()) {
+            remoteConfig.setRevision(revision);
+        }
+        return remoteConfig;
+    }
+}
diff --git a/src/main/java/io/seqera/tower/cli/commands/studios/StudioTemplateOptions.java b/src/main/java/io/seqera/tower/cli/commands/studios/StudioTemplateOptions.java
index ddbd12abd..6c0c1a0ee 100644
--- a/src/main/java/io/seqera/tower/cli/commands/studios/StudioTemplateOptions.java
+++ b/src/main/java/io/seqera/tower/cli/commands/studios/StudioTemplateOptions.java
@@ -20,7 +20,7 @@
 
 public class StudioTemplateOptions {
 
-    @CommandLine.ArgGroup(multiplicity = "1")
+    @CommandLine.ArgGroup(multiplicity = "0..1")
     public StudioTemplate template;
 
     public static class StudioTemplate {
@@ -32,6 +32,9 @@ public static class StudioTemplate {
     }
 
     public String getTemplate() {
+        if (template == null) {
+            return null;
+        }
         return template.standardTemplate != null ? template.standardTemplate : template.customTemplate;
     }
 }
diff --git a/src/main/java/io/seqera/tower/cli/commands/studios/UpdateCmd.java b/src/main/java/io/seqera/tower/cli/commands/studios/UpdateCmd.java
index a3f4bfb77..9154b682f 100644
--- a/src/main/java/io/seqera/tower/cli/commands/studios/UpdateCmd.java
+++ b/src/main/java/io/seqera/tower/cli/commands/studios/UpdateCmd.java
@@ -25,9 +25,11 @@
 import io.seqera.tower.cli.exceptions.TowerException;
 import io.seqera.tower.cli.responses.Response;
 import io.seqera.tower.cli.responses.studios.StudioUpdated;
+import io.seqera.tower.model.ComputeEnvResponseDto;
 import io.seqera.tower.model.DataStudioConfiguration;
 import io.seqera.tower.model.DataStudioDto;
 import io.seqera.tower.model.DataStudioUpdateRequest;
+import io.seqera.tower.model.ListComputeEnvsResponseEntry;
 import picocli.CommandLine;
 
 @CommandLine.Command(
@@ -54,19 +56,39 @@ public class UpdateCmd extends AbstractStudiosCmd {
     @CommandLine.Option(names = {"--new-name"}, description = "Optional new name for the studio.")
     public String newName;
 
+    @CommandLine.Option(names = {"--ssh"}, description = "Optional override to enable or disable SSH connectivity to the studio.")
+    public Boolean ssh;
+
+    @CommandLine.Option(names = {"-c", "--compute-env"}, description = "Move the studio to a different (compatible) compute environment. Only allowed while the studio is stopped.")
+    public String computeEnv;
+
     @Override
     protected Response exec() throws ApiException {
         Long wspId = workspaceId(workspace.workspace);
 
+        DataStudioDto studioDto;
         try {
-            DataStudioDto studioDto = fetchStudio(studioRefOptions, wspId);
+            studioDto = fetchStudio(studioRefOptions, wspId);
+        } catch (ApiException e) {
+            if (e.getCode() == 404) {
+                throw new StudioNotFoundException(studioRefOptions.getStudioIdentifier(), workspace.workspace);
+            }
+            if (e.getCode() == 403) {
+                throw new TowerException(String.format("User not entitled to view studio '%s' at %s workspace", studioRefOptions.getStudioIdentifier(), workspace.workspace));
+            }
+            throw e;
+        }
 
-            DataStudioUpdateRequest request = getUpdateRequestWithOverridesApplied(studioDto);
+        DataStudioUpdateRequest request = getUpdateRequestWithOverridesApplied(studioDto);
 
+        try {
             DataStudioDto updatedStudio = studiosApi().updateDataStudio(studioDto.getSessionId(), request, wspId);
 
             return new StudioUpdated(updatedStudio.getSessionId(), studioRefOptions.getStudioIdentifier(), wspId, workspaceRef(wspId));
         } catch (ApiException e) {
+            if (computeEnv != null && e.getCode() == 400 && isIncompatibleComputeEnvError(e)) {
+                throw incompatibleComputeEnvException(studioDto.getSessionId(), wspId);
+            }
             if (e.getCode() == 404) {
                 throw new StudioNotFoundException(studioRefOptions.getStudioIdentifier(), workspace.workspace);
             }
@@ -77,8 +99,35 @@ protected Response exec() throws ApiException {
         }
     }
 
+    private boolean isIncompatibleComputeEnvError(ApiException e) {
+        String body = e.getResponseBody();
+        return body != null && body.contains("is not compatible with the Studio's current compute environment");
+    }
+
+    private TowerException incompatibleComputeEnvException(String sessionId, Long wspId) {
+        StringBuilder message = new StringBuilder(String.format(
+                "Compute environment '%s' is not compatible with studio '%s'.", computeEnv, studioRefOptions.getStudioIdentifier()));
+        try {
+            List<ListComputeEnvsResponseEntry> compatible = studiosApi()
+                    .listDataStudioCompatibleComputeEnvs(sessionId, wspId, null)
+                    .getComputeEnvs();
+            if (compatible == null || compatible.isEmpty()) {
+                message.append(" No compatible compute environments are available in this workspace.");
+            } else {
+                message.append(" Choose one of the following compatible compute environments:");
+                compatible.forEach(ce -> message.append(String.format("%n  - %s (%s)", ce.getName(), ce.getId())));
+            }
+        } catch (ApiException ex) {
+            message.append(" (unable to retrieve the list of compatible compute environments)");
+        }
+        return new TowerException(message.toString());
+    }
+
     private DataStudioUpdateRequest getUpdateRequestWithOverridesApplied(DataStudioDto studioDto) throws ApiException {
         DataStudioConfiguration newConfig = studioConfigurationFrom(studioDto.getWorkspaceId(), studioDto, studioConfigOptions);
+        if (ssh != null) {
+            newConfig.setSshEnabled(ssh);
+        }
         String appliedDescription = description == null
                 ? studioDto.getDescription()
                 : description;
@@ -93,6 +142,11 @@ private DataStudioUpdateRequest getUpdateRequestWithOverridesApplied(DataStudioD
             request.setName(newName);
         }
 
+        if (computeEnv != null) {
+            ComputeEnvResponseDto ceResponse = computeEnvByRef(studioDto.getWorkspaceId(), computeEnv);
+            request.setComputeEnvId(ceResponse.getId());
+        }
+
         return request;
     }
 
diff --git a/src/main/java/io/seqera/tower/cli/responses/studios/StudiosView.java b/src/main/java/io/seqera/tower/cli/responses/studios/StudiosView.java
index 9bbc53e06..9f62bd9aa 100644
--- a/src/main/java/io/seqera/tower/cli/responses/studios/StudiosView.java
+++ b/src/main/java/io/seqera/tower/cli/responses/studios/StudiosView.java
@@ -22,6 +22,7 @@
 import io.seqera.tower.model.DataStudioConfiguration;
 import io.seqera.tower.model.DataStudioDto;
 import io.seqera.tower.model.DataStudioStatusInfo;
+import io.seqera.tower.model.StudioSshDetailsDto;
 import io.seqera.tower.model.UserInfo;
 
 import java.io.PrintWriter;
@@ -66,6 +67,23 @@ public void toString(PrintWriter out){
         table.addRow("CPU allocated",  config == null ? "-" : String.valueOf(config.getCpu()));
         table.addRow("Memory allocated", config == null ? "-" : String.valueOf(config.getMemory()));
         table.addRow("Build reports", studio.getWaveBuildUrl() == null ? "NA" : studio.getWaveBuildUrl());
+        table.addRow("Private", studio.getIsPrivate() == null ? "NA" : String.valueOf(studio.getIsPrivate()));
+        table.addRow("Lifespan (hours)", studio.getEffectiveLifespanHours() == null ? "Unlimited" : String.valueOf(studio.getEffectiveLifespanHours()));
+        table.addRow("SSH enabled", config == null || config.getSshEnabled() == null ? "false" : String.valueOf(config.getSshEnabled()));
+
+        StudioSshDetailsDto sshDetails = studio.getSshDetails();
+        if (sshDetails != null) {
+            table.addRow("SSH host", sshDetails.getHost());
+            table.addRow("SSH port", sshDetails.getPort() == null ? "NA" : String.valueOf(sshDetails.getPort()));
+            table.addRow("SSH command", sshDetails.getCommand());
+        }
+
+        if (config != null && config.getEnvironment() != null && !config.getEnvironment().isEmpty()) {
+            String envVars = config.getEnvironment().entrySet().stream()
+                    .map(e -> String.format("%s=%s", e.getKey(), e.getValue()))
+                    .collect(java.util.stream.Collectors.joining(", "));
+            table.addRow("Environment variables", envVars);
+        }
 
         table.print();
         if (config != null && config.getCondaEnvironment() != null && !config.getCondaEnvironment().isEmpty()) {
diff --git a/src/test/java/io/seqera/tower/cli/studios/StudiosCmdTest.java b/src/test/java/io/seqera/tower/cli/studios/StudiosCmdTest.java
index 052e3b1a3..97c3a277c 100644
--- a/src/test/java/io/seqera/tower/cli/studios/StudiosCmdTest.java
+++ b/src/test/java/io/seqera/tower/cli/studios/StudiosCmdTest.java
@@ -53,6 +53,7 @@
 
 import static io.seqera.tower.cli.utils.JsonHelper.parseJson;
 import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertTrue;
 import static org.mockserver.matchers.Times.exactly;
 import static org.mockserver.model.HttpRequest.request;
 import static org.mockserver.model.HttpResponse.response;
@@ -2231,4 +2232,330 @@ void testUpdateWithConfigOverride(OutputType format, MockServerClient mock) {
                 "[organization1 / workspace1]"));
     }
 
+    @ParameterizedTest
+    @EnumSource(OutputType.class)
+    void testAddWithSpotAndSsh(OutputType format, MockServerClient mock) {
+        mock.when(
+                request().withMethod("GET").withPath("/user-info"), exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody(loadResource("user")).withContentType(MediaType.APPLICATION_JSON)
+        );
+        mock.when(
+                request().withMethod("GET").withPath("/user/1264/workspaces"), exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody(loadResource("workspaces/workspaces_list")).withContentType(MediaType.APPLICATION_JSON)
+        );
+        mock.when(
+                request().withMethod("GET").withPath("/studios/templates")
+                        .withQueryStringParameter("workspaceId", "75887156211589")
+                        .withQueryStringParameter("max", "20"), exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody(loadResource("studios/studios_templates_response")).withContentType(MediaType.APPLICATION_JSON)
+        );
+        mock.when(
+                request().withMethod("GET").withPath("/compute-envs")
+                        .withQueryStringParameter("status", "AVAILABLE")
+                        .withQueryStringParameter("workspaceId", "75887156211589"), exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody("{\"computeEnvs\":[{\"id\":\"vYOK4vn7spw7bHHWBDXZ2\",\"name\":\"demo\",\"platform\":\"aws-batch\",\"status\":\"AVAILABLE\",\"message\":null,\"lastUsed\":null,\"primary\":true,\"workspaceName\":null,\"visibility\":null}]}").withContentType(MediaType.APPLICATION_JSON)
+        );
+        mock.when(
+                request().withMethod("GET").withPath("/compute-envs/vYOK4vn7spw7bHHWBDXZ2"), exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody(loadResource("compute_env_demo")).withContentType(MediaType.APPLICATION_JSON)
+        );
+        mock.when(
+                request().withMethod("POST").withPath("/studios")
+                        .withQueryStringParameter("workspaceId", "75887156211589")
+                        .withBody(json("""
+                           {
+                             "spot": true,
+                             "configuration": {
+                               "sshEnabled": true
+                             }
+                           }
+                           """))
+                , exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody(loadResource("studios/studios_created_response")).withContentType(MediaType.APPLICATION_JSON)
+        );
+
+        ExecOut out = exec(format, mock, "studios", "add", "-n", "studio-a66d", "-w", "75887156211589", "-t", "cr.seqera.io/public/data-studio-vscode:1.93.1-snapshot", "-c", "demo", "--spot", "--ssh");
+
+        assertOutput(format, out, new StudiosCreated("3e8370e7", 75887156211589L, "[organization1 / workspace1]",
+                "http://localhost:" + mock.getPort() + "/orgs/organization1/workspaces/workspace1", false));
+    }
+
+    @ParameterizedTest
+    @EnumSource(OutputType.class)
+    void testAddWithRepository(OutputType format, MockServerClient mock) {
+        mock.when(
+                request().withMethod("GET").withPath("/user-info"), exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody(loadResource("user")).withContentType(MediaType.APPLICATION_JSON)
+        );
+        mock.when(
+                request().withMethod("GET").withPath("/user/1264/workspaces"), exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody(loadResource("workspaces/workspaces_list")).withContentType(MediaType.APPLICATION_JSON)
+        );
+        mock.when(
+                request().withMethod("GET").withPath("/studios/templates")
+                        .withQueryStringParameter("workspaceId", "75887156211589")
+                        .withQueryStringParameter("max", "20"), exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody(loadResource("studios/studios_templates_response")).withContentType(MediaType.APPLICATION_JSON)
+        );
+        mock.when(
+                request().withMethod("GET").withPath("/compute-envs")
+                        .withQueryStringParameter("status", "AVAILABLE")
+                        .withQueryStringParameter("workspaceId", "75887156211589"), exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody("{\"computeEnvs\":[{\"id\":\"vYOK4vn7spw7bHHWBDXZ2\",\"name\":\"demo\",\"platform\":\"aws-batch\",\"status\":\"AVAILABLE\",\"message\":null,\"lastUsed\":null,\"primary\":true,\"workspaceName\":null,\"visibility\":null}]}").withContentType(MediaType.APPLICATION_JSON)
+        );
+        mock.when(
+                request().withMethod("GET").withPath("/compute-envs/vYOK4vn7spw7bHHWBDXZ2"), exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody(loadResource("compute_env_demo")).withContentType(MediaType.APPLICATION_JSON)
+        );
+        mock.when(
+                request().withMethod("POST").withPath("/studios")
+                        .withQueryStringParameter("workspaceId", "75887156211589")
+                        .withBody(json("""
+                           {
+                             "remoteConfig": {
+                               "repository": "https://github.com/owner/repo",
+                               "revision": "main"
+                             }
+                           }
+                           """))
+                , exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody(loadResource("studios/studios_created_response")).withContentType(MediaType.APPLICATION_JSON)
+        );
+
+        ExecOut out = exec(format, mock, "studios", "add", "-n", "studio-a66d", "-w", "75887156211589", "-t", "cr.seqera.io/public/data-studio-vscode:1.93.1-snapshot", "-c", "demo", "--url", "https://github.com/owner/repo", "--revision", "main");
+
+        assertOutput(format, out, new StudiosCreated("3e8370e7", 75887156211589L, "[organization1 / workspace1]",
+                "http://localhost:" + mock.getPort() + "/orgs/organization1/workspaces/workspace1", false));
+    }
+
+    @ParameterizedTest
+    @EnumSource(OutputType.class)
+    void testUpdateWithComputeEnv(OutputType format, MockServerClient mock) {
+        mock.when(
+                request().withMethod("GET").withPath("/user-info"), exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody(loadResource("user")).withContentType(MediaType.APPLICATION_JSON)
+        );
+        mock.when(
+                request().withMethod("GET").withPath("/user/1264/workspaces"), exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody(loadResource("workspaces/workspaces_list")).withContentType(MediaType.APPLICATION_JSON)
+        );
+        mock.when(
+                request().withMethod("GET").withPath("/studios/3e8370e7").withQueryStringParameter("workspaceId", "75887156211589"), exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody(loadResource("studios/studios_view_response_studio_stopped")).withContentType(MediaType.APPLICATION_JSON)
+        );
+        mock.when(
+                request().withMethod("GET").withPath("/compute-envs")
+                        .withQueryStringParameter("status", "AVAILABLE")
+                        .withQueryStringParameter("workspaceId", "75887156211589"), exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody("{\"computeEnvs\":[{\"id\":\"vYOK4vn7spw7bHHWBDXZ2\",\"name\":\"demo\",\"platform\":\"aws-batch\",\"status\":\"AVAILABLE\",\"message\":null,\"lastUsed\":null,\"primary\":true,\"workspaceName\":null,\"visibility\":null}]}").withContentType(MediaType.APPLICATION_JSON)
+        );
+        mock.when(
+                request().withMethod("GET").withPath("/compute-envs/vYOK4vn7spw7bHHWBDXZ2"), exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody(loadResource("compute_env_demo")).withContentType(MediaType.APPLICATION_JSON)
+        );
+        mock.when(
+                request().withMethod("PUT").withPath("/studios/3e8370e7").withQueryStringParameter("workspaceId", "75887156211589").withBody(json("""
+                           {
+                             "computeEnvId": "vYOK4vn7spw7bHHWBDXZ2"
+                           }
+                           """))
+                , exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody(loadResource("studios/studios_update_response")).withContentType(MediaType.APPLICATION_JSON)
+        );
+
+        ExecOut out = exec(format, mock, "studios", "update", "-w", "75887156211589", "-i", "3e8370e7", "--compute-env", "demo");
+
+        assertOutput(format, out, new StudioUpdated("3e8370e7", "3e8370e7", 75887156211589L,
+                "[organization1 / workspace1]"));
+    }
+
+    @Test
+    void testUpdateWithIncompatibleComputeEnvListsCompatibleOnes(MockServerClient mock) {
+        mock.when(
+                request().withMethod("GET").withPath("/user-info"), exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody(loadResource("user")).withContentType(MediaType.APPLICATION_JSON)
+        );
+        mock.when(
+                request().withMethod("GET").withPath("/user/1264/workspaces"), exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody(loadResource("workspaces/workspaces_list")).withContentType(MediaType.APPLICATION_JSON)
+        );
+        mock.when(
+                request().withMethod("GET").withPath("/studios/3e8370e7").withQueryStringParameter("workspaceId", "75887156211589"), exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody(loadResource("studios/studios_view_response_studio_stopped")).withContentType(MediaType.APPLICATION_JSON)
+        );
+        mock.when(
+                request().withMethod("GET").withPath("/compute-envs")
+                        .withQueryStringParameter("status", "AVAILABLE")
+                        .withQueryStringParameter("workspaceId", "75887156211589"), exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody("{\"computeEnvs\":[{\"id\":\"vYOK4vn7spw7bHHWBDXZ2\",\"name\":\"demo\",\"platform\":\"aws-batch\",\"status\":\"AVAILABLE\",\"message\":null,\"lastUsed\":null,\"primary\":true,\"workspaceName\":null,\"visibility\":null}]}").withContentType(MediaType.APPLICATION_JSON)
+        );
+        mock.when(
+                request().withMethod("GET").withPath("/compute-envs/vYOK4vn7spw7bHHWBDXZ2"), exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody(loadResource("compute_env_demo")).withContentType(MediaType.APPLICATION_JSON)
+        );
+        // The update is rejected because the target CE is not compatible.
+        mock.when(
+                request().withMethod("PUT").withPath("/studios/3e8370e7").withQueryStringParameter("workspaceId", "75887156211589"), exactly(1)
+        ).respond(
+                response().withStatusCode(400).withBody("{\"message\":\"Compute environment 'vYOK4vn7spw7bHHWBDXZ2' is not compatible with the Studio's current compute environment\"}").withContentType(MediaType.APPLICATION_JSON)
+        );
+        // The CLI then queries the compatible CEs to suggest valid options.
+        mock.when(
+                request().withMethod("GET").withPath("/studios/3e8370e7/compatible-ce").withQueryStringParameter("workspaceId", "75887156211589"), exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody("{\"computeEnvs\":[{\"id\":\"ce-compatible-123\",\"name\":\"compatible-ce\",\"platform\":\"aws-batch\",\"status\":\"AVAILABLE\"}]}").withContentType(MediaType.APPLICATION_JSON)
+        );
+
+        ExecOut out = exec(mock, "studios", "update", "-w", "75887156211589", "-i", "3e8370e7", "--compute-env", "demo");
+
+        assertEquals(1, out.exitCode);
+        assertTrue(out.stdErr.contains("not compatible"), out.stdErr);
+        assertTrue(out.stdErr.contains("compatible-ce"), out.stdErr);
+        assertTrue(out.stdErr.contains("ce-compatible-123"), out.stdErr);
+    }
+
+    @ParameterizedTest
+    @EnumSource(OutputType.class)
+    void testAddWithEnvironmentVariables(OutputType format, MockServerClient mock) {
+        mock.when(
+                request().withMethod("GET").withPath("/user-info"), exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody(loadResource("user")).withContentType(MediaType.APPLICATION_JSON)
+        );
+        mock.when(
+                request().withMethod("GET").withPath("/user/1264/workspaces"), exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody(loadResource("workspaces/workspaces_list")).withContentType(MediaType.APPLICATION_JSON)
+        );
+        mock.when(
+                request().withMethod("GET").withPath("/studios/templates")
+                        .withQueryStringParameter("workspaceId", "75887156211589")
+                        .withQueryStringParameter("max", "20"), exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody(loadResource("studios/studios_templates_response")).withContentType(MediaType.APPLICATION_JSON)
+        );
+        mock.when(
+                request().withMethod("GET").withPath("/compute-envs")
+                        .withQueryStringParameter("status", "AVAILABLE")
+                        .withQueryStringParameter("workspaceId", "75887156211589"), exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody("{\"computeEnvs\":[{\"id\":\"vYOK4vn7spw7bHHWBDXZ2\",\"name\":\"demo\",\"platform\":\"aws-batch\",\"status\":\"AVAILABLE\",\"message\":null,\"lastUsed\":null,\"primary\":true,\"workspaceName\":null,\"visibility\":null}]}").withContentType(MediaType.APPLICATION_JSON)
+        );
+        mock.when(
+                request().withMethod("GET").withPath("/compute-envs/vYOK4vn7spw7bHHWBDXZ2"), exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody(loadResource("compute_env_demo")).withContentType(MediaType.APPLICATION_JSON)
+        );
+        mock.when(
+                request().withMethod("POST").withPath("/studios")
+                        .withQueryStringParameter("workspaceId", "75887156211589")
+                        .withBody(json("""
+                           {
+                             "configuration": {
+                               "environment": {
+                                 "FOO": "bar",
+                                 "BAZ": "qux"
+                               }
+                             }
+                           }
+                           """))
+                , exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody(loadResource("studios/studios_created_response")).withContentType(MediaType.APPLICATION_JSON)
+        );
+
+        ExecOut out = exec(format, mock, "studios", "add", "-n", "studio-a66d", "-w", "75887156211589", "-t", "cr.seqera.io/public/data-studio-vscode:1.93.1-snapshot", "-c", "demo", "-e", "FOO=bar", "-e", "BAZ=qux");
+
+        assertOutput(format, out, new StudiosCreated("3e8370e7", 75887156211589L, "[organization1 / workspace1]",
+                "http://localhost:" + mock.getPort() + "/orgs/organization1/workspaces/workspace1", false));
+    }
+
+    @ParameterizedTest
+    @EnumSource(OutputType.class)
+    void testAddWithUrlAndNoTemplate(OutputType format, MockServerClient mock) {
+        mock.when(
+                request().withMethod("GET").withPath("/user-info"), exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody(loadResource("user")).withContentType(MediaType.APPLICATION_JSON)
+        );
+        mock.when(
+                request().withMethod("GET").withPath("/user/1264/workspaces"), exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody(loadResource("workspaces/workspaces_list")).withContentType(MediaType.APPLICATION_JSON)
+        );
+        mock.when(
+                request().withMethod("GET").withPath("/compute-envs")
+                        .withQueryStringParameter("status", "AVAILABLE")
+                        .withQueryStringParameter("workspaceId", "75887156211589"), exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody("{\"computeEnvs\":[{\"id\":\"vYOK4vn7spw7bHHWBDXZ2\",\"name\":\"demo\",\"platform\":\"aws-batch\",\"status\":\"AVAILABLE\",\"message\":null,\"lastUsed\":null,\"primary\":true,\"workspaceName\":null,\"visibility\":null}]}").withContentType(MediaType.APPLICATION_JSON)
+        );
+        mock.when(
+                request().withMethod("GET").withPath("/compute-envs/vYOK4vn7spw7bHHWBDXZ2"), exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody(loadResource("compute_env_demo")).withContentType(MediaType.APPLICATION_JSON)
+        );
+        mock.when(
+                request().withMethod("POST").withPath("/studios")
+                        .withQueryStringParameter("workspaceId", "75887156211589")
+                        .withBody(json("""
+                           {
+                             "remoteConfig": {
+                               "repository": "https://github.com/owner/repo"
+                             }
+                           }
+                           """))
+                , exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody(loadResource("studios/studios_created_response")).withContentType(MediaType.APPLICATION_JSON)
+        );
+
+        ExecOut out = exec(format, mock, "studios", "add", "-n", "studio-a66d", "-w", "75887156211589", "-c", "demo", "--url", "https://github.com/owner/repo");
+
+        assertOutput(format, out, new StudiosCreated("3e8370e7", 75887156211589L, "[organization1 / workspace1]",
+                "http://localhost:" + mock.getPort() + "/orgs/organization1/workspaces/workspace1", false));
+    }
+
+    @Test
+    void testAddWithoutTemplateOrUrlFails(MockServerClient mock) {
+        mock.when(
+                request().withMethod("GET").withPath("/user-info"), exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody(loadResource("user")).withContentType(MediaType.APPLICATION_JSON)
+        );
+        mock.when(
+                request().withMethod("GET").withPath("/user/1264/workspaces"), exactly(1)
+        ).respond(
+                response().withStatusCode(200).withBody(loadResource("workspaces/workspaces_list")).withContentType(MediaType.APPLICATION_JSON)
+        );
+
+        ExecOut out = exec(mock, "studios", "add", "-n", "studio-a66d", "-w", "75887156211589", "-c", "demo");
+
+        assertEquals(1, out.exitCode);
+        assertTrue(out.stdErr.contains("A studio template is required"), out.stdErr);
+    }
+
 }