- Support JetBrains Fleet as an alternative IDE for dev environments #…

…821 (POC)

src/dstack/_internal/core/models/configurations.py

@@ -128,7 +128,7 @@ def convert_ports(cls, v) -> PortMapping:
 
 class DevEnvironmentConfiguration(BaseConfigurationWithPorts):
     type: Literal["dev-environment"] = "dev-environment"
-    ide: Annotated[Literal["vscode"], Field(description="The IDE to run")]
+    ide: Annotated[Literal["vscode", "fleet"], Field(description="The IDE to run")]
     version: Annotated[Optional[str], Field(description="The version of the IDE")]
     init: Annotated[CommandsList, Field(description="The bash commands to run")] = []
 

src/dstack/_internal/server/services/jobs/configurators/dev.py

@@ -4,6 +4,7 @@
 from dstack._internal.core.models.profiles import ProfileRetryPolicy, SpotPolicy
 from dstack._internal.core.models.runs import RetryPolicy, RunSpec
 from dstack._internal.server.services.jobs.configurators.base import JobConfigurator
+from dstack._internal.server.services.jobs.configurators.extensions.fleet import JetBrainsFleet
 from dstack._internal.server.services.jobs.configurators.extensions.vscode import VSCodeDesktop
 
 DEFAULT_MAX_DURATION_SECONDS = 6 * 3600
@@ -18,10 +19,14 @@ class DevEnvironmentJobConfigurator(JobConfigurator):
     TYPE: ConfigurationType = ConfigurationType.DEV_ENVIRONMENT
 
     def __init__(self, run_spec: RunSpec):
-        self.ide = VSCodeDesktop(
-            run_name=run_spec.run_name,
-            version=run_spec.configuration.version,
-            extensions=["ms-python.python", "ms-toolsai.jupyter"],
+        self.ide = (
+            VSCodeDesktop(
+                run_name=run_spec.run_name,
+                version=run_spec.configuration.version,
+                extensions=["ms-python.python", "ms-toolsai.jupyter"],
+            )
+            if run_spec.configuration.ide == "vscode"
+            else JetBrainsFleet(run_name=run_spec.run_name)
         )
         super().__init__(run_spec)
 

src/dstack/_internal/server/services/jobs/configurators/extensions/fleet.py

@@ -0,0 +1,17 @@
+from typing import List
+
+
+class JetBrainsFleet:
+    def __init__(self, run_name: str):
+        self.run_name = run_name
+
+    def get_install_commands(self) -> List[str]:
+        commands = [
+            f'if [ $(uname -m) = "aarch64" ]; then arch="aarch64"; else arch="x64"; fi',
+            'wget -q --show-progress -O fleet "https://download.jetbrains.com/product?code=FLL&release.type=preview&release.type=eap&platform=linux_$arch" && chmod +x fleet',
+            f"./fleet launch workspace -- --auth=accept-everyone --publish --workspaceName {self.run_name} --projectDir /workflow",
+        ]
+        return commands
+
+    def get_print_readme_commands(self) -> List[str]:
+        return []