From 88349f3262f984ec6d5957834c6e2e3781ce5d55 Mon Sep 17 00:00:00 2001
From: pythongosssss <125205205+pythongosssss@users.noreply.github.com>
Date: Sun, 1 Oct 2023 10:40:07 +0100
Subject: [PATCH] Add swap width/height

---
 web/js/swapResolution.js | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)
 create mode 100644 web/js/swapResolution.js

diff --git a/web/js/swapResolution.js b/web/js/swapResolution.js
new file mode 100644
index 0000000..40f6dcc
--- /dev/null
+++ b/web/js/swapResolution.js
@@ -0,0 +1,30 @@
+import { app } from "../../../scripts/app.js";
+app.registerExtension({
+	name: "pysssss.SwapResolution",
+	async beforeRegisterNodeDef(nodeType, nodeData) {
+		const inputs = { ...nodeData.input?.required, ...nodeData.input?.optional };
+		if (inputs.width && inputs.height) {
+			const origGetExtraMenuOptions = nodeType.prototype.getExtraMenuOptions;
+			nodeType.prototype.getExtraMenuOptions = function (_, options) {
+				const r = origGetExtraMenuOptions?.apply?.(this, arguments);
+
+				options.push(
+					{
+						content: "Swap width/height",
+						callback: () => {
+							const w = this.widgets.find((w) => w.name === "width");
+							const h = this.widgets.find((w) => w.name === "height");
+							const a = w.value;
+							w.value = h.value;
+							h.value = a;
+							app.graph.setDirtyCanvas(true);
+						},
+					},
+					null
+				);
+
+				return r;
+			};
+		}
+	},
+});