Add Comfy Registry store and search hook (#2848)

src/stores/comfyRegistryStore.ts

@@ -0,0 +1,89 @@
+import { defineStore } from 'pinia'
+import { computed, ref } from 'vue'
+
+import { useCachedRequest } from '@/composables/useCachedRequest'
+import { useComfyRegistryService } from '@/services/comfyRegistryService'
+import type { components, operations } from '@/types/comfyRegistryTypes'
+
+const PACK_LIST_CACHE_SIZE = 20
+const PACK_BY_ID_CACHE_SIZE = 50
+
+type NodePack = components['schemas']['Node']
+type ListPacksParams = operations['listAllNodes']['parameters']['query']
+type ListPacksResult =
+  operations['listAllNodes']['responses'][200]['content']['application/json']
+
+/**
+ * Store for managing remote custom nodes
+ */
+export const useComfyRegistryStore = defineStore('comfyRegistry', () => {
+  const registryService = useComfyRegistryService()
+
+  let listAllPacksHandler: ReturnType<
+    typeof useCachedRequest<ListPacksParams, ListPacksResult>
+  >
+  let getPackByIdHandler: ReturnType<typeof useCachedRequest<string, NodePack>>
+
+  const recentListResult = ref<NodePack[]>([])
+  const hasPacks = computed(() => recentListResult.value.length > 0)
+
+  /**
+   * Get a list of all node packs from the registry
+   */
+  const listAllPacks = async (params: ListPacksParams) => {
+    listAllPacksHandler ??= useCachedRequest<ListPacksParams, ListPacksResult>(
+      registryService.listAllPacks,
+      { maxSize: PACK_LIST_CACHE_SIZE }
+    )
+
+    const response = await listAllPacksHandler.call(params)
+    if (response) recentListResult.value = response.nodes || []
+
+    return response
+  }
+
+  /**
+   * Get a pack by its ID from the registry
+   */
+  const getPackById = async (
+    packId: NodePack['id']
+  ): Promise<NodePack | null> => {
+    if (!packId) return null
+
+    getPackByIdHandler ??= useCachedRequest<string, NodePack>(
+      registryService.getPackById,
+      { maxSize: PACK_BY_ID_CACHE_SIZE }
+    )
+
+    return getPackByIdHandler.call(packId)
+  }
+
+  /**
+   * Clear all cached data
+   */
+  const clearCache = () => {
+    listAllPacksHandler?.clear()
+    getPackByIdHandler?.clear()
+  }
+
+  /**
+   * Cancel any in-flight requests
+   */
+  const cancelRequests = () => {
+    listAllPacksHandler?.cancel()
+    getPackByIdHandler?.cancel()
+  }
+
+  return {
+    recentListResult,
+    hasPacks,
+
+    listAllPacks,
+    getPackById,
+    clearCache,
+    cancelRequests,
+
+    isLoading: registryService.isLoading,
+    error: registryService.error
+  }
+})