[Refactor] Move node composables to subfolder (#2712)

This commit is contained in:
bymyself
2025-02-24 12:04:16 -07:00
committed by GitHub
parent 9cb993cd3d
commit d340e634a8
8 changed files with 8 additions and 8 deletions

View File

@@ -0,0 +1,49 @@
import type { LGraphNode } from '@comfyorg/litegraph'
type DragHandler = (e: DragEvent) => boolean
type DropHandler<T> = (files: File[]) => Promise<T[]>
interface DragAndDropOptions<T> {
onDragOver?: DragHandler
onDrop: DropHandler<T>
fileFilter?: (file: File) => boolean
}
/**
* Adds drag and drop file handling to a node
*/
export const useNodeDragAndDrop = <T>(
node: LGraphNode,
options: DragAndDropOptions<T>
) => {
const { onDragOver, onDrop, fileFilter = () => true } = options
const hasFiles = (items: DataTransferItemList) =>
!!Array.from(items).find((f) => f.kind === 'file')
const filterFiles = (files: FileList) => Array.from(files).filter(fileFilter)
const hasValidFiles = (files: FileList) => filterFiles(files).length > 0
const isDraggingFiles = (e: DragEvent | undefined) => {
if (!e?.dataTransfer?.items) return false
return onDragOver?.(e) ?? hasFiles(e.dataTransfer.items)
}
const isDraggingValidFiles = (e: DragEvent | undefined) => {
if (!e?.dataTransfer?.files) return false
return hasValidFiles(e.dataTransfer.files)
}
node.onDragOver = isDraggingFiles
node.onDragDrop = function (e: DragEvent) {
if (!isDraggingValidFiles(e)) return false
const files = filterFiles(e.dataTransfer!.files)
onDrop(files).then((results) => {
if (!results?.length) return
})
return true
}
}