mirror of
https://github.com/Comfy-Org/ComfyUI_frontend.git
synced 2026-03-03 12:10:11 +00:00
[Refactor] Make node badge a vue component (#1317)
* [Refactor] Make node badge a vue component * Simplify badge logic * nit
This commit is contained in:
@@ -13,7 +13,12 @@ import { TreeNode } from 'primevue/treenode'
|
||||
import { buildTree } from '@/utils/treeUtil'
|
||||
import { computed, ref } from 'vue'
|
||||
import axios from 'axios'
|
||||
import { type NodeSource, getNodeSource } from '@/types/nodeSource'
|
||||
import {
|
||||
type NodeSource,
|
||||
NodeSourceType,
|
||||
getNodeSource
|
||||
} from '@/types/nodeSource'
|
||||
import type { LGraphNode } from '@comfyorg/litegraph'
|
||||
|
||||
export interface BaseInputSpec<T = any> {
|
||||
name: string
|
||||
@@ -205,6 +210,16 @@ export class ComfyNodeDefImpl {
|
||||
const nodeFrequency = nodeFrequencyStore.getNodeFrequencyByName(this.name)
|
||||
return [scores[0], -nodeFrequency, ...scores.slice(1)]
|
||||
}
|
||||
|
||||
get isCoreNode(): boolean {
|
||||
return this.nodeSource.type === NodeSourceType.Core
|
||||
}
|
||||
|
||||
get nodeLifeCycleBadgeText(): string {
|
||||
if (this.deprecated) return '[DEPR]'
|
||||
if (this.experimental) return '[BETA]'
|
||||
return ''
|
||||
}
|
||||
}
|
||||
|
||||
export const SYSTEM_NODE_DEFS: Record<string, ComfyNodeDef> = {
|
||||
@@ -331,6 +346,10 @@ export const useNodeDefStore = defineStore('nodeDef', {
|
||||
},
|
||||
inputIsWidget(spec: BaseInputSpec) {
|
||||
return this.getWidgetType(spec.type, spec.name) !== null
|
||||
},
|
||||
fromLGraphNode(node: LGraphNode): ComfyNodeDefImpl | null {
|
||||
// Frontend-only nodes don't have nodeDef
|
||||
return this.nodeDefsByName[node.constructor.nodeData?.name] ?? null
|
||||
}
|
||||
}
|
||||
})
|
||||
|
||||
Reference in New Issue
Block a user