portacode 1.4.12.dev1__py3-none-any.whl → 1.4.15.dev3__py3-none-any.whl

portacode/connection/handlers/system_handlers.py

@@ -10,8 +10,9 @@ import platform
 import shutil
 import subprocess
 import threading
+import time
 from pathlib import Path
-from typing import Any, Dict
+from typing import Any, Dict, Optional
 
 from portacode import __version__
 import psutil
@@ -31,11 +32,26 @@ _cpu_percent = 0.0
 _cpu_thread = None
 _cpu_lock = threading.Lock()
 
+# Cgroup v2 tracking
+_CGROUP_ROOT = Path("/sys/fs/cgroup")
+_cgroup_path: Optional[Path] = None
+_cgroup_v2_supported: Optional[bool] = None
+_CGROUP_CPU_STAT = "cpu.stat"
+_CGROUP_CPU_MAX = "cpu.max"
+_last_cgroup_usage: Optional[int] = None
+_last_cgroup_time: Optional[float] = None
+
 def _cpu_monitor():
     """Background thread to update CPU usage every 5 seconds."""
     global _cpu_percent
     while True:
-        _cpu_percent = psutil.cpu_percent(interval=5.0)
+        percent = _get_cgroup_cpu_percent()
+        if percent is None:
+            # Fall back to psutil when cgroup stats are not available yet.
+            percent = psutil.cpu_percent(interval=5.0)
+        else:
+            time.sleep(5.0)
+        _cpu_percent = percent
 
 def _ensure_cpu_thread():
     """Ensure CPU monitoring thread is running (singleton)."""
@@ -130,6 +146,119 @@ def _get_playwright_info() -> Dict[str, Any]:
     return result
 
 
+def _resolve_cgroup_path() -> Path:
+    global _cgroup_path
+    if _cgroup_path is not None and _cgroup_path.exists():
+        return _cgroup_path
+    path = _CGROUP_ROOT
+    cgroup_file = Path("/proc/self/cgroup")
+    if cgroup_file.exists():
+        try:
+            contents = cgroup_file.read_text()
+        except OSError:
+            pass
+        else:
+            for line in contents.splitlines():
+                line = line.strip()
+                if not line:
+                    continue
+                parts = line.split(":", 2)
+                if len(parts) < 3:
+                    continue
+                rel_path = parts[-1].lstrip("/")
+                candidate = _CGROUP_ROOT / rel_path
+                # Fallback to root path if the relative path is empty
+                candidate = candidate if rel_path else _CGROUP_ROOT
+                if candidate.exists():
+                    path = candidate
+                    break
+    _cgroup_path = path
+    return _cgroup_path
+
+
+def _cgroup_file(name: str) -> Path:
+    return _resolve_cgroup_path() / name
+
+
+def _detect_cgroup_v2() -> bool:
+    global _cgroup_v2_supported
+    if _cgroup_v2_supported is not None:
+        return _cgroup_v2_supported
+    controllers = _cgroup_file("cgroup.controllers")
+    cpu_stat = _cgroup_file(_CGROUP_CPU_STAT)
+    _cgroup_v2_supported = controllers.exists() and cpu_stat.exists()
+    return _cgroup_v2_supported
+
+
+def _read_cgroup_cpu_usage() -> Optional[int]:
+    path = _cgroup_file(_CGROUP_CPU_STAT)
+    try:
+        data = path.read_text()
+    except (OSError, UnicodeDecodeError):
+        return None
+    for line in data.splitlines():
+        parts = line.strip().split()
+        if len(parts) >= 2 and parts[0] == "usage_usec":
+            try:
+                return int(parts[1])
+            except ValueError:
+                return None
+    return None
+
+
+def _read_cgroup_cpu_limit() -> Optional[float]:
+    """Return the allowed CPU (cores) for this cgroup, if limited."""
+    path = _cgroup_file(_CGROUP_CPU_MAX)
+    if not path.exists():
+        return None
+    try:
+        data = path.read_text().strip()
+    except (OSError, UnicodeDecodeError):
+        return None
+    parts = data.split()
+    if len(parts) < 2:
+        return None
+    quota, period = parts[0], parts[1]
+    if quota == "max":
+        return None
+    try:
+        quota_value = int(quota)
+        period_value = int(period)
+    except ValueError:
+        return None
+    if period_value <= 0:
+        return None
+    return quota_value / period_value
+
+
+def _get_cgroup_cpu_percent() -> Optional[float]:
+    if not _detect_cgroup_v2():
+        return None
+    usage = _read_cgroup_cpu_usage()
+    if usage is None:
+        return None
+    now = time.monotonic()
+    global _last_cgroup_usage, _last_cgroup_time
+    prev_usage = _last_cgroup_usage
+    prev_time = _last_cgroup_time
+    _last_cgroup_usage = usage
+    _last_cgroup_time = now
+    if prev_usage is None or prev_time is None:
+        return None
+    delta_usage = usage - prev_usage
+    delta_time = now - prev_time
+    if delta_time <= 0 or delta_usage < 0:
+        return None
+    cpu_ratio = (delta_usage / 1_000_000) / delta_time
+    limit_cpus = _read_cgroup_cpu_limit()
+    if limit_cpus and limit_cpus > 0:
+        percent = (cpu_ratio / limit_cpus) * 100.0
+    else:
+        cpu_count = psutil.cpu_count(logical=True) or 1
+        percent = (cpu_ratio / cpu_count) * 100.0
+    return max(0.0, min(percent, 100.0))
+
+
 def _run_probe_command(cmd: list[str]) -> str | None:
     try:
         result = subprocess.run(cmd, capture_output=True, text=True, check=True, timeout=3)

portacode/connection/handlers/test_proxmox_infra.py

@@ -0,0 +1,13 @@
+from unittest import TestCase
+
+from portacode.connection.handlers.proxmox_infra import _build_bootstrap_steps
+
+
+class ProxmoxInfraHandlerTests(TestCase):
+    def test_build_bootstrap_steps_includes_portacode_connect_by_default(self):
+        steps = _build_bootstrap_steps("svcuser", "pass", "", include_portacode_connect=True)
+        self.assertTrue(any(step.get("name") == "portacode_connect" for step in steps))
+
+    def test_build_bootstrap_steps_skips_portacode_connect_when_requested(self):
+        steps = _build_bootstrap_steps("svcuser", "pass", "", include_portacode_connect=False)
+        self.assertFalse(any(step.get("name") == "portacode_connect" for step in steps))

portacode/connection/terminal.py

@@ -56,6 +56,9 @@ from .handlers import (
     CreateProxmoxContainerHandler,
     RevertProxmoxInfraHandler,
     StartPortacodeServiceHandler,
+    StartProxmoxContainerHandler,
+    StopProxmoxContainerHandler,
+    RemoveProxmoxContainerHandler,
 )
 from .handlers.project_aware_file_handlers import (
     ProjectAwareFileWriteHandler,
@@ -480,6 +483,9 @@ class TerminalManager:
         self._command_registry.register(ConfigureProxmoxInfraHandler)
         self._command_registry.register(CreateProxmoxContainerHandler)
         self._command_registry.register(StartPortacodeServiceHandler)
+        self._command_registry.register(StartProxmoxContainerHandler)
+        self._command_registry.register(StopProxmoxContainerHandler)
+        self._command_registry.register(RemoveProxmoxContainerHandler)
         self._command_registry.register(RevertProxmoxInfraHandler)
         self._command_registry.register(UpdatePortacodeHandler)
 

{portacode-1.4.12.dev1.dist-info → portacode-1.4.15.dev3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: portacode
-Version: 1.4.12.dev1
+Version: 1.4.15.dev3
 Summary: Portacode CLI client and SDK
 Home-page: https://github.com/portacode/portacode
 Author: Meena Erian

{portacode-1.4.12.dev1.dist-info → portacode-1.4.15.dev3.dist-info}/RECORD

@@ -1,7 +1,7 @@
 portacode/README.md,sha256=4dKtpvR8LNgZPVz37GmkQCMWIr_u25Ao63iW56s7Ke4,775
 portacode/__init__.py,sha256=oB3sV1wXr-um-RXio73UG8E5Xx6cF2ZVJveqjNmC-vQ,1086
 portacode/__main__.py,sha256=jmHTGC1hzmo9iKJLv-SSYe9BSIbPPZ2IOpecI03PlTs,296
-portacode/_version.py,sha256=KBZ1gEZmvt1J4XR1FpwaCleg3ayMUckV-yZYhnXlSCU,719
+portacode/_version.py,sha256=1sfx8JK_QXWbIlvcX2Sn3XEnIw9YLhiiBq9EMvNgnss,719
 portacode/cli.py,sha256=mGLKoZ-T2FBF7IA9wUq0zyG0X9__-A1ao7gajjcVRH8,21828
 portacode/data.py,sha256=5-s291bv8J354myaHm1Y7CQZTZyRzMU3TGe5U4hb-FA,1591
 portacode/keypair.py,sha256=0OO4vHDcF1XMxCDqce61xFTlFwlTcmqe5HyGsXFEt7s,5838
@@ -12,22 +12,23 @@ portacode/connection/README.md,sha256=f9rbuIEKa7cTm9C98rCiBbEtbiIXQU11esGSNhSMiJ
 portacode/connection/__init__.py,sha256=atqcVGkViIEd7pRa6cP2do07RJOM0UWpbnz5zXjGktU,250
 portacode/connection/client.py,sha256=jtLb9_YufqPkzi9t8VQH3iz_JEMisbtY6a8L9U5weiU,14181
 portacode/connection/multiplex.py,sha256=L-TxqJ_ZEbfNEfu1cwxgJ5vUdyRzZjsMy2Kx1diiZys,5237
-portacode/connection/terminal.py,sha256=07wxG_55JMy3yQ9TXCBldW9h43qCW3U8rv2yzGMx4FM,44757
+portacode/connection/terminal.py,sha256=oyLPOVLPlUuN_eRvHPGazB51yi8W8JEF3oOEYxucGTE,45069
 portacode/connection/handlers/README.md,sha256=HsLZG1QK1JNm67HsgL6WoDg9nxzKXxwkc5fJPFJdX5g,12169
-portacode/connection/handlers/WEBSOCKET_PROTOCOL.md,sha256=7tBYNEY8EBGAPIMT606BqeHnyMOQIZVlQYpH7me26LY,97962
-portacode/connection/handlers/__init__.py,sha256=WSeBmi65GWFQPYt9M3E10rn0uZ_EPCJzNJOzSf2HZyw,2921
+portacode/connection/handlers/WEBSOCKET_PROTOCOL.md,sha256=5VZSDPvXjtOl-8YZ-2UVtH8WgPnEWe7WB33YPrS8_8Q,104190
+portacode/connection/handlers/__init__.py,sha256=j69jGkf2-mYyCicvYfp2wk8-xB8yqpWktiN5xADXBno,3137
 portacode/connection/handlers/base.py,sha256=oENFb-Fcfzwk99Qx8gJQriEMiwSxwygwjOiuCH36hM4,10231
 portacode/connection/handlers/chunked_content.py,sha256=h6hXRmxSeOgnIxoU8CkmvEf2Odv-ajPrpHIe_W3GKcA,9251
 portacode/connection/handlers/diff_handlers.py,sha256=iYTIRCcpEQ03vIPKZCsMTE5aZbQw6sF04M3dM6rUV8Q,24477
 portacode/connection/handlers/file_handlers.py,sha256=nAJH8nXnX07xxD28ngLpgIUzcTuRwZBNpEGEKdRqohw,39507
 portacode/connection/handlers/project_aware_file_handlers.py,sha256=AqgMnDqX2893T2NsrvUSCwjN5VKj4Pb2TN0S_SuboOE,9803
 portacode/connection/handlers/project_state_handlers.py,sha256=v6ZefGW9i7n1aZLq2jOGumJIjYb6aHlPI4m1jkYewm8,1686
-portacode/connection/handlers/proxmox_infra.py,sha256=Sd25UGM7edWDTIiatZvA2Dh3-4NOzfIDmixG0Fz5m0U,51343
+portacode/connection/handlers/proxmox_infra.py,sha256=mH6xxT9v9X0jwdCBjU3p48lURYisY0T0HPjuzV3eoH8,76070
 portacode/connection/handlers/registry.py,sha256=qXGE60sYEWg6ZtVQzFcZ5YI2XWR6lMgw4hAL9x5qR1I,6181
 portacode/connection/handlers/session.py,sha256=uNGfiO_1B9-_yjJKkpvmbiJhIl6b-UXlT86UTfd6WYE,42219
-portacode/connection/handlers/system_handlers.py,sha256=AKh7IbwptlLYrbSw5f-DHigvlaKHsg9lDP-lkAUm8cE,10755
+portacode/connection/handlers/system_handlers.py,sha256=fr12QpOr_Z8KYGUU-AYrTQwRPAcrLK85hvj3SEq1Kw8,14757
 portacode/connection/handlers/tab_factory.py,sha256=yn93h6GASjD1VpvW1oqpax3EpoT0r7r97zFXxML1wdA,16173
 portacode/connection/handlers/terminal_handlers.py,sha256=HRwHW1GiqG1NtHVEqXHKaYkFfQEzCDDH6YIlHcb4XD8,11866
+portacode/connection/handlers/test_proxmox_infra.py,sha256=d6iBB4pwAqWWdEGRayLxDEexqCElbGZDJlCB4bXba24,682
 portacode/connection/handlers/update_handler.py,sha256=f2K4LmG4sHJZ3LahzzoRtHBULTKkPUNwuyhwuAAg3RA,2054
 portacode/connection/handlers/project_state/README.md,sha256=trdd4ig6ungmwH5SpbSLfyxbL-QgPlGNU-_XrMEiXtw,10114
 portacode/connection/handlers/project_state/__init__.py,sha256=5ucIqk6Iclqg6bKkL8r_wVs5Tlt6B9J7yQH6yQUt7gc,2541
@@ -64,7 +65,7 @@ portacode/utils/__init__.py,sha256=NgBlWTuNJESfIYJzP_3adI1yJQJR0XJLRpSdVNaBAN0,3
 portacode/utils/diff_apply.py,sha256=4Oi7ft3VUCKmiUE4VM-OeqO7Gk6H7PF3WnN4WHXtjxI,15157
 portacode/utils/diff_renderer.py,sha256=S76StnQ2DLfsz4Gg0m07UwPfRp8270PuzbNaQq-rmYk,13850
 portacode/utils/ntp_clock.py,sha256=VqCnWCTehCufE43W23oB-WUdAZGeCcLxkmIOPwInYHc,2499
-portacode-1.4.12.dev1.dist-info/licenses/LICENSE,sha256=2FGbCnUDgRYuQTkB1O1dUUpu5CVAjK1j4_p6ack9Z54,1066
+portacode-1.4.15.dev3.dist-info/licenses/LICENSE,sha256=2FGbCnUDgRYuQTkB1O1dUUpu5CVAjK1j4_p6ack9Z54,1066
 test_modules/README.md,sha256=Do_agkm9WhSzueXjRAkV_xEj6Emy5zB3N3VKY5Roce8,9274
 test_modules/__init__.py,sha256=1LcbHodIHsB0g-g4NGjSn6AMuCoGbymvXPYLOb6Z7F0,53
 test_modules/test_device_online.py,sha256=QtYq0Dq9vME8Gp2O4fGSheqVf8LUtpsSKosXXk56gGM,1654
@@ -90,8 +91,8 @@ testing_framework/core/playwright_manager.py,sha256=Tw46qwxIhOFkS48C2IWIQHHNpEe-
 testing_framework/core/runner.py,sha256=j2QwNJmAxVBmJvcbVS7DgPJUKPNzqfLmt_4NNdaKmZU,19297
 testing_framework/core/shared_cli_manager.py,sha256=BESSNtyQb7BOlaOvZmm04T8Uezjms4KCBs2MzTxvzYQ,8790
 testing_framework/core/test_discovery.py,sha256=2FZ9fJ8Dp5dloA-fkgXoJ_gCMC_nYPBnA3Hs2xlagzM,4928
-portacode-1.4.12.dev1.dist-info/METADATA,sha256=o7psDA6l4eoRJLFQipCS3XcgsqTLxNMf39xhh6qYJ00,13051
-portacode-1.4.12.dev1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-portacode-1.4.12.dev1.dist-info/entry_points.txt,sha256=lLUUL-BM6_wwe44Xv0__5NQ1BnAz6jWjSMFvZdWW3zU,48
-portacode-1.4.12.dev1.dist-info/top_level.txt,sha256=TGhTYUxfW8SyVZc_zGgzjzc24gGT7nSw8Qf73liVRKM,41
-portacode-1.4.12.dev1.dist-info/RECORD,,
+portacode-1.4.15.dev3.dist-info/METADATA,sha256=fhua1K0pTVsxZaun0VkgDDpUBh5AdovF_z9U6_LyktE,13051
+portacode-1.4.15.dev3.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
+portacode-1.4.15.dev3.dist-info/entry_points.txt,sha256=lLUUL-BM6_wwe44Xv0__5NQ1BnAz6jWjSMFvZdWW3zU,48
+portacode-1.4.15.dev3.dist-info/top_level.txt,sha256=TGhTYUxfW8SyVZc_zGgzjzc24gGT7nSw8Qf73liVRKM,41
+portacode-1.4.15.dev3.dist-info/RECORD,,

{portacode-1.4.12.dev1.dist-info → portacode-1.4.15.dev3.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.9.0)
+Generator: setuptools (80.10.1)
 Root-Is-Purelib: true
 Tag: py3-none-any