web-agent-bridge 3.4.0 → 3.8.1

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 (310) hide show
  1. package/LICENSE +84 -84
  2. package/README.ar.md +1563 -1304
  3. package/README.md +137 -298
  4. package/bin/agent-runner.js +474 -474
  5. package/bin/cli.js +237 -237
  6. package/bin/wab-init.js +244 -223
  7. package/bin/wab.js +80 -80
  8. package/examples/azure-dns-wab.js +83 -83
  9. package/examples/bidi-agent.js +119 -119
  10. package/examples/cloudflare-wab-dns.js +121 -121
  11. package/examples/cpanel-wab-dns.js +114 -114
  12. package/examples/cross-site-agent.js +91 -91
  13. package/examples/dns-discovery-agent.js +166 -166
  14. package/examples/gcp-dns-wab.js +76 -76
  15. package/examples/governance-agent.js +169 -169
  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 -103
  19. package/examples/puppeteer-agent.js +108 -108
  20. package/examples/route53-wab-dns.js +144 -144
  21. package/examples/saas-dashboard/README.md +55 -55
  22. package/examples/safe-mode-agent.js +96 -96
  23. package/examples/self-discovery.js +106 -0
  24. package/examples/shopify-hydrogen/README.md +74 -74
  25. package/examples/vision-agent.js +171 -171
  26. package/examples/wab-sign.js +74 -74
  27. package/examples/wab-verify.js +60 -60
  28. package/examples/wordpress-elementor/README.md +77 -77
  29. package/package.json +93 -93
  30. package/public/.well-known/agent-tools.json +180 -180
  31. package/public/.well-known/ai-assets.json +59 -59
  32. package/public/.well-known/security.txt +8 -8
  33. package/public/.well-known/wab.json +28 -28
  34. package/public/activate.html +448 -368
  35. package/public/adopt.html +236 -0
  36. package/public/adoption-metrics.html +188 -188
  37. package/public/agent-workspace.html +359 -349
  38. package/public/ai.html +198 -198
  39. package/public/api.html +397 -413
  40. package/public/azure-dns-integration.html +289 -289
  41. package/public/browser.html +486 -486
  42. package/public/cloudflare-integration.html +380 -380
  43. package/public/commander-dashboard.html +243 -243
  44. package/public/cookies.html +210 -210
  45. package/public/cpanel-integration.html +398 -398
  46. package/public/css/agent-workspace.css +1713 -1713
  47. package/public/css/premium.css +317 -317
  48. package/public/css/styles.css +1401 -1263
  49. package/public/dashboard-shieldlink.html +295 -0
  50. package/public/dashboard.html +711 -707
  51. package/public/dns.html +436 -436
  52. package/public/docs.html +588 -588
  53. package/public/enterprise-mesh.ar.html +80 -0
  54. package/public/enterprise-mesh.html +81 -0
  55. package/public/feed.xml +89 -89
  56. package/public/gcp-dns-integration.html +318 -318
  57. package/public/governance.ar.html +70 -0
  58. package/public/governance.html +69 -0
  59. package/public/growth.html +465 -465
  60. package/public/index.html +1372 -1266
  61. package/public/integrations.html +556 -556
  62. package/public/js/activate.js +449 -145
  63. package/public/js/agent-workspace.js +1740 -1740
  64. package/public/js/auth-nav.js +117 -65
  65. package/public/js/auth-redirect.js +12 -12
  66. package/public/js/cookie-consent.js +56 -56
  67. package/public/js/dns.js +438 -438
  68. package/public/js/wab-demo-page.js +721 -721
  69. package/public/js/ws-client.js +74 -74
  70. package/public/l-preview.html +242 -0
  71. package/public/llms-full.txt +360 -360
  72. package/public/llms.txt +125 -125
  73. package/public/login.html +85 -85
  74. package/public/mesh-dashboard.html +328 -328
  75. package/public/milestones.html +346 -0
  76. package/public/one-click.html +779 -0
  77. package/public/openapi.json +669 -669
  78. package/public/partners.ar.html +145 -0
  79. package/public/partners.html +143 -0
  80. package/public/phone-shield.html +281 -281
  81. package/public/plesk-integration.html +375 -375
  82. package/public/premium-dashboard.html +2489 -2489
  83. package/public/premium.html +793 -793
  84. package/public/privacy.html +297 -297
  85. package/public/provider-onboarding.html +172 -172
  86. package/public/provider-sandbox.html +134 -134
  87. package/public/providers.html +359 -359
  88. package/public/refusals.html +172 -0
  89. package/public/register.html +105 -105
  90. package/public/registrar-integrations.html +141 -141
  91. package/public/ring4.html +292 -0
  92. package/public/robots.txt +99 -99
  93. package/public/route53-integration.html +531 -531
  94. package/public/score.html +263 -0
  95. package/public/script/wab-consent.d.ts +36 -36
  96. package/public/script/wab-consent.js +104 -104
  97. package/public/script/wab-schema.js +131 -131
  98. package/public/script/wab.d.ts +108 -108
  99. package/public/script/wab.min.js +580 -580
  100. package/public/security.txt +8 -8
  101. package/public/shieldlink.html +244 -0
  102. package/public/shieldqr.html +231 -231
  103. package/public/sitemap.xml +13 -1
  104. package/public/terms.html +256 -256
  105. package/public/trust-graph-api.ar.html +92 -0
  106. package/public/trust-graph-api.html +91 -0
  107. package/public/wab-features.html +560 -0
  108. package/public/wab-trust.html +200 -200
  109. package/public/wab-truth.html +375 -0
  110. package/public/wab-vs-protocols.html +210 -210
  111. package/public/whitepaper.html +449 -449
  112. package/script/ai-agent-bridge.js +1754 -1754
  113. package/sdk/README.md +99 -99
  114. package/sdk/agent-mesh.js +449 -449
  115. package/sdk/auto-discovery.js +301 -288
  116. package/sdk/commander.js +262 -262
  117. package/sdk/governance.js +262 -262
  118. package/sdk/index.d.ts +464 -464
  119. package/sdk/index.js +649 -649
  120. package/sdk/multi-agent.js +318 -318
  121. package/sdk/safe-mode.js +221 -221
  122. package/sdk/safety-shield.js +219 -219
  123. package/sdk/schema-discovery.js +83 -83
  124. package/server/adapters/index.js +520 -520
  125. package/server/config/plans.js +412 -367
  126. package/server/config/secrets.js +102 -102
  127. package/server/control-plane/index.js +301 -301
  128. package/server/data-plane/index.js +354 -354
  129. package/server/index.js +790 -670
  130. package/server/llm/index.js +404 -404
  131. package/server/middleware/adminAuth.js +35 -35
  132. package/server/middleware/api-tier.js +170 -0
  133. package/server/middleware/auth.js +50 -50
  134. package/server/middleware/featureGate.js +88 -88
  135. package/server/middleware/rateLimits.js +100 -100
  136. package/server/middleware/sensitiveAction.js +157 -157
  137. package/server/middleware/wab-trust.js +141 -0
  138. package/server/migrations/001_add_analytics_indexes.sql +7 -7
  139. package/server/migrations/002_premium_features.sql +418 -418
  140. package/server/migrations/003_ads_integer_cents.sql +33 -33
  141. package/server/migrations/004_agent_os.sql +158 -158
  142. package/server/migrations/005_marketplace_metering.sql +126 -126
  143. package/server/migrations/006_growth_suite.sql +138 -0
  144. package/server/migrations/007_governance.sql +106 -106
  145. package/server/migrations/008_plans.sql +144 -144
  146. package/server/migrations/009_shieldqr.sql +30 -30
  147. package/server/migrations/010_extended_trust.sql +33 -33
  148. package/server/migrations/011_outreach.sql +47 -0
  149. package/server/migrations/012_shieldlink.sql +116 -0
  150. package/server/migrations/013_ct_monitor.sql +13 -0
  151. package/server/migrations/014_wab_advanced_features.sql +128 -0
  152. package/server/migrations/015_wab_truth_layer.sql +101 -0
  153. package/server/migrations/016_ring4_external_trust.sql +84 -0
  154. package/server/migrations/017_ring4_extensions.sql +69 -0
  155. package/server/migrations/018_commercial_foundations.sql +167 -0
  156. package/server/migrations/019_unify_tier_constraints.sql +133 -0
  157. package/server/models/adapters/index.js +33 -33
  158. package/server/models/adapters/mysql.js +183 -183
  159. package/server/models/adapters/postgresql.js +172 -172
  160. package/server/models/adapters/sqlite.js +7 -7
  161. package/server/models/db.js +740 -740
  162. package/server/observability/failure-analysis.js +337 -337
  163. package/server/observability/index.js +394 -394
  164. package/server/protocol/capabilities.js +223 -223
  165. package/server/protocol/index.js +243 -243
  166. package/server/protocol/schema.js +584 -584
  167. package/server/registry/certification.js +271 -271
  168. package/server/registry/index.js +326 -326
  169. package/server/routes/activate.js +478 -0
  170. package/server/routes/admin-outreach.js +239 -0
  171. package/server/routes/admin-plans.js +76 -76
  172. package/server/routes/admin-premium.js +674 -673
  173. package/server/routes/admin-shieldlink.js +137 -0
  174. package/server/routes/admin-shieldqr.js +90 -90
  175. package/server/routes/admin-trust-monitor.js +139 -83
  176. package/server/routes/admin.js +550 -549
  177. package/server/routes/adopt.js +61 -0
  178. package/server/routes/ads.js +130 -130
  179. package/server/routes/agent-workspace.js +540 -540
  180. package/server/routes/api-keys.js +127 -0
  181. package/server/routes/api.js +150 -150
  182. package/server/routes/auth.js +71 -71
  183. package/server/routes/billing.js +57 -57
  184. package/server/routes/commander.js +316 -316
  185. package/server/routes/customer-shieldlink.js +133 -0
  186. package/server/routes/demo-showcase.js +332 -332
  187. package/server/routes/demo-store.js +154 -154
  188. package/server/routes/diagnose.js +373 -0
  189. package/server/routes/discovery.js +2348 -2348
  190. package/server/routes/enterprise-mesh.js +170 -0
  191. package/server/routes/gateway.js +173 -173
  192. package/server/routes/governance-saas.js +203 -0
  193. package/server/routes/governance.js +208 -208
  194. package/server/routes/growth.js +1048 -0
  195. package/server/routes/intent.js +328 -0
  196. package/server/routes/license.js +251 -251
  197. package/server/routes/mesh.js +469 -469
  198. package/server/routes/noscript.js +543 -543
  199. package/server/routes/partners.js +201 -0
  200. package/server/routes/plans.js +33 -33
  201. package/server/routes/premium-v2.js +686 -686
  202. package/server/routes/premium.js +724 -724
  203. package/server/routes/providers.js +650 -650
  204. package/server/routes/reputation.js +411 -0
  205. package/server/routes/ring4.js +885 -0
  206. package/server/routes/runtime.js +2148 -2148
  207. package/server/routes/shieldlink.js +70 -0
  208. package/server/routes/shieldqr.js +88 -88
  209. package/server/routes/sovereign.js +465 -465
  210. package/server/routes/truth-layer.js +670 -0
  211. package/server/routes/universal.js +200 -200
  212. package/server/routes/unsubscribe.js +51 -0
  213. package/server/routes/wab-api.js +850 -850
  214. package/server/routes/wab-cache.js +282 -0
  215. package/server/runtime/container-worker.js +111 -111
  216. package/server/runtime/container.js +448 -448
  217. package/server/runtime/distributed-worker.js +362 -362
  218. package/server/runtime/event-bus.js +210 -210
  219. package/server/runtime/index.js +253 -253
  220. package/server/runtime/queue.js +599 -599
  221. package/server/runtime/replay.js +666 -666
  222. package/server/runtime/sandbox.js +266 -266
  223. package/server/runtime/scheduler.js +534 -534
  224. package/server/runtime/session-engine.js +293 -293
  225. package/server/runtime/state-manager.js +188 -188
  226. package/server/secrets/wab-signing-key.pem +3 -0
  227. package/server/secrets/wab-signing-pub.pem +3 -0
  228. package/server/security/cross-site-redactor.js +196 -196
  229. package/server/security/dry-run.js +180 -180
  230. package/server/security/human-gate-rate-limit.js +147 -147
  231. package/server/security/human-gate-transports.js +178 -178
  232. package/server/security/human-gate.js +281 -281
  233. package/server/security/index.js +368 -368
  234. package/server/security/intent-engine.js +245 -245
  235. package/server/security/reward-guard.js +171 -171
  236. package/server/security/rollback-store.js +239 -239
  237. package/server/security/token-scope.js +404 -404
  238. package/server/security/url-policy.js +139 -139
  239. package/server/services/adoption-agent.js +182 -0
  240. package/server/services/agent-chat.js +506 -506
  241. package/server/services/agent-learning.js +601 -601
  242. package/server/services/agent-memory.js +625 -625
  243. package/server/services/agent-mesh.js +555 -555
  244. package/server/services/agent-symphony.js +717 -717
  245. package/server/services/agent-tasks.js +1807 -1807
  246. package/server/services/api-key-engine.js +292 -292
  247. package/server/services/cluster.js +894 -894
  248. package/server/services/commander.js +738 -738
  249. package/server/services/edge-compute.js +440 -440
  250. package/server/services/email.js +233 -233
  251. package/server/services/fairness-engine.js +409 -0
  252. package/server/services/fairness.js +420 -0
  253. package/server/services/governance.js +466 -466
  254. package/server/services/hosted-runtime.js +205 -205
  255. package/server/services/lfd.js +635 -635
  256. package/server/services/local-ai.js +389 -389
  257. package/server/services/marketplace.js +270 -270
  258. package/server/services/metering.js +182 -182
  259. package/server/services/modules/affiliate-intelligence.js +93 -93
  260. package/server/services/modules/agent-firewall.js +90 -90
  261. package/server/services/modules/bounty.js +89 -89
  262. package/server/services/modules/collective-bargaining.js +92 -92
  263. package/server/services/modules/dark-pattern.js +66 -66
  264. package/server/services/modules/gov-intelligence.js +45 -45
  265. package/server/services/modules/neural.js +55 -55
  266. package/server/services/modules/notary.js +49 -49
  267. package/server/services/modules/price-time-machine.js +86 -86
  268. package/server/services/modules/protocol.js +104 -104
  269. package/server/services/negotiation.js +439 -439
  270. package/server/services/outreach-agent.js +312 -0
  271. package/server/services/plans.js +214 -214
  272. package/server/services/plugins.js +771 -771
  273. package/server/services/price-intelligence.js +566 -566
  274. package/server/services/price-shield.js +1137 -1137
  275. package/server/services/provider-clients.js +740 -740
  276. package/server/services/reputation.js +465 -465
  277. package/server/services/search-engine.js +357 -357
  278. package/server/services/security.js +513 -513
  279. package/server/services/self-healing.js +843 -843
  280. package/server/services/shieldlink.js +492 -0
  281. package/server/services/shieldqr.js +322 -322
  282. package/server/services/sovereign-shield.js +542 -542
  283. package/server/services/ssl-ct-monitor.js +224 -0
  284. package/server/services/ssl-inspector.js +42 -42
  285. package/server/services/ssl-monitor.js +167 -167
  286. package/server/services/stripe.js +206 -205
  287. package/server/services/swarm.js +788 -788
  288. package/server/services/universal-scraper.js +662 -662
  289. package/server/services/verification.js +481 -481
  290. package/server/services/vision.js +1163 -1163
  291. package/server/services/wab-crypto.js +178 -178
  292. package/server/utils/cache.js +125 -125
  293. package/server/utils/migrate.js +81 -81
  294. package/server/utils/safe-fetch.js +228 -228
  295. package/server/utils/secureFields.js +50 -50
  296. package/server/ws.js +161 -161
  297. package/templates/artisan-marketplace.yaml +104 -104
  298. package/templates/book-price-scout.yaml +98 -98
  299. package/templates/electronics-price-tracker.yaml +108 -108
  300. package/templates/flight-deal-hunter.yaml +113 -113
  301. package/templates/freelancer-direct.yaml +116 -116
  302. package/templates/grocery-price-compare.yaml +93 -93
  303. package/templates/hotel-direct-booking.yaml +113 -113
  304. package/templates/local-services.yaml +98 -98
  305. package/templates/olive-oil-tunisia.yaml +88 -88
  306. package/templates/organic-farm-fresh.yaml +101 -101
  307. package/templates/restaurant-direct.yaml +97 -97
  308. package/templates/ring4/banking-sovereign.yaml +55 -0
  309. package/templates/ring4/ecommerce-sovereign.yaml +58 -0
  310. package/templates/ring4/healthcare-sovereign.yaml +60 -0
@@ -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
+ };