mirror of
https://github.com/Comfy-Org/ComfyUI_frontend.git
synced 2026-02-04 07:00:23 +00:00
c94cedf8ee
## Summary add 'SaveVideo' into saveNodeTypes to fix [ComfyUI 10285](https://github.com/comfyanonymous/ComfyUI/issues/10285) and [ComfyUI 10479](https://github.com/comfyanonymous/ComfyUI/issues/10479) ## Changes add 'SaveVideo' into saveNodeTypes values ## Review Focus is this enough? ┆Issue is synchronized with this [Notion page](https://www.notion.so/PR-6647-add-SaveVideo-into-saveNodeTypes-2a76d73d365081d9bb7eccc358c9836d) by [Unito](https://www.unito.io)
62 lines
1.8 KiB
TypeScript
62 lines
1.8 KiB
TypeScript
import { applyTextReplacements } from '@/utils/searchAndReplace'
|
|
|
|
import { app } from '../../scripts/app'
|
|
|
|
const saveNodeTypes = new Set([
|
|
'SaveImage',
|
|
'SaveVideo',
|
|
'SaveAnimatedWEBP',
|
|
'SaveWEBM',
|
|
'SaveAudio',
|
|
'SaveGLB',
|
|
'SaveAnimatedPNG',
|
|
'CLIPSave',
|
|
'VAESave',
|
|
'ModelSave',
|
|
'LoraSave',
|
|
'SaveLatent'
|
|
])
|
|
|
|
// Use widget values and dates in output filenames
|
|
|
|
app.registerExtension({
|
|
name: 'Comfy.SaveImageExtraOutput',
|
|
async beforeRegisterNodeDef(nodeType, nodeData, app) {
|
|
if (saveNodeTypes.has(nodeData.name)) {
|
|
const onNodeCreated = nodeType.prototype.onNodeCreated
|
|
// When the SaveImage node is created we want to override the serialization of the output name widget to run our S&R
|
|
nodeType.prototype.onNodeCreated = function () {
|
|
const r = onNodeCreated
|
|
? // @ts-expect-error fixme ts strict error
|
|
onNodeCreated.apply(this, arguments)
|
|
: undefined
|
|
|
|
// @ts-expect-error fixme ts strict error
|
|
const widget = this.widgets.find((w) => w.name === 'filename_prefix')
|
|
// @ts-expect-error fixme ts strict error
|
|
widget.serializeValue = () => {
|
|
// @ts-expect-error fixme ts strict error
|
|
return applyTextReplacements(app.graph, widget.value)
|
|
}
|
|
|
|
return r
|
|
}
|
|
} else {
|
|
// When any other node is created add a property to alias the node
|
|
const onNodeCreated = nodeType.prototype.onNodeCreated
|
|
nodeType.prototype.onNodeCreated = function () {
|
|
const r = onNodeCreated
|
|
? // @ts-expect-error fixme ts strict error
|
|
onNodeCreated.apply(this, arguments)
|
|
: undefined
|
|
|
|
if (!this.properties || !('Node name for S&R' in this.properties)) {
|
|
this.addProperty('Node name for S&R', this.constructor.type, 'string')
|
|
}
|
|
|
|
return r
|
|
}
|
|
}
|
|
}
|
|
})
|