agent-relay 2.0.23 → 2.0.24
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/dist/src/cli/index.js +66 -13
- package/package.json +18 -52
- package/packages/api-types/package.json +1 -1
- package/packages/bridge/package.json +8 -8
- package/packages/cli-tester/package.json +1 -1
- package/packages/config/package.json +2 -2
- package/packages/continuity/package.json +1 -1
- package/packages/daemon/package.json +12 -12
- package/packages/hooks/package.json +4 -4
- package/packages/mcp/package.json +2 -2
- package/packages/memory/package.json +2 -2
- package/packages/policy/package.json +2 -2
- package/packages/protocol/package.json +1 -1
- package/packages/resiliency/package.json +1 -1
- package/packages/sdk/package.json +2 -2
- package/packages/spawner/package.json +1 -1
- package/packages/state/package.json +1 -1
- package/packages/storage/package.json +2 -2
- package/packages/telemetry/package.json +1 -1
- package/packages/trajectory/package.json +2 -2
- package/packages/user-directory/package.json +2 -2
- package/packages/utils/package.json +1 -1
- package/packages/wrapper/package.json +6 -6
- package/deploy/init-db.sql +0 -5
- package/deploy/scripts/setup-fly-workspaces.sh +0 -69
- package/deploy/scripts/setup-railway.sh +0 -75
- package/dist/src/cloud/index.d.ts +0 -8
- package/dist/src/cloud/index.js +0 -8
- package/packages/cloud/dist/api/admin.d.ts +0 -8
- package/packages/cloud/dist/api/admin.js +0 -225
- package/packages/cloud/dist/api/auth.d.ts +0 -20
- package/packages/cloud/dist/api/auth.js +0 -138
- package/packages/cloud/dist/api/billing.d.ts +0 -7
- package/packages/cloud/dist/api/billing.js +0 -564
- package/packages/cloud/dist/api/cli-pty-runner.d.ts +0 -53
- package/packages/cloud/dist/api/cli-pty-runner.js +0 -175
- package/packages/cloud/dist/api/codex-auth-helper.d.ts +0 -21
- package/packages/cloud/dist/api/codex-auth-helper.js +0 -327
- package/packages/cloud/dist/api/consensus.d.ts +0 -13
- package/packages/cloud/dist/api/consensus.js +0 -261
- package/packages/cloud/dist/api/coordinators.d.ts +0 -8
- package/packages/cloud/dist/api/coordinators.js +0 -750
- package/packages/cloud/dist/api/daemons.d.ts +0 -12
- package/packages/cloud/dist/api/daemons.js +0 -535
- package/packages/cloud/dist/api/email-auth.d.ts +0 -11
- package/packages/cloud/dist/api/email-auth.js +0 -347
- package/packages/cloud/dist/api/generic-webhooks.d.ts +0 -8
- package/packages/cloud/dist/api/generic-webhooks.js +0 -129
- package/packages/cloud/dist/api/git.d.ts +0 -8
- package/packages/cloud/dist/api/git.js +0 -269
- package/packages/cloud/dist/api/github-app.d.ts +0 -11
- package/packages/cloud/dist/api/github-app.js +0 -223
- package/packages/cloud/dist/api/middleware/planLimits.d.ts +0 -43
- package/packages/cloud/dist/api/middleware/planLimits.js +0 -202
- package/packages/cloud/dist/api/monitoring.d.ts +0 -11
- package/packages/cloud/dist/api/monitoring.js +0 -578
- package/packages/cloud/dist/api/nango-auth.d.ts +0 -9
- package/packages/cloud/dist/api/nango-auth.js +0 -741
- package/packages/cloud/dist/api/onboarding.d.ts +0 -15
- package/packages/cloud/dist/api/onboarding.js +0 -679
- package/packages/cloud/dist/api/policy.d.ts +0 -8
- package/packages/cloud/dist/api/policy.js +0 -229
- package/packages/cloud/dist/api/provider-env.d.ts +0 -26
- package/packages/cloud/dist/api/provider-env.js +0 -141
- package/packages/cloud/dist/api/providers.d.ts +0 -7
- package/packages/cloud/dist/api/providers.js +0 -574
- package/packages/cloud/dist/api/repos.d.ts +0 -8
- package/packages/cloud/dist/api/repos.js +0 -577
- package/packages/cloud/dist/api/sessions.d.ts +0 -11
- package/packages/cloud/dist/api/sessions.js +0 -302
- package/packages/cloud/dist/api/teams.d.ts +0 -7
- package/packages/cloud/dist/api/teams.js +0 -281
- package/packages/cloud/dist/api/test-helpers.d.ts +0 -10
- package/packages/cloud/dist/api/test-helpers.js +0 -745
- package/packages/cloud/dist/api/usage.d.ts +0 -7
- package/packages/cloud/dist/api/usage.js +0 -111
- package/packages/cloud/dist/api/webhooks.d.ts +0 -8
- package/packages/cloud/dist/api/webhooks.js +0 -645
- package/packages/cloud/dist/api/workspaces.d.ts +0 -25
- package/packages/cloud/dist/api/workspaces.js +0 -1799
- package/packages/cloud/dist/billing/index.d.ts +0 -9
- package/packages/cloud/dist/billing/index.js +0 -9
- package/packages/cloud/dist/billing/plans.d.ts +0 -39
- package/packages/cloud/dist/billing/plans.js +0 -245
- package/packages/cloud/dist/billing/service.d.ts +0 -80
- package/packages/cloud/dist/billing/service.js +0 -388
- package/packages/cloud/dist/billing/types.d.ts +0 -141
- package/packages/cloud/dist/billing/types.js +0 -7
- package/packages/cloud/dist/config.d.ts +0 -5
- package/packages/cloud/dist/config.js +0 -5
- package/packages/cloud/dist/db/bulk-ingest.d.ts +0 -89
- package/packages/cloud/dist/db/bulk-ingest.js +0 -268
- package/packages/cloud/dist/db/drizzle.d.ts +0 -290
- package/packages/cloud/dist/db/drizzle.js +0 -1422
- package/packages/cloud/dist/db/index.d.ts +0 -56
- package/packages/cloud/dist/db/index.js +0 -70
- package/packages/cloud/dist/db/schema.d.ts +0 -5117
- package/packages/cloud/dist/db/schema.js +0 -656
- package/packages/cloud/dist/index.d.ts +0 -11
- package/packages/cloud/dist/index.js +0 -38
- package/packages/cloud/dist/provisioner/index.d.ts +0 -207
- package/packages/cloud/dist/provisioner/index.js +0 -2118
- package/packages/cloud/dist/server.d.ts +0 -17
- package/packages/cloud/dist/server.js +0 -2055
- package/packages/cloud/dist/services/auto-scaler.d.ts +0 -152
- package/packages/cloud/dist/services/auto-scaler.js +0 -439
- package/packages/cloud/dist/services/capacity-manager.d.ts +0 -148
- package/packages/cloud/dist/services/capacity-manager.js +0 -449
- package/packages/cloud/dist/services/ci-agent-spawner.d.ts +0 -49
- package/packages/cloud/dist/services/ci-agent-spawner.js +0 -373
- package/packages/cloud/dist/services/cloud-message-bus.d.ts +0 -28
- package/packages/cloud/dist/services/cloud-message-bus.js +0 -19
- package/packages/cloud/dist/services/compute-enforcement.d.ts +0 -57
- package/packages/cloud/dist/services/compute-enforcement.js +0 -175
- package/packages/cloud/dist/services/coordinator.d.ts +0 -62
- package/packages/cloud/dist/services/coordinator.js +0 -389
- package/packages/cloud/dist/services/index.d.ts +0 -17
- package/packages/cloud/dist/services/index.js +0 -25
- package/packages/cloud/dist/services/intro-expiration.d.ts +0 -60
- package/packages/cloud/dist/services/intro-expiration.js +0 -252
- package/packages/cloud/dist/services/mention-handler.d.ts +0 -65
- package/packages/cloud/dist/services/mention-handler.js +0 -405
- package/packages/cloud/dist/services/nango.d.ts +0 -219
- package/packages/cloud/dist/services/nango.js +0 -424
- package/packages/cloud/dist/services/persistence.d.ts +0 -131
- package/packages/cloud/dist/services/persistence.js +0 -200
- package/packages/cloud/dist/services/planLimits.d.ts +0 -147
- package/packages/cloud/dist/services/planLimits.js +0 -335
- package/packages/cloud/dist/services/presence-registry.d.ts +0 -56
- package/packages/cloud/dist/services/presence-registry.js +0 -91
- package/packages/cloud/dist/services/scaling-orchestrator.d.ts +0 -159
- package/packages/cloud/dist/services/scaling-orchestrator.js +0 -502
- package/packages/cloud/dist/services/scaling-policy.d.ts +0 -121
- package/packages/cloud/dist/services/scaling-policy.js +0 -415
- package/packages/cloud/dist/services/ssh-security.d.ts +0 -31
- package/packages/cloud/dist/services/ssh-security.js +0 -63
- package/packages/cloud/dist/services/workspace-keepalive.d.ts +0 -76
- package/packages/cloud/dist/services/workspace-keepalive.js +0 -234
- package/packages/cloud/dist/shims/consensus.d.ts +0 -23
- package/packages/cloud/dist/shims/consensus.js +0 -5
- package/packages/cloud/dist/webhooks/index.d.ts +0 -24
- package/packages/cloud/dist/webhooks/index.js +0 -29
- package/packages/cloud/dist/webhooks/parsers/github.d.ts +0 -8
- package/packages/cloud/dist/webhooks/parsers/github.js +0 -234
- package/packages/cloud/dist/webhooks/parsers/index.d.ts +0 -23
- package/packages/cloud/dist/webhooks/parsers/index.js +0 -30
- package/packages/cloud/dist/webhooks/parsers/linear.d.ts +0 -9
- package/packages/cloud/dist/webhooks/parsers/linear.js +0 -258
- package/packages/cloud/dist/webhooks/parsers/slack.d.ts +0 -9
- package/packages/cloud/dist/webhooks/parsers/slack.js +0 -214
- package/packages/cloud/dist/webhooks/responders/github.d.ts +0 -8
- package/packages/cloud/dist/webhooks/responders/github.js +0 -73
- package/packages/cloud/dist/webhooks/responders/index.d.ts +0 -23
- package/packages/cloud/dist/webhooks/responders/index.js +0 -30
- package/packages/cloud/dist/webhooks/responders/linear.d.ts +0 -9
- package/packages/cloud/dist/webhooks/responders/linear.js +0 -149
- package/packages/cloud/dist/webhooks/responders/slack.d.ts +0 -20
- package/packages/cloud/dist/webhooks/responders/slack.js +0 -178
- package/packages/cloud/dist/webhooks/router.d.ts +0 -25
- package/packages/cloud/dist/webhooks/router.js +0 -504
- package/packages/cloud/dist/webhooks/rules-engine.d.ts +0 -24
- package/packages/cloud/dist/webhooks/rules-engine.js +0 -287
- package/packages/cloud/dist/webhooks/types.d.ts +0 -186
- package/packages/cloud/dist/webhooks/types.js +0 -8
- package/packages/cloud/package.json +0 -60
- package/scripts/run-migrations.js +0 -43
- package/scripts/setup-stripe-products.ts +0 -312
- package/scripts/verify-schema.js +0 -134
|
@@ -1,268 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* Bulk Ingest Utilities
|
|
3
|
-
*
|
|
4
|
-
* Optimized bulk insert operations for high-volume message sync.
|
|
5
|
-
* Uses raw SQL for performance instead of ORM-generated queries.
|
|
6
|
-
*
|
|
7
|
-
* Key optimizations:
|
|
8
|
-
* - Multi-row INSERT with VALUES for batches
|
|
9
|
-
* - Streaming COPY for very large batches (>1000 rows)
|
|
10
|
-
* - Proper JSONB serialization
|
|
11
|
-
* - Connection reuse via pool
|
|
12
|
-
* - Chunk processing for memory efficiency
|
|
13
|
-
*/
|
|
14
|
-
export const DEFAULT_POOL_CONFIG = {
|
|
15
|
-
max: 20,
|
|
16
|
-
idleTimeoutMillis: 30000,
|
|
17
|
-
connectionTimeoutMillis: 10000,
|
|
18
|
-
};
|
|
19
|
-
/**
|
|
20
|
-
* Bulk insert messages using optimized multi-row INSERT.
|
|
21
|
-
*
|
|
22
|
-
* Uses ON CONFLICT DO NOTHING for deduplication.
|
|
23
|
-
* Much faster than individual inserts for batches.
|
|
24
|
-
*
|
|
25
|
-
* @param pool Database connection pool
|
|
26
|
-
* @param messages Messages to insert
|
|
27
|
-
* @param chunkSize Number of rows per INSERT statement (default: 100)
|
|
28
|
-
*/
|
|
29
|
-
export async function bulkInsertMessages(pool, messages, chunkSize = 100) {
|
|
30
|
-
if (messages.length === 0) {
|
|
31
|
-
return { inserted: 0, duplicates: 0, errors: 0, durationMs: 0 };
|
|
32
|
-
}
|
|
33
|
-
const startTime = Date.now();
|
|
34
|
-
let totalInserted = 0;
|
|
35
|
-
let totalErrors = 0;
|
|
36
|
-
// Process in chunks to avoid query size limits
|
|
37
|
-
for (let i = 0; i < messages.length; i += chunkSize) {
|
|
38
|
-
const chunk = messages.slice(i, i + chunkSize);
|
|
39
|
-
const result = await insertMessageChunk(pool, chunk);
|
|
40
|
-
totalInserted += result.inserted;
|
|
41
|
-
totalErrors += result.errors;
|
|
42
|
-
}
|
|
43
|
-
return {
|
|
44
|
-
inserted: totalInserted,
|
|
45
|
-
duplicates: messages.length - totalInserted - totalErrors,
|
|
46
|
-
errors: totalErrors,
|
|
47
|
-
durationMs: Date.now() - startTime,
|
|
48
|
-
};
|
|
49
|
-
}
|
|
50
|
-
/**
|
|
51
|
-
* Insert a chunk of messages with multi-row VALUES.
|
|
52
|
-
*/
|
|
53
|
-
async function insertMessageChunk(pool, messages) {
|
|
54
|
-
if (messages.length === 0) {
|
|
55
|
-
return { inserted: 0, errors: 0 };
|
|
56
|
-
}
|
|
57
|
-
// Build parameterized multi-row INSERT
|
|
58
|
-
const columns = [
|
|
59
|
-
'workspace_id',
|
|
60
|
-
'daemon_id',
|
|
61
|
-
'original_id',
|
|
62
|
-
'from_agent',
|
|
63
|
-
'to_agent',
|
|
64
|
-
'body',
|
|
65
|
-
'kind',
|
|
66
|
-
'topic',
|
|
67
|
-
'thread',
|
|
68
|
-
'channel',
|
|
69
|
-
'is_broadcast',
|
|
70
|
-
'is_urgent',
|
|
71
|
-
'data',
|
|
72
|
-
'payload_meta',
|
|
73
|
-
'message_ts',
|
|
74
|
-
'expires_at',
|
|
75
|
-
];
|
|
76
|
-
const values = [];
|
|
77
|
-
const placeholders = [];
|
|
78
|
-
for (let i = 0; i < messages.length; i++) {
|
|
79
|
-
const msg = messages[i];
|
|
80
|
-
const offset = i * columns.length;
|
|
81
|
-
const rowPlaceholders = columns.map((_, j) => `$${offset + j + 1}`);
|
|
82
|
-
placeholders.push(`(${rowPlaceholders.join(', ')})`);
|
|
83
|
-
values.push(msg.workspaceId, msg.daemonId ?? null, msg.originalId, msg.fromAgent, msg.toAgent, msg.body, msg.kind ?? 'message', msg.topic ?? null, msg.thread ?? null, msg.channel ?? null, msg.isBroadcast ?? false, msg.isUrgent ?? false, msg.data ? JSON.stringify(msg.data) : null, msg.payloadMeta ? JSON.stringify(msg.payloadMeta) : null, msg.messageTs, msg.expiresAt ?? null);
|
|
84
|
-
}
|
|
85
|
-
const query = `
|
|
86
|
-
INSERT INTO agent_messages (${columns.join(', ')})
|
|
87
|
-
VALUES ${placeholders.join(', ')}
|
|
88
|
-
ON CONFLICT (workspace_id, original_id) DO NOTHING
|
|
89
|
-
`;
|
|
90
|
-
try {
|
|
91
|
-
const result = await pool.query(query, values);
|
|
92
|
-
return { inserted: result.rowCount ?? 0, errors: 0 };
|
|
93
|
-
}
|
|
94
|
-
catch (err) {
|
|
95
|
-
console.error('[bulk-ingest] Chunk insert failed:', err);
|
|
96
|
-
return { inserted: 0, errors: messages.length };
|
|
97
|
-
}
|
|
98
|
-
}
|
|
99
|
-
/**
|
|
100
|
-
* Streaming bulk insert using staging table for very large batches.
|
|
101
|
-
*
|
|
102
|
-
* Uses chunked multi-row INSERT into a temp staging table,
|
|
103
|
-
* then a single INSERT SELECT for deduplication.
|
|
104
|
-
* This avoids holding all data in memory and is efficient for large batches.
|
|
105
|
-
*
|
|
106
|
-
* @param pool Database connection pool
|
|
107
|
-
* @param messages Messages to insert
|
|
108
|
-
*/
|
|
109
|
-
export async function streamingBulkInsert(pool, messages) {
|
|
110
|
-
if (messages.length === 0) {
|
|
111
|
-
return { inserted: 0, duplicates: 0, errors: 0, durationMs: 0 };
|
|
112
|
-
}
|
|
113
|
-
const startTime = Date.now();
|
|
114
|
-
const client = await pool.connect();
|
|
115
|
-
try {
|
|
116
|
-
// Start transaction
|
|
117
|
-
await client.query('BEGIN');
|
|
118
|
-
// Create temp staging table
|
|
119
|
-
await client.query(`
|
|
120
|
-
CREATE TEMP TABLE _staging_messages (
|
|
121
|
-
workspace_id UUID NOT NULL,
|
|
122
|
-
daemon_id UUID,
|
|
123
|
-
original_id VARCHAR(255) NOT NULL,
|
|
124
|
-
from_agent VARCHAR(255) NOT NULL,
|
|
125
|
-
to_agent VARCHAR(255) NOT NULL,
|
|
126
|
-
body TEXT NOT NULL,
|
|
127
|
-
kind VARCHAR(50) DEFAULT 'message',
|
|
128
|
-
topic VARCHAR(255),
|
|
129
|
-
thread VARCHAR(255),
|
|
130
|
-
channel VARCHAR(255),
|
|
131
|
-
is_broadcast BOOLEAN DEFAULT false,
|
|
132
|
-
is_urgent BOOLEAN DEFAULT false,
|
|
133
|
-
data JSONB,
|
|
134
|
-
payload_meta JSONB,
|
|
135
|
-
message_ts TIMESTAMP NOT NULL,
|
|
136
|
-
expires_at TIMESTAMP
|
|
137
|
-
) ON COMMIT DROP
|
|
138
|
-
`);
|
|
139
|
-
// Insert into staging table in chunks
|
|
140
|
-
const chunkSize = 200;
|
|
141
|
-
for (let i = 0; i < messages.length; i += chunkSize) {
|
|
142
|
-
const chunk = messages.slice(i, i + chunkSize);
|
|
143
|
-
await insertStagingChunk(client, chunk);
|
|
144
|
-
}
|
|
145
|
-
// Insert from staging to main table with dedup
|
|
146
|
-
const result = await client.query(`
|
|
147
|
-
INSERT INTO agent_messages (
|
|
148
|
-
workspace_id, daemon_id, original_id, from_agent, to_agent, body,
|
|
149
|
-
kind, topic, thread, channel, is_broadcast, is_urgent,
|
|
150
|
-
data, payload_meta, message_ts, expires_at
|
|
151
|
-
)
|
|
152
|
-
SELECT
|
|
153
|
-
workspace_id, daemon_id, original_id, from_agent, to_agent, body,
|
|
154
|
-
kind, topic, thread, channel, is_broadcast, is_urgent,
|
|
155
|
-
data, payload_meta, message_ts, expires_at
|
|
156
|
-
FROM _staging_messages
|
|
157
|
-
ON CONFLICT (workspace_id, original_id) DO NOTHING
|
|
158
|
-
`);
|
|
159
|
-
await client.query('COMMIT');
|
|
160
|
-
return {
|
|
161
|
-
inserted: result.rowCount ?? 0,
|
|
162
|
-
duplicates: messages.length - (result.rowCount ?? 0),
|
|
163
|
-
errors: 0,
|
|
164
|
-
durationMs: Date.now() - startTime,
|
|
165
|
-
};
|
|
166
|
-
}
|
|
167
|
-
catch (err) {
|
|
168
|
-
await client.query('ROLLBACK').catch(() => { });
|
|
169
|
-
console.error('[bulk-ingest] Streaming insert failed:', err);
|
|
170
|
-
return {
|
|
171
|
-
inserted: 0,
|
|
172
|
-
duplicates: 0,
|
|
173
|
-
errors: messages.length,
|
|
174
|
-
durationMs: Date.now() - startTime,
|
|
175
|
-
};
|
|
176
|
-
}
|
|
177
|
-
finally {
|
|
178
|
-
client.release();
|
|
179
|
-
}
|
|
180
|
-
}
|
|
181
|
-
/**
|
|
182
|
-
* Insert a chunk of messages into the staging table.
|
|
183
|
-
*/
|
|
184
|
-
async function insertStagingChunk(client, messages) {
|
|
185
|
-
if (messages.length === 0)
|
|
186
|
-
return;
|
|
187
|
-
const columns = [
|
|
188
|
-
'workspace_id',
|
|
189
|
-
'daemon_id',
|
|
190
|
-
'original_id',
|
|
191
|
-
'from_agent',
|
|
192
|
-
'to_agent',
|
|
193
|
-
'body',
|
|
194
|
-
'kind',
|
|
195
|
-
'topic',
|
|
196
|
-
'thread',
|
|
197
|
-
'channel',
|
|
198
|
-
'is_broadcast',
|
|
199
|
-
'is_urgent',
|
|
200
|
-
'data',
|
|
201
|
-
'payload_meta',
|
|
202
|
-
'message_ts',
|
|
203
|
-
'expires_at',
|
|
204
|
-
];
|
|
205
|
-
const values = [];
|
|
206
|
-
const placeholders = [];
|
|
207
|
-
for (let i = 0; i < messages.length; i++) {
|
|
208
|
-
const msg = messages[i];
|
|
209
|
-
const offset = i * columns.length;
|
|
210
|
-
const rowPlaceholders = columns.map((_, j) => `$${offset + j + 1}`);
|
|
211
|
-
placeholders.push(`(${rowPlaceholders.join(', ')})`);
|
|
212
|
-
values.push(msg.workspaceId, msg.daemonId ?? null, msg.originalId, msg.fromAgent, msg.toAgent, msg.body, msg.kind ?? 'message', msg.topic ?? null, msg.thread ?? null, msg.channel ?? null, msg.isBroadcast ?? false, msg.isUrgent ?? false, msg.data ? JSON.stringify(msg.data) : null, msg.payloadMeta ? JSON.stringify(msg.payloadMeta) : null, msg.messageTs, msg.expiresAt ?? null);
|
|
213
|
-
}
|
|
214
|
-
await client.query(`INSERT INTO _staging_messages (${columns.join(', ')}) VALUES ${placeholders.join(', ')}`, values);
|
|
215
|
-
}
|
|
216
|
-
/**
|
|
217
|
-
* Optimized bulk insert that chooses strategy based on batch size.
|
|
218
|
-
*
|
|
219
|
-
* - Small batches (<100): Use regular ORM insert
|
|
220
|
-
* - Medium batches (100-1000): Use multi-row INSERT
|
|
221
|
-
* - Large batches (>1000): Use streaming COPY
|
|
222
|
-
*
|
|
223
|
-
* @param pool Database connection pool
|
|
224
|
-
* @param messages Messages to insert
|
|
225
|
-
*/
|
|
226
|
-
export async function optimizedBulkInsert(pool, messages) {
|
|
227
|
-
const count = messages.length;
|
|
228
|
-
if (count === 0) {
|
|
229
|
-
return { inserted: 0, duplicates: 0, errors: 0, durationMs: 0 };
|
|
230
|
-
}
|
|
231
|
-
// For very large batches, use streaming COPY
|
|
232
|
-
if (count > 1000) {
|
|
233
|
-
return streamingBulkInsert(pool, messages);
|
|
234
|
-
}
|
|
235
|
-
// For medium batches, use multi-row INSERT
|
|
236
|
-
return bulkInsertMessages(pool, messages);
|
|
237
|
-
}
|
|
238
|
-
/**
|
|
239
|
-
* Get pool statistics for monitoring.
|
|
240
|
-
*/
|
|
241
|
-
export function getPoolStats(pool) {
|
|
242
|
-
return {
|
|
243
|
-
total: pool.totalCount,
|
|
244
|
-
idle: pool.idleCount,
|
|
245
|
-
waiting: pool.waitingCount,
|
|
246
|
-
};
|
|
247
|
-
}
|
|
248
|
-
/**
|
|
249
|
-
* Health check for the connection pool.
|
|
250
|
-
*/
|
|
251
|
-
export async function checkPoolHealth(pool) {
|
|
252
|
-
const start = Date.now();
|
|
253
|
-
try {
|
|
254
|
-
await pool.query('SELECT 1');
|
|
255
|
-
return {
|
|
256
|
-
healthy: true,
|
|
257
|
-
latencyMs: Date.now() - start,
|
|
258
|
-
};
|
|
259
|
-
}
|
|
260
|
-
catch (err) {
|
|
261
|
-
return {
|
|
262
|
-
healthy: false,
|
|
263
|
-
latencyMs: Date.now() - start,
|
|
264
|
-
error: String(err),
|
|
265
|
-
};
|
|
266
|
-
}
|
|
267
|
-
}
|
|
268
|
-
//# sourceMappingURL=bulk-ingest.js.map
|
|
@@ -1,290 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* Agent Relay Cloud - Drizzle Database Client
|
|
3
|
-
*
|
|
4
|
-
* Type-safe database access using Drizzle ORM.
|
|
5
|
-
* Use this instead of the raw pg client for new code.
|
|
6
|
-
*/
|
|
7
|
-
import { Pool } from 'pg';
|
|
8
|
-
import * as schema from './schema.js';
|
|
9
|
-
export type { User, NewUser, UserEmail, NewUserEmail, GitHubInstallation, NewGitHubInstallation, Credential, NewCredential, Workspace, NewWorkspace, WorkspaceConfig, WorkspaceMember, NewWorkspaceMember, ProjectGroup, NewProjectGroup, CoordinatorAgentConfig, ProjectAgentConfig, Repository, NewRepository, LinkedDaemon, NewLinkedDaemon, Subscription, NewSubscription, UsageRecord, NewUsageRecord, AgentSession, NewAgentSession, AgentSummary, NewAgentSummary, } from './schema.js';
|
|
10
|
-
export * from './schema.js';
|
|
11
|
-
/**
|
|
12
|
-
* Get the raw connection pool for bulk operations.
|
|
13
|
-
* Use this for optimized bulk inserts that bypass the ORM.
|
|
14
|
-
*/
|
|
15
|
-
export declare function getRawPool(): Pool;
|
|
16
|
-
export declare function getDb(): import("drizzle-orm/node-postgres").NodePgDatabase<Record<string, unknown>> & {
|
|
17
|
-
$client: Pool;
|
|
18
|
-
};
|
|
19
|
-
export interface UserQueries {
|
|
20
|
-
findById(id: string): Promise<schema.User | null>;
|
|
21
|
-
findByGithubId(githubId: string): Promise<schema.User | null>;
|
|
22
|
-
findByGithubUsername(username: string): Promise<schema.User | null>;
|
|
23
|
-
findByEmail(email: string): Promise<schema.User | null>;
|
|
24
|
-
findByNangoConnectionId(connectionId: string): Promise<schema.User | null>;
|
|
25
|
-
findByIncomingConnectionId(connectionId: string): Promise<schema.User | null>;
|
|
26
|
-
findByPlan(plan: string): Promise<schema.User[]>;
|
|
27
|
-
upsert(data: schema.NewUser): Promise<schema.User>;
|
|
28
|
-
update(id: string, data: Partial<Omit<schema.User, 'id' | 'createdAt'>>): Promise<void>;
|
|
29
|
-
completeOnboarding(userId: string): Promise<void>;
|
|
30
|
-
clearIncomingConnectionId(userId: string): Promise<void>;
|
|
31
|
-
setPendingInstallationRequest(userId: string): Promise<void>;
|
|
32
|
-
clearPendingInstallationRequest(userId: string): Promise<void>;
|
|
33
|
-
createEmailUser(data: {
|
|
34
|
-
email: string;
|
|
35
|
-
passwordHash: string;
|
|
36
|
-
displayName?: string;
|
|
37
|
-
}): Promise<schema.User>;
|
|
38
|
-
verifyEmail(userId: string): Promise<void>;
|
|
39
|
-
setEmailVerificationToken(userId: string, token: string, expires: Date): Promise<void>;
|
|
40
|
-
findByEmailVerificationToken(token: string): Promise<schema.User | null>;
|
|
41
|
-
updatePassword(userId: string, passwordHash: string): Promise<void>;
|
|
42
|
-
}
|
|
43
|
-
export declare const userQueries: UserQueries;
|
|
44
|
-
export interface UserEmailQueries {
|
|
45
|
-
/** Find all emails for a user */
|
|
46
|
-
findByUserId(userId: string): Promise<schema.UserEmail[]>;
|
|
47
|
-
/** Find a user by any of their linked emails (for login reconciliation) */
|
|
48
|
-
findUserByEmail(email: string): Promise<schema.User | null>;
|
|
49
|
-
/** Add or update an email for a user */
|
|
50
|
-
upsert(data: {
|
|
51
|
-
userId: string;
|
|
52
|
-
email: string;
|
|
53
|
-
verified: boolean;
|
|
54
|
-
primary: boolean;
|
|
55
|
-
source?: string;
|
|
56
|
-
}): Promise<schema.UserEmail>;
|
|
57
|
-
/** Sync all emails from GitHub for a user (replaces GitHub-sourced emails) */
|
|
58
|
-
syncFromGitHub(userId: string, emails: Array<{
|
|
59
|
-
email: string;
|
|
60
|
-
verified: boolean;
|
|
61
|
-
primary: boolean;
|
|
62
|
-
}>): Promise<void>;
|
|
63
|
-
/** Delete a specific email for a user */
|
|
64
|
-
delete(userId: string, email: string): Promise<void>;
|
|
65
|
-
/** Check if an email is already linked to another user */
|
|
66
|
-
isEmailLinkedToOtherUser(email: string, excludeUserId: string): Promise<boolean>;
|
|
67
|
-
}
|
|
68
|
-
export declare const userEmailQueries: UserEmailQueries;
|
|
69
|
-
export interface GitHubInstallationQueries {
|
|
70
|
-
findById(id: string): Promise<schema.GitHubInstallation | null>;
|
|
71
|
-
findByInstallationId(installationId: string): Promise<schema.GitHubInstallation | null>;
|
|
72
|
-
findByAccountLogin(accountLogin: string): Promise<schema.GitHubInstallation | null>;
|
|
73
|
-
findByInstalledBy(userId: string): Promise<schema.GitHubInstallation[]>;
|
|
74
|
-
findAll(): Promise<schema.GitHubInstallation[]>;
|
|
75
|
-
upsert(data: schema.NewGitHubInstallation): Promise<schema.GitHubInstallation>;
|
|
76
|
-
updatePermissions(installationId: string, permissions: Record<string, string>, events: string[]): Promise<void>;
|
|
77
|
-
suspend(installationId: string, suspendedBy: string): Promise<void>;
|
|
78
|
-
unsuspend(installationId: string): Promise<void>;
|
|
79
|
-
delete(installationId: string): Promise<void>;
|
|
80
|
-
}
|
|
81
|
-
export declare const githubInstallationQueries: GitHubInstallationQueries;
|
|
82
|
-
export interface CredentialQueries {
|
|
83
|
-
/** Get all credentials for a user (across all workspaces) */
|
|
84
|
-
findByUserId(userId: string): Promise<schema.Credential[]>;
|
|
85
|
-
/** Get all credentials for a user in a specific workspace */
|
|
86
|
-
findByUserAndWorkspace(userId: string, workspaceId: string): Promise<schema.Credential[]>;
|
|
87
|
-
/** Get a specific provider credential for a user in a workspace */
|
|
88
|
-
findByUserWorkspaceAndProvider(userId: string, workspaceId: string, provider: string): Promise<schema.Credential | null>;
|
|
89
|
-
/** @deprecated Use findByUserWorkspaceAndProvider for workspace-scoped queries */
|
|
90
|
-
findByUserAndProvider(userId: string, provider: string): Promise<schema.Credential | null>;
|
|
91
|
-
/** Upsert credential (unique on userId + provider + workspaceId) */
|
|
92
|
-
upsert(data: schema.NewCredential): Promise<schema.Credential>;
|
|
93
|
-
/** Delete credential for a specific workspace */
|
|
94
|
-
deleteForWorkspace(userId: string, workspaceId: string, provider: string): Promise<void>;
|
|
95
|
-
/** @deprecated Use deleteForWorkspace for workspace-scoped deletions */
|
|
96
|
-
delete(userId: string, provider: string): Promise<void>;
|
|
97
|
-
}
|
|
98
|
-
export declare const credentialQueries: CredentialQueries;
|
|
99
|
-
export interface WorkspaceQueries {
|
|
100
|
-
findById(id: string): Promise<schema.Workspace | null>;
|
|
101
|
-
findByUserId(userId: string): Promise<schema.Workspace[]>;
|
|
102
|
-
findByCustomDomain(domain: string): Promise<schema.Workspace | null>;
|
|
103
|
-
findByRepoFullName(repoFullName: string): Promise<schema.Workspace | null>;
|
|
104
|
-
findAll(): Promise<schema.Workspace[]>;
|
|
105
|
-
create(data: schema.NewWorkspace): Promise<schema.Workspace>;
|
|
106
|
-
update(id: string, data: Partial<Pick<schema.Workspace, 'name' | 'config'>>): Promise<void>;
|
|
107
|
-
updateStatus(id: string, status: string, options?: {
|
|
108
|
-
computeId?: string;
|
|
109
|
-
publicUrl?: string;
|
|
110
|
-
errorMessage?: string;
|
|
111
|
-
}): Promise<void>;
|
|
112
|
-
updateConfig(id: string, config: schema.WorkspaceConfig): Promise<void>;
|
|
113
|
-
setCustomDomain(id: string, customDomain: string, status?: string): Promise<void>;
|
|
114
|
-
updateCustomDomainStatus(id: string, status: string): Promise<void>;
|
|
115
|
-
removeCustomDomain(id: string): Promise<void>;
|
|
116
|
-
delete(id: string): Promise<void>;
|
|
117
|
-
}
|
|
118
|
-
export declare const workspaceQueries: WorkspaceQueries;
|
|
119
|
-
export interface WorkspaceMemberQueries {
|
|
120
|
-
findByWorkspaceId(workspaceId: string): Promise<schema.WorkspaceMember[]>;
|
|
121
|
-
findByUserId(userId: string): Promise<schema.WorkspaceMember[]>;
|
|
122
|
-
findMembership(workspaceId: string, userId: string): Promise<schema.WorkspaceMember | null>;
|
|
123
|
-
addMember(data: {
|
|
124
|
-
workspaceId: string;
|
|
125
|
-
userId: string;
|
|
126
|
-
role: string;
|
|
127
|
-
invitedBy: string;
|
|
128
|
-
}): Promise<schema.WorkspaceMember>;
|
|
129
|
-
acceptInvite(workspaceId: string, userId: string): Promise<void>;
|
|
130
|
-
updateRole(workspaceId: string, userId: string, role: string): Promise<void>;
|
|
131
|
-
removeMember(workspaceId: string, userId: string): Promise<void>;
|
|
132
|
-
getPendingInvites(userId: string): Promise<schema.WorkspaceMember[]>;
|
|
133
|
-
isOwner(workspaceId: string, userId: string): Promise<boolean>;
|
|
134
|
-
canEdit(workspaceId: string, userId: string): Promise<boolean>;
|
|
135
|
-
canView(workspaceId: string, userId: string): Promise<boolean>;
|
|
136
|
-
}
|
|
137
|
-
export declare const workspaceMemberQueries: WorkspaceMemberQueries;
|
|
138
|
-
export interface DaemonAgentInfo {
|
|
139
|
-
daemonId: string;
|
|
140
|
-
daemonName: string;
|
|
141
|
-
machineId: string;
|
|
142
|
-
agents: Array<{
|
|
143
|
-
name: string;
|
|
144
|
-
status: string;
|
|
145
|
-
}>;
|
|
146
|
-
}
|
|
147
|
-
export interface DaemonUpdate {
|
|
148
|
-
type: string;
|
|
149
|
-
payload: unknown;
|
|
150
|
-
}
|
|
151
|
-
export interface LinkedDaemonQueries {
|
|
152
|
-
findById(id: string): Promise<schema.LinkedDaemon | null>;
|
|
153
|
-
findByUserId(userId: string): Promise<schema.LinkedDaemon[]>;
|
|
154
|
-
findByWorkspaceId(workspaceId: string): Promise<schema.LinkedDaemon[]>;
|
|
155
|
-
findByMachineId(userId: string, machineId: string): Promise<schema.LinkedDaemon | null>;
|
|
156
|
-
findByApiKeyHash(apiKeyHash: string): Promise<schema.LinkedDaemon | null>;
|
|
157
|
-
create(data: schema.NewLinkedDaemon): Promise<schema.LinkedDaemon>;
|
|
158
|
-
update(id: string, data: Partial<schema.LinkedDaemon>): Promise<void>;
|
|
159
|
-
updateLastSeen(id: string): Promise<void>;
|
|
160
|
-
delete(id: string): Promise<void>;
|
|
161
|
-
markStale(): Promise<number>;
|
|
162
|
-
getAllAgentsForUser(userId: string): Promise<DaemonAgentInfo[]>;
|
|
163
|
-
getPendingUpdates(id: string): Promise<DaemonUpdate[]>;
|
|
164
|
-
queueUpdate(id: string, update: DaemonUpdate): Promise<void>;
|
|
165
|
-
queueMessage(id: string, message: Record<string, unknown>): Promise<void>;
|
|
166
|
-
getQueuedMessages(id: string): Promise<Array<Record<string, unknown>>>;
|
|
167
|
-
clearMessageQueue(id: string): Promise<void>;
|
|
168
|
-
}
|
|
169
|
-
export declare const linkedDaemonQueries: LinkedDaemonQueries;
|
|
170
|
-
export interface ProjectGroupWithRepositories extends schema.ProjectGroup {
|
|
171
|
-
repositories: schema.Repository[];
|
|
172
|
-
}
|
|
173
|
-
export interface AllGroupsResult {
|
|
174
|
-
groups: ProjectGroupWithRepositories[];
|
|
175
|
-
ungroupedRepositories: schema.Repository[];
|
|
176
|
-
}
|
|
177
|
-
export interface ProjectGroupQueries {
|
|
178
|
-
findById(id: string): Promise<schema.ProjectGroup | null>;
|
|
179
|
-
findByUserId(userId: string): Promise<schema.ProjectGroup[]>;
|
|
180
|
-
findByName(userId: string, name: string): Promise<schema.ProjectGroup | null>;
|
|
181
|
-
create(data: schema.NewProjectGroup): Promise<schema.ProjectGroup>;
|
|
182
|
-
update(id: string, data: Partial<Omit<schema.ProjectGroup, 'id' | 'userId' | 'createdAt'>>): Promise<void>;
|
|
183
|
-
delete(id: string): Promise<void>;
|
|
184
|
-
findWithRepositories(id: string): Promise<ProjectGroupWithRepositories | null>;
|
|
185
|
-
findAllWithRepositories(userId: string): Promise<AllGroupsResult>;
|
|
186
|
-
updateCoordinatorAgent(id: string, config: schema.CoordinatorAgentConfig): Promise<void>;
|
|
187
|
-
reorder(userId: string, orderedIds: string[]): Promise<void>;
|
|
188
|
-
}
|
|
189
|
-
export declare const projectGroupQueries: ProjectGroupQueries;
|
|
190
|
-
export interface RepositoryQueries {
|
|
191
|
-
findById(id: string): Promise<schema.Repository | null>;
|
|
192
|
-
findByFullName(fullName: string): Promise<schema.Repository | null>;
|
|
193
|
-
findByGithubFullName(fullName: string): Promise<schema.Repository[]>;
|
|
194
|
-
findByUserId(userId: string): Promise<schema.Repository[]>;
|
|
195
|
-
findByWorkspaceId(workspaceId: string): Promise<schema.Repository[]>;
|
|
196
|
-
findByProjectGroupId(projectGroupId: string): Promise<schema.Repository[]>;
|
|
197
|
-
upsert(data: schema.NewRepository): Promise<schema.Repository>;
|
|
198
|
-
assignToWorkspace(repoId: string, workspaceId: string | null): Promise<void>;
|
|
199
|
-
assignToGroup(repoId: string, projectGroupId: string | null): Promise<void>;
|
|
200
|
-
updateProjectAgent(id: string, config: schema.ProjectAgentConfig): Promise<void>;
|
|
201
|
-
updateSyncStatus(id: string, status: string, lastSyncedAt?: Date): Promise<void>;
|
|
202
|
-
delete(id: string): Promise<void>;
|
|
203
|
-
}
|
|
204
|
-
export declare const repositoryQueries: RepositoryQueries;
|
|
205
|
-
export interface AgentSessionQueries {
|
|
206
|
-
findById(id: string): Promise<schema.AgentSession | null>;
|
|
207
|
-
findByWorkspaceId(workspaceId: string): Promise<schema.AgentSession[]>;
|
|
208
|
-
findActiveByWorkspace(workspaceId: string): Promise<schema.AgentSession[]>;
|
|
209
|
-
create(data: schema.NewAgentSession): Promise<schema.AgentSession>;
|
|
210
|
-
endSession(id: string, endMarker?: {
|
|
211
|
-
summary?: string;
|
|
212
|
-
completedTasks?: string[];
|
|
213
|
-
}): Promise<void>;
|
|
214
|
-
delete(id: string): Promise<void>;
|
|
215
|
-
}
|
|
216
|
-
export declare const agentSessionQueries: AgentSessionQueries;
|
|
217
|
-
export interface AgentSummaryQueries {
|
|
218
|
-
findBySessionId(sessionId: string): Promise<schema.AgentSummary[]>;
|
|
219
|
-
findLatestByAgent(agentName: string): Promise<schema.AgentSummary | null>;
|
|
220
|
-
create(data: schema.NewAgentSummary): Promise<schema.AgentSummary>;
|
|
221
|
-
deleteBySession(sessionId: string): Promise<void>;
|
|
222
|
-
}
|
|
223
|
-
export declare const agentSummaryQueries: AgentSummaryQueries;
|
|
224
|
-
export interface CIFailureEventQueries {
|
|
225
|
-
findById(id: string): Promise<schema.CIFailureEvent | null>;
|
|
226
|
-
findByRepository(repository: string, limit?: number): Promise<schema.CIFailureEvent[]>;
|
|
227
|
-
findByPR(repository: string, prNumber: number): Promise<schema.CIFailureEvent[]>;
|
|
228
|
-
findRecentUnprocessed(limit?: number): Promise<schema.CIFailureEvent[]>;
|
|
229
|
-
create(data: schema.NewCIFailureEvent): Promise<schema.CIFailureEvent>;
|
|
230
|
-
markProcessed(id: string, agentSpawned: boolean): Promise<void>;
|
|
231
|
-
delete(id: string): Promise<void>;
|
|
232
|
-
}
|
|
233
|
-
export declare const ciFailureEventQueries: CIFailureEventQueries;
|
|
234
|
-
export interface CIFixAttemptQueries {
|
|
235
|
-
findById(id: string): Promise<schema.CIFixAttempt | null>;
|
|
236
|
-
findByFailureEvent(failureEventId: string): Promise<schema.CIFixAttempt[]>;
|
|
237
|
-
findActiveByRepository(repository: string): Promise<schema.CIFixAttempt[]>;
|
|
238
|
-
create(data: schema.NewCIFixAttempt): Promise<schema.CIFixAttempt>;
|
|
239
|
-
updateStatus(id: string, status: string, errorMessage?: string): Promise<void>;
|
|
240
|
-
complete(id: string, status: 'success' | 'failed', commitSha?: string, errorMessage?: string): Promise<void>;
|
|
241
|
-
}
|
|
242
|
-
export declare const ciFixAttemptQueries: CIFixAttemptQueries;
|
|
243
|
-
export interface IssueAssignmentQueries {
|
|
244
|
-
findById(id: string): Promise<schema.IssueAssignment | null>;
|
|
245
|
-
findByRepository(repository: string, limit?: number): Promise<schema.IssueAssignment[]>;
|
|
246
|
-
findByIssue(repository: string, issueNumber: number): Promise<schema.IssueAssignment | null>;
|
|
247
|
-
findByAgent(agentId: string): Promise<schema.IssueAssignment[]>;
|
|
248
|
-
findPending(limit?: number): Promise<schema.IssueAssignment[]>;
|
|
249
|
-
create(data: schema.NewIssueAssignment): Promise<schema.IssueAssignment>;
|
|
250
|
-
assignAgent(id: string, agentId: string, agentName: string): Promise<void>;
|
|
251
|
-
updateStatus(id: string, status: string, resolution?: string): Promise<void>;
|
|
252
|
-
linkPR(id: string, prNumber: number): Promise<void>;
|
|
253
|
-
}
|
|
254
|
-
export declare const issueAssignmentQueries: IssueAssignmentQueries;
|
|
255
|
-
export interface CommentMentionQueries {
|
|
256
|
-
findById(id: string): Promise<schema.CommentMention | null>;
|
|
257
|
-
findByRepository(repository: string, limit?: number): Promise<schema.CommentMention[]>;
|
|
258
|
-
findBySource(sourceType: string, sourceId: number): Promise<schema.CommentMention | null>;
|
|
259
|
-
findPending(limit?: number): Promise<schema.CommentMention[]>;
|
|
260
|
-
findByMentionedAgent(mentionedAgent: string, limit?: number): Promise<schema.CommentMention[]>;
|
|
261
|
-
create(data: schema.NewCommentMention): Promise<schema.CommentMention>;
|
|
262
|
-
markProcessing(id: string, agentId: string, agentName: string): Promise<void>;
|
|
263
|
-
markResponded(id: string, responseCommentId: number, responseBody: string): Promise<void>;
|
|
264
|
-
markIgnored(id: string): Promise<void>;
|
|
265
|
-
}
|
|
266
|
-
export declare const commentMentionQueries: CommentMentionQueries;
|
|
267
|
-
export interface ChannelQueries {
|
|
268
|
-
findById(id: string): Promise<schema.Channel | null>;
|
|
269
|
-
findByWorkspaceId(workspaceId: string): Promise<schema.Channel[]>;
|
|
270
|
-
findByWorkspaceAndChannelId(workspaceId: string, channelId: string): Promise<schema.Channel | null>;
|
|
271
|
-
create(channel: schema.NewChannel): Promise<schema.Channel>;
|
|
272
|
-
update(id: string, updates: Partial<schema.NewChannel>): Promise<void>;
|
|
273
|
-
archive(id: string): Promise<void>;
|
|
274
|
-
unarchive(id: string): Promise<void>;
|
|
275
|
-
delete(id: string): Promise<void>;
|
|
276
|
-
}
|
|
277
|
-
export declare const channelQueries: ChannelQueries;
|
|
278
|
-
export interface ChannelMemberQueries {
|
|
279
|
-
findByChannelId(channelId: string): Promise<schema.ChannelMember[]>;
|
|
280
|
-
findByMemberId(memberId: string): Promise<schema.ChannelMember[]>;
|
|
281
|
-
findMembership(channelId: string, memberId: string): Promise<schema.ChannelMember | null>;
|
|
282
|
-
addMember(member: schema.NewChannelMember): Promise<schema.ChannelMember>;
|
|
283
|
-
removeMember(channelId: string, memberId: string): Promise<void>;
|
|
284
|
-
updateRole(channelId: string, memberId: string, role: string): Promise<void>;
|
|
285
|
-
countByChannelIds(channelIds: string[]): Promise<Map<string, number>>;
|
|
286
|
-
}
|
|
287
|
-
export declare const channelMemberQueries: ChannelMemberQueries;
|
|
288
|
-
export declare function runMigrations(): Promise<void>;
|
|
289
|
-
export declare function closeDb(): Promise<void>;
|
|
290
|
-
//# sourceMappingURL=drizzle.d.ts.map
|