feat: add Node Marker system with right-click marking

web/comfyui/node_marker.js

@@ -0,0 +1,126 @@
+import { app } from "../../scripts/app.js";
+
+// =============================================================================
+// Node Marker – right-click node marking (no dedicated node required)
+//
+// Adds a "Mark as →" submenu with role options to any node's context menu.
+// Roles are stored in ``node.properties.lm_marker_role`` and automatically
+// persist with the workflow JSON.
+//
+// The workflow registry reads these markers and makes them available to the
+// standalone UI (e.g. ``sendEmbeddingToWorkflow`` also considers nodes marked
+// as ``send_prompt_target``).
+// =============================================================================
+
+const ROLES = {
+  send_prompt_target: {
+    label: "Send Prompt Target",
+    emoji: "\uD83D\uDCDD",
+  },
+  send_gen_params: {
+    label: "Send Gen Params Target",
+    emoji: "\uD83D\uDD27",
+  },
+};
+
+// ---- Helpers ----------------------------------------------------------------
+
+function getMarker(node) {
+  return node?.properties?.lm_marker_role ?? null;
+}
+
+function setMarker(node, roleKey) {
+  if (!node || !ROLES[roleKey]) return;
+  node.properties = node.properties || {};
+  node.properties.lm_marker_role = roleKey;
+
+  // Save original title if not already saved, then prefix with emoji
+  if (!node.properties.lm_marker_original_title) {
+    node.properties.lm_marker_original_title = node.title || "";
+  }
+  const def = ROLES[roleKey];
+  node.title = `${def.emoji} ${node.properties.lm_marker_original_title}`;
+
+  if (typeof node.setDirtyCanvas === "function") {
+    node.setDirtyCanvas(true, true);
+  }
+  triggerRegistryRefresh();
+}
+
+function clearMarker(node) {
+  if (!node) return;
+  delete node.properties.lm_marker_role;
+
+  // Restore original title: prefer stripping emoji from current title
+  // (captures user renames after marking), fall back to saved original.
+  const cleaned = node.title?.replace(
+    /^(\u2709\uFE0F?|\u2699\uFE0F?|\uD83D\uDCDD|\uD83C\uDF9B\uFE0F?|\uD83D\uDD27)\s*/,
+    ''
+  );
+  if (cleaned && cleaned !== node.title) {
+    node.title = cleaned;
+  } else {
+    const orig = node.properties.lm_marker_original_title;
+    if (orig !== undefined) {
+      node.title = orig;
+    }
+  }
+  delete node.properties.lm_marker_original_title;
+
+  if (typeof node.setDirtyCanvas === "function") {
+    node.setDirtyCanvas(true, true);
+  }
+  triggerRegistryRefresh();
+}
+
+function triggerRegistryRefresh() {
+  // workflow_registry.js listens for this event to re-scan the graph.
+  window.dispatchEvent(new CustomEvent("lm_marker_changed"));
+}
+
+// ---- Submenu builder --------------------------------------------------------
+
+function buildSubmenuOptions(node) {
+  const currentRole = getMarker(node);
+  const options = [];
+
+  for (const [key, def] of Object.entries(ROLES)) {
+    const isActive = currentRole === key;
+    options.push({
+      content: `${isActive ? "\u2713 " : ""}${def.label}`,
+      disabled: isActive,
+      callback: () => setMarker(node, key),
+    });
+  }
+
+  if (currentRole) {
+    options.push({
+      content: "Clear marker",
+      callback: () => clearMarker(node),
+    });
+  }
+
+  return options;
+}
+
+function buildMenuItems(node) {
+  return [
+    null,
+    {
+      content: "Mark as",
+      has_submenu: true,
+      submenu: {
+        options: buildSubmenuOptions(node),
+      },
+    },
+  ];
+}
+
+// ---- Extension --------------------------------------------------------------
+
+app.registerExtension({
+  name: "LoraManager.NodeMarker",
+  getNodeMenuItems(node) {
+    return buildMenuItems(node);
+  },
+});