From 1f0c9b2409307f1b875458a7309b98f07b6847ca Mon Sep 17 00:00:00 2001
From: pythongosssss <125205205+pythongosssss@users.noreply.github.com>
Date: Sun, 23 Apr 2023 17:37:57 +0100
Subject: [PATCH] Added widget default values extension

---
 widget-defaults/widgetDefaults.js | 221 ++++++++++++++++++++++++++++++
 1 file changed, 221 insertions(+)
 create mode 100644 widget-defaults/widgetDefaults.js

diff --git a/widget-defaults/widgetDefaults.js b/widget-defaults/widgetDefaults.js
new file mode 100644
index 0000000..8991bab
--- /dev/null
+++ b/widget-defaults/widgetDefaults.js
@@ -0,0 +1,221 @@
+import { app } from "/scripts/app.js";
+import { $el, ComfyDialog } from "/scripts/ui.js";
+
+// Allows you to specify custom default values for any widget on any node
+
+const id = "pysssss.WidgetDefaults";
+const nodeDataKey = Symbol();
+
+app.registerExtension({
+	name: id,
+	beforeRegisterNodeDef(nodeType, nodeData) {
+		nodeType[nodeDataKey] = nodeData;
+	},
+	setup() {
+		let defaults;
+		let setting;
+
+		const applyDefaults = (defaults) => {
+			for (const node of Object.values(LiteGraph.registered_node_types)) {
+				const nodeData = node[nodeDataKey];
+				if (!nodeData) continue;
+				const nodeDefaults = defaults[node.type];
+				if (!nodeDefaults) continue;
+				const inputs = { ...(nodeData.input?.required || {}), ...(nodeData.input?.optional || {}) };
+
+				for (const w in nodeDefaults) {
+					const widgetDef = inputs[w];
+					if (widgetDef) {
+						if (widgetDef[1]) {
+							widgetDef[1].default = nodeDefaults[w];
+						} else {
+							widgetDef[1] = { default: nodeDefaults[w] };
+						}
+					}
+				}
+			}
+		};
+
+		const getDefaults = () => {
+			let items;
+			try {
+				items = JSON.parse(setting.value);
+				items = items.reduce((p, n) => {
+					if (!p[n.node]) p[n.node] = {};
+					p[n.node][n.widget] = n.value;
+					return p;
+				}, {});
+			} catch (error) {}
+			if (!items) {
+				items = {};
+			}
+			applyDefaults(items);
+			return items;
+		};
+
+		// app.graph.onNodeAdded = function (node) {
+		// 	const nodeDefaults = defaults[node.constructor.type];
+		// 	if (nodeDefaults) {
+		// 		for (const w of node.widgets || []) {
+		// 			let value = nodeDefaults[w.name];
+		// 			if (value != null) {
+		// 				debugger
+		// 				if (typeof w.value === "number") {
+		// 					value = +value;
+		// 				}
+		// 				w.value = value;
+		// 				debugger
+		// 				w.callback?.(w.value);
+		// 			}
+		// 		}
+		// 	}
+		// };
+
+		class WidgetDefaultsDialog extends ComfyDialog {
+			constructor() {
+				super();
+				this.element.classList.add("comfy-manage-templates");
+				this.grid = $el(
+					"div",
+					{
+						style: {
+							display: "grid",
+							gridTemplateColumns: "1fr auto auto auto",
+							gap: "5px",
+						},
+						className: "pysssss-widget-defaults",
+					},
+					[
+						$el("label", {
+							textContent: "Node Class",
+						}),
+						$el("label", {
+							textContent: "Widget Name",
+						}),
+						$el("label", {
+							textContent: "Default Value",
+						}),
+						$el("label"),
+						(this.rows = $el("div", {
+							style: {
+								display: "contents",
+							},
+						})),
+					]
+				);
+			}
+
+			createButtons() {
+				const btns = super.createButtons();
+				btns[0].textContent = "Cancel";
+				btns.unshift(
+					$el("button", {
+						type: "button",
+						textContent: "Add New",
+						onclick: () => this.addRow(),
+					}),
+					$el("button", {
+						type: "button",
+						textContent: "Save",
+						onclick: () => this.save(),
+					})
+				);
+				return btns;
+			}
+
+			addRow(node = "", widget = "", value = "") {
+				let nameInput;
+				this.rows.append(
+					$el(
+						"div",
+						{
+							style: {
+								display: "contents",
+							},
+							className: "pysssss-widget-defaults-row",
+						},
+						[
+							$el("input", {
+								placeholder: "e.g. CheckpointLoaderSimple",
+								value: node,
+							}),
+							$el("input", {
+								placeholder: "e.g. ckpt_name",
+								value: widget,
+								$: (el) => (nameInput = el),
+							}),
+							$el("input", {
+								placeholder: "e.g. myBestModel.safetensors",
+								value,
+							}),
+							$el("button", {
+								textContent: "Delete",
+								style: {
+									fontSize: "12px",
+									color: "red",
+									fontWeight: "normal",
+								},
+								onclick: (e) => {
+									nameInput.value = "";
+									e.target.parentElement.style.display = "none";
+								},
+							}),
+						]
+					)
+				);
+			}
+
+			save() {
+				const rows = this.rows.children;
+				const items = [];
+
+				for (const row of rows) {
+					const inputs = row.querySelectorAll("input");
+					const node = inputs[0].value.trim();
+					const widget = inputs[1].value.trim();
+					const value = inputs[2].value;
+					if (node && widget) {
+						items.push({ node, widget, value });
+					}
+				}
+
+				setting.value = JSON.stringify(items);
+				defaults = getDefaults();
+
+				this.close();
+			}
+
+			show() {
+				this.rows.replaceChildren();
+				for (const nodeName in defaults) {
+					const node = defaults[nodeName];
+					for (const widgetName in node) {
+						this.addRow(nodeName, widgetName, node[widgetName]);
+					}
+				}
+
+				this.addRow();
+				super.show(this.grid);
+			}
+		}
+
+		setting = app.ui.settings.addSetting({
+			id,
+			name: "[pysss] Widget Defaults",
+			type: () => {
+				return $el("button", {
+					textContent: "[pysss] Manage Widget Defaults",
+					onclick: () => {
+						app.ui.settings.close();
+						const dialog = new WidgetDefaultsDialog();
+						dialog.show();
+					},
+					style: {
+						fontSize: "14px",
+					},
+				});
+			},
+		});
+		defaults = getDefaults();
+	},
+});