mcp-yieldshell 0.1.7__py3-none-any.whl

mcp_yieldshell/__init__.py

@@ -0,0 +1 @@
+"""YieldShell MCP — A drop-in shell MCP that auto-yields long-running commands."""

mcp_yieldshell/__main__.py

@@ -0,0 +1,17 @@
+"""CLI entrypoint: start the YieldShell MCP stdio server."""
+
+from __future__ import annotations
+
+from .config import Config
+from .server import create_server
+
+
+def main() -> None:
+    """Start the YieldShell MCP server on stdio."""
+    config = Config()
+    server = create_server(config)
+    server.run(transport="stdio")
+
+
+if __name__ == "__main__":
+    main()

mcp_yieldshell/config.py

@@ -0,0 +1,66 @@
+"""Configuration parsed from environment variables at server startup."""
+
+from __future__ import annotations
+
+import os
+import re
+
+
+class Config:
+    def __init__(self) -> None:
+        self.default_cwd: str = os.environ.get("YIELDSHELL_DEFAULT_CWD", os.getcwd())
+        self.allowed_cwd_roots: list[str] = _parse_pathsep(
+            os.environ.get("YIELDSHELL_ALLOWED_CWDS", "")
+        )
+        self.max_output_bytes: int = _parse_int(
+            os.environ.get("YIELDSHELL_MAX_OUTPUT_BYTES", ""), 20000
+        )
+        self.max_processes: int = _parse_int(
+            os.environ.get("YIELDSHELL_MAX_PROCESSES", ""), 50
+        )
+        self.default_yield_ms: int = _parse_int(
+            os.environ.get("YIELDSHELL_DEFAULT_YIELD_MS", ""), 5000
+        )
+        self.max_yield_ms: int = _parse_int(
+            os.environ.get("YIELDSHELL_MAX_YIELD_MS", ""), 300000
+        )
+        self.default_timeout_ms: int = _parse_int(
+            os.environ.get("YIELDSHELL_DEFAULT_TIMEOUT_MS", ""), 0
+        )
+        self.deny_command_regex: re.Pattern[str] | None = _parse_regex(
+            os.environ.get("YIELDSHELL_DENY_COMMAND_REGEX", "")
+        )
+        self.allow_command_regex: re.Pattern[str] | None = _parse_regex(
+            os.environ.get("YIELDSHELL_ALLOW_COMMAND_REGEX", "")
+        )
+        self.redact_env_regex: re.Pattern[str] = _parse_regex_required(
+            os.environ.get("YIELDSHELL_REDACT_ENV_REGEX", ""),
+            r"TOKEN|KEY|SECRET|PASSWORD",
+        )
+
+
+def _parse_pathsep(value: str) -> list[str]:
+    if not value:
+        return []
+    return [p for p in value.split(os.pathsep) if p]
+
+
+def _parse_int(value: str, default: int) -> int:
+    if not value:
+        return default
+    try:
+        return int(value)
+    except ValueError:
+        return default
+
+
+def _parse_regex(value: str) -> re.Pattern[str] | None:
+    if not value:
+        return None
+    return re.compile(value)
+
+
+def _parse_regex_required(value: str, default: str) -> re.Pattern[str]:
+    if not value:
+        return re.compile(default)
+    return re.compile(value)

mcp_yieldshell/process/__init__.py

@@ -0,0 +1 @@
+"""Process management sub-package."""