67 lines
2.1 KiB
JavaScript
67 lines
2.1 KiB
JavaScript
import { app } from "../../scripts/app.js";
|
|
|
|
const COLOR_THEMES = {
|
|
red: { nodeColor: "#332222", nodeBgColor: "#553333" },
|
|
green: { nodeColor: "#223322", nodeBgColor: "#335533" },
|
|
blue: { nodeColor: "#222233", nodeBgColor: "#333355" },
|
|
pale_blue: { nodeColor: "#2a363b", nodeBgColor: "#3f5159" },
|
|
cyan: { nodeColor: "#223333", nodeBgColor: "#335555" },
|
|
purple: { nodeColor: "#332233", nodeBgColor: "#553355" },
|
|
yellow: { nodeColor: "#443322", nodeBgColor: "#665533" },
|
|
// purple_custom: { nodeColor: "#7345b0" , nodeBgColor: "#482b6e" },
|
|
purple_custom: { nodeColor: "#111111" , nodeBgColor: "#482b6e" },
|
|
|
|
none: { nodeColor: null, nodeBgColor: null } // no color
|
|
};
|
|
|
|
const NODE_COLORS = {
|
|
"FrameMaker": "purple_custom",
|
|
"FrameMakerBatch": "purple_custom",
|
|
|
|
|
|
};
|
|
|
|
function shuffleArray(array) {
|
|
for (let i = array.length - 1; i > 0; i--) {
|
|
const j = Math.floor(Math.random() * (i + 1));
|
|
[array[i], array[j]] = [array[j], array[i]]; // Swap elements
|
|
}
|
|
}
|
|
|
|
let colorKeys = Object.keys(COLOR_THEMES).filter(key => key !== "none");
|
|
shuffleArray(colorKeys); // Shuffle the color themes initially
|
|
|
|
function setNodeColors(node, theme) {
|
|
if (!theme) {return;}
|
|
node.shape = "box";
|
|
if(theme.nodeColor && theme.nodeBgColor) {
|
|
node.color = theme.nodeColor;
|
|
node.bgcolor = theme.nodeBgColor;
|
|
}
|
|
}
|
|
|
|
const ext = {
|
|
name: "disty.appearance",
|
|
|
|
nodeCreated(node) {
|
|
const nclass = node.comfyClass;
|
|
if (NODE_COLORS.hasOwnProperty(nclass)) {
|
|
let colorKey = NODE_COLORS[nclass];
|
|
|
|
if (colorKey === "random") {
|
|
// Check for a valid color key before popping
|
|
if (colorKeys.length === 0 || !COLOR_THEMES[colorKeys[colorKeys.length - 1]]) {
|
|
colorKeys = Object.keys(COLOR_THEMES).filter(key => key !== "none");
|
|
shuffleArray(colorKeys);
|
|
}
|
|
colorKey = colorKeys.pop();
|
|
}
|
|
|
|
const theme = COLOR_THEMES[colorKey];
|
|
setNodeColors(node, theme);
|
|
}
|
|
}
|
|
};
|
|
|
|
app.registerExtension(ext);
|