feat: add pure ssh2 remote operations

This commit is contained in:
云服务部-叶林立
2026-08-21 19:51:43 +08:00
parent d3bf562189
commit 0ac50eb581
62 changed files with 3701 additions and 47 deletions
@@ -0,0 +1,47 @@
import { describe, expect, it } from "vitest";
import {
applyShellDecisionFloor,
SHELL_TOOL_DECISION_FLOOR_PATTERN,
} from "#src/handlers/gates/shell-decision-floor";
import type { PermissionCheckResult, PermissionState } from "#src/types";
function check(state: PermissionState, source: PermissionCheckResult["source"] = "bash") {
return {
state,
toolName: "bash",
source,
origin: source === "session" ? ("session" as const) : ("global" as const),
command: "pwd",
matchedPattern: source === "session" ? "pwd" : "*",
} satisfies PermissionCheckResult;
}
describe("applyShellDecisionFloor", () => {
it("raises allow to ask", () => {
expect(applyShellDecisionFloor(check("allow"), "ask")).toEqual({
...check("allow"),
state: "ask",
source: "bash",
matchedPattern: SHELL_TOOL_DECISION_FLOOR_PATTERN,
reason: undefined,
});
});
it("raises a session allow to ask without retaining the session fast path", () => {
const result = applyShellDecisionFloor(check("allow", "session"), "ask");
expect(result.state).toBe("ask");
expect(result.source).toBe("bash");
expect(result.matchedPattern).toBe(SHELL_TOOL_DECISION_FLOOR_PATTERN);
});
it.each(["ask", "deny"] as const)("preserves an existing %s", (state) => {
const original = check(state);
expect(applyShellDecisionFloor(original, "ask")).toBe(original);
});
it("does nothing when no floor is configured", () => {
const original = check("allow");
expect(applyShellDecisionFloor(original, undefined)).toBe(original);
});
});
@@ -120,6 +120,36 @@ describe("describeToolGate", () => {
expect(desc.payload.request.invokedToolName).toBe("exec_command");
});
it("records a shell alias formatter preview for review and authorization context", () => {
const shell: ShellInvocation = { command: "rm -rf dist", workdir: undefined };
const formatter = new ToolPreviewFormatter(
{
toolInputPreviewMaxLength: TOOL_INPUT_PREVIEW_MAX_LENGTH,
toolTextSummaryMaxLength: TOOL_TEXT_SUMMARY_MAX_LENGTH,
},
{
get: (name) =>
name === "ssh_bash"
? () => "SSH target 'packaging-server' in remote cwd '/srv/build'"
: undefined,
},
);
const desc = describeToolGate(
makeTcc({ toolName: "ssh_bash", input: { command: "rm -rf dist" } }),
makeCheckResult("ask", { toolName: "bash", source: "bash", command: "rm -rf dist" }),
formatter,
undefined,
shell,
);
expect(desc.promptDetails.toolInputPreview).toBe(
"SSH target 'packaging-server' in remote cwd '/srv/build'",
);
expect(desc.logContext.toolInputPreview).toBe(
"SSH target 'packaging-server' in remote cwd '/srv/build'",
);
});
it("returns mcp surface with target in decision.value for MCP tools", () => {
const check = makeCheckResult("ask", {
toolName: "mcp",