Use store to manage nodeSearchService (#214)

This commit is contained in:
Chenlei Hu
2024-07-24 12:00:23 -04:00
committed by GitHub
parent b73fe80761
commit ebdd7b8e40
7 changed files with 33 additions and 43 deletions

View File

@@ -1,3 +1,4 @@
import { NodeSearchService } from "@/services/nodeSearchService";
import { ComfyNodeDef } from "@/types/apiTypes";
import { defineStore } from "pinia";
@@ -54,6 +55,9 @@ export const useNodeDefStore = defineStore("nodeDef", {
nodeDefs(state) {
return Object.values(state.nodeDefsByName);
},
nodeSearchService(state) {
return new NodeSearchService(Object.values(state.nodeDefsByName));
},
},
actions: {
addNodeDef(nodeDef: ComfyNodeDef) {
@@ -61,7 +65,7 @@ export const useNodeDefStore = defineStore("nodeDef", {
},
addNodeDefs(nodeDefs: ComfyNodeDef[]) {
for (const nodeDef of nodeDefs) {
this.addNodeDef(nodeDef);
this.nodeDefsByName[nodeDef.name] = nodeDef;
}
},
},