agentremote-cli 1.0.0__py3-none-any.whl

remotecli/shell.py

@@ -0,0 +1,71 @@
+"""
+remotecli shell — 远程 Shell 执行子命令.
+"""
+
+from __future__ import annotations
+
+import re
+from typing import Any, Dict
+
+from .client import RemoteCliClient
+
+
+def _repair_powershell_current_object_expansion(script: str) -> str:
+    """Repair common outer-shell expansion of PowerShell's $_ in Get-StartApps filters."""
+    if "Get-StartApps" not in script or "Where-Object" not in script or "-like" not in script:
+        return script
+
+    matches = list(re.finditer(r"-like\s+(?P<quote>['\"])(?P<pattern>.*?)(?P=quote)", script))
+    if not matches:
+        return script
+
+    has_nested_expansion = (
+        re.search(r"Where-Object\s*\{.*Where-Object\s*\{.*\.Name\s+-like\b", script) is not None
+    )
+    if ("$_.Name" in script or "$PSItem.Name" in script) and not has_nested_expansion:
+        return script
+
+    has_expanded_current_object = (
+        re.search(r"Where-Object\s*\{\s*\.[A-Za-z_][\w]*\s+-like\b", script) is not None
+        or re.search(r"Where-Object\s*\{\s*[^${};|]+\.Name\s+-like\b", script) is not None
+        or has_nested_expansion
+    )
+    if not has_expanded_current_object:
+        return script
+
+    match = matches[-1]
+    quote = match.group("quote")
+    pattern = match.group("pattern")
+    return f"Get-StartApps | Where-Object {{$_.Name -like {quote}{pattern}{quote}}}"
+
+
+class ShellCommands:
+    """远程 Shell 命令构建器。"""
+
+    def __init__(self, client: RemoteCliClient) -> None:
+        self._client = client
+        self._capability_id = "native.shell"
+
+    def exec(self, cmd: str, cwd: str = "", timeout_sec: int = 60) -> Dict[str, Any]:
+        """执行单条 shell 命令。"""
+        args: Dict[str, Any] = {"cmd": cmd}
+        if cwd:
+            args["cwd"] = cwd
+        return self._client.submit(self._capability_id, "exec", args, timeout_sec=timeout_sec)
+
+    def exec_script(self, script: str, shell: str = "", cwd: str = "",
+                    timeout_sec: int = 60) -> Dict[str, Any]:
+        """执行多行脚本 (写入临时文件, 执行后清理)。"""
+        if shell in ("powershell", "pwsh"):
+            script = _repair_powershell_current_object_expansion(script)
+
+        args: Dict[str, Any] = {"script": script}
+        if shell:
+            args["shell"] = shell
+        if cwd:
+            args["cwd"] = cwd
+        return self._client.submit(self._capability_id, "exec-script", args, timeout_sec=timeout_sec)
+
+    def info(self) -> Dict[str, Any]:
+        """获取 Shell/OS 环境信息。"""
+        return self._client.submit(self._capability_id, "info", {})