Renamed to node finder

Added go to executing, and got to node via type
2023-05-06 15:35:38 +01:00
parent f10a840de6
commit c317b65fd8
2 changed files with 82 additions and 35 deletions
@@ -1,35 +0,0 @@
-import { app } from "/scripts/app.js";
-import { api } from "/scripts/api.js";
-
-// Adds a menu option to toggle follow the executing node
-
-app.registerExtension({
-	name: "pysssss.FollowExecution",
-	setup() {
-		let followExecution = false;
-
-		const centerNode = (id) => {
-			if (!followExecution || !id) return;
-			const node = app.graph.getNodeById(id);
-			if (!node) return;
-			app.canvas.centerOnNode(node);
-		};
-
-		api.addEventListener("executing", ({ detail }) => centerNode(detail));
-
-		// Add canvas menu options
-		const orig = LGraphCanvas.prototype.getCanvasMenuOptions;
-		LGraphCanvas.prototype.getCanvasMenuOptions = function () {
-			const options = orig.apply(this, arguments);
-			options.push(null, {
-				content: followExecution ? "Stop following execution" : "Follow execution",
-				callback: () => {
-					if ((followExecution = !followExecution)) {
-						centerNode(app.runningNodeId);
-					}
-				},
-			});
-			return options;
-		};
-	},
-});
@@ -0,0 +1,82 @@
+import { app } from "/scripts/app.js";
+import { api } from "/scripts/api.js";
+
+// Adds a menu option to toggle follow the executing node
+// Adds a menu option to go to the currently executing node
+// Adds a menu option to go to a node by type
+
+app.registerExtension({
+	name: "pysssss.NodeFinder",
+	setup() {
+		let followExecution = false;
+
+		const centerNode = (id) => {
+			if (!followExecution || !id) return;
+			const node = app.graph.getNodeById(id);
+			if (!node) return;
+			app.canvas.centerOnNode(node);
+		};
+
+		api.addEventListener("executing", ({ detail }) => centerNode(detail));
+
+		// Add canvas menu options
+		const orig = LGraphCanvas.prototype.getCanvasMenuOptions;
+		LGraphCanvas.prototype.getCanvasMenuOptions = function () {
+			const options = orig.apply(this, arguments);
+			options.push(null, {
+				content: followExecution ? "Stop following execution" : "Follow execution",
+				callback: () => {
+					if ((followExecution = !followExecution)) {
+						centerNode(app.runningNodeId);
+					}
+				},
+			});
+			if (app.runningNodeId) {
+				options.push({
+					content: "Show executing node",
+					callback: () => {
+						const node = app.graph.getNodeById(app.runningNodeId);
+						if (!node) return;
+						app.canvas.centerOnNode(node);
+					},
+				});
+			}
+
+			const nodes = app.graph._nodes;
+			const types = nodes.reduce((p, n) => {
+				if (n.type in p) {
+					p[n.type].push(n);
+				} else {
+					p[n.type] = [n];
+				}
+				return p;
+			}, {});
+			options.push({
+				content: "Go to node",
+				has_submenu: true,
+				submenu: {
+					options: Object.keys(types)
+						.sort()
+						.map((t) => ({
+							content: t,
+							has_submenu: true,
+							submenu: {
+								options: types[t]
+									.sort((a, b) => {
+										return a.pos[0] - b.pos[0];
+									})
+									.map((n) => ({
+										content: `${n.getTitle()} - #${n.id} (${n.pos[0]}, ${n.pos[1]})`,
+										callback: () => {
+											app.canvas.centerOnNode(n);
+										},
+									})),
+							},
+						})),
+				},
+			});
+
+			return options;
+		};
+	},
+});