dayhoff-tools 1.3.5__tar.gz → 1.3.6__tar.gz

{dayhoff_tools-1.3.5 → dayhoff_tools-1.3.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: dayhoff-tools
-Version: 1.3.5
+Version: 1.3.6
 Summary: Common tools for all the repos at Dayhoff Labs
 Author: Daniel Martin-Alarcon
 Author-email: dma@dayhofflabs.com

{dayhoff_tools-1.3.5 → dayhoff_tools-1.3.6}/dayhoff_tools/cli/engine_commands.py

@@ -338,11 +338,10 @@ def launch_engine(
 @engine_app.command("list")
 def list_engines(
     user: Optional[str] = typer.Option(None, "--user", "-u", help="Filter by user"),
-    all_users: bool = typer.Option(False, "--all", "-a", help="Show all users' engines"),
     running_only: bool = typer.Option(False, "--running", help="Show only running engines"),
     stopped_only: bool = typer.Option(False, "--stopped", help="Show only stopped engines"),
 ):
-    """List all engines."""
+    """List engines (shows all engines by default)."""
     current_user = check_aws_sso()
 
     params = {}
@@ -618,14 +617,18 @@ def ssh_engine(
 @engine_app.command("config-ssh")
 def config_ssh(
     clean: bool = typer.Option(False, "--clean", help="Remove all managed entries"),
+    all_engines: bool = typer.Option(False, "--all", "-a", help="Include all engines from all users"),
 ):
-    """Update SSH config with all available engines."""
+    """Update SSH config with available engines."""
     username = check_aws_sso()
 
     if clean:
         console.print("Removing all managed SSH entries...")
     else:
-        console.print("Updating SSH config with all running engines...")
+        if all_engines:
+            console.print("Updating SSH config with all running engines...")
+        else:
+            console.print(f"Updating SSH config with running engines for [cyan]{username}[/cyan] and [cyan]shared[/cyan]...")
 
     # Get all engines
     response = make_api_request("GET", "/engines")
@@ -635,6 +638,14 @@ def config_ssh(
 
     engines = response.json().get("engines", [])
     running_engines = [e for e in engines if e["state"].lower() == "running"]
+    
+    # Filter engines based on options
+    if not all_engines:
+        # Show only current user's engines and shared engines
+        running_engines = [
+            e for e in running_engines 
+            if e["user"] == username or e["user"] == "shared"
+        ]
 
     # Read existing config
     config_path = Path.home() / ".ssh" / "config"
@@ -685,7 +696,8 @@ def config_ssh(
             f"[green]✓ Updated SSH config with {len(running_engines)} engines[/green]"
         )
         for engine in running_engines:
-            console.print(f"  • {engine['name']} → {engine['instance_id']}")
+            user_display = f"[dim]({engine['user']})[/dim]" if engine['user'] != username else ""
+            console.print(f"  • {engine['name']} → {engine['instance_id']} {user_display}")
 
 
 @engine_app.command("keep-awake")
@@ -1020,7 +1032,7 @@ def studio_status():
     # Format status with colors
     status = studio['status']
     if status == "in-use":
-        status_display = "[magenta]attached[/magenta]"
+        status_display = "[bright_blue]attached[/bright_blue]"
     elif status in ["attaching", "detaching"]:
         status_display = f"[yellow]{status}[/yellow]"
     else:
@@ -1258,7 +1270,7 @@ def list_studios(
         for studio in studios:
             # Change status display
             if studio["status"] == "in-use":
-                status_display = "[magenta]attached[/magenta]"
+                status_display = "[bright_blue]attached[/bright_blue]"
             elif studio["status"] in ["attaching", "detaching"]:
                 status_display = "[yellow]" + studio["status"] + "[/yellow]"
             else:

{dayhoff_tools-1.3.5 → dayhoff_tools-1.3.6}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "poetry.core.masonry.api"
 
 [project]
 name = "dayhoff-tools"
-version = "1.3.5"
+version = "1.3.6"
 description = "Common tools for all the repos at Dayhoff Labs"
 authors = [
     {name = "Daniel Martin-Alarcon", email = "dma@dayhofflabs.com"}