dayhoff-tools 1.9.1__tar.gz → 1.9.2__tar.gz

{dayhoff_tools-1.9.1 → dayhoff_tools-1.9.2}/PKG-INFO

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

{dayhoff_tools-1.9.1 → dayhoff_tools-1.9.2}/dayhoff_tools/cli/engine_commands.py

@@ -319,9 +319,6 @@ def format_status(state: str, ready: Optional[bool]) -> str:
         return state
 
 
-
-
-
 def resolve_engine(name_or_id: str, engines: List[Dict]) -> Dict:
     """Resolve engine by name or ID with interactive selection."""
     # Exact ID match
@@ -927,23 +924,55 @@ def engine_status(
             if not content:
                 return None
             data = json.loads(content)
-            # Convert last_state schema to idle_detector schema used by CLI output
+            # Convert last_state schema (new or old) to idle_detector schema used by CLI output
             idle_info: Dict[str, Any] = {"available": True}
-            idle_info["status"] = "active" if not data.get("idle", True) else "idle"
-            # thresholds if present
+
+            # Active/idle
+            idle_flag = bool(data.get("idle", False))
+            idle_info["status"] = "idle" if idle_flag else "active"
+
+            # Threshold and elapsed
             if isinstance(data.get("timeout_sec"), (int, float)):
                 idle_info["idle_threshold"] = int(data["timeout_sec"])  # seconds
-            # keep raw reasons for sensor display
+            if isinstance(data.get("idle_seconds"), (int, float)):
+                idle_info["idle_seconds"] = int(data["idle_seconds"])
+
+            # Keep raw reasons for sensor display when available (new schema)
             if isinstance(data.get("reasons"), list):
                 idle_info["_reasons_raw"] = data["reasons"]
-            # derive details from sensors
-            for r in data.get("reasons", []):
+            else:
+                # Fallback: synthesize reasons from the old forensics layout
+                f_all = data.get("forensics", {}) or {}
+                synthesized = []
+
+                def _mk(sensor_name: str, key: str):
+                    entry = f_all.get(key, {}) or {}
+                    synthesized.append(
+                        {
+                            "sensor": sensor_name,
+                            "active": bool(entry.get("active", False)),
+                            "reason": entry.get("reason", ""),
+                            "forensic": entry.get("forensic", {}),
+                        }
+                    )
+
+                _mk("CoffeeLockSensor", "coffee")
+                _mk("ActiveLoginSensor", "ssh")
+                _mk("IDEConnectionSensor", "ide")
+                _mk("DockerWorkloadSensor", "docker")
+                idle_info["_reasons_raw"] = synthesized
+
+            # Derive details from sensors
+            for r in idle_info.get("_reasons_raw", []):
                 if not r.get("active"):
                     continue
                 sensor = (r.get("sensor") or "").lower()
                 forensic = r.get("forensic") or {}
                 if sensor == "ideconnectionsensor":
-                    cnt = forensic.get("matches")
+                    # Prefer unique_pid_count written by new detector
+                    cnt = forensic.get("unique_pid_count")
+                    if not isinstance(cnt, int):
+                        cnt = forensic.get("matches")
                     if isinstance(cnt, int):
                         idle_info["ide_connections"] = {"connection_count": cnt}
                     else:
@@ -955,12 +984,11 @@ def engine_status(
                             timedelta(seconds=int(rem))
                         )
                 elif sensor == "activeloginsensor":
-                    # Provide a single summarized SSH session if available
                     sess = {
-                        "tty": r.get("forensic", {}).get("tty", "pts/?"),
-                        "pid": r.get("forensic", {}).get("pid", "?"),
-                        "idle_time": r.get("forensic", {}).get("idle_sec", 0),
-                        "from_ip": r.get("forensic", {}).get("remote_addr", "unknown"),
+                        "tty": forensic.get("tty", "pts/?"),
+                        "pid": forensic.get("pid", "?"),
+                        "idle_time": forensic.get("idle_sec", 0),
+                        "from_ip": forensic.get("remote_addr", "unknown"),
                     }
                     idle_info.setdefault("ssh_sessions", []).append(sess)
             return idle_info
@@ -1002,7 +1030,23 @@ def engine_status(
         status_lines.append(_sensor_line(" IDE   ", "IDEConnectionSensor", "🖥"))
         status_lines.append(_sensor_line("Docker", "DockerWorkloadSensor", "🐳"))
 
-
+        # If we have elapsed idle seconds and threshold, reflect that in the header
+        try:
+            if idle_detector.get("status") == "idle":
+                idle_secs = int(idle_detector.get("idle_seconds", 0))
+                thresh_secs = int(idle_detector.get("idle_threshold", 0))
+                if thresh_secs > 0:
+                    active_disp = (
+                        f"[yellow]Idle {idle_secs//60}m/{thresh_secs//60}m[/yellow]"
+                    )
+                    # Rewrite top header line (index 0) to include updated display
+                    all_header = top_lines[0]
+                    # Replace the portion after two spaces (name and running state fixed)
+                    top_lines[0] = (
+                        f"[blue]{engine['name']}[/blue]  {run_disp}  {active_disp}\n"
+                    )
+        except Exception:
+            pass
 
     # Combine top summary and details
     all_lines = top_lines + status_lines
@@ -1041,9 +1085,6 @@ def engine_status(
             console.print(f"[red]❌ Error fetching log: {e}[/red]")
 
 
-
-
-
 @engine_app.command("stop")
 def stop_engine(
     name_or_id: str = typer.Argument(help="Engine name or instance ID"),

{dayhoff_tools-1.9.1 → dayhoff_tools-1.9.2}/pyproject.toml

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