@acquiredintelligence/attend 0.0.0-stage → 0.2.0
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/LICENSE +21 -0
- package/README.md +174 -2
- package/dist/attend.d.ts +30 -0
- package/dist/attend.d.ts.map +1 -0
- package/dist/attend.js +464 -0
- package/dist/attend.js.map +1 -0
- package/dist/cli.d.ts +3 -0
- package/dist/cli.d.ts.map +1 -0
- package/dist/cli.js +4 -0
- package/dist/cli.js.map +1 -0
- package/dist/core/ids.d.ts +12 -0
- package/dist/core/ids.d.ts.map +1 -0
- package/dist/core/ids.js +97 -0
- package/dist/core/ids.js.map +1 -0
- package/dist/core/reducer.d.ts +26 -0
- package/dist/core/reducer.d.ts.map +1 -0
- package/dist/core/reducer.js +437 -0
- package/dist/core/reducer.js.map +1 -0
- package/dist/core/seen.d.ts +22 -0
- package/dist/core/seen.d.ts.map +1 -0
- package/dist/core/seen.js +61 -0
- package/dist/core/seen.js.map +1 -0
- package/dist/core/state.d.ts +12 -0
- package/dist/core/state.d.ts.map +1 -0
- package/dist/core/state.js +90 -0
- package/dist/core/state.js.map +1 -0
- package/dist/core/step.d.ts +14 -0
- package/dist/core/step.d.ts.map +1 -0
- package/dist/core/step.js +17 -0
- package/dist/core/step.js.map +1 -0
- package/dist/core/threshold.d.ts +40 -0
- package/dist/core/threshold.d.ts.map +1 -0
- package/dist/core/threshold.js +72 -0
- package/dist/core/threshold.js.map +1 -0
- package/dist/core/types.d.ts +704 -0
- package/dist/core/types.d.ts.map +1 -0
- package/dist/core/types.js +24 -0
- package/dist/core/types.js.map +1 -0
- package/dist/core/window.d.ts +38 -0
- package/dist/core/window.d.ts.map +1 -0
- package/dist/core/window.js +322 -0
- package/dist/core/window.js.map +1 -0
- package/dist/decisions.d.ts +53 -0
- package/dist/decisions.d.ts.map +1 -0
- package/dist/decisions.js +115 -0
- package/dist/decisions.js.map +1 -0
- package/dist/errors.d.ts +36 -0
- package/dist/errors.d.ts.map +1 -0
- package/dist/errors.js +53 -0
- package/dist/errors.js.map +1 -0
- package/dist/index.d.ts +10 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +4 -0
- package/dist/index.js.map +1 -0
- package/dist/internal/adapter.d.ts +21 -0
- package/dist/internal/adapter.d.ts.map +1 -0
- package/dist/internal/adapter.js +62 -0
- package/dist/internal/adapter.js.map +1 -0
- package/dist/internal/assert.d.ts +5 -0
- package/dist/internal/assert.d.ts.map +1 -0
- package/dist/internal/assert.js +14 -0
- package/dist/internal/assert.js.map +1 -0
- package/dist/internal/clock.d.ts +8 -0
- package/dist/internal/clock.d.ts.map +1 -0
- package/dist/internal/clock.js +38 -0
- package/dist/internal/clock.js.map +1 -0
- package/dist/internal/http.d.ts +42 -0
- package/dist/internal/http.d.ts.map +1 -0
- package/dist/internal/http.js +122 -0
- package/dist/internal/http.js.map +1 -0
- package/dist/internal/images.d.ts +18 -0
- package/dist/internal/images.d.ts.map +1 -0
- package/dist/internal/images.js +75 -0
- package/dist/internal/images.js.map +1 -0
- package/dist/internal/json.d.ts +3 -0
- package/dist/internal/json.d.ts.map +1 -0
- package/dist/internal/json.js +5 -0
- package/dist/internal/json.js.map +1 -0
- package/dist/internal/prompt.d.ts +31 -0
- package/dist/internal/prompt.d.ts.map +1 -0
- package/dist/internal/prompt.js +149 -0
- package/dist/internal/prompt.js.map +1 -0
- package/dist/internal/shared.d.ts +12 -0
- package/dist/internal/shared.d.ts.map +1 -0
- package/dist/internal/shared.js +138 -0
- package/dist/internal/shared.js.map +1 -0
- package/dist/jev.d.ts +74 -0
- package/dist/jev.d.ts.map +1 -0
- package/dist/jev.js +156 -0
- package/dist/jev.js.map +1 -0
- package/dist/llm.d.ts +164 -0
- package/dist/llm.d.ts.map +1 -0
- package/dist/llm.js +79 -0
- package/dist/llm.js.map +1 -0
- package/dist/mcp.d.ts +130 -0
- package/dist/mcp.d.ts.map +1 -0
- package/dist/mcp.js +550 -0
- package/dist/mcp.js.map +1 -0
- package/dist/memory.d.ts +10 -0
- package/dist/memory.d.ts.map +1 -0
- package/dist/memory.js +50 -0
- package/dist/memory.js.map +1 -0
- package/dist/postgres.d.ts +25 -0
- package/dist/postgres.d.ts.map +1 -0
- package/dist/postgres.js +56 -0
- package/dist/postgres.js.map +1 -0
- package/dist/redis.d.ts +20 -0
- package/dist/redis.d.ts.map +1 -0
- package/dist/redis.js +83 -0
- package/dist/redis.js.map +1 -0
- package/dist/server/compat.d.ts +13 -0
- package/dist/server/compat.d.ts.map +1 -0
- package/dist/server/compat.js +5 -0
- package/dist/server/compat.js.map +1 -0
- package/dist/server/cursor.d.ts +15 -0
- package/dist/server/cursor.d.ts.map +1 -0
- package/dist/server/cursor.js +50 -0
- package/dist/server/cursor.js.map +1 -0
- package/dist/server/events.d.ts +42 -0
- package/dist/server/events.d.ts.map +1 -0
- package/dist/server/events.js +189 -0
- package/dist/server/events.js.map +1 -0
- package/dist/server/fanout.d.ts +28 -0
- package/dist/server/fanout.d.ts.map +1 -0
- package/dist/server/fanout.js +184 -0
- package/dist/server/fanout.js.map +1 -0
- package/dist/server/guard.d.ts +19 -0
- package/dist/server/guard.d.ts.map +1 -0
- package/dist/server/guard.js +27 -0
- package/dist/server/guard.js.map +1 -0
- package/dist/server/http.d.ts +4 -0
- package/dist/server/http.d.ts.map +1 -0
- package/dist/server/http.js +323 -0
- package/dist/server/http.js.map +1 -0
- package/dist/server/ingest.d.ts +4 -0
- package/dist/server/ingest.d.ts.map +1 -0
- package/dist/server/ingest.js +400 -0
- package/dist/server/ingest.js.map +1 -0
- package/dist/server/install.d.ts +5 -0
- package/dist/server/install.d.ts.map +1 -0
- package/dist/server/install.js +91 -0
- package/dist/server/install.js.map +1 -0
- package/dist/server/poll.d.ts +4 -0
- package/dist/server/poll.d.ts.map +1 -0
- package/dist/server/poll.js +45 -0
- package/dist/server/poll.js.map +1 -0
- package/dist/server/serve.d.ts +19 -0
- package/dist/server/serve.d.ts.map +1 -0
- package/dist/server/serve.js +238 -0
- package/dist/server/serve.js.map +1 -0
- package/dist/server/sessions.d.ts +14 -0
- package/dist/server/sessions.d.ts.map +1 -0
- package/dist/server/sessions.js +54 -0
- package/dist/server/sessions.js.map +1 -0
- package/dist/server/stream.d.ts +11 -0
- package/dist/server/stream.d.ts.map +1 -0
- package/dist/server/stream.js +216 -0
- package/dist/server/stream.js.map +1 -0
- package/dist/server/tools.d.ts +22 -0
- package/dist/server/tools.d.ts.map +1 -0
- package/dist/server/tools.js +55 -0
- package/dist/server/tools.js.map +1 -0
- package/dist/server/types.d.ts +351 -0
- package/dist/server/types.d.ts.map +1 -0
- package/dist/server/types.js +5 -0
- package/dist/server/types.js.map +1 -0
- package/dist/server.d.ts +11 -0
- package/dist/server.d.ts.map +1 -0
- package/dist/server.js +10 -0
- package/dist/server.js.map +1 -0
- package/dist/shell/budget.d.ts +4 -0
- package/dist/shell/budget.d.ts.map +1 -0
- package/dist/shell/budget.js +5 -0
- package/dist/shell/budget.js.map +1 -0
- package/dist/shell/copy.d.ts +20 -0
- package/dist/shell/copy.d.ts.map +1 -0
- package/dist/shell/copy.js +73 -0
- package/dist/shell/copy.js.map +1 -0
- package/dist/shell/decisions.d.ts +6 -0
- package/dist/shell/decisions.d.ts.map +1 -0
- package/dist/shell/decisions.js +42 -0
- package/dist/shell/decisions.js.map +1 -0
- package/dist/shell/read.d.ts +7 -0
- package/dist/shell/read.d.ts.map +1 -0
- package/dist/shell/read.js +21 -0
- package/dist/shell/read.js.map +1 -0
- package/dist/shell/scheduler.d.ts +4 -0
- package/dist/shell/scheduler.d.ts.map +1 -0
- package/dist/shell/scheduler.js +90 -0
- package/dist/shell/scheduler.js.map +1 -0
- package/dist/shell/scope.d.ts +27 -0
- package/dist/shell/scope.d.ts.map +1 -0
- package/dist/shell/scope.js +343 -0
- package/dist/shell/scope.js.map +1 -0
- package/dist/shell/types.d.ts +168 -0
- package/dist/shell/types.d.ts.map +1 -0
- package/dist/shell/types.js +2 -0
- package/dist/shell/types.js.map +1 -0
- package/dist/sqlite.d.ts +14 -0
- package/dist/sqlite.d.ts.map +1 -0
- package/dist/sqlite.js +39 -0
- package/dist/sqlite.js.map +1 -0
- package/dist/step.d.ts +18 -0
- package/dist/step.d.ts.map +1 -0
- package/dist/step.js +14 -0
- package/dist/step.js.map +1 -0
- package/dist/store/memory.d.ts +12 -0
- package/dist/store/memory.d.ts.map +1 -0
- package/dist/store/memory.js +264 -0
- package/dist/store/memory.js.map +1 -0
- package/dist/store/postgres.d.ts +24 -0
- package/dist/store/postgres.d.ts.map +1 -0
- package/dist/store/postgres.js +364 -0
- package/dist/store/postgres.js.map +1 -0
- package/dist/store/redis.d.ts +12 -0
- package/dist/store/redis.d.ts.map +1 -0
- package/dist/store/redis.js +540 -0
- package/dist/store/redis.js.map +1 -0
- package/dist/store/sqlite-worker.d.ts +91 -0
- package/dist/store/sqlite-worker.d.ts.map +1 -0
- package/dist/store/sqlite-worker.js +322 -0
- package/dist/store/sqlite-worker.js.map +1 -0
- package/dist/store/sqlite.d.ts +34 -0
- package/dist/store/sqlite.d.ts.map +1 -0
- package/dist/store/sqlite.js +238 -0
- package/dist/store/sqlite.js.map +1 -0
- package/dist/store/types.d.ts +137 -0
- package/dist/store/types.d.ts.map +1 -0
- package/dist/store/types.js +2 -0
- package/dist/store/types.js.map +1 -0
- package/dist/testing/conformance.d.ts +13 -0
- package/dist/testing/conformance.d.ts.map +1 -0
- package/dist/testing/conformance.js +542 -0
- package/dist/testing/conformance.js.map +1 -0
- package/dist/testing/wire.d.ts +16 -0
- package/dist/testing/wire.d.ts.map +1 -0
- package/dist/testing/wire.js +304 -0
- package/dist/testing/wire.js.map +1 -0
- package/dist/testing.d.ts +113 -0
- package/dist/testing.d.ts.map +1 -0
- package/dist/testing.js +655 -0
- package/dist/testing.js.map +1 -0
- package/dist/types.d.ts +5 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +5 -0
- package/dist/types.js.map +1 -0
- package/package.json +130 -4
- package/skills/attend/SKILL.md +231 -0
- package/skills/attend/references/judges.md +283 -0
- package/skills/attend/references/kinds.md +62 -0
- package/skills/attend/references/mcp.md +132 -0
- package/skills/attend/references/sinks.md +73 -0
- package/skills/attend/references/stores.md +90 -0
- package/skills/attend/references/testing.md +120 -0
|
@@ -0,0 +1,238 @@
|
|
|
1
|
+
import { createServer } from 'node:http';
|
|
2
|
+
import { readFileSync } from 'node:fs';
|
|
3
|
+
import { createAttend } from '../attend.js';
|
|
4
|
+
import { KINDS } from '../core/types.js';
|
|
5
|
+
import { memoryStore } from '../memory.js';
|
|
6
|
+
import { eventsServer } from './install.js';
|
|
7
|
+
import { ndjsonIngest } from './ingest.js';
|
|
8
|
+
/** What `attend serve` runs: attend.run, eventsServer over store.log, ndjsonIngest, one HTTP listener. */
|
|
9
|
+
export function serve(attend, store, options) {
|
|
10
|
+
const events = eventsServer(store.log, { ...options.events, ...(options.principal ? { principal: options.principal } : {}) });
|
|
11
|
+
const ingest = options.principal ? { principal: options.principal } : {};
|
|
12
|
+
const routes = { 'POST /v1/ingest': ndjsonIngest(attend, ingest) };
|
|
13
|
+
if (options.ingest?.stream !== false)
|
|
14
|
+
routes['POST /v1/ingest/stream'] = ndjsonIngest(attend, { ...ingest, stream: true });
|
|
15
|
+
const handler = events.http({ path: '/mcp', routes });
|
|
16
|
+
const server = createServer((req, res) => { void Promise.resolve(handler(req, res)).catch(() => res.destroy()); });
|
|
17
|
+
const controller = new AbortController();
|
|
18
|
+
let failure;
|
|
19
|
+
let closing;
|
|
20
|
+
let address = '';
|
|
21
|
+
const running = attend.run({ signal: controller.signal }).catch((error) => { failure = { error }; void close().catch(() => undefined); });
|
|
22
|
+
const ready = new Promise((resolve, reject) => {
|
|
23
|
+
server.once('error', reject);
|
|
24
|
+
server.once('listening', () => {
|
|
25
|
+
const bound = server.address();
|
|
26
|
+
address = typeof bound === 'string' || bound === null
|
|
27
|
+
? `http+unix://${encodeURIComponent(bound ?? '')}`
|
|
28
|
+
: `http://${bound.family === 'IPv6' ? `[${bound.address}]` : bound.address}:${bound.port}`;
|
|
29
|
+
resolve();
|
|
30
|
+
});
|
|
31
|
+
});
|
|
32
|
+
ready.catch(() => { void close().catch(() => undefined); });
|
|
33
|
+
const closed = new Promise((resolve) => server.once('close', () => resolve()));
|
|
34
|
+
if ('path' in options.listen)
|
|
35
|
+
server.listen(options.listen.path);
|
|
36
|
+
else
|
|
37
|
+
server.listen(options.listen.port, options.listen.host ?? '127.0.0.1');
|
|
38
|
+
function close() {
|
|
39
|
+
closing ??= (async () => {
|
|
40
|
+
controller.abort();
|
|
41
|
+
await running;
|
|
42
|
+
await events.close();
|
|
43
|
+
server.closeAllConnections();
|
|
44
|
+
if (server.listening)
|
|
45
|
+
await new Promise((resolve) => server.close(() => resolve()));
|
|
46
|
+
if (failure)
|
|
47
|
+
throw failure.error;
|
|
48
|
+
})();
|
|
49
|
+
return closing;
|
|
50
|
+
}
|
|
51
|
+
return {
|
|
52
|
+
url() {
|
|
53
|
+
if (!address)
|
|
54
|
+
throw new Error('attend: serve is not listening yet; await ready');
|
|
55
|
+
return address;
|
|
56
|
+
},
|
|
57
|
+
ready,
|
|
58
|
+
closed,
|
|
59
|
+
close,
|
|
60
|
+
};
|
|
61
|
+
}
|
|
62
|
+
const USAGE = 'usage: attend serve [--store memory|sqlite:<file>|postgres:<url>|redis:<url>] [--judge rule|jev|decisions] [--kinds <json|file>] [--host h] [--port n] [--socket path] [--act-threshold n] [--token t]';
|
|
63
|
+
const HELP = `${USAGE}
|
|
64
|
+
|
|
65
|
+
flag env default meaning
|
|
66
|
+
--store ATTEND_STORE memory where state and the decision log live
|
|
67
|
+
--judge ATTEND_JUDGE rule rule | jev (TYPESAFE_API_KEY) | decisions (OPENAI_API_KEY, --judge-model)
|
|
68
|
+
--judge-model ATTEND_JUDGE_MODEL model id for the decisions judge
|
|
69
|
+
--kinds ATTEND_KINDS event name -> kind, inline JSON or a JSON file
|
|
70
|
+
--default-kind ATTEND_DEFAULT_KIND consequence kind for names --kinds does not list
|
|
71
|
+
--host ATTEND_HOST 127.0.0.1
|
|
72
|
+
--port ATTEND_PORT 8787 0 picks a free port
|
|
73
|
+
--socket ATTEND_SOCKET listen on a unix socket instead
|
|
74
|
+
--act-threshold ATTEND_ACT_THRESHOLD off number in (0, 1]
|
|
75
|
+
--token ATTEND_TOKEN require Authorization: Bearer <token>
|
|
76
|
+
`;
|
|
77
|
+
class UsageError extends Error {
|
|
78
|
+
}
|
|
79
|
+
const FLAGS = {
|
|
80
|
+
store: 'ATTEND_STORE', judge: 'ATTEND_JUDGE', 'judge-model': 'ATTEND_JUDGE_MODEL', kinds: 'ATTEND_KINDS', 'default-kind': 'ATTEND_DEFAULT_KIND',
|
|
81
|
+
host: 'ATTEND_HOST', port: 'ATTEND_PORT', socket: 'ATTEND_SOCKET', 'act-threshold': 'ATTEND_ACT_THRESHOLD', token: 'ATTEND_TOKEN',
|
|
82
|
+
};
|
|
83
|
+
function parseFlags(args, env) {
|
|
84
|
+
const given = {};
|
|
85
|
+
for (let i = 0; i < args.length; i += 1) {
|
|
86
|
+
const arg = args[i];
|
|
87
|
+
const match = /^--([a-z-]+)(?:=(.*))?$/s.exec(arg);
|
|
88
|
+
if (!match || !(match[1] in FLAGS))
|
|
89
|
+
throw new UsageError(`unknown argument ${arg}`);
|
|
90
|
+
const value = match[2] ?? args[++i];
|
|
91
|
+
if (value === undefined)
|
|
92
|
+
throw new UsageError(`--${match[1]} needs a value`);
|
|
93
|
+
given[match[1]] = value;
|
|
94
|
+
}
|
|
95
|
+
return Object.fromEntries(Object.entries(FLAGS).map(([flag, name]) => [flag, given[flag] ?? env[name]]));
|
|
96
|
+
}
|
|
97
|
+
function kindFunction(flags) {
|
|
98
|
+
const fallback = flags['default-kind'] ?? 'consequence';
|
|
99
|
+
if (!KINDS.includes(fallback))
|
|
100
|
+
throw new UsageError(`--default-kind must be one of ${KINDS.join(', ')}`);
|
|
101
|
+
let map = {};
|
|
102
|
+
const source = flags.kinds;
|
|
103
|
+
if (source !== undefined && source !== '') {
|
|
104
|
+
try {
|
|
105
|
+
map = JSON.parse(source.trimStart().startsWith('{') ? source : readFileSync(source, 'utf8'));
|
|
106
|
+
}
|
|
107
|
+
catch (error) {
|
|
108
|
+
throw new UsageError(`--kinds is not a JSON object or a readable JSON file: ${error instanceof Error ? error.message : String(error)}`);
|
|
109
|
+
}
|
|
110
|
+
if (typeof map !== 'object' || map === null || Array.isArray(map))
|
|
111
|
+
throw new UsageError('--kinds must be a JSON object of event name to kind');
|
|
112
|
+
for (const [name, kind] of Object.entries(map)) {
|
|
113
|
+
if (typeof kind !== 'string' || !KINDS.includes(kind))
|
|
114
|
+
throw new UsageError(`--kinds: "${name}" must map to one of ${KINDS.join(', ')}`);
|
|
115
|
+
}
|
|
116
|
+
}
|
|
117
|
+
return (event) => (Object.hasOwn(map, event.name) ? map[event.name] : fallback);
|
|
118
|
+
}
|
|
119
|
+
async function judgeFor(flags, env) {
|
|
120
|
+
switch (flags.judge ?? 'rule') {
|
|
121
|
+
case 'rule': return async ({ window }) => ({ route: window.reason === 'flush' ? 'interrupt' : 'queue' });
|
|
122
|
+
case 'jev': {
|
|
123
|
+
if (!env.TYPESAFE_API_KEY)
|
|
124
|
+
throw new UsageError('--judge jev needs TYPESAFE_API_KEY in the environment');
|
|
125
|
+
return (await import('../jev.js')).jevJudge({ apiKey: env.TYPESAFE_API_KEY });
|
|
126
|
+
}
|
|
127
|
+
case 'decisions': {
|
|
128
|
+
if (!env.OPENAI_API_KEY)
|
|
129
|
+
throw new UsageError('--judge decisions needs OPENAI_API_KEY in the environment');
|
|
130
|
+
if (!flags['judge-model'])
|
|
131
|
+
throw new UsageError('--judge decisions needs --judge-model (or ATTEND_JUDGE_MODEL)');
|
|
132
|
+
return (await import('../decisions.js')).decisionsJudge({ apiKey: env.OPENAI_API_KEY, model: flags['judge-model'] });
|
|
133
|
+
}
|
|
134
|
+
default: throw new UsageError(`--judge must be rule, jev or decisions, not "${flags.judge}"`);
|
|
135
|
+
}
|
|
136
|
+
}
|
|
137
|
+
async function storeFor(spec, env, cleanup) {
|
|
138
|
+
if (spec === 'memory')
|
|
139
|
+
return memoryStore();
|
|
140
|
+
const missing = (name) => new UsageError(`--store ${spec.split(':')[0]} needs the "${name}" package: npm install ${name}`);
|
|
141
|
+
if (spec.startsWith('sqlite:')) {
|
|
142
|
+
const file = spec.slice('sqlite:'.length);
|
|
143
|
+
if (!file)
|
|
144
|
+
throw new UsageError('--store sqlite:<file> needs a file path');
|
|
145
|
+
const { DatabaseSync } = await import('node:sqlite');
|
|
146
|
+
const db = new DatabaseSync(file);
|
|
147
|
+
cleanup.push(async () => db.close());
|
|
148
|
+
return (await import('../sqlite.js')).sqliteStore(db);
|
|
149
|
+
}
|
|
150
|
+
if (spec === 'postgres' || spec.startsWith('postgres:') || spec.startsWith('postgresql:')) {
|
|
151
|
+
const url = spec === 'postgres' ? env.DATABASE_URL : spec.startsWith('postgres://') || spec.startsWith('postgresql://') ? spec : spec.slice('postgres:'.length);
|
|
152
|
+
if (!url)
|
|
153
|
+
throw new UsageError('--store postgres needs a URL: postgres:<url> or DATABASE_URL');
|
|
154
|
+
const pg = await import('pg').catch(() => { throw missing('pg'); });
|
|
155
|
+
const pool = new (pg.default ?? pg).Pool({ connectionString: url });
|
|
156
|
+
cleanup.push(() => pool.end());
|
|
157
|
+
const { migrate, postgresStore } = await import('../postgres.js');
|
|
158
|
+
await migrate(pool);
|
|
159
|
+
return postgresStore(pool);
|
|
160
|
+
}
|
|
161
|
+
if (spec === 'redis' || spec.startsWith('redis:') || spec.startsWith('rediss:')) {
|
|
162
|
+
const url = spec === 'redis' ? env.REDIS_URL : spec.startsWith('redis://') || spec.startsWith('rediss://') ? spec : spec.slice('redis:'.length);
|
|
163
|
+
if (!url)
|
|
164
|
+
throw new UsageError('--store redis needs a URL: redis:<url> or REDIS_URL');
|
|
165
|
+
const redis = await import('redis').catch(() => { throw missing('redis'); });
|
|
166
|
+
const client = redis.createClient({ url });
|
|
167
|
+
await client.connect();
|
|
168
|
+
cleanup.push(async () => { client.destroy(); });
|
|
169
|
+
return (await import('../redis.js')).redisStore((args) => client.sendCommand(args));
|
|
170
|
+
}
|
|
171
|
+
throw new UsageError(`--store must be memory, sqlite:<file>, postgres:<url> or redis:<url>, not "${spec}"`);
|
|
172
|
+
}
|
|
173
|
+
/** The `attend` command. Returns the exit code once the server has shut down. */
|
|
174
|
+
export async function cli(argv, env, io = {}) {
|
|
175
|
+
const out = io.out ?? ((text) => void process.stdout.write(text));
|
|
176
|
+
const err = io.err ?? ((text) => void process.stderr.write(text));
|
|
177
|
+
if (argv.includes('--help') || argv.includes('-h')) {
|
|
178
|
+
out(HELP);
|
|
179
|
+
return 0;
|
|
180
|
+
}
|
|
181
|
+
const cleanup = [];
|
|
182
|
+
let store;
|
|
183
|
+
let server;
|
|
184
|
+
try {
|
|
185
|
+
if (argv[0] !== 'serve')
|
|
186
|
+
throw new UsageError(argv[0] === undefined ? 'missing command' : `unknown command "${argv[0]}"`);
|
|
187
|
+
const flags = parseFlags(argv.slice(1), env);
|
|
188
|
+
const kind = kindFunction(flags);
|
|
189
|
+
const threshold = flags['act-threshold'] === undefined ? undefined : Number(flags['act-threshold']);
|
|
190
|
+
if (threshold !== undefined && !(threshold > 0 && threshold <= 1))
|
|
191
|
+
throw new UsageError('--act-threshold must be a number in (0, 1]');
|
|
192
|
+
const port = flags.port === undefined ? 8787 : Number(flags.port);
|
|
193
|
+
if (!Number.isInteger(port) || port < 0 || port > 65535)
|
|
194
|
+
throw new UsageError('--port must be an integer from 0 to 65535');
|
|
195
|
+
const judge = await judgeFor(flags, env);
|
|
196
|
+
await import('@modelcontextprotocol/server').catch(() => { throw new Error('attend serve needs the MCP server package: npm install @modelcontextprotocol/server zod'); });
|
|
197
|
+
store = await storeFor(flags.store ?? 'memory', env, cleanup);
|
|
198
|
+
const token = flags.token;
|
|
199
|
+
const principal = token ? (req) => (req.headers.authorization === `Bearer ${token}` ? { id: 'attend', scopes: [''] } : null) : undefined;
|
|
200
|
+
const attend = createAttend({ store, judge, kind, sink: 'log', deadLetter: 'log', ...(threshold === undefined ? {} : { actThreshold: threshold }) });
|
|
201
|
+
server = serve(attend, store, { listen: flags.socket ? { path: flags.socket } : { port, host: flags.host ?? '127.0.0.1' }, ...(principal ? { principal } : {}) });
|
|
202
|
+
await server.ready;
|
|
203
|
+
const url = server.url();
|
|
204
|
+
out(`attend serving ${url} (ingest POST ${url}/v1/ingest, MCP ${url}/mcp)\n`);
|
|
205
|
+
const stop = new AbortController();
|
|
206
|
+
const onSignal = () => stop.abort();
|
|
207
|
+
if (io.signal)
|
|
208
|
+
io.signal.addEventListener('abort', onSignal, { once: true });
|
|
209
|
+
else {
|
|
210
|
+
process.once('SIGINT', onSignal);
|
|
211
|
+
process.once('SIGTERM', onSignal);
|
|
212
|
+
}
|
|
213
|
+
if (io.signal?.aborted)
|
|
214
|
+
stop.abort();
|
|
215
|
+
await Promise.race([new Promise((resolve) => stop.signal.addEventListener('abort', () => resolve(), { once: true })), server.closed]);
|
|
216
|
+
process.off('SIGINT', onSignal);
|
|
217
|
+
process.off('SIGTERM', onSignal);
|
|
218
|
+
const finished = server;
|
|
219
|
+
server = undefined;
|
|
220
|
+
await finished.close();
|
|
221
|
+
return 0;
|
|
222
|
+
}
|
|
223
|
+
catch (error) {
|
|
224
|
+
if (error instanceof UsageError) {
|
|
225
|
+
err(`attend: ${error.message}\n${USAGE}\n`);
|
|
226
|
+
return 2;
|
|
227
|
+
}
|
|
228
|
+
err(`attend: ${error instanceof Error ? error.message : String(error)}\n`);
|
|
229
|
+
return 1;
|
|
230
|
+
}
|
|
231
|
+
finally {
|
|
232
|
+
await server?.close().catch((error) => { err(`attend: ${error instanceof Error ? error.message : String(error)}\n`); });
|
|
233
|
+
await store?.close?.();
|
|
234
|
+
for (const close of cleanup)
|
|
235
|
+
await close().catch(() => undefined);
|
|
236
|
+
}
|
|
237
|
+
}
|
|
238
|
+
//# sourceMappingURL=serve.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"serve.js","sourceRoot":"","sources":["../../src/server/serve.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAe,MAAM,WAAW,CAAC;AACtD,OAAO,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AAEvC,OAAO,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AAC5C,OAAO,EAAE,KAAK,EAAqC,MAAM,kBAAkB,CAAC;AAC5E,OAAO,EAAE,WAAW,EAAE,MAAM,cAAc,CAAC;AAG3C,OAAO,EAAE,YAAY,EAAE,MAAM,cAAc,CAAC;AAC5C,OAAO,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AAG3C,0GAA0G;AAC1G,MAAM,UAAU,KAAK,CAAC,MAAc,EAAE,KAAiB,EAAE,OAAqB;IAC5E,MAAM,MAAM,GAAG,YAAY,CAAC,KAAK,CAAC,GAAG,EAAE,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;IAC9H,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IACzE,MAAM,MAAM,GAA8B,EAAE,iBAAiB,EAAE,YAAY,CAAC,MAAM,EAAE,MAAM,CAAC,EAAE,CAAC;IAC9F,IAAI,OAAO,CAAC,MAAM,EAAE,MAAM,KAAK,KAAK;QAAE,MAAM,CAAC,wBAAwB,CAAC,GAAG,YAAY,CAAC,MAAM,EAAE,EAAE,GAAG,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;IAC3H,MAAM,OAAO,GAAG,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;IACtD,MAAM,MAAM,GAAW,YAAY,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,GAAG,KAAK,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC3H,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;IACzC,IAAI,OAAuC,CAAC;IAC5C,IAAI,OAAkC,CAAC;IACvC,IAAI,OAAO,GAAG,EAAE,CAAC;IAEjB,MAAM,OAAO,GAAG,MAAM,CAAC,GAAG,CAAC,EAAE,MAAM,EAAE,UAAU,CAAC,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE,GAAG,OAAO,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACnJ,MAAM,KAAK,GAAG,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QAClD,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC7B,MAAM,CAAC,IAAI,CAAC,WAAW,EAAE,GAAG,EAAE;YAC5B,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC;YAC/B,OAAO,GAAG,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI;gBACnD,CAAC,CAAC,eAAe,kBAAkB,CAAC,KAAK,IAAI,EAAE,CAAC,EAAE;gBAClD,CAAC,CAAC,UAAU,KAAK,CAAC,MAAM,KAAK,MAAM,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,OAAO,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,IAAK,KAAqB,CAAC,IAAI,EAAE,CAAC;YAC9G,OAAO,EAAE,CAAC;QACZ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IACH,KAAK,CAAC,KAAK,CAAC,GAAG,EAAE,GAAG,KAAK,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC5D,MAAM,MAAM,GAAG,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;IACrF,IAAI,MAAM,IAAI,OAAO,CAAC,MAAM;QAAE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;;QAC5D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,EAAE,OAAO,CAAC,MAAM,CAAC,IAAI,IAAI,WAAW,CAAC,CAAC;IAE5E,SAAS,KAAK;QACZ,OAAO,KAAK,CAAC,KAAK,IAAI,EAAE;YACtB,UAAU,CAAC,KAAK,EAAE,CAAC;YACnB,MAAM,OAAO,CAAC;YACd,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;YACrB,MAAM,CAAC,mBAAmB,EAAE,CAAC;YAC7B,IAAI,MAAM,CAAC,SAAS;gBAAE,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;YAC1F,IAAI,OAAO;gBAAE,MAAM,OAAO,CAAC,KAAK,CAAC;QACnC,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,OAAO;QACL,GAAG;YACD,IAAI,CAAC,OAAO;gBAAE,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;YACjF,OAAO,OAAO,CAAC;QACjB,CAAC;QACD,KAAK;QACL,MAAM;QACN,KAAK;KACN,CAAC;AACJ,CAAC;AASD,MAAM,KAAK,GAAG,wMAAwM,CAAC;AACvN,MAAM,IAAI,GAAG,GAAG,KAAK;;;;;;;;;;;;;CAapB,CAAC;AAEF,MAAM,UAAW,SAAQ,KAAK;CAAG;AAEjC,MAAM,KAAK,GAA2B;IACpC,KAAK,EAAE,cAAc,EAAE,KAAK,EAAE,cAAc,EAAE,aAAa,EAAE,oBAAoB,EAAE,KAAK,EAAE,cAAc,EAAE,cAAc,EAAE,qBAAqB;IAC/I,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,aAAa,EAAE,MAAM,EAAE,eAAe,EAAE,eAAe,EAAE,sBAAsB,EAAE,KAAK,EAAE,cAAc;CAClI,CAAC;AAEF,SAAS,UAAU,CAAC,IAAc,EAAE,GAAsB;IACxD,MAAM,KAAK,GAA2B,EAAE,CAAC;IACzC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC;QACxC,MAAM,GAAG,GAAG,IAAI,CAAC,CAAC,CAAE,CAAC;QACrB,MAAM,KAAK,GAAG,0BAA0B,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACnD,IAAI,CAAC,KAAK,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,CAAE,IAAI,KAAK,CAAC;YAAE,MAAM,IAAI,UAAU,CAAC,oBAAoB,GAAG,EAAE,CAAC,CAAC;QACrF,MAAM,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC;QACpC,IAAI,KAAK,KAAK,SAAS;YAAE,MAAM,IAAI,UAAU,CAAC,KAAK,KAAK,CAAC,CAAC,CAAC,gBAAgB,CAAC,CAAC;QAC7E,KAAK,CAAC,KAAK,CAAC,CAAC,CAAE,CAAC,GAAG,KAAK,CAAC;IAC3B,CAAC;IACD,OAAO,MAAM,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;AAC3G,CAAC;AAED,SAAS,YAAY,CAAC,KAAyC;IAC7D,MAAM,QAAQ,GAAG,KAAK,CAAC,cAAc,CAAC,IAAI,aAAa,CAAC;IACxD,IAAI,CAAE,KAA2B,CAAC,QAAQ,CAAC,QAAQ,CAAC;QAAE,MAAM,IAAI,UAAU,CAAC,iCAAiC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAChI,IAAI,GAAG,GAA4B,EAAE,CAAC;IACtC,MAAM,MAAM,GAAG,KAAK,CAAC,KAAK,CAAC;IAC3B,IAAI,MAAM,KAAK,SAAS,IAAI,MAAM,KAAK,EAAE,EAAE,CAAC;QAC1C,IAAI,CAAC;YAAC,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,YAAY,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;QAAC,CAAC;QACrG,OAAO,KAAK,EAAE,CAAC;YAAC,MAAM,IAAI,UAAU,CAAC,yDAAyD,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAAC,CAAC;QAC1J,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,GAAG,KAAK,IAAI,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC;YAAE,MAAM,IAAI,UAAU,CAAC,qDAAqD,CAAC,CAAC;QAC/I,KAAK,MAAM,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;YAC/C,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,CAAE,KAA2B,CAAC,QAAQ,CAAC,IAAI,CAAC;gBAAE,MAAM,IAAI,UAAU,CAAC,aAAa,IAAI,wBAAwB,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAClK,CAAC;IACH,CAAC;IACD,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAS,CAAC;AAC1F,CAAC;AAED,KAAK,UAAU,QAAQ,CAAC,KAAyC,EAAE,GAAsB;IACvF,QAAQ,KAAK,CAAC,KAAK,IAAI,MAAM,EAAE,CAAC;QAC9B,KAAK,MAAM,EAAE,OAAO,KAAK,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,MAAM,KAAK,OAAO,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;QACzG,KAAK,KAAK,EAAE,CAAC;YACX,IAAI,CAAC,GAAG,CAAC,gBAAgB;gBAAE,MAAM,IAAI,UAAU,CAAC,uDAAuD,CAAC,CAAC;YACzG,OAAO,CAAC,MAAM,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,CAAC,gBAAgB,EAAE,CAAC,CAAC;QAChF,CAAC;QACD,KAAK,WAAW,EAAE,CAAC;YACjB,IAAI,CAAC,GAAG,CAAC,cAAc;gBAAE,MAAM,IAAI,UAAU,CAAC,2DAA2D,CAAC,CAAC;YAC3G,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC;gBAAE,MAAM,IAAI,UAAU,CAAC,+DAA+D,CAAC,CAAC;YACjH,OAAO,CAAC,MAAM,MAAM,CAAC,iBAAiB,CAAC,CAAC,CAAC,cAAc,CAAC,EAAE,MAAM,EAAE,GAAG,CAAC,cAAc,EAAE,KAAK,EAAE,KAAK,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;QACvH,CAAC;QACD,SAAS,MAAM,IAAI,UAAU,CAAC,gDAAgD,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC;IAChG,CAAC;AACH,CAAC;AAED,KAAK,UAAU,QAAQ,CAAC,IAAY,EAAE,GAAsB,EAAE,OAAgC;IAC5F,IAAI,IAAI,KAAK,QAAQ;QAAE,OAAO,WAAW,EAAE,CAAC;IAC5C,MAAM,OAAO,GAAG,CAAC,IAAY,EAAE,EAAE,CAAC,IAAI,UAAU,CAAC,WAAW,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,eAAe,IAAI,0BAA0B,IAAI,EAAE,CAAC,CAAC;IACnI,IAAI,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QAC/B,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QAC1C,IAAI,CAAC,IAAI;YAAE,MAAM,IAAI,UAAU,CAAC,yCAAyC,CAAC,CAAC;QAC3E,MAAM,EAAE,YAAY,EAAE,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,CAAC;QACrD,MAAM,EAAE,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,CAAC;QAClC,OAAO,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC,CAAC;QACrC,OAAO,CAAC,MAAM,MAAM,CAAC,cAAc,CAAC,CAAC,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;IACxD,CAAC;IACD,IAAI,IAAI,KAAK,UAAU,IAAI,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,EAAE,CAAC;QAC1F,MAAM,GAAG,GAAG,IAAI,KAAK,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,aAAa,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;QAChK,IAAI,CAAC,GAAG;YAAE,MAAM,IAAI,UAAU,CAAC,8DAA8D,CAAC,CAAC;QAC/F,MAAM,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACpE,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,EAAE,gBAAgB,EAAE,GAAG,EAAE,CAAC,CAAC;QACpE,OAAO,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC;QAC/B,MAAM,EAAE,OAAO,EAAE,aAAa,EAAE,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAAC,CAAC;QAClE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;QACpB,OAAO,aAAa,CAAC,IAAI,CAAC,CAAC;IAC7B,CAAC;IACD,IAAI,IAAI,KAAK,OAAO,IAAI,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QAChF,MAAM,GAAG,GAAG,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;QAChJ,IAAI,CAAC,GAAG;YAAE,MAAM,IAAI,UAAU,CAAC,qDAAqD,CAAC,CAAC;QACtF,MAAM,KAAK,GAAG,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC7E,MAAM,MAAM,GAAG,KAAK,CAAC,YAAY,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;QAC3C,MAAM,MAAM,CAAC,OAAO,EAAE,CAAC;QACvB,OAAO,CAAC,IAAI,CAAC,KAAK,IAAI,EAAE,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAChD,OAAO,CAAC,MAAM,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC;IACtF,CAAC;IACD,MAAM,IAAI,UAAU,CAAC,8EAA8E,IAAI,GAAG,CAAC,CAAC;AAC9G,CAAC;AAED,iFAAiF;AACjF,MAAM,CAAC,KAAK,UAAU,GAAG,CAAC,IAAc,EAAE,GAAsB,EAAE,EAAE,GAAU,EAAE;IAC9E,MAAM,GAAG,GAAG,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,KAAK,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;IAClE,MAAM,GAAG,GAAG,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,KAAK,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;IAClE,IAAI,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;QAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QAAC,OAAO,CAAC,CAAC;IAAC,CAAC;IAC5E,MAAM,OAAO,GAA4B,EAAE,CAAC;IAC5C,IAAI,KAA6B,CAAC;IAClC,IAAI,MAA4C,CAAC;IACjD,IAAI,CAAC;QACH,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,OAAO;YAAE,MAAM,IAAI,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,SAAS,CAAC,CAAC,CAAC,iBAAiB,CAAC,CAAC,CAAC,oBAAoB,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC;QAC1H,MAAM,KAAK,GAAG,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;QAC7C,MAAM,IAAI,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC;QACjC,MAAM,SAAS,GAAG,KAAK,CAAC,eAAe,CAAC,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC,CAAC;QACpG,IAAI,SAAS,KAAK,SAAS,IAAI,CAAC,CAAC,SAAS,GAAG,CAAC,IAAI,SAAS,IAAI,CAAC,CAAC;YAAE,MAAM,IAAI,UAAU,CAAC,4CAA4C,CAAC,CAAC;QACtI,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAClE,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,IAAI,GAAG,CAAC,IAAI,IAAI,GAAG,KAAK;YAAE,MAAM,IAAI,UAAU,CAAC,2CAA2C,CAAC,CAAC;QAC3H,MAAM,KAAK,GAAG,MAAM,QAAQ,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;QACzC,MAAM,MAAM,CAAC,8BAA8B,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,GAAG,MAAM,IAAI,KAAK,CAAC,yFAAyF,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC1K,KAAK,GAAG,MAAM,QAAQ,CAAC,KAAK,CAAC,KAAK,IAAI,QAAQ,EAAE,GAAG,EAAE,OAAO,CAAC,CAAC;QAC9D,MAAM,KAAK,GAAG,KAAK,CAAC,KAAK,CAAC;QAC1B,MAAM,SAAS,GAAkC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,aAAa,KAAK,UAAU,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QACxK,MAAM,MAAM,GAAG,YAAY,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,UAAU,EAAE,KAAK,EAAE,GAAG,CAAC,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC;QACrJ,MAAM,GAAG,KAAK,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,IAAI,WAAW,EAAE,EAAE,GAAG,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC;QAClK,MAAM,MAAM,CAAC,KAAK,CAAC;QACnB,MAAM,GAAG,GAAG,MAAM,CAAC,GAAG,EAAE,CAAC;QACzB,GAAG,CAAC,kBAAkB,GAAG,kBAAkB,GAAG,mBAAmB,GAAG,SAAS,CAAC,CAAC;QAC/E,MAAM,IAAI,GAAG,IAAI,eAAe,EAAE,CAAC;QACnC,MAAM,QAAQ,GAAG,GAAG,EAAE,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;QACpC,IAAI,EAAE,CAAC,MAAM;YAAE,EAAE,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;aACxE,CAAC;YAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;YAAC,OAAO,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;QAAC,CAAC;QAC7E,IAAI,EAAE,CAAC,MAAM,EAAE,OAAO;YAAE,IAAI,CAAC,KAAK,EAAE,CAAC;QACrC,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,GAAG,EAAE,CAAC,OAAO,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;QAC5I,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QAChC,OAAO,CAAC,GAAG,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;QACjC,MAAM,QAAQ,GAAG,MAAM,CAAC;QACxB,MAAM,GAAG,SAAS,CAAC;QACnB,MAAM,QAAQ,CAAC,KAAK,EAAE,CAAC;QACvB,OAAO,CAAC,CAAC;IACX,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,KAAK,YAAY,UAAU,EAAE,CAAC;YAAC,GAAG,CAAC,WAAW,KAAK,CAAC,OAAO,KAAK,KAAK,IAAI,CAAC,CAAC;YAAC,OAAO,CAAC,CAAC;QAAC,CAAC;QAC3F,GAAG,CAAC,WAAW,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC3E,OAAO,CAAC,CAAC;IACX,CAAC;YAAS,CAAC;QACT,MAAM,MAAM,EAAE,KAAK,EAAE,CAAC,KAAK,CAAC,CAAC,KAAc,EAAE,EAAE,GAAG,GAAG,CAAC,WAAW,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACjI,MAAM,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC;QACvB,KAAK,MAAM,KAAK,IAAI,OAAO;YAAE,MAAM,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;IACpE,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
export interface Sessions {
|
|
2
|
+
open(): string | null;
|
|
3
|
+
touch(id: string): boolean;
|
|
4
|
+
hold(id: string): () => void;
|
|
5
|
+
close(id: string): void;
|
|
6
|
+
dispose(): void;
|
|
7
|
+
}
|
|
8
|
+
/** attend's session table, with a cap and idle expiry. */
|
|
9
|
+
export declare function createSessions(options: {
|
|
10
|
+
max: number;
|
|
11
|
+
idleMs: number;
|
|
12
|
+
onClose?: (id: string) => void;
|
|
13
|
+
}): Sessions;
|
|
14
|
+
//# sourceMappingURL=sessions.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sessions.d.ts","sourceRoot":"","sources":["../../src/server/sessions.ts"],"names":[],"mappings":"AAEA,MAAM,WAAW,QAAQ;IACvB,IAAI,IAAI,MAAM,GAAG,IAAI,CAAC;IACtB,KAAK,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC;IAC3B,IAAI,CAAC,EAAE,EAAE,MAAM,GAAG,MAAM,IAAI,CAAC;IAC7B,KAAK,CAAC,EAAE,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,OAAO,IAAI,IAAI,CAAC;CACjB;AAED,0DAA0D;AAC1D,wBAAgB,cAAc,CAAC,OAAO,EAAE;IAAE,GAAG,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,MAAM,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC,EAAE,EAAE,MAAM,KAAK,IAAI,CAAA;CAAE,GAAG,QAAQ,CAmCjH"}
|
|
@@ -0,0 +1,54 @@
|
|
|
1
|
+
import { randomUUID } from 'node:crypto';
|
|
2
|
+
/** attend's session table, with a cap and idle expiry. */
|
|
3
|
+
export function createSessions(options) {
|
|
4
|
+
const entries = new Map();
|
|
5
|
+
function close(id) { if (entries.delete(id))
|
|
6
|
+
options.onClose?.(id); }
|
|
7
|
+
function expire() {
|
|
8
|
+
const now = Date.now();
|
|
9
|
+
for (const [id, entry] of entries)
|
|
10
|
+
if (!entry.holds && now - entry.touched >= options.idleMs)
|
|
11
|
+
close(id);
|
|
12
|
+
}
|
|
13
|
+
const timer = setInterval(expire, Math.max(1, Math.min(options.idleMs, 60_000)));
|
|
14
|
+
timer.unref();
|
|
15
|
+
return {
|
|
16
|
+
open() {
|
|
17
|
+
expire();
|
|
18
|
+
if (entries.size >= options.max)
|
|
19
|
+
return null;
|
|
20
|
+
const id = randomUUID();
|
|
21
|
+
entries.set(id, { touched: Date.now(), holds: 0 });
|
|
22
|
+
return id;
|
|
23
|
+
},
|
|
24
|
+
touch(id) {
|
|
25
|
+
const entry = entries.get(id);
|
|
26
|
+
if (!entry)
|
|
27
|
+
return false;
|
|
28
|
+
if (!entry.holds && Date.now() - entry.touched >= options.idleMs) {
|
|
29
|
+
close(id);
|
|
30
|
+
return false;
|
|
31
|
+
}
|
|
32
|
+
entry.touched = Date.now();
|
|
33
|
+
return true;
|
|
34
|
+
},
|
|
35
|
+
hold(id) {
|
|
36
|
+
const entry = entries.get(id);
|
|
37
|
+
if (!entry)
|
|
38
|
+
return () => undefined;
|
|
39
|
+
entry.holds++;
|
|
40
|
+
let released = false;
|
|
41
|
+
return () => {
|
|
42
|
+
if (released)
|
|
43
|
+
return;
|
|
44
|
+
released = true;
|
|
45
|
+
if (--entry.holds === 0)
|
|
46
|
+
entry.touched = Date.now();
|
|
47
|
+
};
|
|
48
|
+
},
|
|
49
|
+
close,
|
|
50
|
+
dispose() { clearInterval(timer); for (const id of entries.keys())
|
|
51
|
+
close(id); },
|
|
52
|
+
};
|
|
53
|
+
}
|
|
54
|
+
//# sourceMappingURL=sessions.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sessions.js","sourceRoot":"","sources":["../../src/server/sessions.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAUzC,0DAA0D;AAC1D,MAAM,UAAU,cAAc,CAAC,OAAwE;IACrG,MAAM,OAAO,GAAG,IAAI,GAAG,EAA8C,CAAC;IACtE,SAAS,KAAK,CAAC,EAAU,IAAI,IAAI,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;QAAE,OAAO,CAAC,OAAO,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IAC7E,SAAS,MAAM;QACb,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,KAAK,MAAM,CAAC,EAAE,EAAE,KAAK,CAAC,IAAI,OAAO;YAAE,IAAI,CAAC,KAAK,CAAC,KAAK,IAAI,GAAG,GAAG,KAAK,CAAC,OAAO,IAAI,OAAO,CAAC,MAAM;gBAAE,KAAK,CAAC,EAAE,CAAC,CAAC;IAC1G,CAAC;IACD,MAAM,KAAK,GAAG,WAAW,CAAC,MAAM,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC;IACjF,KAAK,CAAC,KAAK,EAAE,CAAC;IACd,OAAO;QACL,IAAI;YACF,MAAM,EAAE,CAAC;YACT,IAAI,OAAO,CAAC,IAAI,IAAI,OAAO,CAAC,GAAG;gBAAE,OAAO,IAAI,CAAC;YAC7C,MAAM,EAAE,GAAG,UAAU,EAAE,CAAC;YAAC,OAAO,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;YAAC,OAAO,EAAE,CAAC;QACzF,CAAC;QACD,KAAK,CAAC,EAAE;YACN,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAC9B,IAAI,CAAC,KAAK;gBAAE,OAAO,KAAK,CAAC;YACzB,IAAI,CAAC,KAAK,CAAC,KAAK,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC,OAAO,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;gBAAC,KAAK,CAAC,EAAE,CAAC,CAAC;gBAAC,OAAO,KAAK,CAAC;YAAC,CAAC;YAC9F,KAAK,CAAC,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAAC,OAAO,IAAI,CAAC;QAC1C,CAAC;QACD,IAAI,CAAC,EAAE;YACL,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;YAC9B,IAAI,CAAC,KAAK;gBAAE,OAAO,GAAG,EAAE,CAAC,SAAS,CAAC;YACnC,KAAK,CAAC,KAAK,EAAE,CAAC;YACd,IAAI,QAAQ,GAAG,KAAK,CAAC;YACrB,OAAO,GAAG,EAAE;gBACV,IAAI,QAAQ;oBAAE,OAAO;gBACrB,QAAQ,GAAG,IAAI,CAAC;gBAChB,IAAI,EAAE,KAAK,CAAC,KAAK,KAAK,CAAC;oBAAE,KAAK,CAAC,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACtD,CAAC,CAAC;QACJ,CAAC;QACD,KAAK;QACL,OAAO,KAAK,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,MAAM,EAAE,IAAI,OAAO,CAAC,IAAI,EAAE;YAAE,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;KAChF,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
import { type Principal, type ServerContext, type StreamEventsResult, type StreamNotification, type StreamParams } from './types.js';
|
|
2
|
+
/** Owned queue and in-flight work, for checking stream memory bounds. */
|
|
3
|
+
export declare function streamUsage(context: ServerContext): {
|
|
4
|
+
count: number;
|
|
5
|
+
queuedBytes: number;
|
|
6
|
+
writeBytes: number;
|
|
7
|
+
pageRows: number;
|
|
8
|
+
};
|
|
9
|
+
/** Sends a subscription in order, replaying the log whenever its live outbox fills. */
|
|
10
|
+
export declare function streamEvents(context: ServerContext, params: StreamParams, principal: Principal, subscriptionId: string | number, send: (notification: StreamNotification) => Promise<void>, signal: AbortSignal): Promise<StreamEventsResult>;
|
|
11
|
+
//# sourceMappingURL=stream.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"stream.d.ts","sourceRoot":"","sources":["../../src/server/stream.ts"],"names":[],"mappings":"AAKA,OAAO,EAAwB,KAAK,SAAS,EAAE,KAAK,aAAa,EAAE,KAAK,kBAAkB,EAAE,KAAK,kBAAkB,EAAE,KAAK,YAAY,EAAE,MAAM,YAAY,CAAC;AAK3J,yEAAyE;AACzE,wBAAgB,WAAW,CAAC,OAAO,EAAE,aAAa,GAAG;IAAE,KAAK,EAAE,MAAM,CAAC;IAAC,WAAW,EAAE,MAAM,CAAC;IAAC,UAAU,EAAE,MAAM,CAAC;IAAC,QAAQ,EAAE,MAAM,CAAA;CAAE,CAGhI;AAWD,uFAAuF;AACvF,wBAAsB,YAAY,CAAC,OAAO,EAAE,aAAa,EAAE,MAAM,EAAE,YAAY,EAAE,SAAS,EAAE,SAAS,EAAE,cAAc,EAAE,MAAM,GAAG,MAAM,EAAE,IAAI,EAAE,CAAC,YAAY,EAAE,kBAAkB,KAAK,OAAO,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,kBAAkB,CAAC,CA0InP"}
|
|
@@ -0,0 +1,216 @@
|
|
|
1
|
+
import { setImmediate as yieldTurn } from 'node:timers/promises';
|
|
2
|
+
import { encodeCursor, lostBefore, resolveStart } from './cursor.js';
|
|
3
|
+
import { eventsError, matches, parseSubscription, queryFor } from './events.js';
|
|
4
|
+
import { eventNotification, notificationJSON } from './fanout.js';
|
|
5
|
+
import { SUBSCRIPTION_ID_META } from './types.js';
|
|
6
|
+
const streams = new WeakMap();
|
|
7
|
+
/** Owned queue and in-flight work, for checking stream memory bounds. */
|
|
8
|
+
export function streamUsage(context) {
|
|
9
|
+
const active = streams.get(context);
|
|
10
|
+
return { count: active?.size ?? 0, queuedBytes: [...active ?? []].reduce((sum, stream) => sum + stream.bytes, 0), writeBytes: [...active ?? []].reduce((sum, stream) => sum + stream.writeBytes, 0), pageRows: [...active ?? []].reduce((sum, stream) => sum + stream.pageRows, 0) };
|
|
11
|
+
}
|
|
12
|
+
function abortable(promise, signal) {
|
|
13
|
+
return new Promise((resolve, reject) => {
|
|
14
|
+
const abort = () => reject(new DOMException('Stream aborted', 'AbortError'));
|
|
15
|
+
signal.addEventListener('abort', abort, { once: true });
|
|
16
|
+
promise.then(resolve, reject).finally(() => signal.removeEventListener('abort', abort));
|
|
17
|
+
if (signal.aborted)
|
|
18
|
+
abort();
|
|
19
|
+
});
|
|
20
|
+
}
|
|
21
|
+
/** Sends a subscription in order, replaying the log whenever its live outbox fills. */
|
|
22
|
+
export async function streamEvents(context, params, principal, subscriptionId, send, signal) {
|
|
23
|
+
const { filter, cursor, maxAgeMs } = parseSubscription(params, principal);
|
|
24
|
+
const max = context.options.maxStreamsPerPrincipal ?? 256;
|
|
25
|
+
const active = streams.get(context) ?? new Set();
|
|
26
|
+
streams.set(context, active);
|
|
27
|
+
if ([...active].filter((stream) => stream.principal === principal.id).length >= max)
|
|
28
|
+
throw eventsError('ResourceExhausted', 'Stream quota exceeded', { limit: 'streams', max });
|
|
29
|
+
const outbox = { principal: principal.id, bytes: 0, writeBytes: 0, pageRows: 0 };
|
|
30
|
+
active.add(outbox);
|
|
31
|
+
const combined = AbortSignal.any([signal, context.signal]);
|
|
32
|
+
const _meta = { [SUBSCRIPTION_ID_META]: subscriptionId };
|
|
33
|
+
const complete = { resultType: 'complete', _meta };
|
|
34
|
+
let unsubscribe = () => undefined;
|
|
35
|
+
let wake = () => undefined;
|
|
36
|
+
try {
|
|
37
|
+
const now = Date.now();
|
|
38
|
+
const epoch = await abortable(context.epoch(), combined);
|
|
39
|
+
const start = await abortable(resolveStart(context.log, epoch, cursor), combined);
|
|
40
|
+
let checked = start.after;
|
|
41
|
+
const sinceAt = maxAgeMs !== undefined && !start.fromNow ? now - maxAgeMs : undefined;
|
|
42
|
+
let agePending = sinceAt;
|
|
43
|
+
let truncated = start.truncated;
|
|
44
|
+
let behind = !start.fromNow;
|
|
45
|
+
let pendingError;
|
|
46
|
+
let lastFrame = Date.now();
|
|
47
|
+
let sent = 0;
|
|
48
|
+
const queue = [];
|
|
49
|
+
const budget = context.options.streamBufferBytes ?? 1_048_576;
|
|
50
|
+
const heartbeatMs = Math.max(1, Math.min(context.options.heartbeatMs ?? 15_000, 30_000));
|
|
51
|
+
const fanout = context.fanout;
|
|
52
|
+
const clear = () => { queue.length = 0; outbox.bytes = 0; };
|
|
53
|
+
const later = (position, than) => position !== null && (than === null || context.log.compare(position, than) > 0);
|
|
54
|
+
const positionCursor = () => encodeCursor({ epoch, position: checked ?? '' });
|
|
55
|
+
async function write(notification) {
|
|
56
|
+
combined.throwIfAborted();
|
|
57
|
+
outbox.writeBytes = Buffer.byteLength(notificationJSON(notification)) + 8;
|
|
58
|
+
try {
|
|
59
|
+
await abortable(send(notification), combined);
|
|
60
|
+
}
|
|
61
|
+
finally {
|
|
62
|
+
outbox.writeBytes = 0;
|
|
63
|
+
}
|
|
64
|
+
lastFrame = Date.now();
|
|
65
|
+
if (++sent % 64 === 0)
|
|
66
|
+
await abortable(yieldTurn(), combined);
|
|
67
|
+
}
|
|
68
|
+
async function pause(ms) {
|
|
69
|
+
let timer;
|
|
70
|
+
try {
|
|
71
|
+
await abortable(new Promise((resolve) => { wake = resolve; timer = setTimeout(resolve, ms); }), combined);
|
|
72
|
+
}
|
|
73
|
+
finally {
|
|
74
|
+
clearTimeout(timer);
|
|
75
|
+
wake = () => undefined;
|
|
76
|
+
}
|
|
77
|
+
}
|
|
78
|
+
async function reading(promise) {
|
|
79
|
+
let settled;
|
|
80
|
+
promise.then((value) => { settled = { value }; wake(); }, (error) => { settled = { error }; wake(); });
|
|
81
|
+
const outcome = () => settled;
|
|
82
|
+
let result = outcome();
|
|
83
|
+
while (result === undefined) {
|
|
84
|
+
await pause(Math.max(1, heartbeatMs - (Date.now() - lastFrame)));
|
|
85
|
+
result = outcome();
|
|
86
|
+
if (result === undefined && Date.now() - lastFrame >= heartbeatMs)
|
|
87
|
+
await write({ method: 'notifications/events/heartbeat', params: { cursor: positionCursor(), _meta } });
|
|
88
|
+
}
|
|
89
|
+
if ('error' in result)
|
|
90
|
+
throw result.error;
|
|
91
|
+
return result.value;
|
|
92
|
+
}
|
|
93
|
+
unsubscribe = fanout.subscribe(filter, (row) => {
|
|
94
|
+
if (behind || !later(row.position, checked) || combined.aborted)
|
|
95
|
+
return;
|
|
96
|
+
const notification = eventNotification(row, filter, context.options.payload?.maxEventsBytes, epoch, subscriptionId);
|
|
97
|
+
const bytes = Buffer.byteLength(notificationJSON(notification)) + 8;
|
|
98
|
+
if (outbox.bytes + bytes > budget) {
|
|
99
|
+
clear();
|
|
100
|
+
behind = true;
|
|
101
|
+
}
|
|
102
|
+
else {
|
|
103
|
+
queue.push({ position: row.position, notification, bytes });
|
|
104
|
+
outbox.bytes += bytes;
|
|
105
|
+
}
|
|
106
|
+
wake();
|
|
107
|
+
}, {
|
|
108
|
+
change: () => wake(),
|
|
109
|
+
gap: () => { clear(); behind = true; wake(); },
|
|
110
|
+
error: (error) => { pendingError = { reason: error }; behind = true; clear(); wake(); },
|
|
111
|
+
});
|
|
112
|
+
async function read() {
|
|
113
|
+
const cached = sinceAt === undefined ? fanout.cached(checked) : null;
|
|
114
|
+
return cached ?? context.log.read({ after: checked, ...queryFor(filter), ...(sinceAt === undefined ? {} : { sinceAt }), limit: 256, scanLimit: 1024 });
|
|
115
|
+
}
|
|
116
|
+
await abortable(fanout.ready, combined);
|
|
117
|
+
if (later(fanout.checked(), checked))
|
|
118
|
+
behind = true;
|
|
119
|
+
let page;
|
|
120
|
+
if (behind) {
|
|
121
|
+
try {
|
|
122
|
+
page = await abortable(read(), combined);
|
|
123
|
+
if (await abortable(lostBefore(context.log, checked, page, agePending), combined)) {
|
|
124
|
+
checked = null;
|
|
125
|
+
truncated = true;
|
|
126
|
+
page = undefined;
|
|
127
|
+
}
|
|
128
|
+
agePending = undefined;
|
|
129
|
+
}
|
|
130
|
+
catch (error) {
|
|
131
|
+
if (combined.aborted)
|
|
132
|
+
throw error;
|
|
133
|
+
pendingError = { reason: error };
|
|
134
|
+
}
|
|
135
|
+
}
|
|
136
|
+
await write({ method: 'notifications/events/active', params: { cursor: positionCursor(), truncated, _meta } });
|
|
137
|
+
while (!combined.aborted) {
|
|
138
|
+
if (!context.descriptors.some((descriptor) => descriptor.name === filter.name)) {
|
|
139
|
+
await write({ method: 'notifications/events/terminated', params: { error: eventsError('NotFound', 'Event type removed', { kind: 'event' }).toJSON(), _meta } });
|
|
140
|
+
throw new DOMException('Event type removed', 'AbortError');
|
|
141
|
+
}
|
|
142
|
+
if (pendingError !== undefined) {
|
|
143
|
+
const error = pendingError.reason;
|
|
144
|
+
pendingError = undefined;
|
|
145
|
+
await write({ method: 'notifications/events/error', params: { error: eventsError('Transient', 'Store temporarily unavailable', { reason: error instanceof Error ? error.message : String(error) }).toJSON(), _meta } });
|
|
146
|
+
await pause(Math.min(100, heartbeatMs));
|
|
147
|
+
}
|
|
148
|
+
if (behind) {
|
|
149
|
+
let lost = false;
|
|
150
|
+
try {
|
|
151
|
+
page ??= await reading(read());
|
|
152
|
+
lost = await reading(lostBefore(context.log, checked, page, agePending));
|
|
153
|
+
agePending = undefined;
|
|
154
|
+
}
|
|
155
|
+
catch (error) {
|
|
156
|
+
if (combined.aborted)
|
|
157
|
+
throw error;
|
|
158
|
+
pendingError = { reason: error };
|
|
159
|
+
continue;
|
|
160
|
+
}
|
|
161
|
+
if (lost) {
|
|
162
|
+
checked = null;
|
|
163
|
+
page = undefined;
|
|
164
|
+
await write({ method: 'notifications/events/active', params: { cursor: positionCursor(), truncated: true, _meta } });
|
|
165
|
+
continue;
|
|
166
|
+
}
|
|
167
|
+
outbox.pageRows = page.rows.length;
|
|
168
|
+
for (const row of page.rows) {
|
|
169
|
+
if (later(row.position, checked) && matches(filter, row) && (sinceAt === undefined || row.at >= sinceAt)) {
|
|
170
|
+
await write(eventNotification(row, filter, context.options.payload?.maxEventsBytes, epoch, subscriptionId));
|
|
171
|
+
checked = row.position;
|
|
172
|
+
}
|
|
173
|
+
}
|
|
174
|
+
if (later(page.checked, checked))
|
|
175
|
+
checked = page.checked;
|
|
176
|
+
const more = page.more;
|
|
177
|
+
page = undefined;
|
|
178
|
+
outbox.pageRows = 0;
|
|
179
|
+
if (Date.now() - lastFrame >= heartbeatMs)
|
|
180
|
+
await write({ method: 'notifications/events/heartbeat', params: { cursor: positionCursor(), _meta } });
|
|
181
|
+
if (more || later(fanout.checked(), checked))
|
|
182
|
+
continue;
|
|
183
|
+
behind = false;
|
|
184
|
+
continue;
|
|
185
|
+
}
|
|
186
|
+
const next = queue.shift();
|
|
187
|
+
if (next) {
|
|
188
|
+
outbox.bytes -= next.bytes;
|
|
189
|
+
if (later(next.position, checked)) {
|
|
190
|
+
await write(next.notification);
|
|
191
|
+
checked = next.position;
|
|
192
|
+
}
|
|
193
|
+
continue;
|
|
194
|
+
}
|
|
195
|
+
if (later(fanout.checked(), checked))
|
|
196
|
+
checked = fanout.checked();
|
|
197
|
+
if (Date.now() - lastFrame >= heartbeatMs)
|
|
198
|
+
await write({ method: 'notifications/events/heartbeat', params: { cursor: positionCursor(), _meta } });
|
|
199
|
+
await pause(Math.max(1, heartbeatMs - (Date.now() - lastFrame)));
|
|
200
|
+
}
|
|
201
|
+
if (signal.aborted)
|
|
202
|
+
throw new DOMException('Stream aborted', 'AbortError');
|
|
203
|
+
return complete;
|
|
204
|
+
}
|
|
205
|
+
catch (error) {
|
|
206
|
+
if (context.signal.aborted && !signal.aborted)
|
|
207
|
+
return complete;
|
|
208
|
+
throw error;
|
|
209
|
+
}
|
|
210
|
+
finally {
|
|
211
|
+
unsubscribe();
|
|
212
|
+
wake();
|
|
213
|
+
active.delete(outbox);
|
|
214
|
+
}
|
|
215
|
+
}
|
|
216
|
+
//# sourceMappingURL=stream.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"stream.js","sourceRoot":"","sources":["../../src/server/stream.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,IAAI,SAAS,EAAE,MAAM,sBAAsB,CAAC;AACjE,OAAO,EAAE,YAAY,EAAE,UAAU,EAAE,YAAY,EAAE,MAAM,aAAa,CAAC;AACrE,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE,iBAAiB,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AAChF,OAAO,EAAE,iBAAiB,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAElE,OAAO,EAAE,oBAAoB,EAA2G,MAAM,YAAY,CAAC;AAG3J,MAAM,OAAO,GAAG,IAAI,OAAO,EAA8B,CAAC;AAE1D,yEAAyE;AACzE,MAAM,UAAU,WAAW,CAAC,OAAsB;IAChD,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IACpC,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,IAAI,IAAI,CAAC,EAAE,WAAW,EAAE,CAAC,GAAG,MAAM,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,GAAG,GAAG,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC,GAAG,MAAM,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,GAAG,GAAG,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,CAAC,GAAG,MAAM,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,GAAG,GAAG,MAAM,CAAC,QAAQ,EAAE,CAAC,CAAC,EAAE,CAAC;AACvR,CAAC;AAED,SAAS,SAAS,CAAI,OAAmB,EAAE,MAAmB;IAC5D,OAAO,IAAI,OAAO,CAAI,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACxC,MAAM,KAAK,GAAG,GAAG,EAAE,CAAC,MAAM,CAAC,IAAI,YAAY,CAAC,gBAAgB,EAAE,YAAY,CAAC,CAAC,CAAC;QAC7E,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QACxD,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC;QACxF,IAAI,MAAM,CAAC,OAAO;YAAE,KAAK,EAAE,CAAC;IAC9B,CAAC,CAAC,CAAC;AACL,CAAC;AAED,uFAAuF;AACvF,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,OAAsB,EAAE,MAAoB,EAAE,SAAoB,EAAE,cAA+B,EAAE,IAAyD,EAAE,MAAmB;IACpN,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,iBAAiB,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;IAC1E,MAAM,GAAG,GAAG,OAAO,CAAC,OAAO,CAAC,sBAAsB,IAAI,GAAG,CAAC;IAC1D,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,IAAI,GAAG,EAAU,CAAC;IACzD,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;IAC7B,IAAI,CAAC,GAAG,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,SAAS,KAAK,SAAS,CAAC,EAAE,CAAC,CAAC,MAAM,IAAI,GAAG;QAAE,MAAM,WAAW,CAAC,mBAAmB,EAAE,uBAAuB,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,GAAG,EAAE,CAAC,CAAC;IAChL,MAAM,MAAM,GAAW,EAAE,SAAS,EAAE,SAAS,CAAC,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC;IACzF,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IACnB,MAAM,QAAQ,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC;IAC3D,MAAM,KAAK,GAAG,EAAE,CAAC,oBAAoB,CAAC,EAAE,cAAc,EAAE,CAAC;IACzD,MAAM,QAAQ,GAAuB,EAAE,UAAU,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC;IACvE,IAAI,WAAW,GAAe,GAAG,EAAE,CAAC,SAAS,CAAC;IAC9C,IAAI,IAAI,GAAe,GAAG,EAAE,CAAC,SAAS,CAAC;IACvC,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,KAAK,GAAG,MAAM,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,QAAQ,CAAC,CAAC;QACzD,MAAM,KAAK,GAAG,MAAM,SAAS,CAAC,YAAY,CAAC,OAAO,CAAC,GAAG,EAAE,KAAK,EAAE,MAAM,CAAC,EAAE,QAAQ,CAAC,CAAC;QAClF,IAAI,OAAO,GAAG,KAAK,CAAC,KAAK,CAAC;QAC1B,MAAM,OAAO,GAAG,QAAQ,KAAK,SAAS,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,GAAG,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC;QACtF,IAAI,UAAU,GAAG,OAAO,CAAC;QACzB,IAAI,SAAS,GAAG,KAAK,CAAC,SAAS,CAAC;QAChC,IAAI,MAAM,GAAG,CAAC,KAAK,CAAC,OAAO,CAAC;QAC5B,IAAI,YAA6C,CAAC;QAClD,IAAI,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAC3B,IAAI,IAAI,GAAG,CAAC,CAAC;QACb,MAAM,KAAK,GAA8E,EAAE,CAAC;QAC5F,MAAM,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,iBAAiB,IAAI,SAAS,CAAC;QAC9D,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC,WAAW,IAAI,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;QACzF,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;QAC9B,MAAM,KAAK,GAAG,GAAG,EAAE,GAAG,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QAC5D,MAAM,KAAK,GAAG,CAAC,QAAyB,EAAE,IAAqB,EAAE,EAAE,CAAC,QAAQ,KAAK,IAAI,IAAI,CAAC,IAAI,KAAK,IAAI,IAAI,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,QAAQ,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;QACpJ,MAAM,cAAc,GAAG,GAAG,EAAE,CAAC,YAAY,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,OAAO,IAAI,EAAE,EAAE,CAAC,CAAC;QAC9E,KAAK,UAAU,KAAK,CAAC,YAAgC;YACnD,QAAQ,CAAC,cAAc,EAAE,CAAC;YAC1B,MAAM,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,CAAC,gBAAgB,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,CAAC;YAC1E,IAAI,CAAC;gBAAC,MAAM,SAAS,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,QAAQ,CAAC,CAAC;YAAC,CAAC;oBAAS,CAAC;gBAAC,MAAM,CAAC,UAAU,GAAG,CAAC,CAAC;YAAC,CAAC;YACzF,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACvB,IAAI,EAAE,IAAI,GAAG,EAAE,KAAK,CAAC;gBAAE,MAAM,SAAS,CAAC,SAAS,EAAE,EAAE,QAAQ,CAAC,CAAC;QAChE,CAAC;QACD,KAAK,UAAU,KAAK,CAAC,EAAU;YAC7B,IAAI,KAAgD,CAAC;YACrD,IAAI,CAAC;gBACH,MAAM,SAAS,CAAC,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,EAAE,GAAG,IAAI,GAAG,OAAO,CAAC,CAAC,KAAK,GAAG,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC;YAClH,CAAC;oBAAS,CAAC;gBAAC,YAAY,CAAC,KAAK,CAAC,CAAC;gBAAC,IAAI,GAAG,GAAG,EAAE,CAAC,SAAS,CAAC;YAAC,CAAC;QAC5D,CAAC;QACD,KAAK,UAAU,OAAO,CAAI,OAAmB;YAC3C,IAAI,OAAsD,CAAC;YAC3D,OAAO,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,GAAG,OAAO,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,KAAc,EAAE,EAAE,GAAG,OAAO,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YAChH,MAAM,OAAO,GAAG,GAAG,EAAE,CAAC,OAAO,CAAC;YAC9B,IAAI,MAAM,GAAG,OAAO,EAAE,CAAC;YACvB,OAAO,MAAM,KAAK,SAAS,EAAE,CAAC;gBAC5B,MAAM,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,WAAW,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC;gBACjE,MAAM,GAAG,OAAO,EAAE,CAAC;gBACnB,IAAI,MAAM,KAAK,SAAS,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,IAAI,WAAW;oBAAE,MAAM,KAAK,CAAC,EAAE,MAAM,EAAE,gCAAgC,EAAE,MAAM,EAAE,EAAE,MAAM,EAAE,cAAc,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;YAC5K,CAAC;YACD,IAAI,OAAO,IAAI,MAAM;gBAAE,MAAM,MAAM,CAAC,KAAK,CAAC;YAC1C,OAAO,MAAM,CAAC,KAAK,CAAC;QACtB,CAAC;QACD,WAAW,GAAG,MAAM,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,EAAE;YAC7C,IAAI,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,IAAI,QAAQ,CAAC,OAAO;gBAAE,OAAO;YACxE,MAAM,YAAY,GAAG,iBAAiB,CAAC,GAAG,EAAE,MAAM,EAAE,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE,cAAc,EAAE,KAAK,EAAE,cAAc,CAAC,CAAC;YACpH,MAAM,KAAK,GAAG,MAAM,CAAC,UAAU,CAAC,gBAAgB,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,CAAC;YACpE,IAAI,MAAM,CAAC,KAAK,GAAG,KAAK,GAAG,MAAM,EAAE,CAAC;gBAAC,KAAK,EAAE,CAAC;gBAAC,MAAM,GAAG,IAAI,CAAC;YAAC,CAAC;iBACzD,CAAC;gBAAC,KAAK,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,GAAG,CAAC,QAAQ,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC,CAAC;gBAAC,MAAM,CAAC,KAAK,IAAI,KAAK,CAAC;YAAC,CAAC;YAC5F,IAAI,EAAE,CAAC;QACT,CAAC,EAAE;YACD,MAAM,EAAE,GAAG,EAAE,CAAC,IAAI,EAAE;YACpB,GAAG,EAAE,GAAG,EAAE,GAAG,KAAK,EAAE,CAAC,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;YAC9C,KAAK,EAAE,CAAC,KAAK,EAAE,EAAE,GAAG,YAAY,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;SACxF,CAAC,CAAC;QACH,KAAK,UAAU,IAAI;YACjB,MAAM,MAAM,GAAG,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;YACrE,OAAO,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,QAAQ,CAAC,MAAM,CAAC,EAAE,GAAG,CAAC,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACzJ,CAAC;QACD,MAAM,SAAS,CAAC,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;QACxC,IAAI,KAAK,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,OAAO,CAAC;YAAE,MAAM,GAAG,IAAI,CAAC;QACpD,IAAI,IAA8B,CAAC;QACnC,IAAI,MAAM,EAAE,CAAC;YACX,IAAI,CAAC;gBACH,IAAI,GAAG,MAAM,SAAS,CAAC,IAAI,EAAE,EAAE,QAAQ,CAAC,CAAC;gBACzC,IAAI,MAAM,SAAS,CAAC,UAAU,CAAC,OAAO,CAAC,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,UAAU,CAAC,EAAE,QAAQ,CAAC,EAAE,CAAC;oBAAC,OAAO,GAAG,IAAI,CAAC;oBAAC,SAAS,GAAG,IAAI,CAAC;oBAAC,IAAI,GAAG,SAAS,CAAC;gBAAC,CAAC;gBAC1I,UAAU,GAAG,SAAS,CAAC;YACzB,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBAAC,IAAI,QAAQ,CAAC,OAAO;oBAAE,MAAM,KAAK,CAAC;gBAAC,YAAY,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;YAAC,CAAC;QAC1F,CAAC;QACD,MAAM,KAAK,CAAC,EAAE,MAAM,EAAE,6BAA6B,EAAE,MAAM,EAAE,EAAE,MAAM,EAAE,cAAc,EAAE,EAAE,SAAS,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;QAC/G,OAAO,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;YACzB,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,IAAI,KAAK,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC/E,MAAM,KAAK,CAAC,EAAE,MAAM,EAAE,iCAAiC,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,WAAW,CAAC,UAAU,EAAE,oBAAoB,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;gBAChK,MAAM,IAAI,YAAY,CAAC,oBAAoB,EAAE,YAAY,CAAC,CAAC;YAC7D,CAAC;YACD,IAAI,YAAY,KAAK,SAAS,EAAE,CAAC;gBAC/B,MAAM,KAAK,GAAG,YAAY,CAAC,MAAM,CAAC;gBAAC,YAAY,GAAG,SAAS,CAAC;gBAC5D,MAAM,KAAK,CAAC,EAAE,MAAM,EAAE,4BAA4B,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,WAAW,CAAC,WAAW,EAAE,+BAA+B,EAAE,EAAE,MAAM,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;gBACxN,MAAM,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,WAAW,CAAC,CAAC,CAAC;YAC1C,CAAC;YACD,IAAI,MAAM,EAAE,CAAC;gBACX,IAAI,IAAI,GAAG,KAAK,CAAC;gBACjB,IAAI,CAAC;oBACH,IAAI,KAAK,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;oBAC/B,IAAI,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,OAAO,CAAC,GAAG,EAAE,OAAO,EAAE,IAAI,EAAE,UAAU,CAAC,CAAC,CAAC;oBACzE,UAAU,GAAG,SAAS,CAAC;gBACzB,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBAAC,IAAI,QAAQ,CAAC,OAAO;wBAAE,MAAM,KAAK,CAAC;oBAAC,YAAY,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;oBAAC,SAAS;gBAAC,CAAC;gBAClG,IAAI,IAAI,EAAE,CAAC;oBACT,OAAO,GAAG,IAAI,CAAC;oBAAC,IAAI,GAAG,SAAS,CAAC;oBACjC,MAAM,KAAK,CAAC,EAAE,MAAM,EAAE,6BAA6B,EAAE,MAAM,EAAE,EAAE,MAAM,EAAE,cAAc,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;oBACrH,SAAS;gBACX,CAAC;gBACD,MAAM,CAAC,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC;gBACnC,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC;oBAC5B,IAAI,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,IAAI,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,IAAI,CAAC,OAAO,KAAK,SAAS,IAAI,GAAG,CAAC,EAAE,IAAI,OAAO,CAAC,EAAE,CAAC;wBACzG,MAAM,KAAK,CAAC,iBAAiB,CAAC,GAAG,EAAE,MAAM,EAAE,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE,cAAc,EAAE,KAAK,EAAE,cAAc,CAAC,CAAC,CAAC;wBAC5G,OAAO,GAAG,GAAG,CAAC,QAAQ,CAAC;oBACzB,CAAC;gBACH,CAAC;gBACD,IAAI,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC;oBAAE,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;gBACzD,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,CAAC;gBACvB,IAAI,GAAG,SAAS,CAAC;gBAAC,MAAM,CAAC,QAAQ,GAAG,CAAC,CAAC;gBACtC,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,IAAI,WAAW;oBAAE,MAAM,KAAK,CAAC,EAAE,MAAM,EAAE,gCAAgC,EAAE,MAAM,EAAE,EAAE,MAAM,EAAE,cAAc,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;gBAClJ,IAAI,IAAI,IAAI,KAAK,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,OAAO,CAAC;oBAAE,SAAS;gBACvD,MAAM,GAAG,KAAK,CAAC;gBACf,SAAS;YACX,CAAC;YACD,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,EAAE,CAAC;YAC3B,IAAI,IAAI,EAAE,CAAC;gBACT,MAAM,CAAC,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC;gBAC3B,IAAI,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,CAAC,EAAE,CAAC;oBAAC,MAAM,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;oBAAC,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC;gBAAC,CAAC;gBAC/F,SAAS;YACX,CAAC;YACD,IAAI,KAAK,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,OAAO,CAAC;gBAAE,OAAO,GAAG,MAAM,CAAC,OAAO,EAAE,CAAC;YACjE,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,IAAI,WAAW;gBAAE,MAAM,KAAK,CAAC,EAAE,MAAM,EAAE,gCAAgC,EAAE,MAAM,EAAE,EAAE,MAAM,EAAE,cAAc,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,CAAC;YAClJ,MAAM,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,WAAW,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC;QACnE,CAAC;QACD,IAAI,MAAM,CAAC,OAAO;YAAE,MAAM,IAAI,YAAY,CAAC,gBAAgB,EAAE,YAAY,CAAC,CAAC;QAC3E,OAAO,QAAQ,CAAC;IAClB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,IAAI,OAAO,CAAC,MAAM,CAAC,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO;YAAE,OAAO,QAAQ,CAAC;QAC/D,MAAM,KAAK,CAAC;IACd,CAAC;YAAS,CAAC;QAAC,WAAW,EAAE,CAAC;QAAC,IAAI,EAAE,CAAC;QAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IAAC,CAAC;AAC7D,CAAC"}
|
|
@@ -0,0 +1,22 @@
|
|
|
1
|
+
import type { JsonObject } from '../core/types.js';
|
|
2
|
+
import type { Principal, ServerContext } from './types.js';
|
|
3
|
+
/** An MCP tool offered by the event server. */
|
|
4
|
+
export interface ToolDefinition {
|
|
5
|
+
name: string;
|
|
6
|
+
description: string;
|
|
7
|
+
inputSchema: JsonObject;
|
|
8
|
+
}
|
|
9
|
+
/** Text and structured representations of a tool result. */
|
|
10
|
+
export type ToolResult = {
|
|
11
|
+
content: {
|
|
12
|
+
type: 'text';
|
|
13
|
+
text: string;
|
|
14
|
+
}[];
|
|
15
|
+
structuredContent?: JsonObject;
|
|
16
|
+
isError?: boolean;
|
|
17
|
+
};
|
|
18
|
+
/** attend_wait and attend_window. */
|
|
19
|
+
export declare function listTools(): ToolDefinition[];
|
|
20
|
+
/** Calls a tool, preserving application errors as structured tool results. */
|
|
21
|
+
export declare function callTool(context: ServerContext, name: unknown, args: unknown, principal: Principal, signal: AbortSignal): Promise<ToolResult>;
|
|
22
|
+
//# sourceMappingURL=tools.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tools.d.ts","sourceRoot":"","sources":["../../src/server/tools.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAGnD,OAAO,KAAK,EAAE,SAAS,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAE3D,+CAA+C;AAC/C,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,UAAU,CAAC;CACzB;AAED,4DAA4D;AAC5D,MAAM,MAAM,UAAU,GAAG;IACvB,OAAO,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,EAAE,CAAC;IAC1C,iBAAiB,CAAC,EAAE,UAAU,CAAC;IAC/B,OAAO,CAAC,EAAE,OAAO,CAAC;CACnB,CAAC;AAEF,qCAAqC;AACrC,wBAAgB,SAAS,IAAI,cAAc,EAAE,CAO5C;AAED,8EAA8E;AAC9E,wBAAsB,QAAQ,CAAC,OAAO,EAAE,aAAa,EAAE,IAAI,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,SAAS,EAAE,SAAS,EAAE,MAAM,EAAE,WAAW,GAAG,OAAO,CAAC,UAAU,CAAC,CA8BnJ"}
|