@neoline/hostbridge 0.2.0 → 0.2.2

package/README.md

@@ -121,9 +121,10 @@ Recommended agent flow:
 
 1. Call `hosts_list`.
 2. If the target host exists, call `session_open`.
-3. If it does not exist and the user identifies a similar bastion-backed host, run `hostbridge setup <new_id> --from-host <source_id> --menu-selection <ip-or-name> --final-prompt <regex>`.
-4. Call `hosts_list` again, then `session_open(<new_id>)`.
-5. Do not inspect `~/.ssh`, old expect scripts, or shell snapshots unless the user explicitly asks or the setup command lacks required information.
+3. If `session_open` times out or fails, run `hostbridge check <host_id>` to verify the recorded login path, then retry or repair setup.
+4. If it does not exist and the user identifies a similar bastion-backed host, run `hostbridge setup <new_id> --from-host <source_id> --menu-selection <ip-or-name> --final-prompt <regex>`.
+5. Call `hosts_list` again, then `session_open(<new_id>)`.
+6. Do not inspect `~/.ssh`, old expect scripts, or shell snapshots unless the user explicitly asks or the setup command lacks required information.
 
 ### Encrypted login automation
 
@@ -327,11 +328,12 @@ See `SECURITY.md` and `docs/THREAT_MODEL.md` for the trust model and audited eve
 1. Call `hosts_list` and choose a configured host id.
 2. If the host is missing, ask for only the missing fields and use `hostbridge setup <host_id>` rather than manually editing config.
 3. Call `session_open` once for the chosen host.
-4. Use `command_run` for quick commands such as `nvidia-smi`, `pwd`, `df -h`, or `tmux ls`.
-5. Use `task_start` for long training jobs, downloads, builds, or scripts.
-6. Poll `task_status` until `state` is `finished`; inspect `exit_code` and `log_tail`.
-7. Call `task_stop` early if a job should be aborted.
-8. Call `session_close` when finished, or `sessions_close_all` to tear down every session.
+4. Only if `session_open` times out or fails, run `hostbridge check <host_id>` and retry after fixing the login path.
+5. Use `command_run` for quick commands such as `nvidia-smi`, `pwd`, `df -h`, or `tmux ls`.
+6. Use `task_start` for long training jobs, downloads, builds, or scripts.
+7. Poll `task_status` until `state` is `finished`; inspect `exit_code` and `log_tail`.
+8. Call `task_stop` early if a job should be aborted.
+9. Call `session_close` when finished, or `sessions_close_all` to tear down every session.
 
 ## Security notes
 
@@ -347,4 +349,4 @@ See `CONTRIBUTING.md`. Run tests with `python -m pytest -q`. The full suite shou
 
 ## License
 
-MIT. See `LICENSE`.
+MIT. See `LICENSE`.

package/bin/hostbridge.js

@@ -19,6 +19,8 @@ if (!existsSync(moduleDir)) {
 
 const env = { ...process.env };
 env.PYTHONPATH = env.PYTHONPATH ? `${bundledSrc}${delimiter}${env.PYTHONPATH}` : bundledSrc;
+env.HOSTBRIDGE_DEFAULT_MCP_COMMAND = env.HOSTBRIDGE_DEFAULT_MCP_COMMAND || 'npx';
+env.HOSTBRIDGE_DEFAULT_MCP_ARGS = env.HOSTBRIDGE_DEFAULT_MCP_ARGS || JSON.stringify(['-y', '@neoline/hostbridge']);
 
 const child = spawn(pythonPath, ['-m', 'server_control_mcp', ...process.argv.slice(2)], {
   stdio: 'inherit',

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@neoline/hostbridge",
-  "version": "0.2.0",
+  "version": "0.2.2",
   "description": "HostBridge MCP server for persistent allowlisted SSH and shell sessions.",
   "type": "module",
   "license": "MIT",

package/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "hostbridge"
-version = "0.2.0"
+version = "0.2.2"
 description = "HostBridge MCP server for persistent allowlisted SSH and shell sessions."
 readme = "README.md"
 requires-python = ">=3.11"

package/src/server_control_mcp/__init__.py

@@ -6,7 +6,7 @@ import sys
 from typing import NoReturn
 
 __all__ = ["__version__", "ensure_runtime"]
-__version__ = "0.2.0"
+__version__ = "0.2.2"
 
 # (import_name, pip_name) pairs for runtime dependency probing.
 _REQUIRED_DEPENDENCIES = (

package/src/server_control_mcp/cli.py

@@ -35,7 +35,7 @@ Use HostBridge when the user asks to connect to, inspect, or operate an allowlis
 
 - First call `hosts_list` to see configured hosts; then use `session_open("<host_id>")`.
 - If the host is missing, ask only for the missing connection details and run `hostbridge setup <host_id>` instead of editing JSON by hand.
-- Validate host entries with `hostbridge check <host_id>` before relying on them.
+- Do not run `hostbridge check` before every connection; only use `hostbridge check <host_id>` when `session_open` times out or fails.
 - Do not search shell history, SSH config, or old scripts unless the setup flow lacks required information or the user explicitly asks.
 - Keep long-running work in `task_start`/`task_status`; use `session_close` or `sessions_close_all` when done.
 {HOSTBRIDGE_GUIDANCE_END}
@@ -228,12 +228,23 @@ def _looks_like_shell_script(command: str, args: list[str]) -> bool:
 
 
 def _default_mcp_command_args() -> tuple[str, list[str]]:
+    env_command = os.environ.get("HOSTBRIDGE_DEFAULT_MCP_COMMAND")
+    if env_command:
+        raw_args = os.environ.get("HOSTBRIDGE_DEFAULT_MCP_ARGS", "[]")
+        try:
+            parsed_args = json.loads(raw_args)
+        except json.JSONDecodeError:
+            parsed_args = []
+        if not isinstance(parsed_args, list) or not all(isinstance(arg, str) for arg in parsed_args):
+            parsed_args = []
+        return env_command, parsed_args
+
     repo_root = Path(__file__).resolve().parents[2]
     run_server = repo_root / "run_server.py"
     if run_server.exists():
         return "python3", [str(run_server)]
     command = Path(sys.argv[0]).name or PUBLIC_NAME
-    if command in (LEGACY_PUBLIC_NAME, LEGACY_PACKAGE_NAME):
+    if command in (LEGACY_PUBLIC_NAME, LEGACY_PACKAGE_NAME, "__main__.py"):
         command = PUBLIC_NAME
     return command, []