Replace magic numbers with LGraphEventMode (#1809)

This commit is contained in:
Chenlei Hu
2024-12-05 08:27:43 -08:00
committed by GitHub
parent 6cdc524174
commit 679405e0a2
2 changed files with 23 additions and 15 deletions

View File

@@ -16,6 +16,7 @@ import { useWorkspaceStore } from '@/stores/workspaceStore'
import { LGraphGroup } from '@comfyorg/litegraph'
import { LiteGraph } from '@comfyorg/litegraph'
import { LGraphNode } from '@comfyorg/litegraph'
import { LGraphEventMode } from '@comfyorg/litegraph/dist/types/globalEnums'
export function useCoreCommands(): ComfyCommand[] {
const getTracker = () => useWorkflowStore()?.activeWorkflow?.changeTracker
@@ -32,10 +33,10 @@ export function useCoreCommands(): ComfyCommand[] {
return result
}
const toggleSelectedNodesMode = (mode: number) => {
const toggleSelectedNodesMode = (mode: LGraphEventMode) => {
getSelectedNodes().forEach((node) => {
if (node.mode === mode) {
node.mode = 0 // always
node.mode = LGraphEventMode.ALWAYS
} else {
node.mode = mode
}
@@ -340,7 +341,7 @@ export function useCoreCommands(): ComfyCommand[] {
label: 'Mute/Unmute Selected Nodes',
versionAdded: '1.3.11',
function: () => {
toggleSelectedNodesMode(2) // muted
toggleSelectedNodesMode(LGraphEventMode.NEVER)
}
},
{
@@ -349,7 +350,7 @@ export function useCoreCommands(): ComfyCommand[] {
label: 'Bypass/Unbypass Selected Nodes',
versionAdded: '1.3.11',
function: () => {
toggleSelectedNodesMode(4) // bypassed
toggleSelectedNodesMode(LGraphEventMode.BYPASS)
}
},
{