mercury-agent 0.4.5

package/src/core/trigger.ts

@@ -0,0 +1,142 @@
+import type { Db } from "../storage/db.js";
+import type { TriggerConfig, TriggerMatch } from "../types.js";
+
+export interface TriggerResult {
+  matched: boolean;
+  prompt: string; // text with trigger stripped
+}
+
+const VALID_MATCHES: Set<string> = new Set(["prefix", "mention", "always"]);
+
+export function loadTriggerConfig(
+  db: Db,
+  spaceId: string,
+  defaults: { patterns: string[]; match: string },
+): TriggerConfig {
+  const match = db.getSpaceConfig(
+    spaceId,
+    "trigger.match",
+  ) as TriggerMatch | null;
+  const patternsRaw = db.getSpaceConfig(spaceId, "trigger.patterns");
+  const caseSensitive = db.getSpaceConfig(spaceId, "trigger.case_sensitive");
+  const mediaInGroups =
+    db.getSpaceConfig(spaceId, "trigger.media_in_groups") === "true";
+
+  const defaultMatch = VALID_MATCHES.has(defaults.match)
+    ? (defaults.match as TriggerMatch)
+    : "mention";
+
+  const patterns = patternsRaw
+    ? patternsRaw
+        .split(",")
+        .map((s) => s.trim())
+        .filter(Boolean)
+    : defaults.patterns;
+
+  return {
+    match: match && VALID_MATCHES.has(match) ? match : defaultMatch,
+    patterns: patterns.length > 0 ? patterns : defaults.patterns,
+    caseSensitive: caseSensitive === "true",
+    mediaInGroups,
+  };
+}
+
+export function matchTrigger(
+  text: string,
+  config: TriggerConfig,
+  isDM: boolean,
+  hasAttachments = false,
+): TriggerResult {
+  const trimmed = text.trim();
+  if (!trimmed) {
+    if (hasAttachments) {
+      if (isDM) return { matched: true, prompt: "" };
+      if (config.match === "always") return { matched: true, prompt: "" };
+      if (config.mediaInGroups) return { matched: true, prompt: "" };
+    }
+    return { matched: false, prompt: "" };
+  }
+
+  if (config.match === "always") {
+    return { matched: true, prompt: trimmed };
+  }
+
+  // DMs: trigger is optional — try to match and strip, but if no match, use full text
+  if (isDM) {
+    const stripped = tryMatch(trimmed, config);
+    return { matched: true, prompt: stripped ?? trimmed };
+  }
+
+  // Groups: trigger is required
+  const stripped = tryMatch(trimmed, config);
+  if (stripped !== null) return { matched: true, prompt: stripped };
+  return { matched: false, prompt: "" };
+}
+
+/**
+ * Try to match any of the patterns against the text using the configured mode.
+ * Returns the prompt with the matched pattern removed, or null if no match.
+ * Patterns are tried longest-first to avoid partial matches (e.g. "@Pi" before "Pi").
+ */
+function tryMatch(text: string, config: TriggerConfig): string | null {
+  const sorted = [...config.patterns].sort((a, b) => b.length - a.length);
+
+  for (const pattern of sorted) {
+    const result = matchSinglePattern(text, pattern, config);
+    if (result !== null) return result;
+  }
+
+  return null;
+}
+
+function matchSinglePattern(
+  text: string,
+  pattern: string,
+  config: TriggerConfig,
+): string | null {
+  switch (config.match) {
+    case "prefix":
+      return stripPrefix(text, pattern, config.caseSensitive);
+
+    case "mention": {
+      // Word-boundary match — works for "@Mick", "Mick", "@Mercury", etc.
+      const escaped = pattern.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
+      const flags = config.caseSensitive ? "" : "i";
+      // Use \b for word chars, but @ isn't a word char so we handle it:
+      // Match at start-of-string or after whitespace, then end the trigger at
+      // whitespace, common punctuation (e.g. "@Mercury, hi"), or EOS — but not
+      // another letter/digit (avoids "Pi" inside "Pixel").
+      const regex = new RegExp(
+        `(?:^|(?<=\\s))${escaped}(?=\\s|[,;:.!?]|$)`,
+        flags,
+      );
+      const match = regex.exec(text);
+      if (!match) return null;
+      const before = text.slice(0, match.index).trim();
+      const after = text
+        .slice(match.index + match[0].length)
+        .replace(/^[,;:.!?\s]+/, "")
+        .trim();
+      return [before, after].filter(Boolean).join(" ") || text;
+    }
+
+    default:
+      return null;
+  }
+}
+
+function stripPrefix(
+  text: string,
+  prefix: string,
+  caseSensitive: boolean,
+): string | null {
+  const textCmp = caseSensitive ? text : text.toLowerCase();
+  const prefixCmp = caseSensitive ? prefix : prefix.toLowerCase();
+
+  if (!textCmp.startsWith(prefixCmp)) return null;
+  const rest = text.slice(prefix.length);
+  // For prefix mode, require a space or end-of-string after the pattern
+  // to avoid "Pixel art" matching "Pi"
+  if (rest.length > 0 && rest[0] !== " ") return null;
+  return rest.trim();
+}