Function beforeRegisterNodeDef

web/js/quickNodes.js:38–195

(nodeType, nodeData, app)

app.registerExtension({
	name: "pysssss.QuickNodes",
	async beforeRegisterNodeDef(nodeType, nodeData, app) {
		if (nodeData.input && nodeData.input.required) {
			const keys = Object.keys(nodeData.input.required);
			for (let i = 0; i < keys.length; i++) {
				if (nodeData.input.required[keys[i]][0] === "VAE") {
					addMenuHandler(nodeType, function (_, options) {
						options.unshift({
							content: "Use VAE",
							callback: () => {
								getOrAddVAELoader(this).connect(0, this, i);
							},
						});
					});
					break;
				}
			}
		}

		if (nodeData.name === "KSampler") {
			addMenuHandler(nodeType, function (_, options) {
				options.unshift(
					{
						content: "Add Blank Input",
						callback: () => {
							const imageNode = addNode("EmptyLatentImage", this, { before: true });
							imageNode.connect(0, this, 3);
						},
					},
					{
						content: "Add Hi-res Fix",
						callback: () => {
							const upscaleNode = addNode("LatentUpscale", this);
							this.connect(0, upscaleNode, 0);

							const sampleNode = addNode("KSampler", upscaleNode);

							for (let i = 0; i < 3; i++) {
								const l = this.getInputLink(i);
								if (l) {
									app.graph.getNodeById(l.origin_id).connect(l.origin_slot, sampleNode, i);
								}
							}

							upscaleNode.connect(0, sampleNode, 3);
						},
					},
					{
						content: "Add 2nd Pass",
						callback: () => {
							const upscaleNode = addNode("LatentUpscale", this);
							this.connect(0, upscaleNode, 0);

							const ckptNode = addNode("CheckpointLoaderSimple", this);
							const sampleNode = addNode("KSampler", ckptNode);

							const positiveLink = this.getInputLink(1);
							const negativeLink = this.getInputLink(2);
							const positiveNode = positiveLink