71 lines
2.4 KiB
TypeScript
71 lines
2.4 KiB
TypeScript
import { app } from '../../../scripts/app.js';
|
|
import { api } from '../../../scripts/api.js';
|
|
import { ComfyWidgets } from '../../../scripts/widgets.js';
|
|
import { commonPrefix, displayContext } from './common.js';
|
|
|
|
// "Show any" Node
|
|
app.registerExtension({
|
|
name: 'Crystools.Debugger.ConsoleAny',
|
|
async beforeRegisterNodeDef(nodeType, nodeData, app) {
|
|
if (nodeData.name === 'Show any [Crystools]') {
|
|
// 3 is the index of the text field in the node
|
|
displayContext(nodeType, app, 3);
|
|
}
|
|
},
|
|
});
|
|
|
|
app.registerExtension({
|
|
name: 'Crystools.Debugger.Metadata',
|
|
|
|
registerCustomNodes() {
|
|
class MetadataNode {
|
|
constructor() {
|
|
this.serialize_widgets = false;
|
|
this.isVirtualNode = true;
|
|
|
|
const w = ComfyWidgets.STRING(this, '', ['', {default: '', multiline: true}], app).widget;
|
|
w.inputEl.readOnly = true;
|
|
ComfyWidgets.BOOLEAN(this, 'Active', ['', {default: true}], app);
|
|
ComfyWidgets.BOOLEAN(this, 'Parsed', ['', {default: true}], app);
|
|
ComfyWidgets.COMBO(this, 'What', [['Prompt', 'Workflow'], {default: 'Prompt'}]);
|
|
|
|
// It runs at finish on each prompt queue
|
|
api.addEventListener('executed', this.fillMetadataWidget);
|
|
}
|
|
|
|
fillMetadataWidget = () => {
|
|
app.graphToPrompt()
|
|
.then(workflow => {
|
|
let result = 'inactive';
|
|
// debugger
|
|
const output = this.widgets[0];
|
|
const active = this.widgets[1].value;
|
|
const parsed = this.widgets[2].value;
|
|
let what = this.widgets[3].value.toLowerCase();
|
|
|
|
if (active) {
|
|
what = what === 'prompt' ? 'output' : what; // little fix for better understanding
|
|
result = workflow[what];
|
|
if (parsed) {
|
|
result = JSON.stringify(result, null, 2);
|
|
} else {
|
|
result = JSON.stringify(result);
|
|
}
|
|
}
|
|
|
|
output.value = result;
|
|
});
|
|
}
|
|
}
|
|
|
|
// I'm not sure for what they're using prototype and lots of black magic, don't change the order!
|
|
LiteGraph.registerNodeType('Show Metadata [Crystools]', MetadataNode);
|
|
MetadataNode.category = `crystools ${commonPrefix}/Debugger`;
|
|
MetadataNode.shape = LiteGraph.BOX_SHAPE;
|
|
MetadataNode.title = `Show Metadata ${commonPrefix}`;
|
|
// MetadataNode.collapsable = false;
|
|
// MetadataNode.color = "#FF2222";
|
|
// MetadataNode.bgcolor = "#000000";
|
|
},
|
|
});
|