dayhoff-tools 1.4.21__tar.gz → 1.5.1__tar.gz

{dayhoff_tools-1.4.21 → dayhoff_tools-1.5.1}/PKG-INFO

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

{dayhoff_tools-1.4.21 → dayhoff_tools-1.5.1}/dayhoff_tools/cli/engine_commands.py

@@ -1,6 +1,7 @@
 """Engine and Studio management commands for DHT CLI."""
 
 import json
+import shutil
 import subprocess
 import sys
 import time
@@ -365,6 +366,19 @@ def get_ssh_public_key() -> str:
     )
 
 
+def check_session_manager_plugin():
+    """Check if AWS Session Manager Plugin is available and warn if not."""
+    if shutil.which("session-manager-plugin") is None:
+        console.print(
+            "[bold red]⚠️  AWS Session Manager Plugin not found![/bold red]\n"
+            "SSH connections to engines require the Session Manager Plugin.\n"
+            "Please install it following the setup guide:\n"
+            "[link]https://github.com/dayhofflabs/nutshell/blob/main/REFERENCE/setup_guides/new-laptop.md[/link]"
+        )
+        return False
+    return True
+
+
 def update_ssh_config_entry(engine_name: str, instance_id: str, ssh_user: str):
     """Add or update a single SSH config entry for the given SSH user."""
     config_path = Path.home() / ".ssh" / "config"
@@ -850,6 +864,10 @@ def ssh_engine(
     Pass `--admin` to connect with the underlying [`ec2-user`] account for break-glass or debugging.
     """
     username = check_aws_sso()
+    
+    # Check for Session Manager Plugin
+    if not check_session_manager_plugin():
+        raise typer.Exit(1)
 
     # Get all engines to resolve name
     response = make_api_request("GET", "/engines")
@@ -887,6 +905,10 @@ def config_ssh(
     """Update SSH config with available engines."""
     username = check_aws_sso()
 
+    # Only check for Session Manager Plugin if we're not just cleaning
+    if not clean and not check_session_manager_plugin():
+        raise typer.Exit(1)
+
     if clean:
         console.print("Removing all managed SSH entries...")
     else:
@@ -1568,6 +1590,10 @@ def attach_studio(
     """Attach your studio to an engine."""
     username = check_aws_sso()
     
+    # Check for Session Manager Plugin since we'll update SSH config
+    if not check_session_manager_plugin():
+        raise typer.Exit(1)
+    
     # Use specified user if provided, otherwise use current user
     target_user = user if user else username
     

{dayhoff_tools-1.4.21 → dayhoff_tools-1.5.1}/pyproject.toml

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