Function beforeRegisterNodeDef

web/js/workflows.js:307–342

(nodeType, nodeData, app)

		};
	},
	async beforeRegisterNodeDef(nodeType, nodeData, app) {
		const getExtraMenuOptions = nodeType.prototype.getExtraMenuOptions;
		nodeType.prototype.getExtraMenuOptions = function (_, options) {
			const r = getExtraMenuOptions?.apply?.(this, arguments);
			let img;
			if (this.imageIndex != null) {
				// An image is selected so select that
				img = this.imgs[this.imageIndex];
			} else if (this.overIndex != null) {
				// No image is selected but one is hovered
				img = this.imgs[this.overIndex];
			}

			if (img) {
				let pos = options.findIndex((o) => o.content === "Save Image");
				if (pos === -1) {
					pos = 0;
				} else {
					pos++;
				}

				options.splice(pos, 0, {
					content: "Send to workflow",
					has_submenu: true,
					submenu: {
						options: [
							{ callback: () => sendToWorkflow(img), title: "[Current workflow]" },
							...workflows.getMenuOptions(sendToWorkflow.bind(null, img)),
						],
					},
				});
			}

			return r;
		};
	},
});