From bb50b390f7a6048872234f48439b0b817d4abf51 Mon Sep 17 00:00:00 2001
From: pythongosssss <125205205+pythongosssss@users.noreply.github.com>
Date: Fri, 24 Mar 2023 16:19:52 +0000
Subject: [PATCH] Added subfolder menus to lora loader

---
 lora-subfolders/loraSubfolders.js | 80 +++++++++++++++++++++++++++++++
 1 file changed, 80 insertions(+)
 create mode 100644 lora-subfolders/loraSubfolders.js

diff --git a/lora-subfolders/loraSubfolders.js b/lora-subfolders/loraSubfolders.js
new file mode 100644
index 0000000..29fbb12
--- /dev/null
+++ b/lora-subfolders/loraSubfolders.js
@@ -0,0 +1,80 @@
+import { app } from "/scripts/app.js";
+import { ComfyWidgets } from "/scripts/widgets.js";
+import { api } from "/scripts/api.js";
+
+// Displays the wd14 prompt
+
+app.registerExtension({
+	name: "pysssss.LoraSubfolder",
+	async beforeRegisterNodeDef(nodeType, nodeData, app) {
+		if (nodeData.name === "LoraLoader") {
+			const onNodeCreated = nodeType.prototype.onNodeCreated;
+			nodeType.prototype.onNodeCreated = function () {
+				const r = onNodeCreated ? onNodeCreated.apply(this, arguments) : undefined;
+
+				const name = this.widgets.find((w) => w.name === "lora_name");
+				const callback = name.callback;
+				let prevValue = name.value;
+				name.callback = function () {
+					if (typeof name.value === "object") {
+						// Prevent selection of submenus
+						name.value = prevValue;
+						return;
+					}
+
+					prevValue = name.value;
+					return callback ? callback.call(this, name.value) : undefined;
+				};
+
+				const { values } = name.options;
+				const lookup = {
+					"": { options: [] },
+				};
+
+				// Simple check for the path character
+				const splitBy = (navigator.platform || navigator.userAgent).includes("Win") ? "\\" : "/";
+
+				// Split paths into menu structure
+				for (const value of values) {
+					const split = value.split(splitBy);
+					let path = "";
+					for (let i = 0; i < split.length; i++) {
+						const s = split[i];
+						const last = i === split.length - 1;
+						if (last) {
+							// Leaf node, manually add handler that sets the lora
+							lookup[path].options.push({
+								title: s,
+								callback: () => {
+									name.value = value;
+									name.callback(value);
+									app.graph.setDirtyCanvas(true);
+								},
+							});
+						} else {
+							const prevPath = path;
+							path += s + splitBy;
+							if (!lookup[path]) {
+								const sub = {
+									title: s,
+									submenu: {
+										options: [],
+										title: s,
+									},
+								};
+
+								// Add to tree
+								lookup[path] = sub.submenu;
+								lookup[prevPath].options.push(sub);
+							}
+						}
+					}
+				}
+
+				name.options.values = lookup[""].options;
+
+				return r;
+			};
+		}
+	},
+});