web-agent-bridge 3.2.0 → 3.4.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.
Files changed (256) hide show
  1. package/LICENSE +84 -72
  2. package/README.ar.md +1304 -1152
  3. package/README.md +298 -1635
  4. package/bin/agent-runner.js +474 -474
  5. package/bin/cli.js +237 -138
  6. package/bin/wab-init.js +223 -0
  7. package/bin/wab.js +80 -80
  8. package/examples/azure-dns-wab.js +83 -0
  9. package/examples/bidi-agent.js +119 -119
  10. package/examples/cloudflare-wab-dns.js +121 -0
  11. package/examples/cpanel-wab-dns.js +114 -0
  12. package/examples/cross-site-agent.js +91 -91
  13. package/examples/dns-discovery-agent.js +166 -0
  14. package/examples/gcp-dns-wab.js +76 -0
  15. package/examples/governance-agent.js +169 -0
  16. package/examples/mcp-agent.js +94 -94
  17. package/examples/next-app-router/README.md +44 -44
  18. package/examples/plesk-wab-dns.js +103 -0
  19. package/examples/puppeteer-agent.js +108 -108
  20. package/examples/route53-wab-dns.js +144 -0
  21. package/examples/saas-dashboard/README.md +55 -55
  22. package/examples/safe-mode-agent.js +96 -0
  23. package/examples/shopify-hydrogen/README.md +74 -74
  24. package/examples/vision-agent.js +171 -171
  25. package/examples/wab-sign.js +74 -0
  26. package/examples/wab-verify.js +60 -0
  27. package/examples/wordpress-elementor/README.md +77 -77
  28. package/package.json +19 -6
  29. package/public/.well-known/agent-tools.json +180 -180
  30. package/public/.well-known/ai-assets.json +59 -59
  31. package/public/.well-known/security.txt +8 -0
  32. package/public/.well-known/wab.json +28 -0
  33. package/public/activate.html +368 -0
  34. package/public/adoption-metrics.html +188 -0
  35. package/public/agent-workspace.html +349 -349
  36. package/public/ai.html +198 -198
  37. package/public/api.html +413 -412
  38. package/public/azure-dns-integration.html +289 -0
  39. package/public/browser.html +486 -486
  40. package/public/cloudflare-integration.html +380 -0
  41. package/public/commander-dashboard.html +243 -243
  42. package/public/cookies.html +210 -210
  43. package/public/cpanel-integration.html +398 -0
  44. package/public/css/agent-workspace.css +1713 -1713
  45. package/public/css/premium.css +317 -317
  46. package/public/css/styles.css +1263 -1235
  47. package/public/dashboard.html +707 -706
  48. package/public/dns.html +436 -0
  49. package/public/docs.html +588 -587
  50. package/public/feed.xml +89 -89
  51. package/public/gcp-dns-integration.html +318 -0
  52. package/public/growth.html +465 -463
  53. package/public/index.html +1266 -982
  54. package/public/integrations.html +556 -0
  55. package/public/js/activate.js +145 -0
  56. package/public/js/agent-workspace.js +1740 -1740
  57. package/public/js/auth-nav.js +65 -31
  58. package/public/js/auth-redirect.js +12 -12
  59. package/public/js/cookie-consent.js +56 -56
  60. package/public/js/dns.js +438 -0
  61. package/public/js/wab-demo-page.js +721 -721
  62. package/public/js/ws-client.js +74 -74
  63. package/public/llms-full.txt +360 -360
  64. package/public/llms.txt +125 -125
  65. package/public/login.html +85 -85
  66. package/public/mesh-dashboard.html +328 -328
  67. package/public/openapi.json +669 -580
  68. package/public/phone-shield.html +281 -0
  69. package/public/plesk-integration.html +375 -0
  70. package/public/premium-dashboard.html +2489 -2489
  71. package/public/premium.html +793 -793
  72. package/public/privacy.html +297 -297
  73. package/public/provider-onboarding.html +172 -0
  74. package/public/provider-sandbox.html +134 -0
  75. package/public/providers.html +359 -0
  76. package/public/register.html +105 -105
  77. package/public/registrar-integrations.html +141 -0
  78. package/public/robots.txt +99 -87
  79. package/public/route53-integration.html +531 -0
  80. package/public/script/wab-consent.d.ts +36 -36
  81. package/public/script/wab-consent.js +104 -104
  82. package/public/script/wab-schema.js +131 -131
  83. package/public/script/wab.d.ts +108 -108
  84. package/public/script/wab.min.js +580 -580
  85. package/public/security.txt +8 -0
  86. package/public/shieldqr.html +231 -0
  87. package/public/sitemap.xml +6 -0
  88. package/public/terms.html +256 -256
  89. package/public/wab-trust.html +200 -0
  90. package/public/wab-vs-protocols.html +210 -0
  91. package/public/whitepaper.html +449 -0
  92. package/script/ai-agent-bridge.js +1754 -1754
  93. package/sdk/README.md +99 -99
  94. package/sdk/agent-mesh.js +449 -449
  95. package/sdk/auto-discovery.js +288 -0
  96. package/sdk/commander.js +262 -262
  97. package/sdk/governance.js +262 -0
  98. package/sdk/index.d.ts +464 -464
  99. package/sdk/index.js +25 -1
  100. package/sdk/multi-agent.js +318 -318
  101. package/sdk/package.json +2 -2
  102. package/sdk/safe-mode.js +221 -0
  103. package/sdk/safety-shield.js +219 -0
  104. package/sdk/schema-discovery.js +83 -83
  105. package/server/adapters/index.js +520 -520
  106. package/server/config/plans.js +367 -367
  107. package/server/config/secrets.js +102 -102
  108. package/server/control-plane/index.js +301 -301
  109. package/server/data-plane/index.js +354 -354
  110. package/server/index.js +670 -427
  111. package/server/llm/index.js +404 -404
  112. package/server/middleware/adminAuth.js +35 -35
  113. package/server/middleware/auth.js +50 -50
  114. package/server/middleware/featureGate.js +88 -88
  115. package/server/middleware/rateLimits.js +100 -100
  116. package/server/middleware/sensitiveAction.js +157 -0
  117. package/server/migrations/001_add_analytics_indexes.sql +7 -7
  118. package/server/migrations/002_premium_features.sql +418 -418
  119. package/server/migrations/003_ads_integer_cents.sql +33 -33
  120. package/server/migrations/004_agent_os.sql +158 -158
  121. package/server/migrations/005_marketplace_metering.sql +126 -126
  122. package/server/migrations/007_governance.sql +106 -0
  123. package/server/migrations/008_plans.sql +144 -0
  124. package/server/migrations/009_shieldqr.sql +30 -0
  125. package/server/migrations/010_extended_trust.sql +33 -0
  126. package/server/models/adapters/index.js +33 -33
  127. package/server/models/adapters/mysql.js +183 -183
  128. package/server/models/adapters/postgresql.js +172 -172
  129. package/server/models/adapters/sqlite.js +7 -7
  130. package/server/models/db.js +740 -681
  131. package/server/observability/failure-analysis.js +337 -337
  132. package/server/observability/index.js +394 -394
  133. package/server/protocol/capabilities.js +223 -223
  134. package/server/protocol/index.js +243 -243
  135. package/server/protocol/schema.js +584 -584
  136. package/server/registry/certification.js +271 -271
  137. package/server/registry/index.js +326 -326
  138. package/server/routes/admin-plans.js +76 -0
  139. package/server/routes/admin-premium.js +673 -671
  140. package/server/routes/admin-shieldqr.js +90 -0
  141. package/server/routes/admin-trust-monitor.js +83 -0
  142. package/server/routes/admin.js +549 -261
  143. package/server/routes/ads.js +130 -130
  144. package/server/routes/agent-workspace.js +540 -540
  145. package/server/routes/api.js +150 -150
  146. package/server/routes/auth.js +71 -71
  147. package/server/routes/billing.js +57 -45
  148. package/server/routes/commander.js +316 -316
  149. package/server/routes/demo-showcase.js +332 -332
  150. package/server/routes/demo-store.js +154 -0
  151. package/server/routes/discovery.js +2348 -417
  152. package/server/routes/gateway.js +173 -157
  153. package/server/routes/governance.js +208 -0
  154. package/server/routes/license.js +251 -240
  155. package/server/routes/mesh.js +469 -469
  156. package/server/routes/noscript.js +543 -543
  157. package/server/routes/plans.js +33 -0
  158. package/server/routes/premium-v2.js +686 -686
  159. package/server/routes/premium.js +724 -724
  160. package/server/routes/providers.js +650 -0
  161. package/server/routes/runtime.js +2148 -2147
  162. package/server/routes/shieldqr.js +88 -0
  163. package/server/routes/sovereign.js +465 -385
  164. package/server/routes/universal.js +200 -185
  165. package/server/routes/wab-api.js +850 -501
  166. package/server/runtime/container-worker.js +111 -111
  167. package/server/runtime/container.js +448 -448
  168. package/server/runtime/distributed-worker.js +362 -362
  169. package/server/runtime/event-bus.js +210 -210
  170. package/server/runtime/index.js +253 -253
  171. package/server/runtime/queue.js +599 -599
  172. package/server/runtime/replay.js +666 -666
  173. package/server/runtime/sandbox.js +266 -266
  174. package/server/runtime/scheduler.js +534 -534
  175. package/server/runtime/session-engine.js +293 -293
  176. package/server/runtime/state-manager.js +188 -188
  177. package/server/security/cross-site-redactor.js +196 -0
  178. package/server/security/dry-run.js +180 -0
  179. package/server/security/human-gate-rate-limit.js +147 -0
  180. package/server/security/human-gate-transports.js +178 -0
  181. package/server/security/human-gate.js +281 -0
  182. package/server/security/index.js +368 -368
  183. package/server/security/intent-engine.js +245 -0
  184. package/server/security/reward-guard.js +171 -0
  185. package/server/security/rollback-store.js +239 -0
  186. package/server/security/token-scope.js +404 -0
  187. package/server/security/url-policy.js +139 -0
  188. package/server/services/agent-chat.js +506 -506
  189. package/server/services/agent-learning.js +601 -575
  190. package/server/services/agent-memory.js +625 -625
  191. package/server/services/agent-mesh.js +555 -539
  192. package/server/services/agent-symphony.js +717 -717
  193. package/server/services/agent-tasks.js +1807 -1807
  194. package/server/services/api-key-engine.js +292 -261
  195. package/server/services/cluster.js +894 -894
  196. package/server/services/commander.js +738 -738
  197. package/server/services/edge-compute.js +440 -440
  198. package/server/services/email.js +233 -204
  199. package/server/services/governance.js +466 -0
  200. package/server/services/hosted-runtime.js +205 -205
  201. package/server/services/lfd.js +635 -635
  202. package/server/services/local-ai.js +389 -389
  203. package/server/services/marketplace.js +270 -270
  204. package/server/services/metering.js +182 -182
  205. package/server/services/modules/affiliate-intelligence.js +93 -93
  206. package/server/services/modules/agent-firewall.js +90 -90
  207. package/server/services/modules/bounty.js +89 -89
  208. package/server/services/modules/collective-bargaining.js +92 -92
  209. package/server/services/modules/dark-pattern.js +66 -66
  210. package/server/services/modules/gov-intelligence.js +45 -45
  211. package/server/services/modules/neural.js +55 -55
  212. package/server/services/modules/notary.js +49 -49
  213. package/server/services/modules/price-time-machine.js +86 -86
  214. package/server/services/modules/protocol.js +104 -104
  215. package/server/services/negotiation.js +439 -439
  216. package/server/services/plans.js +214 -0
  217. package/server/services/plugins.js +771 -771
  218. package/server/services/premium.js +1 -1
  219. package/server/services/price-intelligence.js +566 -566
  220. package/server/services/price-shield.js +1137 -1137
  221. package/server/services/provider-clients.js +740 -0
  222. package/server/services/reputation.js +465 -465
  223. package/server/services/search-engine.js +357 -357
  224. package/server/services/security.js +513 -513
  225. package/server/services/self-healing.js +843 -843
  226. package/server/services/shieldqr.js +322 -0
  227. package/server/services/sovereign-shield.js +542 -0
  228. package/server/services/ssl-inspector.js +42 -0
  229. package/server/services/ssl-monitor.js +167 -0
  230. package/server/services/stripe.js +205 -192
  231. package/server/services/swarm.js +788 -788
  232. package/server/services/universal-scraper.js +662 -661
  233. package/server/services/verification.js +481 -481
  234. package/server/services/vision.js +1163 -1163
  235. package/server/services/wab-crypto.js +178 -0
  236. package/server/utils/cache.js +125 -125
  237. package/server/utils/migrate.js +81 -81
  238. package/server/utils/safe-fetch.js +228 -0
  239. package/server/utils/secureFields.js +50 -50
  240. package/server/ws.js +161 -161
  241. package/templates/artisan-marketplace.yaml +104 -104
  242. package/templates/book-price-scout.yaml +98 -98
  243. package/templates/electronics-price-tracker.yaml +108 -108
  244. package/templates/flight-deal-hunter.yaml +113 -113
  245. package/templates/freelancer-direct.yaml +116 -116
  246. package/templates/grocery-price-compare.yaml +93 -93
  247. package/templates/hotel-direct-booking.yaml +113 -113
  248. package/templates/local-services.yaml +98 -98
  249. package/templates/olive-oil-tunisia.yaml +88 -88
  250. package/templates/organic-farm-fresh.yaml +101 -101
  251. package/templates/restaurant-direct.yaml +97 -97
  252. package/public/score.html +0 -263
  253. package/server/migrations/006_growth_suite.sql +0 -138
  254. package/server/routes/growth.js +0 -962
  255. package/server/services/fairness-engine.js +0 -409
  256. package/server/services/fairness.js +0 -420
@@ -1,788 +1,788 @@
1
- const crypto = require('crypto');
2
- const https = require('https');
3
- const http = require('http');
4
- const { URL } = require('url');
5
- const { db } = require('../models/db');
6
-
7
- // ─── Schema ──────────────────────────────────────────────────────────
8
-
9
- db.exec(`
10
- CREATE TABLE IF NOT EXISTS swarm_configs (
11
- id TEXT PRIMARY KEY,
12
- site_id TEXT UNIQUE,
13
- name TEXT,
14
- strategy TEXT DEFAULT 'parallel',
15
- max_agents INTEGER DEFAULT 3,
16
- timeout_ms INTEGER DEFAULT 30000,
17
- merge_strategy TEXT DEFAULT 'best_score',
18
- enabled INTEGER DEFAULT 1,
19
- created_at TEXT DEFAULT (datetime('now')),
20
- updated_at TEXT DEFAULT (datetime('now'))
21
- );
22
-
23
- CREATE TABLE IF NOT EXISTS swarm_tasks (
24
- id TEXT PRIMARY KEY,
25
- site_id TEXT,
26
- config_id TEXT,
27
- task_type TEXT,
28
- objective TEXT,
29
- parameters TEXT DEFAULT '{}',
30
- status TEXT DEFAULT 'pending',
31
- created_by TEXT,
32
- created_at TEXT DEFAULT (datetime('now')),
33
- started_at TEXT,
34
- completed_at TEXT
35
- );
36
-
37
- CREATE TABLE IF NOT EXISTS swarm_agents (
38
- id TEXT PRIMARY KEY,
39
- task_id TEXT,
40
- agent_role TEXT,
41
- agent_type TEXT,
42
- target TEXT,
43
- status TEXT DEFAULT 'idle',
44
- result TEXT DEFAULT '{}',
45
- score REAL,
46
- error TEXT,
47
- started_at TEXT,
48
- completed_at TEXT
49
- );
50
-
51
- CREATE TABLE IF NOT EXISTS swarm_results (
52
- id TEXT PRIMARY KEY,
53
- task_id TEXT,
54
- merged_result TEXT DEFAULT '{}',
55
- fairness_applied INTEGER DEFAULT 0,
56
- total_sources INTEGER DEFAULT 0,
57
- best_source TEXT,
58
- comparison TEXT DEFAULT '{}',
59
- created_at TEXT DEFAULT (datetime('now'))
60
- );
61
-
62
- CREATE INDEX IF NOT EXISTS idx_swarm_configs_site ON swarm_configs(site_id);
63
- CREATE INDEX IF NOT EXISTS idx_swarm_tasks_site ON swarm_tasks(site_id);
64
- CREATE INDEX IF NOT EXISTS idx_swarm_tasks_status ON swarm_tasks(status);
65
- CREATE INDEX IF NOT EXISTS idx_swarm_tasks_config ON swarm_tasks(config_id);
66
- CREATE INDEX IF NOT EXISTS idx_swarm_agents_task ON swarm_agents(task_id);
67
- CREATE INDEX IF NOT EXISTS idx_swarm_agents_status ON swarm_agents(status);
68
- CREATE INDEX IF NOT EXISTS idx_swarm_results_task ON swarm_results(task_id);
69
- `);
70
-
71
- // ─── Prepared Statements ─────────────────────────────────────────────
72
-
73
- const stmts = {
74
- upsertConfig: db.prepare(`
75
- INSERT INTO swarm_configs (id, site_id, name, strategy, max_agents, timeout_ms, merge_strategy, enabled)
76
- VALUES (@id, @site_id, @name, @strategy, @max_agents, @timeout_ms, @merge_strategy, @enabled)
77
- ON CONFLICT(site_id) DO UPDATE SET
78
- name = excluded.name,
79
- strategy = excluded.strategy,
80
- max_agents = excluded.max_agents,
81
- timeout_ms = excluded.timeout_ms,
82
- merge_strategy = excluded.merge_strategy,
83
- enabled = excluded.enabled,
84
- updated_at = datetime('now')
85
- `),
86
- getConfig: db.prepare('SELECT * FROM swarm_configs WHERE site_id = ?'),
87
- getConfigById: db.prepare('SELECT * FROM swarm_configs WHERE id = ?'),
88
- insertTask: db.prepare(`
89
- INSERT INTO swarm_tasks (id, site_id, config_id, task_type, objective, parameters, status, created_by)
90
- VALUES (@id, @site_id, @config_id, @task_type, @objective, @parameters, @status, @created_by)
91
- `),
92
- getTask: db.prepare('SELECT * FROM swarm_tasks WHERE id = ?'),
93
- updateTaskStatus: db.prepare('UPDATE swarm_tasks SET status = ?, started_at = COALESCE(started_at, ?), completed_at = ? WHERE id = ?'),
94
- insertAgent: db.prepare(`
95
- INSERT INTO swarm_agents (id, task_id, agent_role, agent_type, target, status)
96
- VALUES (@id, @task_id, @agent_role, @agent_type, @target, @status)
97
- `),
98
- getAgentsByTask: db.prepare('SELECT * FROM swarm_agents WHERE task_id = ? ORDER BY started_at ASC'),
99
- updateAgent: db.prepare(`
100
- UPDATE swarm_agents SET status = ?, result = ?, score = ?, error = ?, started_at = COALESCE(started_at, ?), completed_at = ?
101
- WHERE id = ?
102
- `),
103
- insertResult: db.prepare(`
104
- INSERT INTO swarm_results (id, task_id, merged_result, fairness_applied, total_sources, best_source, comparison)
105
- VALUES (@id, @task_id, @merged_result, @fairness_applied, @total_sources, @best_source, @comparison)
106
- `),
107
- getResult: db.prepare('SELECT * FROM swarm_results WHERE task_id = ?'),
108
- cancelPendingAgents: db.prepare("UPDATE swarm_agents SET status = 'cancelled' WHERE task_id = ? AND status IN ('idle', 'running')"),
109
- taskHistory: db.prepare('SELECT * FROM swarm_tasks WHERE site_id = ? ORDER BY created_at DESC LIMIT ?'),
110
- taskHistoryByType: db.prepare('SELECT * FROM swarm_tasks WHERE site_id = ? AND task_type = ? ORDER BY created_at DESC LIMIT ?'),
111
- };
112
-
113
- // ─── 1. configureSwarm ───────────────────────────────────────────────
114
-
115
- function configureSwarm(siteId, { name, strategy, maxAgents, timeoutMs, mergeStrategy } = {}) {
116
- const existing = stmts.getConfig.get(siteId);
117
- const id = existing ? existing.id : crypto.randomUUID();
118
-
119
- const params = {
120
- id,
121
- site_id: siteId,
122
- name: name ?? (existing ? existing.name : siteId),
123
- strategy: strategy ?? (existing ? existing.strategy : 'parallel'),
124
- max_agents: maxAgents ?? (existing ? existing.max_agents : 3),
125
- timeout_ms: timeoutMs ?? (existing ? existing.timeout_ms : 30000),
126
- merge_strategy: mergeStrategy ?? (existing ? existing.merge_strategy : 'best_score'),
127
- enabled: 1,
128
- };
129
-
130
- stmts.upsertConfig.run(params);
131
- return stmts.getConfig.get(siteId);
132
- }
133
-
134
- // ─── 2. getSwarmConfig ───────────────────────────────────────────────
135
-
136
- function getSwarmConfig(siteId) {
137
- return stmts.getConfig.get(siteId) || null;
138
- }
139
-
140
- // ─── 3. createTask ───────────────────────────────────────────────────
141
-
142
- function createTask(siteId, { taskType, objective, parameters, createdBy } = {}) {
143
- const config = stmts.getConfig.get(siteId);
144
- const id = crypto.randomUUID();
145
-
146
- stmts.insertTask.run({
147
- id,
148
- site_id: siteId,
149
- config_id: config ? config.id : null,
150
- task_type: taskType || 'general',
151
- objective: objective || '',
152
- parameters: JSON.stringify(parameters || {}),
153
- status: 'pending',
154
- created_by: createdBy || null,
155
- });
156
-
157
- return stmts.getTask.get(id);
158
- }
159
-
160
- // ─── 4. assignAgents ─────────────────────────────────────────────────
161
-
162
- function assignAgents(taskId, agentDefinitions) {
163
- const task = stmts.getTask.get(taskId);
164
- if (!task) throw new Error(`Task ${taskId} not found`);
165
-
166
- const config = task.config_id ? stmts.getConfigById.get(task.config_id) : null;
167
- const maxAgents = config ? config.max_agents : 10;
168
- const defs = agentDefinitions.slice(0, maxAgents);
169
-
170
- const agents = [];
171
- const insertMany = db.transaction((items) => {
172
- for (const def of items) {
173
- const agentId = crypto.randomUUID();
174
- stmts.insertAgent.run({
175
- id: agentId,
176
- task_id: taskId,
177
- agent_role: def.role || 'worker',
178
- agent_type: def.type || 'fetch',
179
- target: def.target || '',
180
- status: 'idle',
181
- });
182
- agents.push({ id: agentId, task_id: taskId, agent_role: def.role, agent_type: def.type, target: def.target, status: 'idle' });
183
- }
184
- });
185
-
186
- insertMany(defs);
187
- return agents;
188
- }
189
-
190
- // ─── 5. runSwarmTask ─────────────────────────────────────────────────
191
-
192
- async function runSwarmTask(taskId) {
193
- const task = stmts.getTask.get(taskId);
194
- if (!task) throw new Error(`Task ${taskId} not found`);
195
- if (task.status === 'cancelled') throw new Error(`Task ${taskId} is cancelled`);
196
-
197
- const config = task.config_id ? stmts.getConfigById.get(task.config_id) : null;
198
- const strategy = config ? config.strategy : 'parallel';
199
- const mergeStrategy = config ? config.merge_strategy : 'best_score';
200
- const timeoutMs = config ? config.timeout_ms : 30000;
201
-
202
- const now = new Date().toISOString();
203
- stmts.updateTaskStatus.run('running', now, null, taskId);
204
-
205
- let agents = stmts.getAgentsByTask.all(taskId);
206
- if (agents.length === 0) throw new Error(`No agents assigned to task ${taskId}`);
207
-
208
- try {
209
- switch (strategy) {
210
- case 'sequential':
211
- await runSequential(agents, timeoutMs);
212
- break;
213
- case 'competitive':
214
- await runCompetitive(agents, timeoutMs);
215
- break;
216
- case 'collaborative':
217
- await runCollaborative(agents, timeoutMs);
218
- break;
219
- case 'parallel':
220
- default:
221
- await runParallel(agents, timeoutMs);
222
- break;
223
- }
224
-
225
- agents = stmts.getAgentsByTask.all(taskId);
226
- const completedAgents = agents.filter(a => a.status === 'completed');
227
-
228
- const rawResults = completedAgents.map(a => {
229
- let parsed = {};
230
- try { parsed = JSON.parse(a.result || '{}'); } catch (_) {}
231
- return { ...parsed, _agentId: a.id, _role: a.agent_role, _score: a.score, _source: a.target };
232
- });
233
-
234
- const fairResults = applyFairnessToResults(rawResults);
235
- const merged = mergeResults(fairResults, mergeStrategy);
236
-
237
- const bestAgent = completedAgents.reduce((best, cur) =>
238
- (cur.score || 0) > (best.score || 0) ? cur : best, completedAgents[0]);
239
-
240
- const comparison = completedAgents.map(a => ({
241
- agentId: a.id,
242
- role: a.agent_role,
243
- target: a.target,
244
- score: a.score,
245
- status: a.status,
246
- }));
247
-
248
- const resultId = crypto.randomUUID();
249
- stmts.insertResult.run({
250
- id: resultId,
251
- task_id: taskId,
252
- merged_result: JSON.stringify(merged),
253
- fairness_applied: 1,
254
- total_sources: completedAgents.length,
255
- best_source: bestAgent ? bestAgent.target : null,
256
- comparison: JSON.stringify(comparison),
257
- });
258
-
259
- const doneAt = new Date().toISOString();
260
- stmts.updateTaskStatus.run('completed', now, doneAt, taskId);
261
-
262
- return { taskId, status: 'completed', result: merged, sources: completedAgents.length, bestSource: bestAgent ? bestAgent.target : null };
263
- } catch (err) {
264
- const failAt = new Date().toISOString();
265
- stmts.updateTaskStatus.run('failed', now, failAt, taskId);
266
- throw err;
267
- }
268
- }
269
-
270
- // ─── Strategy Runners ────────────────────────────────────────────────
271
-
272
- async function runParallel(agents, timeoutMs) {
273
- const promises = agents.map(agent => runSingleAgent(agent, timeoutMs));
274
- await Promise.allSettled(promises);
275
- }
276
-
277
- async function runSequential(agents, timeoutMs) {
278
- let previousResult = null;
279
- for (const agent of agents) {
280
- previousResult = await runSingleAgent(agent, timeoutMs, previousResult);
281
- }
282
- }
283
-
284
- async function runCompetitive(agents, timeoutMs) {
285
- const promises = agents.map(agent => runSingleAgent(agent, timeoutMs));
286
- await Promise.allSettled(promises);
287
- }
288
-
289
- async function runCollaborative(agents, timeoutMs) {
290
- const promises = agents.map(agent => runSingleAgent(agent, timeoutMs));
291
- await Promise.allSettled(promises);
292
- }
293
-
294
- async function runSingleAgent(agent, timeoutMs, previousResult) {
295
- const startedAt = new Date().toISOString();
296
- stmts.updateAgent.run('running', agent.result || '{}', null, null, startedAt, null, agent.id);
297
-
298
- try {
299
- const agentInput = { ...agent, timeoutMs, previousResult };
300
- const result = await executeAgent(agentInput);
301
- const completedAt = new Date().toISOString();
302
- stmts.updateAgent.run('completed', JSON.stringify(result), result.relevance || 0, null, startedAt, completedAt, agent.id);
303
- return result;
304
- } catch (err) {
305
- const completedAt = new Date().toISOString();
306
- stmts.updateAgent.run('failed', '{}', 0, err.message, startedAt, completedAt, agent.id);
307
- return null;
308
- }
309
- }
310
-
311
- // ─── 6. executeAgent ─────────────────────────────────────────────────
312
-
313
- function executeAgent(agent) {
314
- const timeout = agent.timeoutMs || 30000;
315
- const targetUrl = agent.target;
316
-
317
- if (!targetUrl) {
318
- return Promise.resolve({ title: '', description: '', prices: [], relevance: 0, source: '' });
319
- }
320
-
321
- return new Promise((resolve, reject) => {
322
- let parsed;
323
- try {
324
- parsed = new URL(targetUrl);
325
- } catch (_) {
326
- return resolve({ title: '', description: '', prices: [], relevance: 0, source: targetUrl, error: 'Invalid URL' });
327
- }
328
-
329
- const transport = parsed.protocol === 'https:' ? https : http;
330
- const timer = setTimeout(() => {
331
- req.destroy();
332
- resolve({ title: '', description: '', prices: [], relevance: 0, source: targetUrl, error: 'Timeout' });
333
- }, timeout);
334
-
335
- const req = transport.get(targetUrl, { headers: { 'User-Agent': 'WAB-Swarm-Agent/1.0' } }, (res) => {
336
- if (res.statusCode >= 300 && res.statusCode < 400 && res.headers.location) {
337
- clearTimeout(timer);
338
- const redirectAgent = { ...agent, target: res.headers.location };
339
- resolve(executeAgent(redirectAgent));
340
- res.resume();
341
- return;
342
- }
343
-
344
- let body = '';
345
- const maxBytes = 2 * 1024 * 1024;
346
- let received = 0;
347
-
348
- res.on('data', (chunk) => {
349
- received += chunk.length;
350
- if (received <= maxBytes) body += chunk;
351
- });
352
-
353
- res.on('end', () => {
354
- clearTimeout(timer);
355
-
356
- const title = extractFirst(body, /<title[^>]*>([\s\S]*?)<\/title>/i) || '';
357
- const metaDesc = extractFirst(body, /<meta\s+name=["']description["']\s+content=["']([\s\S]*?)["']/i)
358
- || extractFirst(body, /<meta\s+content=["']([\s\S]*?)["']\s+name=["']description["']/i)
359
- || '';
360
- const h1 = extractFirst(body, /<h1[^>]*>([\s\S]*?)<\/h1>/i) || '';
361
- const priceMatches = body.match(/(?:\$|USD\s?)(\d{1,7}(?:[.,]\d{2})?)/g) || [];
362
- const prices = [...new Set(priceMatches.slice(0, 20))];
363
-
364
- const objective = agent.previousResult?.objective || '';
365
- const relevance = computeRelevanceScore(title, metaDesc, h1, prices, objective, targetUrl);
366
-
367
- resolve({
368
- title: cleanHtml(title).slice(0, 500),
369
- description: cleanHtml(metaDesc).slice(0, 1000),
370
- h1: cleanHtml(h1).slice(0, 500),
371
- prices,
372
- relevance,
373
- source: targetUrl,
374
- statusCode: res.statusCode,
375
- });
376
- });
377
-
378
- res.on('error', (err) => {
379
- clearTimeout(timer);
380
- resolve({ title: '', description: '', prices: [], relevance: 0, source: targetUrl, error: err.message });
381
- });
382
- });
383
-
384
- req.on('error', (err) => {
385
- clearTimeout(timer);
386
- resolve({ title: '', description: '', prices: [], relevance: 0, source: targetUrl, error: err.message });
387
- });
388
- });
389
- }
390
-
391
- function extractFirst(html, regex) {
392
- const m = html.match(regex);
393
- return m ? m[1].trim() : null;
394
- }
395
-
396
- function cleanHtml(text) {
397
- return text.replace(/<[^>]+>/g, '').replace(/&amp;/g, '&').replace(/&lt;/g, '<')
398
- .replace(/&gt;/g, '>').replace(/&quot;/g, '"').replace(/&#039;/g, "'")
399
- .replace(/&nbsp;/g, ' ').replace(/\s+/g, ' ').trim();
400
- }
401
-
402
- function computeRelevanceScore(title, description, h1, prices, objective, source) {
403
- let score = 0;
404
-
405
- if (title) score += 15;
406
- if (description) score += 15;
407
- if (h1) score += 10;
408
- if (prices.length > 0) score += 10;
409
-
410
- if (objective) {
411
- const terms = objective.toLowerCase().split(/\s+/).filter(Boolean);
412
- const corpus = `${title} ${description} ${h1}`.toLowerCase();
413
- let hits = 0;
414
- for (const t of terms) {
415
- if (corpus.includes(t)) hits++;
416
- }
417
- const termRatio = terms.length > 0 ? hits / terms.length : 0;
418
- score += Math.round(termRatio * 40);
419
- } else {
420
- score += 20;
421
- }
422
-
423
- if (source) {
424
- try {
425
- const u = new URL(source);
426
- if (u.protocol === 'https:') score += 5;
427
- if (u.pathname !== '/' && u.pathname.length > 1) score += 5;
428
- } catch (_) {}
429
- }
430
-
431
- return Math.min(Math.max(Math.round(score * 100) / 100, 0), 100);
432
- }
433
-
434
- // ─── 7. mergeResults ─────────────────────────────────────────────────
435
-
436
- function mergeResults(agents, strategy) {
437
- if (!agents || agents.length === 0) return { merged: true, data: [], strategy };
438
-
439
- switch (strategy) {
440
- case 'best_score': return mergeBestScore(agents);
441
- case 'weighted_average': return mergeWeightedAverage(agents);
442
- case 'union': return mergeUnion(agents);
443
- case 'consensus': return mergeConsensus(agents);
444
- default: return mergeBestScore(agents);
445
- }
446
- }
447
-
448
- function mergeBestScore(agents) {
449
- const sorted = [...agents].sort((a, b) => (b._score || b.relevance || 0) - (a._score || a.relevance || 0));
450
- const best = sorted[0];
451
- return {
452
- merged: true,
453
- strategy: 'best_score',
454
- title: best.title || '',
455
- description: best.description || '',
456
- h1: best.h1 || '',
457
- prices: best.prices || [],
458
- relevance: best._score || best.relevance || 0,
459
- source: best._source || best.source || '',
460
- allSources: sorted.map(a => ({ source: a._source || a.source, score: a._score || a.relevance || 0 })),
461
- };
462
- }
463
-
464
- function mergeWeightedAverage(agents) {
465
- const totalWeight = agents.reduce((s, a) => s + (a._score || a.relevance || 0), 0);
466
- if (totalWeight === 0) return mergeBestScore(agents);
467
-
468
- const allPrices = [];
469
- let weightedRelevance = 0;
470
- let bestTitle = '';
471
- let bestDesc = '';
472
- let bestH1 = '';
473
- let highestWeight = -1;
474
-
475
- for (const a of agents) {
476
- const w = a._score || a.relevance || 0;
477
- weightedRelevance += w * w;
478
- if (a.prices) allPrices.push(...a.prices);
479
- if (w > highestWeight) {
480
- highestWeight = w;
481
- bestTitle = a.title || bestTitle;
482
- bestDesc = a.description || bestDesc;
483
- bestH1 = a.h1 || bestH1;
484
- }
485
- }
486
-
487
- return {
488
- merged: true,
489
- strategy: 'weighted_average',
490
- title: bestTitle,
491
- description: bestDesc,
492
- h1: bestH1,
493
- prices: [...new Set(allPrices)],
494
- relevance: Math.round((weightedRelevance / totalWeight) * 100) / 100,
495
- sourceCount: agents.length,
496
- allSources: agents.map(a => ({ source: a._source || a.source, score: a._score || a.relevance || 0 })),
497
- };
498
- }
499
-
500
- function mergeUnion(agents) {
501
- const allPrices = [];
502
- const allTitles = [];
503
- const allDescriptions = [];
504
- let maxRelevance = 0;
505
-
506
- for (const a of agents) {
507
- if (a.title) allTitles.push(a.title);
508
- if (a.description) allDescriptions.push(a.description);
509
- if (a.prices) allPrices.push(...a.prices);
510
- maxRelevance = Math.max(maxRelevance, a._score || a.relevance || 0);
511
- }
512
-
513
- return {
514
- merged: true,
515
- strategy: 'union',
516
- titles: [...new Set(allTitles)],
517
- descriptions: [...new Set(allDescriptions)],
518
- prices: [...new Set(allPrices)],
519
- relevance: maxRelevance,
520
- sourceCount: agents.length,
521
- allSources: agents.map(a => ({ source: a._source || a.source, score: a._score || a.relevance || 0 })),
522
- };
523
- }
524
-
525
- function mergeConsensus(agents) {
526
- const titleFreq = {};
527
- const priceFreq = {};
528
- let bestDesc = '';
529
- let bestDescScore = -1;
530
-
531
- for (const a of agents) {
532
- const t = (a.title || '').toLowerCase().trim();
533
- if (t) titleFreq[t] = (titleFreq[t] || 0) + 1;
534
- if (a.prices) {
535
- for (const p of a.prices) priceFreq[p] = (priceFreq[p] || 0) + 1;
536
- }
537
- const s = a._score || a.relevance || 0;
538
- if (s > bestDescScore && a.description) {
539
- bestDescScore = s;
540
- bestDesc = a.description;
541
- }
542
- }
543
-
544
- const threshold = Math.max(Math.ceil(agents.length / 2), 1);
545
-
546
- const consensusTitles = Object.entries(titleFreq).filter(([, c]) => c >= threshold).map(([t]) => t);
547
- const consensusPrices = Object.entries(priceFreq).filter(([, c]) => c >= threshold).map(([p]) => p);
548
-
549
- const avgRelevance = agents.reduce((s, a) => s + (a._score || a.relevance || 0), 0) / agents.length;
550
-
551
- return {
552
- merged: true,
553
- strategy: 'consensus',
554
- titles: consensusTitles,
555
- description: bestDesc,
556
- prices: consensusPrices,
557
- relevance: Math.round(avgRelevance * 100) / 100,
558
- agreement: {
559
- titleAgreement: consensusTitles.length > 0,
560
- priceAgreement: consensusPrices.length > 0,
561
- threshold,
562
- totalAgents: agents.length,
563
- },
564
- allSources: agents.map(a => ({ source: a._source || a.source, score: a._score || a.relevance || 0 })),
565
- };
566
- }
567
-
568
- // ─── 8. applyFairnessToResults ───────────────────────────────────────
569
-
570
- function applyFairnessToResults(results) {
571
- if (!results || results.length === 0) return results;
572
-
573
- const domainCounts = {};
574
- for (const r of results) {
575
- const domain = extractDomain(r._source || r.source || '');
576
- domainCounts[domain] = (domainCounts[domain] || 0) + 1;
577
- }
578
-
579
- const totalResults = results.length;
580
-
581
- return results.map(r => {
582
- const domain = extractDomain(r._source || r.source || '');
583
- const score = r._score || r.relevance || 0;
584
- let adjusted = score;
585
-
586
- const isIndie = isIndependentSite(domain);
587
- if (isIndie) adjusted += score * 0.12;
588
-
589
- const domainShare = (domainCounts[domain] || 1) / totalResults;
590
- if (domainShare > 0.4) {
591
- adjusted -= score * 0.10;
592
- }
593
-
594
- const isSmall = isSmallSite(domain);
595
- if (isSmall) adjusted += score * 0.08;
596
-
597
- adjusted = Math.min(Math.max(Math.round(adjusted * 100) / 100, 0), 100);
598
-
599
- return { ...r, _score: adjusted, _fairnessApplied: true, _originalScore: score };
600
- });
601
- }
602
-
603
- function extractDomain(url) {
604
- try { return new URL(url).hostname; } catch (_) { return url; }
605
- }
606
-
607
- const MAJOR_DOMAINS = [
608
- 'amazon.com', 'google.com', 'facebook.com', 'apple.com', 'microsoft.com',
609
- 'walmart.com', 'ebay.com', 'target.com', 'bestbuy.com', 'alibaba.com',
610
- 'aliexpress.com', 'rakuten.com', 'shopify.com', 'etsy.com',
611
- ];
612
-
613
- function isIndependentSite(domain) {
614
- const lower = domain.toLowerCase().replace(/^www\./, '');
615
- return !MAJOR_DOMAINS.some(d => lower === d || lower.endsWith('.' + d));
616
- }
617
-
618
- function isSmallSite(domain) {
619
- const lower = domain.toLowerCase().replace(/^www\./, '');
620
- const parts = lower.split('.');
621
- if (parts.length > 3) return false;
622
- return !MAJOR_DOMAINS.some(d => lower === d || lower.endsWith('.' + d));
623
- }
624
-
625
- // ─── 9. getTaskStatus / getTaskResult ────────────────────────────────
626
-
627
- function getTaskStatus(taskId) {
628
- const task = stmts.getTask.get(taskId);
629
- if (!task) return null;
630
-
631
- const agents = stmts.getAgentsByTask.all(taskId);
632
- const agentSummary = agents.map(a => ({
633
- id: a.id,
634
- role: a.agent_role,
635
- type: a.agent_type,
636
- target: a.target,
637
- status: a.status,
638
- score: a.score,
639
- error: a.error,
640
- }));
641
-
642
- return {
643
- id: task.id,
644
- siteId: task.site_id,
645
- taskType: task.task_type,
646
- objective: task.objective,
647
- status: task.status,
648
- createdAt: task.created_at,
649
- startedAt: task.started_at,
650
- completedAt: task.completed_at,
651
- agents: agentSummary,
652
- };
653
- }
654
-
655
- function getTaskResult(taskId) {
656
- const task = stmts.getTask.get(taskId);
657
- if (!task) return null;
658
-
659
- const resultRow = stmts.getResult.get(taskId);
660
- if (!resultRow) return { taskId, status: task.status, result: null };
661
-
662
- let mergedResult = {};
663
- let comparison = {};
664
- try { mergedResult = JSON.parse(resultRow.merged_result || '{}'); } catch (_) {}
665
- try { comparison = JSON.parse(resultRow.comparison || '{}'); } catch (_) {}
666
-
667
- return {
668
- taskId,
669
- status: task.status,
670
- result: mergedResult,
671
- fairnessApplied: !!resultRow.fairness_applied,
672
- totalSources: resultRow.total_sources,
673
- bestSource: resultRow.best_source,
674
- comparison,
675
- createdAt: resultRow.created_at,
676
- };
677
- }
678
-
679
- // ─── 10. cancelTask ──────────────────────────────────────────────────
680
-
681
- function cancelTask(taskId) {
682
- const task = stmts.getTask.get(taskId);
683
- if (!task) return null;
684
- if (task.status === 'completed' || task.status === 'cancelled') {
685
- return { id: task.id, status: task.status, changed: false };
686
- }
687
-
688
- const now = new Date().toISOString();
689
- stmts.updateTaskStatus.run('cancelled', task.started_at || now, now, taskId);
690
- stmts.cancelPendingAgents.run(taskId);
691
-
692
- return { id: task.id, status: 'cancelled', changed: true };
693
- }
694
-
695
- // ─── 11. getSwarmHistory ─────────────────────────────────────────────
696
-
697
- function getSwarmHistory(siteId, { limit = 50, taskType } = {}) {
698
- let tasks;
699
- if (taskType) {
700
- tasks = stmts.taskHistoryByType.all(siteId, taskType, limit);
701
- } else {
702
- tasks = stmts.taskHistory.all(siteId, limit);
703
- }
704
-
705
- return tasks.map(t => {
706
- let params = {};
707
- try { params = JSON.parse(t.parameters || '{}'); } catch (_) {}
708
- const agents = stmts.getAgentsByTask.all(t.id);
709
- return {
710
- id: t.id,
711
- taskType: t.task_type,
712
- objective: t.objective,
713
- parameters: params,
714
- status: t.status,
715
- agentCount: agents.length,
716
- createdBy: t.created_by,
717
- createdAt: t.created_at,
718
- startedAt: t.started_at,
719
- completedAt: t.completed_at,
720
- };
721
- });
722
- }
723
-
724
- // ─── 12. getSwarmStats ───────────────────────────────────────────────
725
-
726
- function getSwarmStats(siteId) {
727
- const totalTasks = db.prepare('SELECT COUNT(*) as c FROM swarm_tasks WHERE site_id = ?').get(siteId).c;
728
- const completedTasks = db.prepare("SELECT COUNT(*) as c FROM swarm_tasks WHERE site_id = ? AND status = 'completed'").get(siteId).c;
729
- const failedTasks = db.prepare("SELECT COUNT(*) as c FROM swarm_tasks WHERE site_id = ? AND status = 'failed'").get(siteId).c;
730
- const cancelledTasks = db.prepare("SELECT COUNT(*) as c FROM swarm_tasks WHERE site_id = ? AND status = 'cancelled'").get(siteId).c;
731
- const runningTasks = db.prepare("SELECT COUNT(*) as c FROM swarm_tasks WHERE site_id = ? AND status = 'running'").get(siteId).c;
732
-
733
- const avgAgentsRow = db.prepare(`
734
- SELECT AVG(agent_count) as avg_agents FROM (
735
- SELECT COUNT(*) as agent_count FROM swarm_agents a
736
- JOIN swarm_tasks t ON a.task_id = t.id
737
- WHERE t.site_id = ? GROUP BY a.task_id
738
- )
739
- `).get(siteId);
740
- const avgAgents = avgAgentsRow ? Math.round((avgAgentsRow.avg_agents || 0) * 100) / 100 : 0;
741
-
742
- const successRate = totalTasks > 0 ? Math.round((completedTasks / totalTasks) * 10000) / 100 : 0;
743
-
744
- const latencyRow = db.prepare(`
745
- SELECT AVG(
746
- CAST((julianday(completed_at) - julianday(started_at)) * 86400000 AS INTEGER)
747
- ) as avg_latency
748
- FROM swarm_tasks
749
- WHERE site_id = ? AND status = 'completed' AND started_at IS NOT NULL AND completed_at IS NOT NULL
750
- `).get(siteId);
751
- const avgLatencyMs = latencyRow ? Math.round(latencyRow.avg_latency || 0) : 0;
752
-
753
- const taskTypeBreakdown = db.prepare(`
754
- SELECT task_type, COUNT(*) as count,
755
- SUM(CASE WHEN status = 'completed' THEN 1 ELSE 0 END) as completed
756
- FROM swarm_tasks WHERE site_id = ? GROUP BY task_type
757
- `).all(siteId);
758
-
759
- return {
760
- totalTasks,
761
- completedTasks,
762
- failedTasks,
763
- cancelledTasks,
764
- runningTasks,
765
- avgAgents,
766
- successRate,
767
- avgLatencyMs,
768
- taskTypeBreakdown,
769
- };
770
- }
771
-
772
- // ─── Exports ─────────────────────────────────────────────────────────
773
-
774
- module.exports = {
775
- configureSwarm,
776
- getSwarmConfig,
777
- createTask,
778
- assignAgents,
779
- runSwarmTask,
780
- executeAgent,
781
- mergeResults,
782
- applyFairnessToResults,
783
- getTaskStatus,
784
- getTaskResult,
785
- cancelTask,
786
- getSwarmHistory,
787
- getSwarmStats,
788
- };
1
+ const crypto = require('crypto');
2
+ const https = require('https');
3
+ const http = require('http');
4
+ const { URL } = require('url');
5
+ const { db } = require('../models/db');
6
+
7
+ // ─── Schema ──────────────────────────────────────────────────────────
8
+
9
+ db.exec(`
10
+ CREATE TABLE IF NOT EXISTS swarm_configs (
11
+ id TEXT PRIMARY KEY,
12
+ site_id TEXT UNIQUE,
13
+ name TEXT,
14
+ strategy TEXT DEFAULT 'parallel',
15
+ max_agents INTEGER DEFAULT 3,
16
+ timeout_ms INTEGER DEFAULT 30000,
17
+ merge_strategy TEXT DEFAULT 'best_score',
18
+ enabled INTEGER DEFAULT 1,
19
+ created_at TEXT DEFAULT (datetime('now')),
20
+ updated_at TEXT DEFAULT (datetime('now'))
21
+ );
22
+
23
+ CREATE TABLE IF NOT EXISTS swarm_tasks (
24
+ id TEXT PRIMARY KEY,
25
+ site_id TEXT,
26
+ config_id TEXT,
27
+ task_type TEXT,
28
+ objective TEXT,
29
+ parameters TEXT DEFAULT '{}',
30
+ status TEXT DEFAULT 'pending',
31
+ created_by TEXT,
32
+ created_at TEXT DEFAULT (datetime('now')),
33
+ started_at TEXT,
34
+ completed_at TEXT
35
+ );
36
+
37
+ CREATE TABLE IF NOT EXISTS swarm_agents (
38
+ id TEXT PRIMARY KEY,
39
+ task_id TEXT,
40
+ agent_role TEXT,
41
+ agent_type TEXT,
42
+ target TEXT,
43
+ status TEXT DEFAULT 'idle',
44
+ result TEXT DEFAULT '{}',
45
+ score REAL,
46
+ error TEXT,
47
+ started_at TEXT,
48
+ completed_at TEXT
49
+ );
50
+
51
+ CREATE TABLE IF NOT EXISTS swarm_results (
52
+ id TEXT PRIMARY KEY,
53
+ task_id TEXT,
54
+ merged_result TEXT DEFAULT '{}',
55
+ fairness_applied INTEGER DEFAULT 0,
56
+ total_sources INTEGER DEFAULT 0,
57
+ best_source TEXT,
58
+ comparison TEXT DEFAULT '{}',
59
+ created_at TEXT DEFAULT (datetime('now'))
60
+ );
61
+
62
+ CREATE INDEX IF NOT EXISTS idx_swarm_configs_site ON swarm_configs(site_id);
63
+ CREATE INDEX IF NOT EXISTS idx_swarm_tasks_site ON swarm_tasks(site_id);
64
+ CREATE INDEX IF NOT EXISTS idx_swarm_tasks_status ON swarm_tasks(status);
65
+ CREATE INDEX IF NOT EXISTS idx_swarm_tasks_config ON swarm_tasks(config_id);
66
+ CREATE INDEX IF NOT EXISTS idx_swarm_agents_task ON swarm_agents(task_id);
67
+ CREATE INDEX IF NOT EXISTS idx_swarm_agents_status ON swarm_agents(status);
68
+ CREATE INDEX IF NOT EXISTS idx_swarm_results_task ON swarm_results(task_id);
69
+ `);
70
+
71
+ // ─── Prepared Statements ─────────────────────────────────────────────
72
+
73
+ const stmts = {
74
+ upsertConfig: db.prepare(`
75
+ INSERT INTO swarm_configs (id, site_id, name, strategy, max_agents, timeout_ms, merge_strategy, enabled)
76
+ VALUES (@id, @site_id, @name, @strategy, @max_agents, @timeout_ms, @merge_strategy, @enabled)
77
+ ON CONFLICT(site_id) DO UPDATE SET
78
+ name = excluded.name,
79
+ strategy = excluded.strategy,
80
+ max_agents = excluded.max_agents,
81
+ timeout_ms = excluded.timeout_ms,
82
+ merge_strategy = excluded.merge_strategy,
83
+ enabled = excluded.enabled,
84
+ updated_at = datetime('now')
85
+ `),
86
+ getConfig: db.prepare('SELECT * FROM swarm_configs WHERE site_id = ?'),
87
+ getConfigById: db.prepare('SELECT * FROM swarm_configs WHERE id = ?'),
88
+ insertTask: db.prepare(`
89
+ INSERT INTO swarm_tasks (id, site_id, config_id, task_type, objective, parameters, status, created_by)
90
+ VALUES (@id, @site_id, @config_id, @task_type, @objective, @parameters, @status, @created_by)
91
+ `),
92
+ getTask: db.prepare('SELECT * FROM swarm_tasks WHERE id = ?'),
93
+ updateTaskStatus: db.prepare('UPDATE swarm_tasks SET status = ?, started_at = COALESCE(started_at, ?), completed_at = ? WHERE id = ?'),
94
+ insertAgent: db.prepare(`
95
+ INSERT INTO swarm_agents (id, task_id, agent_role, agent_type, target, status)
96
+ VALUES (@id, @task_id, @agent_role, @agent_type, @target, @status)
97
+ `),
98
+ getAgentsByTask: db.prepare('SELECT * FROM swarm_agents WHERE task_id = ? ORDER BY started_at ASC'),
99
+ updateAgent: db.prepare(`
100
+ UPDATE swarm_agents SET status = ?, result = ?, score = ?, error = ?, started_at = COALESCE(started_at, ?), completed_at = ?
101
+ WHERE id = ?
102
+ `),
103
+ insertResult: db.prepare(`
104
+ INSERT INTO swarm_results (id, task_id, merged_result, fairness_applied, total_sources, best_source, comparison)
105
+ VALUES (@id, @task_id, @merged_result, @fairness_applied, @total_sources, @best_source, @comparison)
106
+ `),
107
+ getResult: db.prepare('SELECT * FROM swarm_results WHERE task_id = ?'),
108
+ cancelPendingAgents: db.prepare("UPDATE swarm_agents SET status = 'cancelled' WHERE task_id = ? AND status IN ('idle', 'running')"),
109
+ taskHistory: db.prepare('SELECT * FROM swarm_tasks WHERE site_id = ? ORDER BY created_at DESC LIMIT ?'),
110
+ taskHistoryByType: db.prepare('SELECT * FROM swarm_tasks WHERE site_id = ? AND task_type = ? ORDER BY created_at DESC LIMIT ?'),
111
+ };
112
+
113
+ // ─── 1. configureSwarm ───────────────────────────────────────────────
114
+
115
+ function configureSwarm(siteId, { name, strategy, maxAgents, timeoutMs, mergeStrategy } = {}) {
116
+ const existing = stmts.getConfig.get(siteId);
117
+ const id = existing ? existing.id : crypto.randomUUID();
118
+
119
+ const params = {
120
+ id,
121
+ site_id: siteId,
122
+ name: name ?? (existing ? existing.name : siteId),
123
+ strategy: strategy ?? (existing ? existing.strategy : 'parallel'),
124
+ max_agents: maxAgents ?? (existing ? existing.max_agents : 3),
125
+ timeout_ms: timeoutMs ?? (existing ? existing.timeout_ms : 30000),
126
+ merge_strategy: mergeStrategy ?? (existing ? existing.merge_strategy : 'best_score'),
127
+ enabled: 1,
128
+ };
129
+
130
+ stmts.upsertConfig.run(params);
131
+ return stmts.getConfig.get(siteId);
132
+ }
133
+
134
+ // ─── 2. getSwarmConfig ───────────────────────────────────────────────
135
+
136
+ function getSwarmConfig(siteId) {
137
+ return stmts.getConfig.get(siteId) || null;
138
+ }
139
+
140
+ // ─── 3. createTask ───────────────────────────────────────────────────
141
+
142
+ function createTask(siteId, { taskType, objective, parameters, createdBy } = {}) {
143
+ const config = stmts.getConfig.get(siteId);
144
+ const id = crypto.randomUUID();
145
+
146
+ stmts.insertTask.run({
147
+ id,
148
+ site_id: siteId,
149
+ config_id: config ? config.id : null,
150
+ task_type: taskType || 'general',
151
+ objective: objective || '',
152
+ parameters: JSON.stringify(parameters || {}),
153
+ status: 'pending',
154
+ created_by: createdBy || null,
155
+ });
156
+
157
+ return stmts.getTask.get(id);
158
+ }
159
+
160
+ // ─── 4. assignAgents ─────────────────────────────────────────────────
161
+
162
+ function assignAgents(taskId, agentDefinitions) {
163
+ const task = stmts.getTask.get(taskId);
164
+ if (!task) throw new Error(`Task ${taskId} not found`);
165
+
166
+ const config = task.config_id ? stmts.getConfigById.get(task.config_id) : null;
167
+ const maxAgents = config ? config.max_agents : 10;
168
+ const defs = agentDefinitions.slice(0, maxAgents);
169
+
170
+ const agents = [];
171
+ const insertMany = db.transaction((items) => {
172
+ for (const def of items) {
173
+ const agentId = crypto.randomUUID();
174
+ stmts.insertAgent.run({
175
+ id: agentId,
176
+ task_id: taskId,
177
+ agent_role: def.role || 'worker',
178
+ agent_type: def.type || 'fetch',
179
+ target: def.target || '',
180
+ status: 'idle',
181
+ });
182
+ agents.push({ id: agentId, task_id: taskId, agent_role: def.role, agent_type: def.type, target: def.target, status: 'idle' });
183
+ }
184
+ });
185
+
186
+ insertMany(defs);
187
+ return agents;
188
+ }
189
+
190
+ // ─── 5. runSwarmTask ─────────────────────────────────────────────────
191
+
192
+ async function runSwarmTask(taskId) {
193
+ const task = stmts.getTask.get(taskId);
194
+ if (!task) throw new Error(`Task ${taskId} not found`);
195
+ if (task.status === 'cancelled') throw new Error(`Task ${taskId} is cancelled`);
196
+
197
+ const config = task.config_id ? stmts.getConfigById.get(task.config_id) : null;
198
+ const strategy = config ? config.strategy : 'parallel';
199
+ const mergeStrategy = config ? config.merge_strategy : 'best_score';
200
+ const timeoutMs = config ? config.timeout_ms : 30000;
201
+
202
+ const now = new Date().toISOString();
203
+ stmts.updateTaskStatus.run('running', now, null, taskId);
204
+
205
+ let agents = stmts.getAgentsByTask.all(taskId);
206
+ if (agents.length === 0) throw new Error(`No agents assigned to task ${taskId}`);
207
+
208
+ try {
209
+ switch (strategy) {
210
+ case 'sequential':
211
+ await runSequential(agents, timeoutMs);
212
+ break;
213
+ case 'competitive':
214
+ await runCompetitive(agents, timeoutMs);
215
+ break;
216
+ case 'collaborative':
217
+ await runCollaborative(agents, timeoutMs);
218
+ break;
219
+ case 'parallel':
220
+ default:
221
+ await runParallel(agents, timeoutMs);
222
+ break;
223
+ }
224
+
225
+ agents = stmts.getAgentsByTask.all(taskId);
226
+ const completedAgents = agents.filter(a => a.status === 'completed');
227
+
228
+ const rawResults = completedAgents.map(a => {
229
+ let parsed = {};
230
+ try { parsed = JSON.parse(a.result || '{}'); } catch (_) {}
231
+ return { ...parsed, _agentId: a.id, _role: a.agent_role, _score: a.score, _source: a.target };
232
+ });
233
+
234
+ const fairResults = applyFairnessToResults(rawResults);
235
+ const merged = mergeResults(fairResults, mergeStrategy);
236
+
237
+ const bestAgent = completedAgents.reduce((best, cur) =>
238
+ (cur.score || 0) > (best.score || 0) ? cur : best, completedAgents[0]);
239
+
240
+ const comparison = completedAgents.map(a => ({
241
+ agentId: a.id,
242
+ role: a.agent_role,
243
+ target: a.target,
244
+ score: a.score,
245
+ status: a.status,
246
+ }));
247
+
248
+ const resultId = crypto.randomUUID();
249
+ stmts.insertResult.run({
250
+ id: resultId,
251
+ task_id: taskId,
252
+ merged_result: JSON.stringify(merged),
253
+ fairness_applied: 1,
254
+ total_sources: completedAgents.length,
255
+ best_source: bestAgent ? bestAgent.target : null,
256
+ comparison: JSON.stringify(comparison),
257
+ });
258
+
259
+ const doneAt = new Date().toISOString();
260
+ stmts.updateTaskStatus.run('completed', now, doneAt, taskId);
261
+
262
+ return { taskId, status: 'completed', result: merged, sources: completedAgents.length, bestSource: bestAgent ? bestAgent.target : null };
263
+ } catch (err) {
264
+ const failAt = new Date().toISOString();
265
+ stmts.updateTaskStatus.run('failed', now, failAt, taskId);
266
+ throw err;
267
+ }
268
+ }
269
+
270
+ // ─── Strategy Runners ────────────────────────────────────────────────
271
+
272
+ async function runParallel(agents, timeoutMs) {
273
+ const promises = agents.map(agent => runSingleAgent(agent, timeoutMs));
274
+ await Promise.allSettled(promises);
275
+ }
276
+
277
+ async function runSequential(agents, timeoutMs) {
278
+ let previousResult = null;
279
+ for (const agent of agents) {
280
+ previousResult = await runSingleAgent(agent, timeoutMs, previousResult);
281
+ }
282
+ }
283
+
284
+ async function runCompetitive(agents, timeoutMs) {
285
+ const promises = agents.map(agent => runSingleAgent(agent, timeoutMs));
286
+ await Promise.allSettled(promises);
287
+ }
288
+
289
+ async function runCollaborative(agents, timeoutMs) {
290
+ const promises = agents.map(agent => runSingleAgent(agent, timeoutMs));
291
+ await Promise.allSettled(promises);
292
+ }
293
+
294
+ async function runSingleAgent(agent, timeoutMs, previousResult) {
295
+ const startedAt = new Date().toISOString();
296
+ stmts.updateAgent.run('running', agent.result || '{}', null, null, startedAt, null, agent.id);
297
+
298
+ try {
299
+ const agentInput = { ...agent, timeoutMs, previousResult };
300
+ const result = await executeAgent(agentInput);
301
+ const completedAt = new Date().toISOString();
302
+ stmts.updateAgent.run('completed', JSON.stringify(result), result.relevance || 0, null, startedAt, completedAt, agent.id);
303
+ return result;
304
+ } catch (err) {
305
+ const completedAt = new Date().toISOString();
306
+ stmts.updateAgent.run('failed', '{}', 0, err.message, startedAt, completedAt, agent.id);
307
+ return null;
308
+ }
309
+ }
310
+
311
+ // ─── 6. executeAgent ─────────────────────────────────────────────────
312
+
313
+ function executeAgent(agent) {
314
+ const timeout = agent.timeoutMs || 30000;
315
+ const targetUrl = agent.target;
316
+
317
+ if (!targetUrl) {
318
+ return Promise.resolve({ title: '', description: '', prices: [], relevance: 0, source: '' });
319
+ }
320
+
321
+ return new Promise((resolve, reject) => {
322
+ let parsed;
323
+ try {
324
+ parsed = new URL(targetUrl);
325
+ } catch (_) {
326
+ return resolve({ title: '', description: '', prices: [], relevance: 0, source: targetUrl, error: 'Invalid URL' });
327
+ }
328
+
329
+ const transport = parsed.protocol === 'https:' ? https : http;
330
+ const timer = setTimeout(() => {
331
+ req.destroy();
332
+ resolve({ title: '', description: '', prices: [], relevance: 0, source: targetUrl, error: 'Timeout' });
333
+ }, timeout);
334
+
335
+ const req = transport.get(targetUrl, { headers: { 'User-Agent': 'WAB-Swarm-Agent/1.0' } }, (res) => {
336
+ if (res.statusCode >= 300 && res.statusCode < 400 && res.headers.location) {
337
+ clearTimeout(timer);
338
+ const redirectAgent = { ...agent, target: res.headers.location };
339
+ resolve(executeAgent(redirectAgent));
340
+ res.resume();
341
+ return;
342
+ }
343
+
344
+ let body = '';
345
+ const maxBytes = 2 * 1024 * 1024;
346
+ let received = 0;
347
+
348
+ res.on('data', (chunk) => {
349
+ received += chunk.length;
350
+ if (received <= maxBytes) body += chunk;
351
+ });
352
+
353
+ res.on('end', () => {
354
+ clearTimeout(timer);
355
+
356
+ const title = extractFirst(body, /<title[^>]*>([\s\S]*?)<\/title>/i) || '';
357
+ const metaDesc = extractFirst(body, /<meta\s+name=["']description["']\s+content=["']([\s\S]*?)["']/i)
358
+ || extractFirst(body, /<meta\s+content=["']([\s\S]*?)["']\s+name=["']description["']/i)
359
+ || '';
360
+ const h1 = extractFirst(body, /<h1[^>]*>([\s\S]*?)<\/h1>/i) || '';
361
+ const priceMatches = body.match(/(?:\$|USD\s?)(\d{1,7}(?:[.,]\d{2})?)/g) || [];
362
+ const prices = [...new Set(priceMatches.slice(0, 20))];
363
+
364
+ const objective = agent.previousResult?.objective || '';
365
+ const relevance = computeRelevanceScore(title, metaDesc, h1, prices, objective, targetUrl);
366
+
367
+ resolve({
368
+ title: cleanHtml(title).slice(0, 500),
369
+ description: cleanHtml(metaDesc).slice(0, 1000),
370
+ h1: cleanHtml(h1).slice(0, 500),
371
+ prices,
372
+ relevance,
373
+ source: targetUrl,
374
+ statusCode: res.statusCode,
375
+ });
376
+ });
377
+
378
+ res.on('error', (err) => {
379
+ clearTimeout(timer);
380
+ resolve({ title: '', description: '', prices: [], relevance: 0, source: targetUrl, error: err.message });
381
+ });
382
+ });
383
+
384
+ req.on('error', (err) => {
385
+ clearTimeout(timer);
386
+ resolve({ title: '', description: '', prices: [], relevance: 0, source: targetUrl, error: err.message });
387
+ });
388
+ });
389
+ }
390
+
391
+ function extractFirst(html, regex) {
392
+ const m = html.match(regex);
393
+ return m ? m[1].trim() : null;
394
+ }
395
+
396
+ function cleanHtml(text) {
397
+ return text.replace(/<[^>]+>/g, '').replace(/&amp;/g, '&').replace(/&lt;/g, '<')
398
+ .replace(/&gt;/g, '>').replace(/&quot;/g, '"').replace(/&#039;/g, "'")
399
+ .replace(/&nbsp;/g, ' ').replace(/\s+/g, ' ').trim();
400
+ }
401
+
402
+ function computeRelevanceScore(title, description, h1, prices, objective, source) {
403
+ let score = 0;
404
+
405
+ if (title) score += 15;
406
+ if (description) score += 15;
407
+ if (h1) score += 10;
408
+ if (prices.length > 0) score += 10;
409
+
410
+ if (objective) {
411
+ const terms = objective.toLowerCase().split(/\s+/).filter(Boolean);
412
+ const corpus = `${title} ${description} ${h1}`.toLowerCase();
413
+ let hits = 0;
414
+ for (const t of terms) {
415
+ if (corpus.includes(t)) hits++;
416
+ }
417
+ const termRatio = terms.length > 0 ? hits / terms.length : 0;
418
+ score += Math.round(termRatio * 40);
419
+ } else {
420
+ score += 20;
421
+ }
422
+
423
+ if (source) {
424
+ try {
425
+ const u = new URL(source);
426
+ if (u.protocol === 'https:') score += 5;
427
+ if (u.pathname !== '/' && u.pathname.length > 1) score += 5;
428
+ } catch (_) {}
429
+ }
430
+
431
+ return Math.min(Math.max(Math.round(score * 100) / 100, 0), 100);
432
+ }
433
+
434
+ // ─── 7. mergeResults ─────────────────────────────────────────────────
435
+
436
+ function mergeResults(agents, strategy) {
437
+ if (!agents || agents.length === 0) return { merged: true, data: [], strategy };
438
+
439
+ switch (strategy) {
440
+ case 'best_score': return mergeBestScore(agents);
441
+ case 'weighted_average': return mergeWeightedAverage(agents);
442
+ case 'union': return mergeUnion(agents);
443
+ case 'consensus': return mergeConsensus(agents);
444
+ default: return mergeBestScore(agents);
445
+ }
446
+ }
447
+
448
+ function mergeBestScore(agents) {
449
+ const sorted = [...agents].sort((a, b) => (b._score || b.relevance || 0) - (a._score || a.relevance || 0));
450
+ const best = sorted[0];
451
+ return {
452
+ merged: true,
453
+ strategy: 'best_score',
454
+ title: best.title || '',
455
+ description: best.description || '',
456
+ h1: best.h1 || '',
457
+ prices: best.prices || [],
458
+ relevance: best._score || best.relevance || 0,
459
+ source: best._source || best.source || '',
460
+ allSources: sorted.map(a => ({ source: a._source || a.source, score: a._score || a.relevance || 0 })),
461
+ };
462
+ }
463
+
464
+ function mergeWeightedAverage(agents) {
465
+ const totalWeight = agents.reduce((s, a) => s + (a._score || a.relevance || 0), 0);
466
+ if (totalWeight === 0) return mergeBestScore(agents);
467
+
468
+ const allPrices = [];
469
+ let weightedRelevance = 0;
470
+ let bestTitle = '';
471
+ let bestDesc = '';
472
+ let bestH1 = '';
473
+ let highestWeight = -1;
474
+
475
+ for (const a of agents) {
476
+ const w = a._score || a.relevance || 0;
477
+ weightedRelevance += w * w;
478
+ if (a.prices) allPrices.push(...a.prices);
479
+ if (w > highestWeight) {
480
+ highestWeight = w;
481
+ bestTitle = a.title || bestTitle;
482
+ bestDesc = a.description || bestDesc;
483
+ bestH1 = a.h1 || bestH1;
484
+ }
485
+ }
486
+
487
+ return {
488
+ merged: true,
489
+ strategy: 'weighted_average',
490
+ title: bestTitle,
491
+ description: bestDesc,
492
+ h1: bestH1,
493
+ prices: [...new Set(allPrices)],
494
+ relevance: Math.round((weightedRelevance / totalWeight) * 100) / 100,
495
+ sourceCount: agents.length,
496
+ allSources: agents.map(a => ({ source: a._source || a.source, score: a._score || a.relevance || 0 })),
497
+ };
498
+ }
499
+
500
+ function mergeUnion(agents) {
501
+ const allPrices = [];
502
+ const allTitles = [];
503
+ const allDescriptions = [];
504
+ let maxRelevance = 0;
505
+
506
+ for (const a of agents) {
507
+ if (a.title) allTitles.push(a.title);
508
+ if (a.description) allDescriptions.push(a.description);
509
+ if (a.prices) allPrices.push(...a.prices);
510
+ maxRelevance = Math.max(maxRelevance, a._score || a.relevance || 0);
511
+ }
512
+
513
+ return {
514
+ merged: true,
515
+ strategy: 'union',
516
+ titles: [...new Set(allTitles)],
517
+ descriptions: [...new Set(allDescriptions)],
518
+ prices: [...new Set(allPrices)],
519
+ relevance: maxRelevance,
520
+ sourceCount: agents.length,
521
+ allSources: agents.map(a => ({ source: a._source || a.source, score: a._score || a.relevance || 0 })),
522
+ };
523
+ }
524
+
525
+ function mergeConsensus(agents) {
526
+ const titleFreq = {};
527
+ const priceFreq = {};
528
+ let bestDesc = '';
529
+ let bestDescScore = -1;
530
+
531
+ for (const a of agents) {
532
+ const t = (a.title || '').toLowerCase().trim();
533
+ if (t) titleFreq[t] = (titleFreq[t] || 0) + 1;
534
+ if (a.prices) {
535
+ for (const p of a.prices) priceFreq[p] = (priceFreq[p] || 0) + 1;
536
+ }
537
+ const s = a._score || a.relevance || 0;
538
+ if (s > bestDescScore && a.description) {
539
+ bestDescScore = s;
540
+ bestDesc = a.description;
541
+ }
542
+ }
543
+
544
+ const threshold = Math.max(Math.ceil(agents.length / 2), 1);
545
+
546
+ const consensusTitles = Object.entries(titleFreq).filter(([, c]) => c >= threshold).map(([t]) => t);
547
+ const consensusPrices = Object.entries(priceFreq).filter(([, c]) => c >= threshold).map(([p]) => p);
548
+
549
+ const avgRelevance = agents.reduce((s, a) => s + (a._score || a.relevance || 0), 0) / agents.length;
550
+
551
+ return {
552
+ merged: true,
553
+ strategy: 'consensus',
554
+ titles: consensusTitles,
555
+ description: bestDesc,
556
+ prices: consensusPrices,
557
+ relevance: Math.round(avgRelevance * 100) / 100,
558
+ agreement: {
559
+ titleAgreement: consensusTitles.length > 0,
560
+ priceAgreement: consensusPrices.length > 0,
561
+ threshold,
562
+ totalAgents: agents.length,
563
+ },
564
+ allSources: agents.map(a => ({ source: a._source || a.source, score: a._score || a.relevance || 0 })),
565
+ };
566
+ }
567
+
568
+ // ─── 8. applyFairnessToResults ───────────────────────────────────────
569
+
570
+ function applyFairnessToResults(results) {
571
+ if (!results || results.length === 0) return results;
572
+
573
+ const domainCounts = {};
574
+ for (const r of results) {
575
+ const domain = extractDomain(r._source || r.source || '');
576
+ domainCounts[domain] = (domainCounts[domain] || 0) + 1;
577
+ }
578
+
579
+ const totalResults = results.length;
580
+
581
+ return results.map(r => {
582
+ const domain = extractDomain(r._source || r.source || '');
583
+ const score = r._score || r.relevance || 0;
584
+ let adjusted = score;
585
+
586
+ const isIndie = isIndependentSite(domain);
587
+ if (isIndie) adjusted += score * 0.12;
588
+
589
+ const domainShare = (domainCounts[domain] || 1) / totalResults;
590
+ if (domainShare > 0.4) {
591
+ adjusted -= score * 0.10;
592
+ }
593
+
594
+ const isSmall = isSmallSite(domain);
595
+ if (isSmall) adjusted += score * 0.08;
596
+
597
+ adjusted = Math.min(Math.max(Math.round(adjusted * 100) / 100, 0), 100);
598
+
599
+ return { ...r, _score: adjusted, _fairnessApplied: true, _originalScore: score };
600
+ });
601
+ }
602
+
603
+ function extractDomain(url) {
604
+ try { return new URL(url).hostname; } catch (_) { return url; }
605
+ }
606
+
607
+ const MAJOR_DOMAINS = [
608
+ 'amazon.com', 'google.com', 'facebook.com', 'apple.com', 'microsoft.com',
609
+ 'walmart.com', 'ebay.com', 'target.com', 'bestbuy.com', 'alibaba.com',
610
+ 'aliexpress.com', 'rakuten.com', 'shopify.com', 'etsy.com',
611
+ ];
612
+
613
+ function isIndependentSite(domain) {
614
+ const lower = domain.toLowerCase().replace(/^www\./, '');
615
+ return !MAJOR_DOMAINS.some(d => lower === d || lower.endsWith('.' + d));
616
+ }
617
+
618
+ function isSmallSite(domain) {
619
+ const lower = domain.toLowerCase().replace(/^www\./, '');
620
+ const parts = lower.split('.');
621
+ if (parts.length > 3) return false;
622
+ return !MAJOR_DOMAINS.some(d => lower === d || lower.endsWith('.' + d));
623
+ }
624
+
625
+ // ─── 9. getTaskStatus / getTaskResult ────────────────────────────────
626
+
627
+ function getTaskStatus(taskId) {
628
+ const task = stmts.getTask.get(taskId);
629
+ if (!task) return null;
630
+
631
+ const agents = stmts.getAgentsByTask.all(taskId);
632
+ const agentSummary = agents.map(a => ({
633
+ id: a.id,
634
+ role: a.agent_role,
635
+ type: a.agent_type,
636
+ target: a.target,
637
+ status: a.status,
638
+ score: a.score,
639
+ error: a.error,
640
+ }));
641
+
642
+ return {
643
+ id: task.id,
644
+ siteId: task.site_id,
645
+ taskType: task.task_type,
646
+ objective: task.objective,
647
+ status: task.status,
648
+ createdAt: task.created_at,
649
+ startedAt: task.started_at,
650
+ completedAt: task.completed_at,
651
+ agents: agentSummary,
652
+ };
653
+ }
654
+
655
+ function getTaskResult(taskId) {
656
+ const task = stmts.getTask.get(taskId);
657
+ if (!task) return null;
658
+
659
+ const resultRow = stmts.getResult.get(taskId);
660
+ if (!resultRow) return { taskId, status: task.status, result: null };
661
+
662
+ let mergedResult = {};
663
+ let comparison = {};
664
+ try { mergedResult = JSON.parse(resultRow.merged_result || '{}'); } catch (_) {}
665
+ try { comparison = JSON.parse(resultRow.comparison || '{}'); } catch (_) {}
666
+
667
+ return {
668
+ taskId,
669
+ status: task.status,
670
+ result: mergedResult,
671
+ fairnessApplied: !!resultRow.fairness_applied,
672
+ totalSources: resultRow.total_sources,
673
+ bestSource: resultRow.best_source,
674
+ comparison,
675
+ createdAt: resultRow.created_at,
676
+ };
677
+ }
678
+
679
+ // ─── 10. cancelTask ──────────────────────────────────────────────────
680
+
681
+ function cancelTask(taskId) {
682
+ const task = stmts.getTask.get(taskId);
683
+ if (!task) return null;
684
+ if (task.status === 'completed' || task.status === 'cancelled') {
685
+ return { id: task.id, status: task.status, changed: false };
686
+ }
687
+
688
+ const now = new Date().toISOString();
689
+ stmts.updateTaskStatus.run('cancelled', task.started_at || now, now, taskId);
690
+ stmts.cancelPendingAgents.run(taskId);
691
+
692
+ return { id: task.id, status: 'cancelled', changed: true };
693
+ }
694
+
695
+ // ─── 11. getSwarmHistory ─────────────────────────────────────────────
696
+
697
+ function getSwarmHistory(siteId, { limit = 50, taskType } = {}) {
698
+ let tasks;
699
+ if (taskType) {
700
+ tasks = stmts.taskHistoryByType.all(siteId, taskType, limit);
701
+ } else {
702
+ tasks = stmts.taskHistory.all(siteId, limit);
703
+ }
704
+
705
+ return tasks.map(t => {
706
+ let params = {};
707
+ try { params = JSON.parse(t.parameters || '{}'); } catch (_) {}
708
+ const agents = stmts.getAgentsByTask.all(t.id);
709
+ return {
710
+ id: t.id,
711
+ taskType: t.task_type,
712
+ objective: t.objective,
713
+ parameters: params,
714
+ status: t.status,
715
+ agentCount: agents.length,
716
+ createdBy: t.created_by,
717
+ createdAt: t.created_at,
718
+ startedAt: t.started_at,
719
+ completedAt: t.completed_at,
720
+ };
721
+ });
722
+ }
723
+
724
+ // ─── 12. getSwarmStats ───────────────────────────────────────────────
725
+
726
+ function getSwarmStats(siteId) {
727
+ const totalTasks = db.prepare('SELECT COUNT(*) as c FROM swarm_tasks WHERE site_id = ?').get(siteId).c;
728
+ const completedTasks = db.prepare("SELECT COUNT(*) as c FROM swarm_tasks WHERE site_id = ? AND status = 'completed'").get(siteId).c;
729
+ const failedTasks = db.prepare("SELECT COUNT(*) as c FROM swarm_tasks WHERE site_id = ? AND status = 'failed'").get(siteId).c;
730
+ const cancelledTasks = db.prepare("SELECT COUNT(*) as c FROM swarm_tasks WHERE site_id = ? AND status = 'cancelled'").get(siteId).c;
731
+ const runningTasks = db.prepare("SELECT COUNT(*) as c FROM swarm_tasks WHERE site_id = ? AND status = 'running'").get(siteId).c;
732
+
733
+ const avgAgentsRow = db.prepare(`
734
+ SELECT AVG(agent_count) as avg_agents FROM (
735
+ SELECT COUNT(*) as agent_count FROM swarm_agents a
736
+ JOIN swarm_tasks t ON a.task_id = t.id
737
+ WHERE t.site_id = ? GROUP BY a.task_id
738
+ )
739
+ `).get(siteId);
740
+ const avgAgents = avgAgentsRow ? Math.round((avgAgentsRow.avg_agents || 0) * 100) / 100 : 0;
741
+
742
+ const successRate = totalTasks > 0 ? Math.round((completedTasks / totalTasks) * 10000) / 100 : 0;
743
+
744
+ const latencyRow = db.prepare(`
745
+ SELECT AVG(
746
+ CAST((julianday(completed_at) - julianday(started_at)) * 86400000 AS INTEGER)
747
+ ) as avg_latency
748
+ FROM swarm_tasks
749
+ WHERE site_id = ? AND status = 'completed' AND started_at IS NOT NULL AND completed_at IS NOT NULL
750
+ `).get(siteId);
751
+ const avgLatencyMs = latencyRow ? Math.round(latencyRow.avg_latency || 0) : 0;
752
+
753
+ const taskTypeBreakdown = db.prepare(`
754
+ SELECT task_type, COUNT(*) as count,
755
+ SUM(CASE WHEN status = 'completed' THEN 1 ELSE 0 END) as completed
756
+ FROM swarm_tasks WHERE site_id = ? GROUP BY task_type
757
+ `).all(siteId);
758
+
759
+ return {
760
+ totalTasks,
761
+ completedTasks,
762
+ failedTasks,
763
+ cancelledTasks,
764
+ runningTasks,
765
+ avgAgents,
766
+ successRate,
767
+ avgLatencyMs,
768
+ taskTypeBreakdown,
769
+ };
770
+ }
771
+
772
+ // ─── Exports ─────────────────────────────────────────────────────────
773
+
774
+ module.exports = {
775
+ configureSwarm,
776
+ getSwarmConfig,
777
+ createTask,
778
+ assignAgents,
779
+ runSwarmTask,
780
+ executeAgent,
781
+ mergeResults,
782
+ applyFairnessToResults,
783
+ getTaskStatus,
784
+ getTaskResult,
785
+ cancelTask,
786
+ getSwarmHistory,
787
+ getSwarmStats,
788
+ };