Function beforeRegisterNodeDef

web/js/common/binding.js:178–243

(node, nodeData)

app.registerExtension({
	name: "pysssss.Binding",
	beforeRegisterNodeDef(node, nodeData) {
		const hasBinding = (v) => {
			if (!v) return false;
			return Object.values(v).find((c) => c[1]?.["pysssss.binding"]);
		};
		const inputs = { ...nodeData.input?.required, ...nodeData.input?.optional };
		if (hasBinding(inputs)) {
			const onAdded = node.prototype.onAdded;
			node.prototype.onAdded = function () {
				const r = onAdded?.apply(this, arguments);

				for (const widget of this.widgets || []) {
					const bindings = inputs[widget.name][1]?.["pysssss.binding"];
					if (!bindings) continue;

					for (const binding of bindings) {
						/**
						 * @type {import("../../../../../web/types/litegraph.d.ts").IWidget}
						 */
						const source = this.widgets.find((w) => w.name === binding.source);
						if (!source) {
							console.warn(
								"%c[🐍 pysssss]",
								"color: limegreen",
								`[binding ${node.comfyClass}.${widget.name}]`,
								"unable to find source binding widget:",
								binding.source,
								binding
							);
							continue;
						}

						let lastValue;
						async function valueChanged() {
							const state = {
								$this: widget,
								$source: source,
								$node: node,
							};

							for (const callback of binding.callback) {
								await invokeCallback(callback, state);
							}

							app.graph.setDirtyCanvas(true, false);
						}

						const cb = source.callback;
						source.callback = function () {
							const v = cb?.apply(this, arguments) ?? source.value;
							if (v !== lastValue) {
								lastValue = v;
								valueChanged();
							}
							return v;
						};

						lastValue = source.value;