@sleep2agi/commhub-server 0.9.0-preview.7 → 0.9.0-preview.71
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 +130 -10
- package/bin/commhub.ts +10 -2
- package/package.json +5 -5
- package/src/ack-create-request-transport.test.ts +159 -0
- package/src/ack-create-request.test.ts +331 -0
- package/src/admin-delete-network-http.test.ts +177 -0
- package/src/admin-networks-http.test.ts +88 -0
- package/src/agent-access.ts +461 -0
- package/src/agent-acl-groups-http.test.ts +247 -0
- package/src/agent-acl-http.test.ts +510 -0
- package/src/agent-acl-slice3-http.test.ts +267 -0
- package/src/agent-groups-perf.test.ts +167 -0
- package/src/alias-filter.test.ts +72 -0
- package/src/alias-filter.ts +32 -0
- package/src/api-host-supervisors-can-create-nodes.test.ts +350 -0
- package/src/api-host-supervisors-fallback.test.ts +271 -0
- package/src/api-nodes-shape.test.ts +26 -10
- package/src/auth-sessions-http.test.ts +250 -0
- package/src/auth-validate.test.ts +5 -1
- package/src/auth.ts +423 -29
- package/src/avatar-validate.test.ts +163 -0
- package/src/avatar-validate.ts +103 -0
- package/src/blocked-exit-on-reply.test.ts +122 -0
- package/src/boot-side-effect.test.ts +124 -0
- package/src/config-apply-sec1.test.ts +107 -0
- package/src/config-apply-validate.test.ts +92 -0
- package/src/config-apply-validate.ts +81 -3
- package/src/create-node-blocked-gate.test.ts +356 -0
- package/src/create-node-tool-schema.test.ts +39 -0
- package/src/create-node-validate.test.ts +89 -1
- package/src/create-node-validate.ts +89 -7
- package/src/create-node-workdir.test.ts +255 -0
- package/src/create-node.ts +51 -1
- package/src/dashboard-slash-routing-http.test.ts +107 -0
- package/src/db-adapter-guard.test.ts +85 -0
- package/src/db-adapter-translate.test.ts +112 -0
- package/src/db-adapter.ts +246 -83
- package/src/db-timestamp.test.ts +38 -0
- package/src/db-timestamp.ts +15 -0
- package/src/db.ts +799 -33
- package/src/delivered-without-push.test.ts +140 -0
- package/src/desktop-message-seam.test.ts +150 -0
- package/src/desktop-message.test.ts +368 -0
- package/src/ext-token-shared.test.ts +60 -0
- package/src/ext-validation-regression.test.ts +131 -0
- package/src/external-schedule-edits-http.test.ts +321 -0
- package/src/external-schedule-edits-timestamps.test.ts +64 -0
- package/src/external-schedule-edits.ts +327 -0
- package/src/file-download-authz-dev-open.test.ts +135 -0
- package/src/file-download-authz.test.ts +404 -0
- package/src/file-download-date-bucket.test.ts +436 -0
- package/src/file-network-scope-dev-open.test.ts +97 -0
- package/src/file-network-scope.test.ts +732 -0
- package/src/health-redaction.test.ts +642 -0
- package/src/http-gzip.test.ts +73 -0
- package/src/http-gzip.ts +60 -0
- package/src/hub-timestamp.test.ts +38 -0
- package/src/hub-timestamp.ts +27 -0
- package/src/human-dm.ts +149 -0
- package/src/inbox-count-sse.test.ts +310 -0
- package/src/inbox-count.ts +29 -0
- package/src/index.ts +15 -2331
- package/src/lifecycle-guard.ts +50 -0
- package/src/mcp-write-network-resolution.test.ts +429 -0
- package/src/messages-ack-by-agent-http.test.ts +142 -0
- package/src/network-name-http.test.ts +75 -0
- package/src/network-scope.ts +224 -0
- package/src/node-attrs-validate.ts +88 -0
- package/src/node-attrs.test.ts +217 -0
- package/src/node-create-request-status-http.test.ts +92 -0
- package/src/node-files-transport.test.ts +344 -0
- package/src/node-id-alias.test.ts +188 -0
- package/src/node-lifecycle-controllable.test.ts +64 -0
- package/src/node-lifecycle-controllable.ts +51 -0
- package/src/node-logs-transport.test.ts +263 -0
- package/src/node-request-retention.test.ts +281 -0
- package/src/node-request-retention.ts +120 -0
- package/src/node-skills-transport.test.ts +209 -0
- package/src/node-token-privilege-http.test.ts +149 -0
- package/src/observer-avatar-http.test.ts +429 -0
- package/src/observer-push.test.ts +252 -0
- package/src/package-version-consistency.test.ts +25 -0
- package/src/peer-reply-atomic.test.ts +455 -0
- package/src/peer-reply-capability.test.ts +70 -0
- package/src/pg-worker.ts +87 -0
- package/src/push.ts +204 -7
- package/src/redact-tokens.test.ts +65 -0
- package/src/redact-tokens.ts +40 -0
- package/src/rename.ts +7 -2
- package/src/reply-target-derivation.test.ts +96 -0
- package/src/report-status-host-schema.test.ts +211 -0
- package/src/report-status-resume-id-handover.test.ts +150 -0
- package/src/require-explicit-test-db.ts +9 -0
- package/src/requirement-tags.ts +16 -0
- package/src/requirements-etag-http.test.ts +91 -0
- package/src/requirements-http.test.ts +708 -0
- package/src/requirements-mcp.test.ts +142 -0
- package/src/requirements-migrate.ts +83 -0
- package/src/requirements-priority-lowest.test.ts +114 -0
- package/src/requirements-start-date.test.ts +125 -0
- package/src/requirements-upgrade-e2e.test.ts +126 -0
- package/src/requirements-upgrade-migration.test.ts +85 -0
- package/src/requirements.ts +899 -0
- package/src/resolve-port.test.ts +57 -0
- package/src/resolve-port.ts +44 -0
- package/src/response-charset.test.ts +130 -0
- package/src/rest-explicit-columns-http.test.ts +216 -0
- package/src/rest-identity.test.ts +129 -0
- package/src/rest-identity.ts +100 -0
- package/src/rest-projections.ts +68 -0
- package/src/rest-write-network-resolution.test.ts +133 -0
- package/src/restricted-files.ts +73 -0
- package/src/retention.ts +10 -1
- package/src/rules-file-duplicate-alias.test.ts +234 -0
- package/src/rules-file-session-target.test.ts +207 -0
- package/src/rules-file-transport.test.ts +237 -0
- package/src/runtime-label.test.ts +22 -0
- package/src/runtime-label.ts +21 -0
- package/src/scheduled-doorbell-nodeless-session.test.ts +130 -0
- package/src/scheduled-run-terminal.test.ts +268 -0
- package/src/scheduled-skip-blocking-task.test.ts +120 -0
- package/src/scheduled-tasks-http.test.ts +625 -0
- package/src/scheduled-tasks.ts +597 -0
- package/src/send-dedup-request-id.test.ts +229 -0
- package/src/send-reply-agent-warning.test.ts +202 -0
- package/src/send-reply-alias-optional.test.ts +131 -0
- package/src/send-reply-attachments.test.ts +458 -0
- package/src/send_dedup.test.ts +54 -0
- package/src/send_dedup.ts +52 -6
- package/src/serve-error.test.ts +38 -0
- package/src/serve-error.ts +27 -0
- package/src/server.ts +4385 -0
- package/src/shared/external-schedule-contract.ts +60 -0
- package/src/side-thread-command-transport.test.ts +302 -0
- package/src/side-thread-command-transport.ts +318 -0
- package/src/side-thread-http-integration.test.ts +233 -0
- package/src/side-thread-http.ts +504 -0
- package/src/side-thread-production.test.ts +72 -0
- package/src/side-thread-production.ts +118 -0
- package/src/side-thread.test.ts +1067 -0
- package/src/side-thread.ts +2495 -0
- package/src/skillhub-http.test.ts +92 -0
- package/src/start-node.test.ts +109 -0
- package/src/status-alias-filter-http.test.ts +106 -0
- package/src/stop-delete-node.test.ts +1164 -0
- package/src/task-auth-origin-http.test.ts +91 -0
- package/src/task-auth-origin.test.ts +24 -0
- package/src/task-auth-origin.ts +7 -0
- package/src/task-consumption.test.ts +482 -0
- package/src/task-diagnostic-http.test.ts +149 -0
- package/src/task-diagnostic.test.ts +115 -0
- package/src/task-diagnostic.ts +118 -0
- package/src/task-idempotency-mcp.test.ts +142 -0
- package/src/task-idempotency.test.ts +60 -0
- package/src/task-idempotency.ts +75 -0
- package/src/task-lifecycle-watcher.test.ts +204 -0
- package/src/task-lifecycle-watcher.ts +49 -0
- package/src/task-network-resolution-http.test.ts +141 -0
- package/src/tasks-pagination-http.test.ts +141 -0
- package/src/tools-list-json-schema.test.ts +66 -0
- package/src/tools.ts +3446 -257
- package/src/update-provider.test.ts +65 -0
- package/src/upload-early-reject-guard.test.ts +155 -0
- package/src/uploads-http.test.ts +97 -15
- package/src/uploads.test.ts +66 -0
- package/src/uploads.ts +130 -5
- package/src/user-inbox-by-agent-http.test.ts +161 -0
- package/src/user-inbox-netid-notnull.test.ts +122 -0
- package/src/user-inbox-read-http.test.ts +223 -0
- package/src/user-inbox-schema.test.ts +54 -0
|
@@ -0,0 +1,207 @@
|
|
|
1
|
+
// app#225 follow-up — rules-file requests to a SESSION that has no `nodes` row
|
|
2
|
+
// (claude-code sessions launched by `anet node start` + the channel server).
|
|
3
|
+
//
|
|
4
|
+
// Hub-side contract verified here:
|
|
5
|
+
// 1. report_status { rules_file_capable: true } marks the session capable only
|
|
6
|
+
// when the caller is a node token bound to that same alias (a user token or
|
|
7
|
+
// another node's token cannot mark it); omitting the field later does not clear it.
|
|
8
|
+
// 2. read/write_node_rules_file { alias } reach such a session: the node pulls
|
|
9
|
+
// with its own token (queue key `session:<alias>`), acks, the client polls done.
|
|
10
|
+
// 3. An alias that has neither a node row nor a capable session is refused up
|
|
11
|
+
// front (no request row, no 60 s timeout for the user to sit through).
|
|
12
|
+
// 4. Network isolation: a same-named alias in another network neither sees nor
|
|
13
|
+
// acks the request; a caller in the other network cannot enqueue to it.
|
|
14
|
+
// 5. The original node_id path is unchanged (node row wins when both exist).
|
|
15
|
+
//
|
|
16
|
+
// 跑法:cd server && COMMHUB_DB=/tmp/rules-session.db bun test src/rules-file-session-target.test.ts
|
|
17
|
+
import { afterAll, beforeEach, describe, expect, test } from "bun:test";
|
|
18
|
+
import { Client } from "@modelcontextprotocol/sdk/client/index.js";
|
|
19
|
+
import { InMemoryTransport } from "@modelcontextprotocol/sdk/inMemory.js";
|
|
20
|
+
import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
|
|
21
|
+
import { db } from "./db.js";
|
|
22
|
+
import { registerTools } from "./tools.js";
|
|
23
|
+
|
|
24
|
+
const NET = "net_rules_session";
|
|
25
|
+
const OTHER_NET = "net_rules_session_other";
|
|
26
|
+
const USER = "u_rules_session_owner";
|
|
27
|
+
const SESSION_ALIAS = "rules-session-cc";
|
|
28
|
+
const SESSION_RESUME = "cc-rules-session-1";
|
|
29
|
+
const OTHER_ALIAS = "rules-session-peer";
|
|
30
|
+
const NODE_ROW_ALIAS = "rules-session-noderow";
|
|
31
|
+
const NODE_ROW_ID = "node_rules_session_noderow";
|
|
32
|
+
const T_SESSION = "tok_rules_session_cc";
|
|
33
|
+
const T_PEER = "tok_rules_session_peer";
|
|
34
|
+
const T_OTHER_NET_SAME_ALIAS = "tok_rules_session_other_net";
|
|
35
|
+
const T_NODE_ROW = "tok_rules_session_noderow";
|
|
36
|
+
|
|
37
|
+
function cleanup() {
|
|
38
|
+
for (const net of [NET, OTHER_NET]) {
|
|
39
|
+
try { db.run("DELETE FROM node_rules_requests WHERE network_id = ?1", [net]); } catch {}
|
|
40
|
+
try { db.run("DELETE FROM sessions WHERE network_id = ?1", [net]); } catch {}
|
|
41
|
+
try { db.run("DELETE FROM nodes WHERE network_id = ?1", [net]); } catch {}
|
|
42
|
+
try { db.run("DELETE FROM api_tokens WHERE network_id = ?1", [net]); } catch {}
|
|
43
|
+
try { db.run("DELETE FROM network_members WHERE network_id = ?1", [net]); } catch {}
|
|
44
|
+
try { db.run("DELETE FROM networks WHERE network_id = ?1", [net]); } catch {}
|
|
45
|
+
}
|
|
46
|
+
try { db.run("DELETE FROM users WHERE user_id = ?1", [USER]); } catch {}
|
|
47
|
+
}
|
|
48
|
+
|
|
49
|
+
function seedWorld() {
|
|
50
|
+
db.run(`INSERT INTO users (user_id, username, password_hash, role, created_at) VALUES (?1, ?2, 'x', 'user', datetime('now'))`, [USER, USER]);
|
|
51
|
+
for (const net of [NET, OTHER_NET]) {
|
|
52
|
+
db.run(`INSERT INTO networks (network_id, network_name, owner_id, created_at) VALUES (?1, ?2, ?3, datetime('now'))`, [net, net, USER]);
|
|
53
|
+
db.run(`INSERT INTO network_members (user_id, network_id, role, joined_at) VALUES (?1, ?2, 'owner', datetime('now'))`, [USER, net]);
|
|
54
|
+
}
|
|
55
|
+
const tok = (id: string, net: string, alias: string) => db.run(
|
|
56
|
+
`INSERT INTO api_tokens (token_id, user_id, network_id, scope, name, token_hash, expires_at, revoked_at) VALUES (?1, ?2, ?3, 'network', ?4, ?5, NULL, NULL)`,
|
|
57
|
+
[id, USER, net, `node:${alias}`, `hash_${id}`],
|
|
58
|
+
);
|
|
59
|
+
tok(T_SESSION, NET, SESSION_ALIAS);
|
|
60
|
+
tok(T_PEER, NET, OTHER_ALIAS);
|
|
61
|
+
tok(T_OTHER_NET_SAME_ALIAS, OTHER_NET, SESSION_ALIAS);
|
|
62
|
+
tok(T_NODE_ROW, NET, NODE_ROW_ALIAS);
|
|
63
|
+
db.run(
|
|
64
|
+
`INSERT INTO nodes (node_id, node_name, alias, network_id, created_at, updated_at) VALUES (?1, ?2, ?3, ?4, datetime('now'), datetime('now'))`,
|
|
65
|
+
[NODE_ROW_ID, NODE_ROW_ALIAS, NODE_ROW_ALIAS, NET],
|
|
66
|
+
);
|
|
67
|
+
}
|
|
68
|
+
|
|
69
|
+
type Identity = { net: string; alias: string; isNetworkToken: boolean; tokenId: string };
|
|
70
|
+
const asUser: Identity = { net: NET, alias: USER, isNetworkToken: false, tokenId: "tok_rules_session_user" };
|
|
71
|
+
const asOtherNetUser: Identity = { net: OTHER_NET, alias: USER, isNetworkToken: false, tokenId: "tok_rules_session_user2" };
|
|
72
|
+
const asSession: Identity = { net: NET, alias: SESSION_ALIAS, isNetworkToken: true, tokenId: T_SESSION };
|
|
73
|
+
const asPeer: Identity = { net: NET, alias: OTHER_ALIAS, isNetworkToken: true, tokenId: T_PEER };
|
|
74
|
+
const asOtherNetSameAlias: Identity = { net: OTHER_NET, alias: SESSION_ALIAS, isNetworkToken: true, tokenId: T_OTHER_NET_SAME_ALIAS };
|
|
75
|
+
const asNodeRow: Identity = { net: NET, alias: NODE_ROW_ALIAS, isNetworkToken: true, tokenId: T_NODE_ROW };
|
|
76
|
+
|
|
77
|
+
async function connect(id: Identity) {
|
|
78
|
+
const server = new McpServer({ name: "rules-session-test", version: "1" });
|
|
79
|
+
registerTools(server, undefined, id.net, USER, id.alias, id.isNetworkToken, id.tokenId);
|
|
80
|
+
const client = new Client({ name: "rules-session-client", version: "1" });
|
|
81
|
+
const [ct, st] = InMemoryTransport.createLinkedPair();
|
|
82
|
+
await server.connect(st);
|
|
83
|
+
await client.connect(ct);
|
|
84
|
+
const call = async (name: string, args: Record<string, unknown>) => {
|
|
85
|
+
const r: any = await client.callTool({ name, arguments: args });
|
|
86
|
+
const first = r.content?.[0];
|
|
87
|
+
if (!first || first.type !== "text") throw new Error(`no text result from ${name}: ${JSON.stringify(r).slice(0, 300)}`);
|
|
88
|
+
return JSON.parse(first.text);
|
|
89
|
+
};
|
|
90
|
+
const close = async () => { await client.close(); await server.close(); };
|
|
91
|
+
return { call, close };
|
|
92
|
+
}
|
|
93
|
+
|
|
94
|
+
const capable = (alias: string, net = NET) =>
|
|
95
|
+
db.get<{ c: number }>("SELECT rules_file_capable AS c FROM sessions WHERE alias = ?1 AND network_id = ?2", alias, net)?.c;
|
|
96
|
+
|
|
97
|
+
async function registerSession(id: Identity, alias: string, resume: string, flag: boolean) {
|
|
98
|
+
const c = await connect(id);
|
|
99
|
+
try {
|
|
100
|
+
return await c.call("report_status", { resume_id: resume, alias, status: "idle", agent: "claude-code", ...(flag ? { rules_file_capable: true } : {}) });
|
|
101
|
+
} finally { await c.close(); }
|
|
102
|
+
}
|
|
103
|
+
|
|
104
|
+
beforeEach(() => { cleanup(); seedWorld(); });
|
|
105
|
+
afterAll(cleanup);
|
|
106
|
+
|
|
107
|
+
describe("rules_file_capable is set only by the session's own node token", () => {
|
|
108
|
+
test("bound node token sets it; a later report without the field does not clear it", async () => {
|
|
109
|
+
await registerSession(asSession, SESSION_ALIAS, SESSION_RESUME, true);
|
|
110
|
+
expect(capable(SESSION_ALIAS)).toBe(1);
|
|
111
|
+
await registerSession(asSession, SESSION_ALIAS, SESSION_RESUME, false);
|
|
112
|
+
expect(capable(SESSION_ALIAS)).toBe(1);
|
|
113
|
+
}, 20_000);
|
|
114
|
+
|
|
115
|
+
// The hub already refuses these reports outright (network_token_required /
|
|
116
|
+
// alias_identity_mismatch); the bound-token check on the UPDATE is defence in
|
|
117
|
+
// depth. The real session row is registered first so "stays 0" is not vacuous.
|
|
118
|
+
test("a user token reporting for the alias is refused and cannot mark it capable", async () => {
|
|
119
|
+
await registerSession(asSession, SESSION_ALIAS, SESSION_RESUME, false);
|
|
120
|
+
expect(capable(SESSION_ALIAS)).toBe(0);
|
|
121
|
+
const r = await registerSession(asUser, SESSION_ALIAS, SESSION_RESUME, true);
|
|
122
|
+
expect(r).toMatchObject({ ok: false, error: "network_token_required" });
|
|
123
|
+
expect(capable(SESSION_ALIAS)).toBe(0);
|
|
124
|
+
}, 20_000);
|
|
125
|
+
|
|
126
|
+
test("another node's token is refused and cannot mark someone else's session capable", async () => {
|
|
127
|
+
await registerSession(asSession, SESSION_ALIAS, SESSION_RESUME, false);
|
|
128
|
+
const r = await registerSession(asPeer, SESSION_ALIAS, SESSION_RESUME, true);
|
|
129
|
+
expect(r).toMatchObject({ ok: false, error: "alias_identity_mismatch" });
|
|
130
|
+
expect(capable(SESSION_ALIAS)).toBe(0);
|
|
131
|
+
}, 20_000);
|
|
132
|
+
});
|
|
133
|
+
|
|
134
|
+
describe("read/write by alias reach a session with no nodes row", () => {
|
|
135
|
+
test("read round trip: enqueue by alias → session pulls with its own token → ack → done", async () => {
|
|
136
|
+
await registerSession(asSession, SESSION_ALIAS, SESSION_RESUME, true);
|
|
137
|
+
const u = await connect(asUser);
|
|
138
|
+
const s = await connect(asSession);
|
|
139
|
+
try {
|
|
140
|
+
const enq = await u.call("read_node_rules_file", { alias: SESSION_ALIAS });
|
|
141
|
+
expect(enq).toMatchObject({ ok: true, op: "read" });
|
|
142
|
+
expect(db.get<{ node_id: string }>("SELECT node_id FROM node_rules_requests WHERE request_id = ?1", enq.request_id)?.node_id).toBe(`session:${SESSION_ALIAS}`);
|
|
143
|
+
const pulled = await s.call("get_rules_file_request", {});
|
|
144
|
+
expect(pulled.request).toEqual({ request_id: enq.request_id, op: "read" });
|
|
145
|
+
expect(await s.call("ack_rules_file_request", { request_id: enq.request_id, status: "done", file_name: "CLAUDE.md", exists: true, content: "# rules\n" })).toMatchObject({ ok: true, status: "done" });
|
|
146
|
+
expect(await u.call("get_rules_file_result", { request_id: enq.request_id })).toMatchObject({ ok: true, status: "done", file_name: "CLAUDE.md", exists: true, content: "# rules\n" });
|
|
147
|
+
} finally { await u.close(); await s.close(); }
|
|
148
|
+
}, 20_000);
|
|
149
|
+
|
|
150
|
+
test("write carries content to the session", async () => {
|
|
151
|
+
await registerSession(asSession, SESSION_ALIAS, SESSION_RESUME, true);
|
|
152
|
+
const u = await connect(asUser);
|
|
153
|
+
const s = await connect(asSession);
|
|
154
|
+
try {
|
|
155
|
+
const enq = await u.call("write_node_rules_file", { alias: SESSION_ALIAS, content: "new body\n" });
|
|
156
|
+
expect(enq).toMatchObject({ ok: true, op: "write" });
|
|
157
|
+
const pulled = await s.call("get_rules_file_request", {});
|
|
158
|
+
expect(pulled.request).toEqual({ request_id: enq.request_id, op: "write", content: "new body\n" });
|
|
159
|
+
} finally { await u.close(); await s.close(); }
|
|
160
|
+
}, 20_000);
|
|
161
|
+
|
|
162
|
+
test("alias with neither a node row nor a capable session is refused before any row is written", async () => {
|
|
163
|
+
await registerSession(asSession, SESSION_ALIAS, SESSION_RESUME, false);
|
|
164
|
+
const u = await connect(asUser);
|
|
165
|
+
try {
|
|
166
|
+
expect(await u.call("read_node_rules_file", { alias: SESSION_ALIAS })).toMatchObject({ ok: false, error: "rules_file_target_not_found" });
|
|
167
|
+
expect(db.get<{ c: number }>("SELECT COUNT(*) AS c FROM node_rules_requests WHERE network_id = ?1", NET)?.c).toBe(0);
|
|
168
|
+
} finally { await u.close(); }
|
|
169
|
+
}, 20_000);
|
|
170
|
+
|
|
171
|
+
test("network isolation: same alias in another network neither pulls nor acks; other-network user cannot enqueue", async () => {
|
|
172
|
+
await registerSession(asSession, SESSION_ALIAS, SESSION_RESUME, true);
|
|
173
|
+
const u = await connect(asUser);
|
|
174
|
+
const x = await connect(asOtherNetSameAlias);
|
|
175
|
+
const ou = await connect(asOtherNetUser);
|
|
176
|
+
try {
|
|
177
|
+
const enq = await u.call("read_node_rules_file", { alias: SESSION_ALIAS });
|
|
178
|
+
expect(enq.ok).toBe(true);
|
|
179
|
+
expect(await x.call("get_rules_file_request", {})).toEqual({ ok: true, request: null });
|
|
180
|
+
expect(await x.call("ack_rules_file_request", { request_id: enq.request_id, status: "done", content: "stolen" })).toMatchObject({ ok: true, ignored: "unknown_or_foreign_request" });
|
|
181
|
+
expect(db.get<{ s: string }>("SELECT status AS s FROM node_rules_requests WHERE request_id = ?1", enq.request_id)?.s).toBe("pending");
|
|
182
|
+
expect(await ou.call("read_node_rules_file", { alias: SESSION_ALIAS })).toMatchObject({ ok: false, error: "rules_file_target_not_found" });
|
|
183
|
+
} finally { await u.close(); await x.close(); await ou.close(); }
|
|
184
|
+
}, 20_000);
|
|
185
|
+
|
|
186
|
+
test("a peer node in the same network cannot pull or ack the session's request", async () => {
|
|
187
|
+
await registerSession(asSession, SESSION_ALIAS, SESSION_RESUME, true);
|
|
188
|
+
const u = await connect(asUser);
|
|
189
|
+
const p = await connect(asPeer);
|
|
190
|
+
try {
|
|
191
|
+
const enq = await u.call("read_node_rules_file", { alias: SESSION_ALIAS });
|
|
192
|
+
expect(await p.call("get_rules_file_request", {})).toEqual({ ok: true, request: null });
|
|
193
|
+
expect(await p.call("ack_rules_file_request", { request_id: enq.request_id, status: "done", content: "x" })).toMatchObject({ ok: true, ignored: "unknown_or_foreign_request" });
|
|
194
|
+
} finally { await u.close(); await p.close(); }
|
|
195
|
+
}, 20_000);
|
|
196
|
+
|
|
197
|
+
test("alias that has a nodes row resolves to that node_id (original queue key)", async () => {
|
|
198
|
+
const u = await connect(asUser);
|
|
199
|
+
const n = await connect(asNodeRow);
|
|
200
|
+
try {
|
|
201
|
+
const enq = await u.call("read_node_rules_file", { alias: NODE_ROW_ALIAS });
|
|
202
|
+
expect(enq.ok).toBe(true);
|
|
203
|
+
expect(db.get<{ node_id: string }>("SELECT node_id FROM node_rules_requests WHERE request_id = ?1", enq.request_id)?.node_id).toBe(NODE_ROW_ID);
|
|
204
|
+
expect((await n.call("get_rules_file_request", {})).request?.request_id).toBe(enq.request_id);
|
|
205
|
+
} finally { await u.close(); await n.close(); }
|
|
206
|
+
}, 20_000);
|
|
207
|
+
});
|
|
@@ -0,0 +1,237 @@
|
|
|
1
|
+
// app#225 — 节点规则文件远程读写:走真 MCP in-process transport 的端到端。
|
|
2
|
+
//
|
|
3
|
+
// 验的是 #225 验收里 hub 能验的那几条:
|
|
4
|
+
// 1. 客户端只用 node_id 就能发起读/写;工具的 inputSchema 里**没有**任何
|
|
5
|
+
// 路径/文件名字段(验收第 5 条「不能借路径参数写任意文件」在 hub 侧的形状)。
|
|
6
|
+
// 2. 节点用网络 token + alias 拉到的请求里也没有路径字段;ack 后客户端轮询拿到内容。
|
|
7
|
+
// 3. SEC-1:跨网络节点发不了、别的节点 ack 不了、用户 token 拉不了。
|
|
8
|
+
// 4. 单飞 + 超时回收。
|
|
9
|
+
//
|
|
10
|
+
// 跑法:cd server && COMMHUB_DB=/tmp/rules.db bun test src/rules-file-transport.test.ts
|
|
11
|
+
import { afterAll, beforeEach, describe, expect, test } from "bun:test";
|
|
12
|
+
import { Client } from "@modelcontextprotocol/sdk/client/index.js";
|
|
13
|
+
import { InMemoryTransport } from "@modelcontextprotocol/sdk/inMemory.js";
|
|
14
|
+
import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
|
|
15
|
+
import { db } from "./db.js";
|
|
16
|
+
import { registerTools } from "./tools.js";
|
|
17
|
+
|
|
18
|
+
const NET = "net_rules_file";
|
|
19
|
+
const OTHER_NET = "net_rules_file_other";
|
|
20
|
+
const USER = "u_rules_file_owner";
|
|
21
|
+
const NODE_ID = "node_rules_file_claude";
|
|
22
|
+
const NODE_ALIAS = "rules-file-claude";
|
|
23
|
+
const OTHER_NODE_ID = "node_rules_file_other";
|
|
24
|
+
const OTHER_NODE_ALIAS = "rules-file-other";
|
|
25
|
+
const NODE_TOKEN = "tok_rules_file_node";
|
|
26
|
+
const OTHER_NODE_TOKEN = "tok_rules_file_other_node";
|
|
27
|
+
|
|
28
|
+
function cleanup() {
|
|
29
|
+
for (const net of [NET, OTHER_NET]) {
|
|
30
|
+
try { db.run("DELETE FROM node_rules_requests WHERE network_id = ?1", [net]); } catch {}
|
|
31
|
+
try { db.run("DELETE FROM audit_log WHERE network_id = ?1", [net]); } catch {}
|
|
32
|
+
try { db.run("DELETE FROM nodes WHERE network_id = ?1", [net]); } catch {}
|
|
33
|
+
try { db.run("DELETE FROM api_tokens WHERE network_id = ?1", [net]); } catch {}
|
|
34
|
+
try { db.run("DELETE FROM network_members WHERE network_id = ?1", [net]); } catch {}
|
|
35
|
+
try { db.run("DELETE FROM networks WHERE network_id = ?1", [net]); } catch {}
|
|
36
|
+
}
|
|
37
|
+
try { db.run("DELETE FROM users WHERE user_id = ?1", [USER]); } catch {}
|
|
38
|
+
}
|
|
39
|
+
|
|
40
|
+
function seedWorld() {
|
|
41
|
+
db.run(
|
|
42
|
+
`INSERT INTO users (user_id, username, password_hash, role, created_at) VALUES (?1, ?2, 'x', 'user', datetime('now'))`,
|
|
43
|
+
[USER, USER],
|
|
44
|
+
);
|
|
45
|
+
for (const net of [NET, OTHER_NET]) {
|
|
46
|
+
db.run(`INSERT INTO networks (network_id, network_name, owner_id, created_at) VALUES (?1, ?2, ?3, datetime('now'))`, [net, net, USER]);
|
|
47
|
+
db.run(`INSERT INTO network_members (user_id, network_id, role, joined_at) VALUES (?1, ?2, 'owner', datetime('now'))`, [USER, net]);
|
|
48
|
+
}
|
|
49
|
+
db.run(
|
|
50
|
+
`INSERT INTO nodes (node_id, node_name, alias, network_id, created_at, updated_at) VALUES (?1, ?2, ?3, ?4, datetime('now'), datetime('now'))`,
|
|
51
|
+
[NODE_ID, NODE_ALIAS, NODE_ALIAS, NET],
|
|
52
|
+
);
|
|
53
|
+
db.run(
|
|
54
|
+
`INSERT INTO nodes (node_id, node_name, alias, network_id, created_at, updated_at) VALUES (?1, ?2, ?3, ?4, datetime('now'), datetime('now'))`,
|
|
55
|
+
[OTHER_NODE_ID, OTHER_NODE_ALIAS, OTHER_NODE_ALIAS, OTHER_NET],
|
|
56
|
+
);
|
|
57
|
+
db.run(
|
|
58
|
+
`INSERT INTO api_tokens (token_id, user_id, network_id, scope, name, token_hash, expires_at, revoked_at) VALUES (?1, ?2, ?3, 'network', ?4, ?5, NULL, NULL)`,
|
|
59
|
+
[NODE_TOKEN, USER, NET, `node:${NODE_ALIAS}`, "hash_rules_file_node"],
|
|
60
|
+
);
|
|
61
|
+
db.run(
|
|
62
|
+
`INSERT INTO api_tokens (token_id, user_id, network_id, scope, name, token_hash, expires_at, revoked_at) VALUES (?1, ?2, ?3, 'network', ?4, ?5, NULL, NULL)`,
|
|
63
|
+
[OTHER_NODE_TOKEN, USER, OTHER_NET, `node:${OTHER_NODE_ALIAS}`, "hash_rules_file_other"],
|
|
64
|
+
);
|
|
65
|
+
}
|
|
66
|
+
|
|
67
|
+
type Identity = { net: string; alias: string; isNetworkToken: boolean; tokenId: string };
|
|
68
|
+
const asUser: Identity = { net: NET, alias: USER, isNetworkToken: false, tokenId: "tok_rules_file_user" };
|
|
69
|
+
const asNode: Identity = { net: NET, alias: NODE_ALIAS, isNetworkToken: true, tokenId: NODE_TOKEN };
|
|
70
|
+
const asOtherNode: Identity = { net: OTHER_NET, alias: OTHER_NODE_ALIAS, isNetworkToken: true, tokenId: OTHER_NODE_TOKEN };
|
|
71
|
+
|
|
72
|
+
async function connect(id: Identity) {
|
|
73
|
+
const server = new McpServer({ name: "rules-file-test", version: "1" });
|
|
74
|
+
registerTools(server, undefined, id.net, USER, id.alias, id.isNetworkToken, id.tokenId);
|
|
75
|
+
const client = new Client({ name: "rules-file-client", version: "1" });
|
|
76
|
+
const [ct, st] = InMemoryTransport.createLinkedPair();
|
|
77
|
+
await server.connect(st);
|
|
78
|
+
await client.connect(ct);
|
|
79
|
+
const call = async (name: string, args: Record<string, unknown>) => {
|
|
80
|
+
const r: any = await client.callTool({ name, arguments: args });
|
|
81
|
+
const first = r.content?.[0];
|
|
82
|
+
if (!first || first.type !== "text") throw new Error(`no text result from ${name}`);
|
|
83
|
+
return JSON.parse(first.text);
|
|
84
|
+
};
|
|
85
|
+
const close = async () => { await client.close(); await server.close(); };
|
|
86
|
+
return { client, server, call, close };
|
|
87
|
+
}
|
|
88
|
+
|
|
89
|
+
beforeEach(() => { cleanup(); seedWorld(); });
|
|
90
|
+
afterAll(cleanup);
|
|
91
|
+
|
|
92
|
+
describe("app#225 rules file — schema has no path", () => {
|
|
93
|
+
test("read/write tool inputs are node id + content only; no path / file_name / dir field anywhere", async () => {
|
|
94
|
+
// 🔴 不走 client.listTools():main 上 report_status / send_desktop_message /
|
|
95
|
+
// update_node_config / create_node 四个存量工具的 zod 形状让 SDK 的
|
|
96
|
+
// toJSONSchema 抛 `schema._zod`,listTools 整个失败(与本 PR 无关,实验见
|
|
97
|
+
// PR 正文)。这里直接看 McpServer 登记表里的 zod shape —— 它就是 hub 校验
|
|
98
|
+
// 入参用的那份,比 JSON schema 更接近真相。
|
|
99
|
+
const u = await connect(asUser);
|
|
100
|
+
try {
|
|
101
|
+
const reg: Record<string, any> = (u.server as any)._registeredTools;
|
|
102
|
+
const shapeKeys = (name: string): string[] => {
|
|
103
|
+
expect(reg[name], name).toBeDefined();
|
|
104
|
+
const schema = reg[name].inputSchema;
|
|
105
|
+
const shape = schema?.shape ?? schema?._def?.shape?.() ?? schema ?? {};
|
|
106
|
+
return Object.keys(shape).sort();
|
|
107
|
+
};
|
|
108
|
+
// `alias` (app#225 follow-up) names a TARGET node/session, not a file —
|
|
109
|
+
// the no-path regex below still covers it.
|
|
110
|
+
expect(shapeKeys("read_node_rules_file")).toEqual(["alias", "child_node_id", "network_id", "node_id"]);
|
|
111
|
+
expect(shapeKeys("write_node_rules_file")).toEqual(["alias", "child_node_id", "content", "network_id", "node_id"]);
|
|
112
|
+
expect(shapeKeys("get_rules_file_request")).toEqual([]);
|
|
113
|
+
// ack 的 file_name 是节点**回报**它用了哪个文件,不是客户端**指定**用哪个文件。
|
|
114
|
+
expect(shapeKeys("ack_rules_file_request")).toEqual(["content", "error", "exists", "file_name", "request_id", "status"]);
|
|
115
|
+
expect(shapeKeys("get_rules_file_result")).toEqual(["network_id", "request_id"]);
|
|
116
|
+
for (const name of ["read_node_rules_file", "write_node_rules_file", "get_rules_file_request", "get_rules_file_result"]) {
|
|
117
|
+
for (const k of shapeKeys(name)) expect(k, `${name}.${k}`).not.toMatch(/path|dir|file|cwd/i);
|
|
118
|
+
}
|
|
119
|
+
} finally { await u.close(); }
|
|
120
|
+
}, 20_000);
|
|
121
|
+
});
|
|
122
|
+
|
|
123
|
+
describe("app#225 rules file — read round trip", () => {
|
|
124
|
+
test("user enqueues read → node pulls (no path in request) → acks content → user polls done", async () => {
|
|
125
|
+
const u = await connect(asUser);
|
|
126
|
+
const n = await connect(asNode);
|
|
127
|
+
try {
|
|
128
|
+
const enq = await u.call("read_node_rules_file", { node_id: NODE_ID });
|
|
129
|
+
expect(enq).toMatchObject({ ok: true, op: "read" });
|
|
130
|
+
expect(enq.request_id).toMatch(/^rf_/);
|
|
131
|
+
|
|
132
|
+
expect(await u.call("get_rules_file_result", { request_id: enq.request_id })).toMatchObject({ ok: true, status: "pending", op: "read" });
|
|
133
|
+
|
|
134
|
+
const pulled = await n.call("get_rules_file_request", {});
|
|
135
|
+
expect(pulled.ok).toBe(true);
|
|
136
|
+
expect(Object.keys(pulled.request).sort()).toEqual(["op", "request_id"]);
|
|
137
|
+
expect(pulled.request).toEqual({ request_id: enq.request_id, op: "read" });
|
|
138
|
+
// 第二次拉:已经 in_progress,不再重复下发
|
|
139
|
+
expect(await n.call("get_rules_file_request", {})).toEqual({ ok: true, request: null });
|
|
140
|
+
|
|
141
|
+
expect(await n.call("ack_rules_file_request", {
|
|
142
|
+
request_id: enq.request_id, status: "done", file_name: "CLAUDE.md", exists: true, content: "# 规则\n",
|
|
143
|
+
})).toMatchObject({ ok: true, status: "done" });
|
|
144
|
+
|
|
145
|
+
const res = await u.call("get_rules_file_result", { request_id: enq.request_id });
|
|
146
|
+
expect(res).toMatchObject({ ok: true, status: "done", op: "read", file_name: "CLAUDE.md", exists: true, content: "# 规则\n", error: null });
|
|
147
|
+
} finally { await u.close(); await n.close(); }
|
|
148
|
+
}, 20_000);
|
|
149
|
+
|
|
150
|
+
test("write: content travels to the node verbatim; result carries no content back", async () => {
|
|
151
|
+
const u = await connect(asUser);
|
|
152
|
+
const n = await connect(asNode);
|
|
153
|
+
try {
|
|
154
|
+
const body = "# new rules\n\n- 不碰生产 DB\n";
|
|
155
|
+
const enq = await u.call("write_node_rules_file", { node_id: NODE_ID, content: body });
|
|
156
|
+
expect(enq).toMatchObject({ ok: true, op: "write" });
|
|
157
|
+
const pulled = await n.call("get_rules_file_request", {});
|
|
158
|
+
expect(pulled.request).toEqual({ request_id: enq.request_id, op: "write", content: body });
|
|
159
|
+
await n.call("ack_rules_file_request", { request_id: enq.request_id, status: "done", file_name: "CLAUDE.md", exists: true });
|
|
160
|
+
const res = await u.call("get_rules_file_result", { request_id: enq.request_id });
|
|
161
|
+
expect(res).toMatchObject({ ok: true, status: "done", op: "write", file_name: "CLAUDE.md" });
|
|
162
|
+
expect(res.content).toBeUndefined();
|
|
163
|
+
} finally { await u.close(); await n.close(); }
|
|
164
|
+
}, 20_000);
|
|
165
|
+
|
|
166
|
+
test("node failure is surfaced with its reason", async () => {
|
|
167
|
+
const u = await connect(asUser);
|
|
168
|
+
const n = await connect(asNode);
|
|
169
|
+
try {
|
|
170
|
+
const enq = await u.call("write_node_rules_file", { node_id: NODE_ID, content: "x" });
|
|
171
|
+
await n.call("get_rules_file_request", {});
|
|
172
|
+
await n.call("ack_rules_file_request", { request_id: enq.request_id, status: "failed", file_name: "CLAUDE.md", error: "EACCES: permission denied" });
|
|
173
|
+
expect(await u.call("get_rules_file_result", { request_id: enq.request_id })).toMatchObject({ ok: true, status: "failed", error: "EACCES: permission denied" });
|
|
174
|
+
} finally { await u.close(); await n.close(); }
|
|
175
|
+
}, 20_000);
|
|
176
|
+
});
|
|
177
|
+
|
|
178
|
+
describe("app#225 rules file — SEC-1 and lifecycle", () => {
|
|
179
|
+
test("cross-network node cannot be targeted", async () => {
|
|
180
|
+
const u = await connect(asUser);
|
|
181
|
+
try {
|
|
182
|
+
expect(await u.call("read_node_rules_file", { node_id: OTHER_NODE_ID })).toMatchObject({ ok: false, error: "cross_network_node" });
|
|
183
|
+
expect(db.get<{ c: number }>("SELECT COUNT(*) AS c FROM node_rules_requests WHERE node_id = ?1", OTHER_NODE_ID)?.c).toBe(0);
|
|
184
|
+
} finally { await u.close(); }
|
|
185
|
+
}, 20_000);
|
|
186
|
+
|
|
187
|
+
test("a node in another network can neither pull nor ack this node's request", async () => {
|
|
188
|
+
const u = await connect(asUser);
|
|
189
|
+
const o = await connect(asOtherNode);
|
|
190
|
+
try {
|
|
191
|
+
const enq = await u.call("read_node_rules_file", { node_id: NODE_ID });
|
|
192
|
+
expect(await o.call("get_rules_file_request", {})).toEqual({ ok: true, request: null });
|
|
193
|
+
expect(await o.call("ack_rules_file_request", { request_id: enq.request_id, status: "done", content: "stolen" })).toMatchObject({ ok: true, ignored: "unknown_or_foreign_request" });
|
|
194
|
+
expect(db.get<{ status: string }>("SELECT status FROM node_rules_requests WHERE request_id = ?1", enq.request_id)?.status).toBe("pending");
|
|
195
|
+
} finally { await u.close(); await o.close(); }
|
|
196
|
+
}, 20_000);
|
|
197
|
+
|
|
198
|
+
test("user token (not a network token) cannot use the node-side pull/ack", async () => {
|
|
199
|
+
const u = await connect(asUser);
|
|
200
|
+
try {
|
|
201
|
+
expect(await u.call("get_rules_file_request", {})).toMatchObject({ ok: false, error: "network_token_required" });
|
|
202
|
+
expect(await u.call("ack_rules_file_request", { request_id: "rf_nope", status: "done" })).toMatchObject({ ok: false, error: "network_token_required" });
|
|
203
|
+
} finally { await u.close(); }
|
|
204
|
+
}, 20_000);
|
|
205
|
+
|
|
206
|
+
test("single flight: a second request while one is pending is refused; a stale one is superseded", async () => {
|
|
207
|
+
const u = await connect(asUser);
|
|
208
|
+
try {
|
|
209
|
+
const first = await u.call("read_node_rules_file", { node_id: NODE_ID });
|
|
210
|
+
expect(await u.call("read_node_rules_file", { node_id: NODE_ID })).toMatchObject({ ok: false, error: "request_in_flight", existing_request_id: first.request_id });
|
|
211
|
+
db.run("UPDATE node_rules_requests SET created_at = ?1 WHERE request_id = ?2", [Date.now() - 120_000, first.request_id]);
|
|
212
|
+
const second = await u.call("read_node_rules_file", { node_id: NODE_ID });
|
|
213
|
+
expect(second.ok).toBe(true);
|
|
214
|
+
expect(db.get<{ status: string }>("SELECT status FROM node_rules_requests WHERE request_id = ?1", first.request_id)?.status).toBe("timeout");
|
|
215
|
+
} finally { await u.close(); }
|
|
216
|
+
}, 20_000);
|
|
217
|
+
|
|
218
|
+
test("polling a request the node never answered flips it to timeout with a readable reason", async () => {
|
|
219
|
+
const u = await connect(asUser);
|
|
220
|
+
try {
|
|
221
|
+
const enq = await u.call("read_node_rules_file", { node_id: NODE_ID });
|
|
222
|
+
db.run("UPDATE node_rules_requests SET created_at = ?1 WHERE request_id = ?2", [Date.now() - 120_000, enq.request_id]);
|
|
223
|
+
const res = await u.call("get_rules_file_result", { request_id: enq.request_id });
|
|
224
|
+
expect(res).toMatchObject({ ok: true, status: "timeout" });
|
|
225
|
+
expect(res.error).toMatch(/did not answer/);
|
|
226
|
+
} finally { await u.close(); }
|
|
227
|
+
}, 20_000);
|
|
228
|
+
|
|
229
|
+
test("oversized content is refused before anything is queued", async () => {
|
|
230
|
+
const u = await connect(asUser);
|
|
231
|
+
try {
|
|
232
|
+
const r: any = await u.client.callTool({ name: "write_node_rules_file", arguments: { node_id: NODE_ID, content: "x".repeat(256 * 1024 + 1) } });
|
|
233
|
+
expect(r.isError).toBe(true);
|
|
234
|
+
expect(db.get<{ c: number }>("SELECT COUNT(*) AS c FROM node_rules_requests WHERE node_id = ?1", NODE_ID)?.c).toBe(0);
|
|
235
|
+
} finally { await u.close(); }
|
|
236
|
+
}, 20_000);
|
|
237
|
+
});
|
|
@@ -0,0 +1,22 @@
|
|
|
1
|
+
import { describe, expect, test } from "bun:test";
|
|
2
|
+
import { normalizeSessionRuntime } from "./runtime-label";
|
|
3
|
+
|
|
4
|
+
describe("session runtime projection", () => {
|
|
5
|
+
test("preserves exact co-presence and CLI runtimes", () => {
|
|
6
|
+
expect(normalizeSessionRuntime("agent-node:codex-app-server")).toBe("codex-app-server");
|
|
7
|
+
expect(normalizeSessionRuntime("agent-node:grok-build-cli")).toBe("grok-build-cli");
|
|
8
|
+
});
|
|
9
|
+
test("keeps legacy labels backward-compatible", () => {
|
|
10
|
+
expect(normalizeSessionRuntime("agent-node:codex")).toBe("codex-sdk");
|
|
11
|
+
expect(normalizeSessionRuntime("agent-node:claude")).toBe("claude-agent-sdk");
|
|
12
|
+
expect(normalizeSessionRuntime("agent-node:grok")).toBe("grok-build-acp");
|
|
13
|
+
expect(normalizeSessionRuntime("agent-node:opencode")).toBe("opencode-cli");
|
|
14
|
+
expect(normalizeSessionRuntime("claude-code")).toBe("claude-code-cli");
|
|
15
|
+
});
|
|
16
|
+
test("does not invent unknown runtimes", () => {
|
|
17
|
+
expect(normalizeSessionRuntime("agent-node:future-runtime")).toBeNull();
|
|
18
|
+
expect(normalizeSessionRuntime("agent-node:codex-app-server-evil")).toBeNull();
|
|
19
|
+
expect(normalizeSessionRuntime("agent-node:grok-build-cli-preview")).toBeNull();
|
|
20
|
+
expect(normalizeSessionRuntime(null)).toBeNull();
|
|
21
|
+
});
|
|
22
|
+
});
|
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
/** Preserve exact agent-node runtimes before applying legacy family aliases. */
|
|
2
|
+
export function normalizeSessionRuntime(agent: unknown): string | null {
|
|
3
|
+
if (typeof agent !== "string" || agent.length === 0) return null;
|
|
4
|
+
if (agent === "claude-code") return "claude-code-cli";
|
|
5
|
+
const reported = agent.startsWith("agent-node:") ? agent.slice("agent-node:".length) : "";
|
|
6
|
+
const exact: Record<string, string> = {
|
|
7
|
+
"codex-app-server": "codex-app-server",
|
|
8
|
+
"grok-build-cli": "grok-build-cli",
|
|
9
|
+
opencode: "opencode-cli",
|
|
10
|
+
"opencode-cli": "opencode-cli",
|
|
11
|
+
codex: "codex-sdk",
|
|
12
|
+
"codex-sdk": "codex-sdk",
|
|
13
|
+
claude: "claude-agent-sdk",
|
|
14
|
+
"claude-agent-sdk": "claude-agent-sdk",
|
|
15
|
+
grok: "grok-build-acp",
|
|
16
|
+
"grok-build-acp": "grok-build-acp",
|
|
17
|
+
};
|
|
18
|
+
if (exact[reported]) return exact[reported];
|
|
19
|
+
if (agent === "http-api" || agent === "http" || agent === "api") return "http-api";
|
|
20
|
+
return null;
|
|
21
|
+
}
|
|
@@ -0,0 +1,130 @@
|
|
|
1
|
+
import { afterAll, beforeEach, describe, expect, test } from "bun:test";
|
|
2
|
+
import { db } from "./db.js";
|
|
3
|
+
import { createSSEStream } from "./push.js";
|
|
4
|
+
import { dispatchScheduledOccurrence } from "./scheduled-tasks.js";
|
|
5
|
+
|
|
6
|
+
// A claude-code node's channel server reports status without node_id, so its
|
|
7
|
+
// sessions row has node_id NULL. The scheduler used to look the session up by
|
|
8
|
+
// node_id only, judged every occurrence "queued" and never rang the doorbell:
|
|
9
|
+
// the task sat in the inbox until an unrelated push drained it (20–36 min in
|
|
10
|
+
// production).
|
|
11
|
+
|
|
12
|
+
const NET = "net_sched_doorbell_nodeless";
|
|
13
|
+
const NODE = "n_sched_doorbell_cc";
|
|
14
|
+
const ALIAS = "sched-doorbell-cc";
|
|
15
|
+
|
|
16
|
+
function cleanup(): void {
|
|
17
|
+
for (const table of ["task_events", "inbox", "scheduled_task_runs", "scheduled_tasks", "tasks", "sessions", "nodes"]) {
|
|
18
|
+
try { db.run(`DELETE FROM ${table} WHERE network_id = ?1`, [NET]); } catch {}
|
|
19
|
+
}
|
|
20
|
+
}
|
|
21
|
+
|
|
22
|
+
function seedNode(): void {
|
|
23
|
+
db.run(
|
|
24
|
+
`INSERT INTO nodes (node_id, node_name, alias, runtime, network_id, lifecycle_state)
|
|
25
|
+
VALUES (?1, ?2, ?2, 'claude-code-cli', ?3, 'active')`,
|
|
26
|
+
[NODE, ALIAS, NET],
|
|
27
|
+
);
|
|
28
|
+
}
|
|
29
|
+
|
|
30
|
+
function seedSession(status: string, nodeId: string | null): void {
|
|
31
|
+
db.run(
|
|
32
|
+
`INSERT INTO sessions (resume_id, alias, status, node_id, network_id, agent, updated_at, last_seen_at)
|
|
33
|
+
VALUES (?1, ?2, ?3, ?4, ?5, 'claude-code', datetime('now'), datetime('now'))`,
|
|
34
|
+
[`cc-${crypto.randomUUID()}`, ALIAS, status, nodeId, NET],
|
|
35
|
+
);
|
|
36
|
+
}
|
|
37
|
+
|
|
38
|
+
function seedSchedule(): any {
|
|
39
|
+
const scheduleId = `sched_${crypto.randomUUID()}`;
|
|
40
|
+
db.run(
|
|
41
|
+
`INSERT INTO scheduled_tasks
|
|
42
|
+
(schedule_id, network_id, name, target_node_id, target_alias, task_content, priority,
|
|
43
|
+
schedule_type, schedule_json, timezone, status, next_run_at)
|
|
44
|
+
VALUES (?1, ?2, 'doorbell', ?3, ?4, 'scheduled doorbell task', 'normal', 'interval',
|
|
45
|
+
'{"type":"interval","every_seconds":3600}', 'UTC', 'active', datetime('now', '+1 hour'))`,
|
|
46
|
+
[scheduleId, NET, NODE, ALIAS],
|
|
47
|
+
);
|
|
48
|
+
return db.get("SELECT * FROM scheduled_tasks WHERE schedule_id = ?1", scheduleId);
|
|
49
|
+
}
|
|
50
|
+
|
|
51
|
+
/** Subscribe like the channel server does and collect decoded SSE frames. */
|
|
52
|
+
function subscribe(): { frames: any[]; close: () => Promise<void> } {
|
|
53
|
+
const res = createSSEStream(ALIAS, NET);
|
|
54
|
+
const reader = res.body!.getReader();
|
|
55
|
+
const decoder = new TextDecoder();
|
|
56
|
+
const frames: any[] = [];
|
|
57
|
+
let buffer = "";
|
|
58
|
+
let open = true;
|
|
59
|
+
(async () => {
|
|
60
|
+
while (open) {
|
|
61
|
+
const { done, value } = await reader.read();
|
|
62
|
+
if (done) break;
|
|
63
|
+
buffer += decoder.decode(value, { stream: true });
|
|
64
|
+
const blocks = buffer.split("\n\n");
|
|
65
|
+
buffer = blocks.pop() || "";
|
|
66
|
+
for (const block of blocks) {
|
|
67
|
+
const line = block.split("\n").find((l) => l.startsWith("data: "));
|
|
68
|
+
if (line) frames.push(JSON.parse(line.slice(6)));
|
|
69
|
+
}
|
|
70
|
+
}
|
|
71
|
+
})().catch(() => {});
|
|
72
|
+
return {
|
|
73
|
+
frames,
|
|
74
|
+
close: async () => {
|
|
75
|
+
open = false;
|
|
76
|
+
await reader.cancel().catch(() => {});
|
|
77
|
+
},
|
|
78
|
+
};
|
|
79
|
+
}
|
|
80
|
+
|
|
81
|
+
async function settle(): Promise<void> {
|
|
82
|
+
await new Promise((resolve) => setTimeout(resolve, 20));
|
|
83
|
+
}
|
|
84
|
+
|
|
85
|
+
beforeEach(() => {
|
|
86
|
+
cleanup();
|
|
87
|
+
seedNode();
|
|
88
|
+
});
|
|
89
|
+
|
|
90
|
+
afterAll(cleanup);
|
|
91
|
+
|
|
92
|
+
describe("scheduler doorbell for sessions reported without node_id", () => {
|
|
93
|
+
test("idle node_id-less session is delivered and its subscriber receives new_task", async () => {
|
|
94
|
+
seedSession("idle", null);
|
|
95
|
+
const sub = subscribe();
|
|
96
|
+
try {
|
|
97
|
+
await settle();
|
|
98
|
+
const result = dispatchScheduledOccurrence(seedSchedule(), new Date().toISOString(), false);
|
|
99
|
+
await settle();
|
|
100
|
+
expect(result.status).toBe("delivered");
|
|
101
|
+
expect(db.get<{ status: string }>("SELECT status FROM scheduled_task_runs WHERE run_id = ?1", result.runId)?.status).toBe("delivered");
|
|
102
|
+
const wake = sub.frames.filter((f) => f.type === "new_task");
|
|
103
|
+
expect(wake).toHaveLength(1);
|
|
104
|
+
expect(wake[0].from).toBe("scheduler");
|
|
105
|
+
expect(wake[0].inbox_count).toBe(1);
|
|
106
|
+
} finally {
|
|
107
|
+
await sub.close();
|
|
108
|
+
}
|
|
109
|
+
});
|
|
110
|
+
|
|
111
|
+
test("a live subscriber counts even when the session row says offline", async () => {
|
|
112
|
+
seedSession("offline", null);
|
|
113
|
+
const sub = subscribe();
|
|
114
|
+
try {
|
|
115
|
+
await settle();
|
|
116
|
+
const result = dispatchScheduledOccurrence(seedSchedule(), new Date().toISOString(), false);
|
|
117
|
+
await settle();
|
|
118
|
+
expect(result.status).toBe("delivered");
|
|
119
|
+
expect(sub.frames.filter((f) => f.type === "new_task")).toHaveLength(1);
|
|
120
|
+
} finally {
|
|
121
|
+
await sub.close();
|
|
122
|
+
}
|
|
123
|
+
});
|
|
124
|
+
|
|
125
|
+
test("offline session with nobody listening stays queued", () => {
|
|
126
|
+
seedSession("offline", null);
|
|
127
|
+
const result = dispatchScheduledOccurrence(seedSchedule(), new Date().toISOString(), false);
|
|
128
|
+
expect(result.status).toBe("queued");
|
|
129
|
+
});
|
|
130
|
+
});
|