64 lines
3.0 KiB
JavaScript
64 lines
3.0 KiB
JavaScript
import { createScope } from '../ui/core.js';
|
|
import { createTextOutputView } from '../views/text-output.js';
|
|
|
|
export function restoreDisplayOutput(data, kind, hasInputWidget) {
|
|
if (data.kikoDisplay?.version === 1 && typeof data.kikoDisplay.text === 'string') return data.kikoDisplay.text;
|
|
if (!Array.isArray(data.widgets_values)) return null;
|
|
const values = data.widgets_values.slice(hasInputWidget && data.widgets_values.length > 1 ? 1 : 0);
|
|
if (kind === 'any' && ['raw value', 'tensor shape'].includes(values[0])) values.shift();
|
|
const strings = values.filter(value => typeof value === 'string');
|
|
if (!strings.length) return null;
|
|
if (kind === 'text' && strings[0] === '✓ Positive Prompt' && strings[2] === '✗ Negative Prompt') {
|
|
return `Positive prompt: ${strings[1]}\nNegative prompt: ${strings[3]}`;
|
|
}
|
|
return kind === 'any' ? strings.join('\n') : strings[0];
|
|
}
|
|
|
|
function updateOutput(node, text) {
|
|
if (node.displayOutput.scope.signal.aborted) return;
|
|
node.displayOutput.text = text;
|
|
node.displayOutput.view.update(text);
|
|
node.setDirtyCanvas(true, true);
|
|
}
|
|
|
|
export function registerDisplayNode(nodeType, app, kind) {
|
|
const onNodeCreated = nodeType.prototype.onNodeCreated;
|
|
nodeType.prototype.onNodeCreated = function () {
|
|
const result = onNodeCreated?.apply(this, arguments);
|
|
const scope = createScope();
|
|
const view = createTextOutputView(scope, kind, app.ui.settings);
|
|
this.displayOutput = { scope, view, text: '' };
|
|
this.addDOMWidget('kiko_display_output', 'div', view.element, {
|
|
serialize: false, hideOnZoom: false,
|
|
getMinHeight: () => kind === 'text' ? 240 : 180,
|
|
});
|
|
this.setSize([Math.max(this.size[0], kind === 'text' ? 400 : 450), Math.max(this.size[1], kind === 'text' ? 300 : 250)]);
|
|
updateOutput(this, kind === 'text' ? 'Text will appear here after execution...' : 'Value will appear here...');
|
|
return result;
|
|
};
|
|
const onExecuted = nodeType.prototype.onExecuted;
|
|
nodeType.prototype.onExecuted = function (message) {
|
|
const result = onExecuted?.apply(this, arguments);
|
|
if (typeof message?.text?.[0] === 'string') updateOutput(this, message.text[0]);
|
|
return result;
|
|
};
|
|
const configure = nodeType.prototype.configure;
|
|
nodeType.prototype.configure = function (data) {
|
|
const text = restoreDisplayOutput(data, kind, Boolean(this.inputs?.[0]?.widget));
|
|
const result = configure?.apply(this, arguments);
|
|
if (text !== null) updateOutput(this, text);
|
|
return result;
|
|
};
|
|
const onSerialize = nodeType.prototype.onSerialize;
|
|
nodeType.prototype.onSerialize = function (data) {
|
|
const result = onSerialize?.apply(this, arguments);
|
|
data.kikoDisplay = { version: 1, text: this.displayOutput.text };
|
|
return result;
|
|
};
|
|
const onRemoved = nodeType.prototype.onRemoved;
|
|
nodeType.prototype.onRemoved = function () {
|
|
this.displayOutput.scope.dispose();
|
|
return onRemoved?.apply(this, arguments);
|
|
};
|
|
}
|