@workerdeck/server 2.11.0 → 2.12.0

package/build/index.mjs

@@ -1,8 +1,8 @@
 import { createServer } from "node:http";
 import { WebSocketServer } from "ws";
-import { BrowserBridgeExecutor, PEER_MESSAGE_MAX_CHARS, PEER_RECENT_DEFAULT, PEER_RECENT_MAX, SessionRunner, attachmentKind, checkClaudeAuth, createEngineSession, getEngineAdapter, installPeerDirectory, normalizeMediaType, peerDirectoryHandle, peerSummary, recentLines } from "@workerdeck/core";
+import { BrowserBridgeExecutor, PEER_MESSAGE_MAX_CHARS, PEER_RECENT_DEFAULT, PEER_RECENT_MAX, SessionRunner, attachmentKind, checkClaudeAuth, createEngineSession, getEngineAdapter, installPeerDirectory, isSlashCommand, normalizeMediaType, peerDirectoryHandle, peerSummary, recentLines } from "@workerdeck/core";
 import { JobQueue } from "@workerdeck/queue";
-import { ENGINE_CAPABILITIES, PROTOCOL_VERSION, SHELL_COMMAND_MAX, costOfByModel, imagePartRef, isJobRun, mergeByModel, sessionState, setPricingOverrides, subscriptionComparison, supportsPermissionMode } from "@workerdeck/protocol";
+import { ENGINE_CAPABILITIES, PEER_MENTION_MAX, PROTOCOL_VERSION, SHELL_COMMAND_MAX, costOfByModel, imagePartRef, isJobRun, mergeByModel, peerMentionKey, peerMentionSlug, scanPeerMentions, sessionState, setPricingOverrides, subscriptionComparison, supportsPermissionMode } from "@workerdeck/protocol";
 import { createHash, randomUUID } from "node:crypto";
 import { closeSync, constants, createReadStream, existsSync, fstatSync, ftruncateSync, lstatSync, mkdirSync, openSync, readFileSync, readdirSync, realpathSync, renameSync, statSync, writeFileSync } from "node:fs";
 import { constants as constants$1, homedir } from "node:os";
@@ -1854,13 +1854,15 @@ async function handleCommand(ctx, frame, runner, access) {
 	const { attachmentStore, bridge } = ctx;
 	switch (frame.type) {
 		case "user_message": {
+			const mentions = ctx.peers && !isSlashCommand(frame.text) ? await ctx.peers.mentions(runner.id, frame.text).catch(() => []) : [];
+			const options = mentions.length > 0 ? { mentions } : void 0;
 			if (!frame.attachmentIds?.length) {
-				runner.sendMessage(frame.text);
+				runner.sendMessage(frame.text, void 0, options);
 				return;
 			}
 			const resolved = attachmentStore.resolve(runner.id, frame.attachmentIds);
 			if (!resolved.ok) throw new Error(`unknown attachment(s): ${resolved.missing.join(", ")}`);
-			runner.sendMessage(frame.text, resolved.attachments);
+			runner.sendMessage(frame.text, resolved.attachments, options);
 			return;
 		}
 		case "permission_decision":
@@ -3164,6 +3166,44 @@ function createPeerService(deps) {
 			queued: before === "running" || before === "awaiting_approval"
 		};
 	};
+	const mentions = async (from, text) => {
+		const tokens = scanPeerMentions(text);
+		if (tokens.length === 0) return [];
+		const rows = await list(from);
+		const byKey = /* @__PURE__ */ new Map();
+		const add = (key, row) => {
+			const held = byKey.get(key);
+			if (held) held.push(row);
+			else byKey.set(key, [row]);
+		};
+		for (const row of rows) {
+			add(peerMentionKey(peerMentionSlug(row.title, row.id)), row);
+			add(peerMentionKey(row.id), row);
+		}
+		const resolved = [];
+		const seen = /* @__PURE__ */ new Set();
+		for (const token of tokens) {
+			const key = peerMentionKey(token.body);
+			const exact = byKey.get(key);
+			const matches = exact ?? (key.length >= 4 ? rows.filter((row) => row.id.startsWith(key)) : []);
+			const target = matches[0];
+			if (!target || matches.length > 1 && !exact) continue;
+			if (seen.has(target.id)) continue;
+			seen.add(target.id);
+			const others = matches.slice(1, 4).map((row) => row.id);
+			resolved.push({
+				typed: token.body,
+				id: target.id,
+				name: target.title,
+				engine: target.engine,
+				status: target.status,
+				cwd: target.cwd,
+				...others.length ? { ambiguousWith: others } : {}
+			});
+			if (resolved.length >= PEER_MENTION_MAX) break;
+		}
+		return resolved;
+	};
 	const watch = (runner) => runner.subscribe((event) => {
 		if (event.type === "user_message" && !event.origin && !event.synthetic && event.parentToolUseId == null && !event.replay) inbound.delete(runner.id);
 	}, runner.info().lastSeq);
@@ -3171,6 +3211,7 @@ function createPeerService(deps) {
 		list,
 		peek,
 		send,
+		mentions,
 		watch
 	};
 }
@@ -3664,6 +3705,7 @@ function createWorkerServer(options = {}) {
 		factory,
 		registry,
 		parking,
+		peers,
 		bridge,
 		projects,
 		queue,