@wingman-ai/gateway 0.4.0 → 0.4.2
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +29 -111
- package/dist/agent/config/agentConfig.cjs +2 -0
- package/dist/agent/config/agentConfig.d.ts +6 -0
- package/dist/agent/config/agentConfig.js +2 -0
- package/dist/agent/config/agentLoader.cjs +21 -18
- package/dist/agent/config/agentLoader.js +22 -19
- package/dist/agent/config/toolRegistry.cjs +19 -0
- package/dist/agent/config/toolRegistry.d.ts +4 -0
- package/dist/agent/config/toolRegistry.js +17 -1
- package/dist/agent/middleware/additional-messages.cjs +115 -11
- package/dist/agent/middleware/additional-messages.d.ts +9 -0
- package/dist/agent/middleware/additional-messages.js +115 -11
- package/dist/agent/tests/agentLoader.test.cjs +45 -0
- package/dist/agent/tests/agentLoader.test.js +45 -0
- package/dist/agent/tests/toolRegistry.test.cjs +2 -0
- package/dist/agent/tests/toolRegistry.test.js +2 -0
- package/dist/agent/tools/node_invoke.cjs +146 -0
- package/dist/agent/tools/node_invoke.d.ts +86 -0
- package/dist/agent/tools/node_invoke.js +109 -0
- package/dist/cli/commands/gateway.cjs +1 -1
- package/dist/cli/commands/gateway.js +1 -1
- package/dist/cli/commands/init.cjs +135 -1
- package/dist/cli/commands/init.js +136 -2
- package/dist/cli/commands/skill.cjs +7 -3
- package/dist/cli/commands/skill.js +7 -3
- package/dist/cli/config/loader.cjs +7 -3
- package/dist/cli/config/loader.js +7 -3
- package/dist/cli/config/schema.cjs +27 -9
- package/dist/cli/config/schema.d.ts +18 -4
- package/dist/cli/config/schema.js +23 -8
- package/dist/cli/core/agentInvoker.cjs +70 -14
- package/dist/cli/core/agentInvoker.d.ts +10 -0
- package/dist/cli/core/agentInvoker.js +70 -14
- package/dist/cli/services/skillRepository.cjs +155 -69
- package/dist/cli/services/skillRepository.d.ts +7 -2
- package/dist/cli/services/skillRepository.js +155 -69
- package/dist/cli/services/skillService.cjs +93 -26
- package/dist/cli/services/skillService.d.ts +7 -0
- package/dist/cli/services/skillService.js +96 -29
- package/dist/cli/types/skill.d.ts +8 -3
- package/dist/gateway/http/nodes.cjs +247 -0
- package/dist/gateway/http/nodes.d.ts +20 -0
- package/dist/gateway/http/nodes.js +210 -0
- package/dist/gateway/node.cjs +10 -1
- package/dist/gateway/node.d.ts +10 -1
- package/dist/gateway/node.js +10 -1
- package/dist/gateway/server.cjs +414 -27
- package/dist/gateway/server.d.ts +34 -0
- package/dist/gateway/server.js +408 -27
- package/dist/gateway/types.d.ts +6 -1
- package/dist/gateway/validation.cjs +2 -0
- package/dist/gateway/validation.d.ts +4 -0
- package/dist/gateway/validation.js +2 -0
- package/dist/skills/activation.cjs +92 -0
- package/dist/skills/activation.d.ts +12 -0
- package/dist/skills/activation.js +58 -0
- package/dist/skills/bin-requirements.cjs +63 -0
- package/dist/skills/bin-requirements.d.ts +3 -0
- package/dist/skills/bin-requirements.js +26 -0
- package/dist/skills/metadata.cjs +141 -0
- package/dist/skills/metadata.d.ts +29 -0
- package/dist/skills/metadata.js +104 -0
- package/dist/skills/overlay.cjs +75 -0
- package/dist/skills/overlay.d.ts +2 -0
- package/dist/skills/overlay.js +38 -0
- package/dist/tests/additionalMessageMiddleware.test.cjs +92 -0
- package/dist/tests/additionalMessageMiddleware.test.js +92 -0
- package/dist/tests/cli-config-loader.test.cjs +7 -3
- package/dist/tests/cli-config-loader.test.js +7 -3
- package/dist/tests/cli-init.test.cjs +54 -0
- package/dist/tests/cli-init.test.js +54 -0
- package/dist/tests/config-json-schema.test.cjs +12 -0
- package/dist/tests/config-json-schema.test.js +12 -0
- package/dist/tests/gateway-http-security.test.cjs +277 -0
- package/dist/tests/gateway-http-security.test.d.ts +1 -0
- package/dist/tests/gateway-http-security.test.js +271 -0
- package/dist/tests/gateway-node-mode.test.cjs +174 -0
- package/dist/tests/gateway-node-mode.test.d.ts +1 -0
- package/dist/tests/gateway-node-mode.test.js +168 -0
- package/dist/tests/gateway-origin-policy.test.cjs +60 -0
- package/dist/tests/gateway-origin-policy.test.d.ts +1 -0
- package/dist/tests/gateway-origin-policy.test.js +54 -0
- package/dist/tests/gateway.test.cjs +1 -0
- package/dist/tests/gateway.test.js +1 -0
- package/dist/tests/node-tools.test.cjs +77 -0
- package/dist/tests/node-tools.test.d.ts +1 -0
- package/dist/tests/node-tools.test.js +71 -0
- package/dist/tests/nodes-api.test.cjs +86 -0
- package/dist/tests/nodes-api.test.d.ts +1 -0
- package/dist/tests/nodes-api.test.js +80 -0
- package/dist/tests/skill-activation.test.cjs +86 -0
- package/dist/tests/skill-activation.test.d.ts +1 -0
- package/dist/tests/skill-activation.test.js +80 -0
- package/dist/tests/skill-metadata.test.cjs +119 -0
- package/dist/tests/skill-metadata.test.d.ts +1 -0
- package/dist/tests/skill-metadata.test.js +113 -0
- package/dist/tests/skill-repository.test.cjs +363 -0
- package/dist/tests/skill-repository.test.js +363 -0
- package/dist/webui/assets/{index-DHbfLOUR.js → index-BMekSELC.js} +106 -106
- package/dist/webui/index.html +1 -1
- package/package.json +4 -4
- package/skills/gog/SKILL.md +1 -1
- package/skills/weather/SKILL.md +1 -1
- package/skills/ui-registry/SKILL.md +0 -35
|
@@ -0,0 +1,174 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __webpack_exports__ = {};
|
|
3
|
+
const external_node_fs_namespaceObject = require("node:fs");
|
|
4
|
+
const external_node_os_namespaceObject = require("node:os");
|
|
5
|
+
const external_node_path_namespaceObject = require("node:path");
|
|
6
|
+
const external_vitest_namespaceObject = require("vitest");
|
|
7
|
+
const server_cjs_namespaceObject = require("../gateway/server.cjs");
|
|
8
|
+
const isBun = void 0 !== globalThis.Bun;
|
|
9
|
+
const describeIfBun = isBun ? external_vitest_namespaceObject.describe : external_vitest_namespaceObject.describe.skip;
|
|
10
|
+
describeIfBun("Gateway node enablement", ()=>{
|
|
11
|
+
let server;
|
|
12
|
+
let port = 0;
|
|
13
|
+
let workspace;
|
|
14
|
+
(0, external_vitest_namespaceObject.beforeAll)(async ()=>{
|
|
15
|
+
workspace = (0, external_node_fs_namespaceObject.mkdtempSync)((0, external_node_path_namespaceObject.join)((0, external_node_os_namespaceObject.tmpdir)(), "wingman-gateway-node-mode-"));
|
|
16
|
+
server = new server_cjs_namespaceObject.GatewayServer({
|
|
17
|
+
port: 0,
|
|
18
|
+
host: "localhost",
|
|
19
|
+
requireAuth: false,
|
|
20
|
+
auth: {
|
|
21
|
+
mode: "none"
|
|
22
|
+
},
|
|
23
|
+
logLevel: "silent",
|
|
24
|
+
workspace,
|
|
25
|
+
configDir: ".wingman-node-test-config",
|
|
26
|
+
stateDir: ".wingman-node-test-state"
|
|
27
|
+
});
|
|
28
|
+
await server.start();
|
|
29
|
+
port = server.getPort();
|
|
30
|
+
if (!port) throw new Error("Unable to determine gateway port");
|
|
31
|
+
});
|
|
32
|
+
(0, external_vitest_namespaceObject.afterAll)(async ()=>{
|
|
33
|
+
await server.stop();
|
|
34
|
+
(0, external_node_fs_namespaceObject.rmSync)(workspace, {
|
|
35
|
+
recursive: true,
|
|
36
|
+
force: true
|
|
37
|
+
});
|
|
38
|
+
});
|
|
39
|
+
const connectClient = (instanceId, clientType = "desktop")=>new Promise((resolve, reject)=>{
|
|
40
|
+
const ws = new WebSocket(`ws://localhost:${port}/ws`);
|
|
41
|
+
const connectId = `connect-${instanceId}-${Date.now()}`;
|
|
42
|
+
const timeout = setTimeout(()=>reject(new Error("Connect timeout")), 5000);
|
|
43
|
+
ws.addEventListener("open", ()=>{
|
|
44
|
+
ws.send(JSON.stringify({
|
|
45
|
+
type: "connect",
|
|
46
|
+
id: connectId,
|
|
47
|
+
client: {
|
|
48
|
+
instanceId,
|
|
49
|
+
clientType,
|
|
50
|
+
version: "test"
|
|
51
|
+
},
|
|
52
|
+
timestamp: Date.now()
|
|
53
|
+
}));
|
|
54
|
+
});
|
|
55
|
+
ws.addEventListener("message", (event)=>{
|
|
56
|
+
const msg = JSON.parse(event.data);
|
|
57
|
+
if ("res" === msg.type && msg.id === connectId && msg.ok) {
|
|
58
|
+
clearTimeout(timeout);
|
|
59
|
+
resolve(ws);
|
|
60
|
+
}
|
|
61
|
+
});
|
|
62
|
+
ws.addEventListener("error", ()=>{
|
|
63
|
+
clearTimeout(timeout);
|
|
64
|
+
reject(new Error("WebSocket error"));
|
|
65
|
+
});
|
|
66
|
+
});
|
|
67
|
+
const waitForMessage = (ws, predicate, timeoutMs = 5000)=>new Promise((resolve, reject)=>{
|
|
68
|
+
const timeout = setTimeout(()=>reject(new Error("Message timeout")), timeoutMs);
|
|
69
|
+
const handler = (event)=>{
|
|
70
|
+
const data = event.data;
|
|
71
|
+
if ("string" != typeof data) return;
|
|
72
|
+
let msg;
|
|
73
|
+
try {
|
|
74
|
+
msg = JSON.parse(data);
|
|
75
|
+
} catch {
|
|
76
|
+
return;
|
|
77
|
+
}
|
|
78
|
+
if (!predicate(msg)) return;
|
|
79
|
+
clearTimeout(timeout);
|
|
80
|
+
ws.removeEventListener("message", handler);
|
|
81
|
+
resolve(msg);
|
|
82
|
+
};
|
|
83
|
+
ws.addEventListener("message", handler);
|
|
84
|
+
});
|
|
85
|
+
(0, external_vitest_namespaceObject.it)("blocks node registration before device enablement", async ()=>{
|
|
86
|
+
const ws = await connectClient("desktop-node-blocked");
|
|
87
|
+
ws.send(JSON.stringify({
|
|
88
|
+
type: "register",
|
|
89
|
+
payload: {
|
|
90
|
+
name: "Blocked Node",
|
|
91
|
+
capabilities: [
|
|
92
|
+
"system.notify"
|
|
93
|
+
]
|
|
94
|
+
},
|
|
95
|
+
timestamp: Date.now()
|
|
96
|
+
}));
|
|
97
|
+
const errorMessage = await waitForMessage(ws, (msg)=>"error" === msg.type && msg.payload?.code === "NODE_NOT_ENABLED");
|
|
98
|
+
(0, external_vitest_namespaceObject.expect)(errorMessage.payload?.message).toContain("not approved");
|
|
99
|
+
ws.close();
|
|
100
|
+
});
|
|
101
|
+
(0, external_vitest_namespaceObject.it)("allows enabled devices to register, execute, and revoke", async ()=>{
|
|
102
|
+
const enableResponse = await fetch(`http://localhost:${port}/api/nodes/${encodeURIComponent("desktop-node-enabled")}`, {
|
|
103
|
+
method: "PUT",
|
|
104
|
+
headers: {
|
|
105
|
+
"Content-Type": "application/json"
|
|
106
|
+
},
|
|
107
|
+
body: JSON.stringify({
|
|
108
|
+
enabled: true,
|
|
109
|
+
name: "Enabled Desktop"
|
|
110
|
+
})
|
|
111
|
+
});
|
|
112
|
+
(0, external_vitest_namespaceObject.expect)(enableResponse.ok).toBe(true);
|
|
113
|
+
const nodeWs = await connectClient("desktop-node-enabled");
|
|
114
|
+
nodeWs.send(JSON.stringify({
|
|
115
|
+
type: "register",
|
|
116
|
+
payload: {
|
|
117
|
+
name: "Enabled Desktop",
|
|
118
|
+
capabilities: [
|
|
119
|
+
"system.notify"
|
|
120
|
+
]
|
|
121
|
+
},
|
|
122
|
+
timestamp: Date.now()
|
|
123
|
+
}));
|
|
124
|
+
const registrationAck = await waitForMessage(nodeWs, (msg)=>"ack" === msg.type && "string" == typeof msg.payload?.nodeId);
|
|
125
|
+
const nodeId = registrationAck.payload.nodeId;
|
|
126
|
+
(0, external_vitest_namespaceObject.expect)(nodeId).toBeTruthy();
|
|
127
|
+
const requesterWs = await connectClient("desktop-requester");
|
|
128
|
+
requesterWs.send(JSON.stringify({
|
|
129
|
+
type: "req:node",
|
|
130
|
+
id: "node-req-1",
|
|
131
|
+
targetNodeId: nodeId,
|
|
132
|
+
payload: {
|
|
133
|
+
tool: "system.notify",
|
|
134
|
+
args: {
|
|
135
|
+
title: "Hello",
|
|
136
|
+
body: "Node test"
|
|
137
|
+
}
|
|
138
|
+
},
|
|
139
|
+
timestamp: Date.now()
|
|
140
|
+
}));
|
|
141
|
+
const forwardedToNode = await waitForMessage(nodeWs, (msg)=>"req:node" === msg.type && "node-req-1" === msg.id);
|
|
142
|
+
(0, external_vitest_namespaceObject.expect)(forwardedToNode.targetNodeId).toBe(nodeId);
|
|
143
|
+
nodeWs.send(JSON.stringify({
|
|
144
|
+
type: "res",
|
|
145
|
+
id: "node-req-1",
|
|
146
|
+
ok: true,
|
|
147
|
+
payload: {
|
|
148
|
+
delivered: true
|
|
149
|
+
},
|
|
150
|
+
timestamp: Date.now()
|
|
151
|
+
}));
|
|
152
|
+
const returnedToRequester = await waitForMessage(requesterWs, (msg)=>"res" === msg.type && "node-req-1" === msg.id);
|
|
153
|
+
(0, external_vitest_namespaceObject.expect)(returnedToRequester.ok).toBe(true);
|
|
154
|
+
(0, external_vitest_namespaceObject.expect)(returnedToRequester.nodeId).toBe(nodeId);
|
|
155
|
+
const closePromise = new Promise((resolve)=>{
|
|
156
|
+
nodeWs.addEventListener("close", ()=>{
|
|
157
|
+
resolve();
|
|
158
|
+
}, {
|
|
159
|
+
once: true
|
|
160
|
+
});
|
|
161
|
+
});
|
|
162
|
+
const revokeResponse = await fetch(`http://localhost:${port}/api/nodes/${encodeURIComponent("desktop-node-enabled")}`, {
|
|
163
|
+
method: "DELETE"
|
|
164
|
+
});
|
|
165
|
+
(0, external_vitest_namespaceObject.expect)(revokeResponse.ok).toBe(true);
|
|
166
|
+
await closePromise;
|
|
167
|
+
requesterWs.close();
|
|
168
|
+
nodeWs.close();
|
|
169
|
+
});
|
|
170
|
+
});
|
|
171
|
+
for(var __rspack_i in __webpack_exports__)exports[__rspack_i] = __webpack_exports__[__rspack_i];
|
|
172
|
+
Object.defineProperty(exports, '__esModule', {
|
|
173
|
+
value: true
|
|
174
|
+
});
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
export {};
|
|
@@ -0,0 +1,168 @@
|
|
|
1
|
+
import { mkdtempSync, rmSync } from "node:fs";
|
|
2
|
+
import { tmpdir } from "node:os";
|
|
3
|
+
import { join } from "node:path";
|
|
4
|
+
import { afterAll, beforeAll, describe, expect, it } from "vitest";
|
|
5
|
+
import { GatewayServer } from "../gateway/server.js";
|
|
6
|
+
const isBun = void 0 !== globalThis.Bun;
|
|
7
|
+
const describeIfBun = isBun ? describe : describe.skip;
|
|
8
|
+
describeIfBun("Gateway node enablement", ()=>{
|
|
9
|
+
let server;
|
|
10
|
+
let port = 0;
|
|
11
|
+
let workspace;
|
|
12
|
+
beforeAll(async ()=>{
|
|
13
|
+
workspace = mkdtempSync(join(tmpdir(), "wingman-gateway-node-mode-"));
|
|
14
|
+
server = new GatewayServer({
|
|
15
|
+
port: 0,
|
|
16
|
+
host: "localhost",
|
|
17
|
+
requireAuth: false,
|
|
18
|
+
auth: {
|
|
19
|
+
mode: "none"
|
|
20
|
+
},
|
|
21
|
+
logLevel: "silent",
|
|
22
|
+
workspace,
|
|
23
|
+
configDir: ".wingman-node-test-config",
|
|
24
|
+
stateDir: ".wingman-node-test-state"
|
|
25
|
+
});
|
|
26
|
+
await server.start();
|
|
27
|
+
port = server.getPort();
|
|
28
|
+
if (!port) throw new Error("Unable to determine gateway port");
|
|
29
|
+
});
|
|
30
|
+
afterAll(async ()=>{
|
|
31
|
+
await server.stop();
|
|
32
|
+
rmSync(workspace, {
|
|
33
|
+
recursive: true,
|
|
34
|
+
force: true
|
|
35
|
+
});
|
|
36
|
+
});
|
|
37
|
+
const connectClient = (instanceId, clientType = "desktop")=>new Promise((resolve, reject)=>{
|
|
38
|
+
const ws = new WebSocket(`ws://localhost:${port}/ws`);
|
|
39
|
+
const connectId = `connect-${instanceId}-${Date.now()}`;
|
|
40
|
+
const timeout = setTimeout(()=>reject(new Error("Connect timeout")), 5000);
|
|
41
|
+
ws.addEventListener("open", ()=>{
|
|
42
|
+
ws.send(JSON.stringify({
|
|
43
|
+
type: "connect",
|
|
44
|
+
id: connectId,
|
|
45
|
+
client: {
|
|
46
|
+
instanceId,
|
|
47
|
+
clientType,
|
|
48
|
+
version: "test"
|
|
49
|
+
},
|
|
50
|
+
timestamp: Date.now()
|
|
51
|
+
}));
|
|
52
|
+
});
|
|
53
|
+
ws.addEventListener("message", (event)=>{
|
|
54
|
+
const msg = JSON.parse(event.data);
|
|
55
|
+
if ("res" === msg.type && msg.id === connectId && msg.ok) {
|
|
56
|
+
clearTimeout(timeout);
|
|
57
|
+
resolve(ws);
|
|
58
|
+
}
|
|
59
|
+
});
|
|
60
|
+
ws.addEventListener("error", ()=>{
|
|
61
|
+
clearTimeout(timeout);
|
|
62
|
+
reject(new Error("WebSocket error"));
|
|
63
|
+
});
|
|
64
|
+
});
|
|
65
|
+
const waitForMessage = (ws, predicate, timeoutMs = 5000)=>new Promise((resolve, reject)=>{
|
|
66
|
+
const timeout = setTimeout(()=>reject(new Error("Message timeout")), timeoutMs);
|
|
67
|
+
const handler = (event)=>{
|
|
68
|
+
const data = event.data;
|
|
69
|
+
if ("string" != typeof data) return;
|
|
70
|
+
let msg;
|
|
71
|
+
try {
|
|
72
|
+
msg = JSON.parse(data);
|
|
73
|
+
} catch {
|
|
74
|
+
return;
|
|
75
|
+
}
|
|
76
|
+
if (!predicate(msg)) return;
|
|
77
|
+
clearTimeout(timeout);
|
|
78
|
+
ws.removeEventListener("message", handler);
|
|
79
|
+
resolve(msg);
|
|
80
|
+
};
|
|
81
|
+
ws.addEventListener("message", handler);
|
|
82
|
+
});
|
|
83
|
+
it("blocks node registration before device enablement", async ()=>{
|
|
84
|
+
const ws = await connectClient("desktop-node-blocked");
|
|
85
|
+
ws.send(JSON.stringify({
|
|
86
|
+
type: "register",
|
|
87
|
+
payload: {
|
|
88
|
+
name: "Blocked Node",
|
|
89
|
+
capabilities: [
|
|
90
|
+
"system.notify"
|
|
91
|
+
]
|
|
92
|
+
},
|
|
93
|
+
timestamp: Date.now()
|
|
94
|
+
}));
|
|
95
|
+
const errorMessage = await waitForMessage(ws, (msg)=>"error" === msg.type && msg.payload?.code === "NODE_NOT_ENABLED");
|
|
96
|
+
expect(errorMessage.payload?.message).toContain("not approved");
|
|
97
|
+
ws.close();
|
|
98
|
+
});
|
|
99
|
+
it("allows enabled devices to register, execute, and revoke", async ()=>{
|
|
100
|
+
const enableResponse = await fetch(`http://localhost:${port}/api/nodes/${encodeURIComponent("desktop-node-enabled")}`, {
|
|
101
|
+
method: "PUT",
|
|
102
|
+
headers: {
|
|
103
|
+
"Content-Type": "application/json"
|
|
104
|
+
},
|
|
105
|
+
body: JSON.stringify({
|
|
106
|
+
enabled: true,
|
|
107
|
+
name: "Enabled Desktop"
|
|
108
|
+
})
|
|
109
|
+
});
|
|
110
|
+
expect(enableResponse.ok).toBe(true);
|
|
111
|
+
const nodeWs = await connectClient("desktop-node-enabled");
|
|
112
|
+
nodeWs.send(JSON.stringify({
|
|
113
|
+
type: "register",
|
|
114
|
+
payload: {
|
|
115
|
+
name: "Enabled Desktop",
|
|
116
|
+
capabilities: [
|
|
117
|
+
"system.notify"
|
|
118
|
+
]
|
|
119
|
+
},
|
|
120
|
+
timestamp: Date.now()
|
|
121
|
+
}));
|
|
122
|
+
const registrationAck = await waitForMessage(nodeWs, (msg)=>"ack" === msg.type && "string" == typeof msg.payload?.nodeId);
|
|
123
|
+
const nodeId = registrationAck.payload.nodeId;
|
|
124
|
+
expect(nodeId).toBeTruthy();
|
|
125
|
+
const requesterWs = await connectClient("desktop-requester");
|
|
126
|
+
requesterWs.send(JSON.stringify({
|
|
127
|
+
type: "req:node",
|
|
128
|
+
id: "node-req-1",
|
|
129
|
+
targetNodeId: nodeId,
|
|
130
|
+
payload: {
|
|
131
|
+
tool: "system.notify",
|
|
132
|
+
args: {
|
|
133
|
+
title: "Hello",
|
|
134
|
+
body: "Node test"
|
|
135
|
+
}
|
|
136
|
+
},
|
|
137
|
+
timestamp: Date.now()
|
|
138
|
+
}));
|
|
139
|
+
const forwardedToNode = await waitForMessage(nodeWs, (msg)=>"req:node" === msg.type && "node-req-1" === msg.id);
|
|
140
|
+
expect(forwardedToNode.targetNodeId).toBe(nodeId);
|
|
141
|
+
nodeWs.send(JSON.stringify({
|
|
142
|
+
type: "res",
|
|
143
|
+
id: "node-req-1",
|
|
144
|
+
ok: true,
|
|
145
|
+
payload: {
|
|
146
|
+
delivered: true
|
|
147
|
+
},
|
|
148
|
+
timestamp: Date.now()
|
|
149
|
+
}));
|
|
150
|
+
const returnedToRequester = await waitForMessage(requesterWs, (msg)=>"res" === msg.type && "node-req-1" === msg.id);
|
|
151
|
+
expect(returnedToRequester.ok).toBe(true);
|
|
152
|
+
expect(returnedToRequester.nodeId).toBe(nodeId);
|
|
153
|
+
const closePromise = new Promise((resolve)=>{
|
|
154
|
+
nodeWs.addEventListener("close", ()=>{
|
|
155
|
+
resolve();
|
|
156
|
+
}, {
|
|
157
|
+
once: true
|
|
158
|
+
});
|
|
159
|
+
});
|
|
160
|
+
const revokeResponse = await fetch(`http://localhost:${port}/api/nodes/${encodeURIComponent("desktop-node-enabled")}`, {
|
|
161
|
+
method: "DELETE"
|
|
162
|
+
});
|
|
163
|
+
expect(revokeResponse.ok).toBe(true);
|
|
164
|
+
await closePromise;
|
|
165
|
+
requesterWs.close();
|
|
166
|
+
nodeWs.close();
|
|
167
|
+
});
|
|
168
|
+
});
|
|
@@ -0,0 +1,60 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __webpack_exports__ = {};
|
|
3
|
+
const external_vitest_namespaceObject = require("vitest");
|
|
4
|
+
const server_cjs_namespaceObject = require("../gateway/server.cjs");
|
|
5
|
+
(0, external_vitest_namespaceObject.describe)("gateway origin policy", ()=>{
|
|
6
|
+
(0, external_vitest_namespaceObject.it)("identifies loopback hostnames", ()=>{
|
|
7
|
+
(0, external_vitest_namespaceObject.expect)((0, server_cjs_namespaceObject.isLoopbackHostname)("127.0.0.1")).toBe(true);
|
|
8
|
+
(0, external_vitest_namespaceObject.expect)((0, server_cjs_namespaceObject.isLoopbackHostname)("localhost")).toBe(true);
|
|
9
|
+
(0, external_vitest_namespaceObject.expect)((0, server_cjs_namespaceObject.isLoopbackHostname)("[::1]")).toBe(true);
|
|
10
|
+
(0, external_vitest_namespaceObject.expect)((0, server_cjs_namespaceObject.isLoopbackHostname)("example.com")).toBe(false);
|
|
11
|
+
});
|
|
12
|
+
(0, external_vitest_namespaceObject.it)("allows loopback development origins", ()=>{
|
|
13
|
+
const allowed = (0, server_cjs_namespaceObject.isGatewayOriginAllowed)({
|
|
14
|
+
origin: "http://localhost:5173",
|
|
15
|
+
requestUrl: "http://127.0.0.1:18789/api/sessions",
|
|
16
|
+
gatewayHost: "127.0.0.1",
|
|
17
|
+
gatewayPort: 18789,
|
|
18
|
+
controlUiEnabled: true,
|
|
19
|
+
controlUiPort: 18790
|
|
20
|
+
});
|
|
21
|
+
(0, external_vitest_namespaceObject.expect)(allowed).toBe(true);
|
|
22
|
+
});
|
|
23
|
+
(0, external_vitest_namespaceObject.it)("rejects unrelated internet origins", ()=>{
|
|
24
|
+
const allowed = (0, server_cjs_namespaceObject.isGatewayOriginAllowed)({
|
|
25
|
+
origin: "https://evil.example",
|
|
26
|
+
requestUrl: "http://127.0.0.1:18789/api/sessions",
|
|
27
|
+
gatewayHost: "127.0.0.1",
|
|
28
|
+
gatewayPort: 18789,
|
|
29
|
+
controlUiEnabled: true,
|
|
30
|
+
controlUiPort: 18790
|
|
31
|
+
});
|
|
32
|
+
(0, external_vitest_namespaceObject.expect)(allowed).toBe(false);
|
|
33
|
+
});
|
|
34
|
+
(0, external_vitest_namespaceObject.it)("allows same-host control UI origin on configured port", ()=>{
|
|
35
|
+
const allowed = (0, server_cjs_namespaceObject.isGatewayOriginAllowed)({
|
|
36
|
+
origin: "http://192.168.1.50:18790",
|
|
37
|
+
requestUrl: "http://192.168.1.50:18789/api/sessions",
|
|
38
|
+
gatewayHost: "0.0.0.0",
|
|
39
|
+
gatewayPort: 18789,
|
|
40
|
+
controlUiEnabled: true,
|
|
41
|
+
controlUiPort: 18790
|
|
42
|
+
});
|
|
43
|
+
(0, external_vitest_namespaceObject.expect)(allowed).toBe(true);
|
|
44
|
+
});
|
|
45
|
+
(0, external_vitest_namespaceObject.it)("rejects same-host origins on unapproved ports", ()=>{
|
|
46
|
+
const allowed = (0, server_cjs_namespaceObject.isGatewayOriginAllowed)({
|
|
47
|
+
origin: "http://192.168.1.50:9999",
|
|
48
|
+
requestUrl: "http://192.168.1.50:18789/api/sessions",
|
|
49
|
+
gatewayHost: "0.0.0.0",
|
|
50
|
+
gatewayPort: 18789,
|
|
51
|
+
controlUiEnabled: true,
|
|
52
|
+
controlUiPort: 18790
|
|
53
|
+
});
|
|
54
|
+
(0, external_vitest_namespaceObject.expect)(allowed).toBe(false);
|
|
55
|
+
});
|
|
56
|
+
});
|
|
57
|
+
for(var __rspack_i in __webpack_exports__)exports[__rspack_i] = __webpack_exports__[__rspack_i];
|
|
58
|
+
Object.defineProperty(exports, '__esModule', {
|
|
59
|
+
value: true
|
|
60
|
+
});
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
export {};
|
|
@@ -0,0 +1,54 @@
|
|
|
1
|
+
import { describe, expect, it } from "vitest";
|
|
2
|
+
import { isGatewayOriginAllowed, isLoopbackHostname } from "../gateway/server.js";
|
|
3
|
+
describe("gateway origin policy", ()=>{
|
|
4
|
+
it("identifies loopback hostnames", ()=>{
|
|
5
|
+
expect(isLoopbackHostname("127.0.0.1")).toBe(true);
|
|
6
|
+
expect(isLoopbackHostname("localhost")).toBe(true);
|
|
7
|
+
expect(isLoopbackHostname("[::1]")).toBe(true);
|
|
8
|
+
expect(isLoopbackHostname("example.com")).toBe(false);
|
|
9
|
+
});
|
|
10
|
+
it("allows loopback development origins", ()=>{
|
|
11
|
+
const allowed = isGatewayOriginAllowed({
|
|
12
|
+
origin: "http://localhost:5173",
|
|
13
|
+
requestUrl: "http://127.0.0.1:18789/api/sessions",
|
|
14
|
+
gatewayHost: "127.0.0.1",
|
|
15
|
+
gatewayPort: 18789,
|
|
16
|
+
controlUiEnabled: true,
|
|
17
|
+
controlUiPort: 18790
|
|
18
|
+
});
|
|
19
|
+
expect(allowed).toBe(true);
|
|
20
|
+
});
|
|
21
|
+
it("rejects unrelated internet origins", ()=>{
|
|
22
|
+
const allowed = isGatewayOriginAllowed({
|
|
23
|
+
origin: "https://evil.example",
|
|
24
|
+
requestUrl: "http://127.0.0.1:18789/api/sessions",
|
|
25
|
+
gatewayHost: "127.0.0.1",
|
|
26
|
+
gatewayPort: 18789,
|
|
27
|
+
controlUiEnabled: true,
|
|
28
|
+
controlUiPort: 18790
|
|
29
|
+
});
|
|
30
|
+
expect(allowed).toBe(false);
|
|
31
|
+
});
|
|
32
|
+
it("allows same-host control UI origin on configured port", ()=>{
|
|
33
|
+
const allowed = isGatewayOriginAllowed({
|
|
34
|
+
origin: "http://192.168.1.50:18790",
|
|
35
|
+
requestUrl: "http://192.168.1.50:18789/api/sessions",
|
|
36
|
+
gatewayHost: "0.0.0.0",
|
|
37
|
+
gatewayPort: 18789,
|
|
38
|
+
controlUiEnabled: true,
|
|
39
|
+
controlUiPort: 18790
|
|
40
|
+
});
|
|
41
|
+
expect(allowed).toBe(true);
|
|
42
|
+
});
|
|
43
|
+
it("rejects same-host origins on unapproved ports", ()=>{
|
|
44
|
+
const allowed = isGatewayOriginAllowed({
|
|
45
|
+
origin: "http://192.168.1.50:9999",
|
|
46
|
+
requestUrl: "http://192.168.1.50:18789/api/sessions",
|
|
47
|
+
gatewayHost: "0.0.0.0",
|
|
48
|
+
gatewayPort: 18789,
|
|
49
|
+
controlUiEnabled: true,
|
|
50
|
+
controlUiPort: 18790
|
|
51
|
+
});
|
|
52
|
+
expect(allowed).toBe(false);
|
|
53
|
+
});
|
|
54
|
+
});
|
|
@@ -459,6 +459,7 @@ describeIfBun("Gateway", ()=>{
|
|
|
459
459
|
const lastOptions = invokerConstructOptions.at(-1) || {};
|
|
460
460
|
(0, external_vitest_namespaceObject.expect)(lastOptions.workspace).toBe(workspaceOverride);
|
|
461
461
|
(0, external_vitest_namespaceObject.expect)(lastOptions.configDir).toBe(configDirOverride);
|
|
462
|
+
(0, external_vitest_namespaceObject.expect)(typeof lastOptions.nodeInvoker).toBe("function");
|
|
462
463
|
requester.close();
|
|
463
464
|
});
|
|
464
465
|
(0, external_vitest_namespaceObject.it)("should cancel an in-flight agent request", async ()=>{
|
|
@@ -457,6 +457,7 @@ describeIfBun("Gateway", ()=>{
|
|
|
457
457
|
const lastOptions = invokerConstructOptions.at(-1) || {};
|
|
458
458
|
expect(lastOptions.workspace).toBe(workspaceOverride);
|
|
459
459
|
expect(lastOptions.configDir).toBe(configDirOverride);
|
|
460
|
+
expect(typeof lastOptions.nodeInvoker).toBe("function");
|
|
460
461
|
requester.close();
|
|
461
462
|
});
|
|
462
463
|
it("should cancel an in-flight agent request", async ()=>{
|
|
@@ -0,0 +1,77 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __webpack_exports__ = {};
|
|
3
|
+
const external_vitest_namespaceObject = require("vitest");
|
|
4
|
+
const toolRegistry_cjs_namespaceObject = require("../agent/config/toolRegistry.cjs");
|
|
5
|
+
const node_invoke_cjs_namespaceObject = require("../agent/tools/node_invoke.cjs");
|
|
6
|
+
(0, external_vitest_namespaceObject.describe)("node tools", ()=>{
|
|
7
|
+
(0, external_vitest_namespaceObject.it)("exposes node tools in available tool names", ()=>{
|
|
8
|
+
const tools = (0, toolRegistry_cjs_namespaceObject.getAvailableTools)();
|
|
9
|
+
(0, external_vitest_namespaceObject.expect)(tools).toContain("node_notify");
|
|
10
|
+
(0, external_vitest_namespaceObject.expect)(tools).toContain("node_run");
|
|
11
|
+
});
|
|
12
|
+
(0, external_vitest_namespaceObject.it)("routes node_notify through the injected node invoker", async ()=>{
|
|
13
|
+
const nodeInvoker = external_vitest_namespaceObject.vi.fn(async ()=>({
|
|
14
|
+
nodeId: "node-123",
|
|
15
|
+
payload: {
|
|
16
|
+
delivered: true
|
|
17
|
+
}
|
|
18
|
+
}));
|
|
19
|
+
const tool = (0, node_invoke_cjs_namespaceObject.createNodeNotifyTool)({
|
|
20
|
+
nodeInvoker,
|
|
21
|
+
defaultTargetClientId: "desktop-abc"
|
|
22
|
+
});
|
|
23
|
+
const result = await tool.invoke({
|
|
24
|
+
body: "Build finished"
|
|
25
|
+
});
|
|
26
|
+
(0, external_vitest_namespaceObject.expect)(result.ok).toBe(true);
|
|
27
|
+
(0, external_vitest_namespaceObject.expect)(result.nodeId).toBe("node-123");
|
|
28
|
+
(0, external_vitest_namespaceObject.expect)(result.delivered).toBe(true);
|
|
29
|
+
(0, external_vitest_namespaceObject.expect)(nodeInvoker).toHaveBeenCalledWith(external_vitest_namespaceObject.expect.objectContaining({
|
|
30
|
+
tool: "system.notify",
|
|
31
|
+
targetClientId: "desktop-abc"
|
|
32
|
+
}));
|
|
33
|
+
});
|
|
34
|
+
(0, external_vitest_namespaceObject.it)("returns command result fields from node_run", async ()=>{
|
|
35
|
+
const nodeInvoker = external_vitest_namespaceObject.vi.fn(async ()=>({
|
|
36
|
+
nodeId: "node-789",
|
|
37
|
+
payload: {
|
|
38
|
+
exitCode: 0,
|
|
39
|
+
stdout: "ok",
|
|
40
|
+
stderr: ""
|
|
41
|
+
}
|
|
42
|
+
}));
|
|
43
|
+
const tool = (0, node_invoke_cjs_namespaceObject.createNodeRunTool)({
|
|
44
|
+
nodeInvoker
|
|
45
|
+
});
|
|
46
|
+
const result = await tool.invoke({
|
|
47
|
+
command: "/bin/echo",
|
|
48
|
+
args: [
|
|
49
|
+
"hello"
|
|
50
|
+
],
|
|
51
|
+
target: {
|
|
52
|
+
nodeId: "node-789"
|
|
53
|
+
}
|
|
54
|
+
});
|
|
55
|
+
(0, external_vitest_namespaceObject.expect)(result.ok).toBe(true);
|
|
56
|
+
(0, external_vitest_namespaceObject.expect)(result.nodeId).toBe("node-789");
|
|
57
|
+
(0, external_vitest_namespaceObject.expect)(result.exitCode).toBe(0);
|
|
58
|
+
(0, external_vitest_namespaceObject.expect)(result.stdout).toBe("ok");
|
|
59
|
+
(0, external_vitest_namespaceObject.expect)(result.stderr).toBe("");
|
|
60
|
+
(0, external_vitest_namespaceObject.expect)(nodeInvoker).toHaveBeenCalledWith(external_vitest_namespaceObject.expect.objectContaining({
|
|
61
|
+
tool: "system.run",
|
|
62
|
+
targetNodeId: "node-789"
|
|
63
|
+
}));
|
|
64
|
+
});
|
|
65
|
+
(0, external_vitest_namespaceObject.it)("returns an availability error outside gateway runtime", async ()=>{
|
|
66
|
+
const tool = (0, node_invoke_cjs_namespaceObject.createNodeRunTool)();
|
|
67
|
+
const result = await tool.invoke({
|
|
68
|
+
command: "pwd"
|
|
69
|
+
});
|
|
70
|
+
(0, external_vitest_namespaceObject.expect)(result.ok).toBe(false);
|
|
71
|
+
(0, external_vitest_namespaceObject.expect)(result.error).toContain("only available");
|
|
72
|
+
});
|
|
73
|
+
});
|
|
74
|
+
for(var __rspack_i in __webpack_exports__)exports[__rspack_i] = __webpack_exports__[__rspack_i];
|
|
75
|
+
Object.defineProperty(exports, '__esModule', {
|
|
76
|
+
value: true
|
|
77
|
+
});
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
export {};
|
|
@@ -0,0 +1,71 @@
|
|
|
1
|
+
import { describe, expect, it, vi } from "vitest";
|
|
2
|
+
import { getAvailableTools } from "../agent/config/toolRegistry.js";
|
|
3
|
+
import { createNodeNotifyTool, createNodeRunTool } from "../agent/tools/node_invoke.js";
|
|
4
|
+
describe("node tools", ()=>{
|
|
5
|
+
it("exposes node tools in available tool names", ()=>{
|
|
6
|
+
const tools = getAvailableTools();
|
|
7
|
+
expect(tools).toContain("node_notify");
|
|
8
|
+
expect(tools).toContain("node_run");
|
|
9
|
+
});
|
|
10
|
+
it("routes node_notify through the injected node invoker", async ()=>{
|
|
11
|
+
const nodeInvoker = vi.fn(async ()=>({
|
|
12
|
+
nodeId: "node-123",
|
|
13
|
+
payload: {
|
|
14
|
+
delivered: true
|
|
15
|
+
}
|
|
16
|
+
}));
|
|
17
|
+
const tool = createNodeNotifyTool({
|
|
18
|
+
nodeInvoker,
|
|
19
|
+
defaultTargetClientId: "desktop-abc"
|
|
20
|
+
});
|
|
21
|
+
const result = await tool.invoke({
|
|
22
|
+
body: "Build finished"
|
|
23
|
+
});
|
|
24
|
+
expect(result.ok).toBe(true);
|
|
25
|
+
expect(result.nodeId).toBe("node-123");
|
|
26
|
+
expect(result.delivered).toBe(true);
|
|
27
|
+
expect(nodeInvoker).toHaveBeenCalledWith(expect.objectContaining({
|
|
28
|
+
tool: "system.notify",
|
|
29
|
+
targetClientId: "desktop-abc"
|
|
30
|
+
}));
|
|
31
|
+
});
|
|
32
|
+
it("returns command result fields from node_run", async ()=>{
|
|
33
|
+
const nodeInvoker = vi.fn(async ()=>({
|
|
34
|
+
nodeId: "node-789",
|
|
35
|
+
payload: {
|
|
36
|
+
exitCode: 0,
|
|
37
|
+
stdout: "ok",
|
|
38
|
+
stderr: ""
|
|
39
|
+
}
|
|
40
|
+
}));
|
|
41
|
+
const tool = createNodeRunTool({
|
|
42
|
+
nodeInvoker
|
|
43
|
+
});
|
|
44
|
+
const result = await tool.invoke({
|
|
45
|
+
command: "/bin/echo",
|
|
46
|
+
args: [
|
|
47
|
+
"hello"
|
|
48
|
+
],
|
|
49
|
+
target: {
|
|
50
|
+
nodeId: "node-789"
|
|
51
|
+
}
|
|
52
|
+
});
|
|
53
|
+
expect(result.ok).toBe(true);
|
|
54
|
+
expect(result.nodeId).toBe("node-789");
|
|
55
|
+
expect(result.exitCode).toBe(0);
|
|
56
|
+
expect(result.stdout).toBe("ok");
|
|
57
|
+
expect(result.stderr).toBe("");
|
|
58
|
+
expect(nodeInvoker).toHaveBeenCalledWith(expect.objectContaining({
|
|
59
|
+
tool: "system.run",
|
|
60
|
+
targetNodeId: "node-789"
|
|
61
|
+
}));
|
|
62
|
+
});
|
|
63
|
+
it("returns an availability error outside gateway runtime", async ()=>{
|
|
64
|
+
const tool = createNodeRunTool();
|
|
65
|
+
const result = await tool.invoke({
|
|
66
|
+
command: "pwd"
|
|
67
|
+
});
|
|
68
|
+
expect(result.ok).toBe(false);
|
|
69
|
+
expect(result.error).toContain("only available");
|
|
70
|
+
});
|
|
71
|
+
});
|