29 lines
725 B
JavaScript
29 lines
725 B
JavaScript
/*"Design-focused node suite for ComfyUI."
|
|
# Copyright 2026 Augment Studio
|
|
Augmentstudio.app */
|
|
|
|
import { app } from "../../../scripts/app.js";
|
|
|
|
const TITLE_BG = "#443322";
|
|
const NODE_BG = "#665533";
|
|
|
|
const PAID_NODES = [
|
|
"AugmentSVGToPNG",
|
|
"AugmentPNGToSVG",
|
|
];
|
|
|
|
app.registerExtension({
|
|
name: "augment.NodeStyle",
|
|
|
|
async beforeRegisterNodeDef(nodeType, nodeData, app) {
|
|
if (!PAID_NODES.includes(nodeData.name)) return;
|
|
|
|
const onDrawForeground = nodeType.prototype.onDrawForeground;
|
|
nodeType.prototype.onDrawForeground = function (ctx) {
|
|
this.color = TITLE_BG;
|
|
this.bgcolor = NODE_BG;
|
|
onDrawForeground?.apply(this, arguments);
|
|
};
|
|
},
|
|
});
|