From f0c6d1513e31134bea3e38efc4c8fc525b48698d Mon Sep 17 00:00:00 2001
From: pixelass <greg@pixelass.com>
Date: Mon, 31 Jul 2023 18:39:55 +0200
Subject: [PATCH] feat(arrange): allow arranging nodes

---
 extensions/arrange.js | 42 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)
 create mode 100644 extensions/arrange.js

diff --git a/extensions/arrange.js b/extensions/arrange.js
new file mode 100644
index 0000000..f75920b
--- /dev/null
+++ b/extensions/arrange.js
@@ -0,0 +1,42 @@
+/**
+ * Coded with love by Failfa.st
+ * LICENSE: AGPL 3.0
+ * https://github.com/failfa-st/failfast-comfyui-extensions/blob/main/LICENSE
+ *
+ * Visit https://github.com/failfa-st/failfast-comfyui-extensions for more info
+ *
+ * Homepage:  https://failfa.st
+ * GitHub: https://github.com/failfa-st
+ * Discord: https://discord.com/invite/m3TBB9XEkb
+ */
+import { app } from "../scripts/app.js";
+import { $el } from "../scripts/ui.js";
+
+/**
+ * Render Shadow
+ */
+
+const alignName = "Failfast.alignNodes";
+
+app.registerExtension({
+  name: alignName,
+  async init(app) {
+    const getCanvasMenuOptions = LGraphCanvas.prototype.getCanvasMenuOptions;
+    LGraphCanvas.prototype.getCanvasMenuOptions = function () {
+      const menuOptions = getCanvasMenuOptions.apply(this, arguments);
+      menuOptions.push({
+        content: "Arrange (vertical)",
+        callback: () =>
+          app.graph.arrange(
+            LiteGraph.CANVAS_GRID_SIZE * 4,
+            LiteGraph.VERTICAL_LAYOUT,
+          ),
+      });
+      menuOptions.push({
+        content: "Arrange (horizontal)",
+        callback: () => app.graph.arrange(LiteGraph.CANVAS_GRID_SIZE * 2),
+      });
+      return menuOptions;
+    };
+  },
+});