dayhoff-tools 1.7.0__py3-none-any.whl → 1.7.1__py3-none-any.whl

dayhoff_tools/cli/engine_commands.py

@@ -1,6 +1,7 @@
 """Engine and Studio management commands for DHT CLI."""
 
 import json
+import os
 import re
 import shutil
 import subprocess
@@ -365,13 +366,58 @@ def resolve_engine(name_or_id: str, engines: List[Dict]) -> Dict:
 
 
 def get_ssh_public_key() -> str:
-    """Get the user's SSH public key."""
+    """Get the user's SSH public key.
+
+    Discovery order (container-friendly):
+    1) DHT_SSH_PUBLIC_KEY env var (direct key content)
+    2) DHT_SSH_PUBLIC_KEY_PATH env var (path to a .pub file)
+    3) ssh-agent via `ssh-add -L` (requires SSH_AUTH_SOCK)
+    4) Conventional files: ~/.ssh/id_ed25519.pub, ~/.ssh/id_rsa.pub
+
+    Raises:
+        FileNotFoundError: If no public key can be discovered.
+    """
+    # 1) Direct env var content
+    env_key = os.environ.get("DHT_SSH_PUBLIC_KEY")
+    if env_key and env_key.strip():
+        return env_key.strip()
+
+    # 2) Env var path
+    env_path = os.environ.get("DHT_SSH_PUBLIC_KEY_PATH")
+    if env_path:
+        p = Path(env_path).expanduser()
+        if p.is_file():
+            try:
+                return p.read_text().strip()
+            except Exception:
+                pass
+
+    # 3) Agent lookup (ssh-add -L)
+    try:
+        if shutil.which("ssh-add") is not None:
+            proc = subprocess.run(["ssh-add", "-L"], capture_output=True, text=True)
+            if proc.returncode == 0 and proc.stdout:
+                keys = [line.strip() for line in proc.stdout.splitlines() if line.strip()]
+                # Prefer ed25519, then rsa
+                for pref in ("ssh-ed25519", "ssh-rsa", "ecdsa-sha2-nistp256"):
+                    for k in keys:
+                        if k.startswith(pref + " "):
+                            return k
+                # Fallback to first key if types not matched
+                if keys:
+                    return keys[0]
+    except Exception:
+        pass
+
+    # 4) Conventional files
     home = Path.home()
     key_paths = [home / ".ssh" / "id_ed25519.pub", home / ".ssh" / "id_rsa.pub"]
-
     for key_path in key_paths:
         if key_path.is_file():
-            return key_path.read_text().strip()
+            try:
+                return key_path.read_text().strip()
+            except Exception:
+                continue
 
     raise FileNotFoundError(
         "No SSH public key found. Please create one with 'ssh-keygen' first."

{dayhoff_tools-1.7.0.dist-info → dayhoff_tools-1.7.1.dist-info}/METADATA

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

{dayhoff_tools-1.7.0.dist-info → dayhoff_tools-1.7.1.dist-info}/RECORD

@@ -3,7 +3,7 @@ dayhoff_tools/chemistry/standardizer.py,sha256=uMn7VwHnx02nc404eO6fRuS4rsl4dvSPf
 dayhoff_tools/chemistry/utils.py,sha256=jt-7JgF-GeeVC421acX-bobKbLU_X94KNOW24p_P-_M,2257
 dayhoff_tools/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dayhoff_tools/cli/cloud_commands.py,sha256=33qcWLmq-FwEXMdL3F0OHm-5Stlh2r65CldyEZgQ1no,40904
-dayhoff_tools/cli/engine_commands.py,sha256=4svrENpNa5OYqooeWuiXvoXe9h4jfuU9iUBoFOVIEMk,94855
+dayhoff_tools/cli/engine_commands.py,sha256=_NQI9x4VKtAC9vTDdA4NGrILf2kb005t3Kw0aZ5oroE,96485
 dayhoff_tools/cli/main.py,sha256=tRN7WCBHg6uyNp6rA54pKTCoVmBntta2i0Yas3bUpZ4,4853
 dayhoff_tools/cli/swarm_commands.py,sha256=5EyKj8yietvT5lfoz8Zx0iQvVaNgc3SJX1z2zQR6o6M,5614
 dayhoff_tools/cli/utility_commands.py,sha256=FRZTPrjsG_qmIIqoNxd1Q1vVkS_5w8aY33IrVYVNCLg,18131
@@ -27,7 +27,7 @@ dayhoff_tools/intake/uniprot.py,sha256=BZYJQF63OtPcBBnQ7_P9gulxzJtqyorgyuDiPeOJq
 dayhoff_tools/logs.py,sha256=DKdeP0k0kliRcilwvX0mUB2eipO5BdWUeHwh-VnsICs,838
 dayhoff_tools/sqlite.py,sha256=jV55ikF8VpTfeQqqlHSbY8OgfyfHj8zgHNpZjBLos_E,18672
 dayhoff_tools/warehouse.py,sha256=heaYc64qplgN3_1WVPFmqj53goStioWwY5NqlWc4c0s,24453
-dayhoff_tools-1.7.0.dist-info/METADATA,sha256=dORGLI3ynceOQVLTkLsVmYgvKvxD8e-05RHSHg-zDMc,2914
-dayhoff_tools-1.7.0.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-dayhoff_tools-1.7.0.dist-info/entry_points.txt,sha256=iAf4jteNqW3cJm6CO6czLxjW3vxYKsyGLZ8WGmxamSc,49
-dayhoff_tools-1.7.0.dist-info/RECORD,,
+dayhoff_tools-1.7.1.dist-info/METADATA,sha256=SfgSoCKU9G3NVk4aeQprKBvjn8-bHUXPy5jT2EIS1sA,2914
+dayhoff_tools-1.7.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+dayhoff_tools-1.7.1.dist-info/entry_points.txt,sha256=iAf4jteNqW3cJm6CO6czLxjW3vxYKsyGLZ8WGmxamSc,49
+dayhoff_tools-1.7.1.dist-info/RECORD,,