update to support comfyui native web file handling

2023-08-23 17:31:11 +01:00
parent 4461d5e29a
commit e98a92ba5d
37 changed files with 31 additions and 9 deletions
@@ -0,0 +1,35 @@
+import { app } from "../../../scripts/app.js";
+import { ComfyWidgets } from "../../../scripts/widgets.js";
+
+app.registerExtension({
+	name: "pysssss.MathExpression",
+	init() {
+		const STRING = ComfyWidgets.STRING;
+		ComfyWidgets.STRING = function (node, inputName, inputData) {
+			const r = STRING.apply(this, arguments);
+			r.widget.dynamicPrompts = inputData?.[1].dynamicPrompts;
+			return r;
+		};
+	},
+	beforeRegisterNodeDef(nodeType) {
+		if (nodeType.comfyClass === "MathExpression|pysssss") {
+			const onDrawForeground = nodeType.prototype.onDrawForeground;
+			nodeType.prototype.onDrawForeground = function (ctx) {
+				const r = onDrawForeground?.apply?.(this, arguments);
+
+				const v = app.nodeOutputs?.[this.id + ""];
+				if (!this.flags.collapsed && v) {
+					const text = v.value[0] + "";
+					ctx.save();
+					ctx.font = "bold 12px sans-serif";
+					ctx.fillStyle = "dodgerblue";
+					const sz = ctx.measureText(text);
+					ctx.fillText(text, this.size[0] - sz.width - 5, LiteGraph.NODE_SLOT_HEIGHT * 3);
+					ctx.restore();
+				}
+
+				return r;
+			};
+		}
+	},
+});