69 lines
3.1 KiB
JavaScript
69 lines
3.1 KiB
JavaScript
import { app } from "../../../scripts/app.js";
|
|
import { ComfyWidgets } from "../../../scripts/widgets.js";
|
|
|
|
// Add new colors to LGraphCanvas.node_colors
|
|
function addNewColors() {
|
|
// Completely replacing the node_colors object
|
|
LGraphCanvas.node_colors = {
|
|
red: { color: "#C62828", bgcolor: "#E53935", groupcolor: "#B71C1C" }, // Red: 800, 600, 900
|
|
pink: { color: "#AD1457", bgcolor: "#D81B60", groupcolor: "#880E4F" }, // Pink: 800, 600, 900
|
|
purple: { color: "#6A1B9A", bgcolor: "#8E24AA", groupcolor: "#4A148C" }, // Purple: 800, 600, 900
|
|
indigo: { color: "#283593", bgcolor: "#3949AB", groupcolor: "#1A237E" }, // Indigo: 800, 600, 900
|
|
blue: { color: "#1565C0", bgcolor: "#1E88E5", groupcolor: "#0D47A1" }, // Blue: 800, 600, 900
|
|
cyan: { color: "#00838F", bgcolor: "#00ACC1", groupcolor: "#006064" }, // Cyan: 800, 600, 900
|
|
teal: { color: "#00695C", bgcolor: "#00897B", groupcolor: "#004D40" }, // Teal: 800, 600, 900
|
|
green: { color: "#2E7D32", bgcolor: "#43A047", groupcolor: "#1B5E20" }, // Green: 800, 600, 900
|
|
"light green": { color: "#558B2F", bgcolor: "#7CB342", groupcolor: "#33691E" }, // Light Green: 800, 600, 900
|
|
"deep orange": { color: "#D84315", bgcolor: "#F4511E", groupcolor: "#BF360C" }, // Deep Orange: 800, 600, 900
|
|
grey: { color: "#424242", bgcolor: "#757575", groupcolor: "#212121" }, // Grey: 800, 600, 900
|
|
|
|
// Adding the 6 extra colors
|
|
"blue grey": { color: "#455A64", bgcolor: "#607D8B", groupcolor: "#263238" }, // Blue Grey
|
|
lime: { color: "#9E9D24", bgcolor: "#C0CA33", groupcolor: "#827717" }, // Lime
|
|
"light blue": { color: "#0277BD", bgcolor: "#039BE5", groupcolor: "#01579B" }, // Light Blue
|
|
"steel blue": { color: "#4682B4", bgcolor: "#5A9BD3", groupcolor: "#2B5A8A" }, // Steel Blue
|
|
"turquoise": { color: "#00796B", bgcolor: "#009688", groupcolor: "#004D40" }, // Turquoise
|
|
navy: { color: "#003366", bgcolor: "#004080", groupcolor: "#001f3f" }, // Navy Blue
|
|
};
|
|
}
|
|
|
|
|
|
// Function to enhance nodes with text display
|
|
function addTextDisplay(nodeType) {
|
|
const onNodeCreated = nodeType.prototype.onNodeCreated;
|
|
nodeType.prototype.onNodeCreated = function () {
|
|
const r = onNodeCreated?.apply(this, arguments);
|
|
const w = ComfyWidgets["STRING"](this, "display", ["STRING", { multiline: true, placeholder: " " }], app).widget;
|
|
w.inputEl.readOnly = true;
|
|
w.inputEl.style.opacity = 0.7;
|
|
w.inputEl.style.cursor = "auto";
|
|
return r;
|
|
};
|
|
|
|
const onExecuted = nodeType.prototype.onExecuted;
|
|
nodeType.prototype.onExecuted = function (message) {
|
|
onExecuted?.apply(this, arguments);
|
|
|
|
for (const widget of this.widgets) {
|
|
if (widget.type === "customtext" && widget.name === "display" && widget.inputEl.readOnly === true) {
|
|
widget.value = message.text.join('');
|
|
}
|
|
}
|
|
|
|
this.onResize?.(this.size);
|
|
};
|
|
}
|
|
|
|
app.registerExtension({
|
|
name: "rka.GetRandomDimensions",
|
|
setup() {
|
|
// Call the function to add new colors
|
|
addNewColors();
|
|
},
|
|
async beforeRegisterNodeDef(nodeType, nodeData, app) {
|
|
if (["GetRandomDimensions"].includes(nodeData.name)) {
|
|
addTextDisplay(nodeType);
|
|
}
|
|
},
|
|
});
|